aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/de/hysky/skyblocker/utils/NEURepoManager.java
blob: 870e94da82b3772cb91ea69d1547646bed177641 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
package de.hysky.skyblocker.utils;

import de.hysky.skyblocker.SkyblockerMod;
import de.hysky.skyblocker.skyblock.itemlist.ItemRepository;
import io.github.moulberry.repo.NEURepository;
import net.fabricmc.fabric.api.client.command.v2.ClientCommandManager;
import net.fabricmc.fabric.api.client.command.v2.ClientCommandRegistrationCallback;
import net.minecraft.client.MinecraftClient;
import net.minecraft.text.Text;
import org.eclipse.jgit.api.Git;
import org.eclipse.jgit.api.errors.TransportException;
import org.eclipse.jgit.errors.RepositoryNotFoundException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.stream.Stream;

/**
 * Initializes the NEU repo, which contains item metadata and fairy souls location data. Clones the repo if it does not exist and checks for updates. Use {@link #runAsyncAfterLoad(Runnable)} to run code after the repo is initialized.
 */
public class NEURepoManager {
    private static final Logger LOGGER = LoggerFactory.getLogger(NEURepoManager.class);
    public static final String REMOTE_REPO_URL = "https://github.com/NotEnoughUpdates/NotEnoughUpdates-REPO.git";
    /**
     * Use {@link #NEU_REPO}.
     */
    private static final Path LOCAL_REPO_DIR = SkyblockerMod.CONFIG_DIR.resolve("item-repo"); // TODO rename to NotEnoughUpdates-REPO
    private static final CompletableFuture<Void> REPO_INITIALIZED = loadRepository();
    public static final NEURepository NEU_REPO = NEURepository.of(LOCAL_REPO_DIR);

    /**
     * Adds command to update repository manually from ingame.
     * <p></p>
     * TODO A button could be added to the settings menu that will trigger this command.
     */
    public static void init() {
        ClientCommandRegistrationCallback.EVENT.register((dispatcher, registryAccess) ->
                dispatcher.register(ClientCommandManager.literal(SkyblockerMod.NAMESPACE)
                        .then(ClientCommandManager.literal("updaterepository").executes(context -> {
                            deleteAndDownloadRepository();
                            return 1;
                        }))));
    }

    private static CompletableFuture<Void> loadRepository() {
        return CompletableFuture.runAsync(() -> {
            try {
                if (Files.isDirectory(NEURepoManager.LOCAL_REPO_DIR)) {
                    try (Git localRepo = Git.open(NEURepoManager.LOCAL_REPO_DIR.toFile())) {
                        localRepo.pull().setRebase(true).call();
                        LOGGER.info("[Skyblocker] NEU Repository Updated");
                    }
                } else {
                    Git.cloneRepository().setURI(REMOTE_REPO_URL).setDirectory(NEURepoManager.LOCAL_REPO_DIR.toFile()).setBranchesToClone(List.of("refs/heads/master")).setBranch("refs/heads/master").call().close();
                    LOGGER.info("[Skyblocker] NEU Repository Downloaded");
                }
                NEU_REPO.reload();
            } catch (TransportException e){
                LOGGER.error("[Skyblocker] Transport operation failed. Most likely unable to connect to the remote NEU repo on github", e);
            } catch (RepositoryNotFoundException e) {
                LOGGER.warn("[Skyblocker] Local NEU Repository not found or corrupted, downloading new one", e);
                deleteAndDownloadRepository();
            } catch (Exception e) {
                LOGGER.error("[Skyblocker] Encountered unknown exception while initializing NEU Repository", e);
            }
        });
    }

    private static void deleteAndDownloadRepository() {
        CompletableFuture.runAsync(() -> {
            try {
                ItemRepository.setFilesImported(false);
                recursiveDelete(NEURepoManager.LOCAL_REPO_DIR);
            } catch (Exception ex) {
                if (MinecraftClient.getInstance().player != null)
                    MinecraftClient.getInstance().player.sendMessage(Constants.PREFIX.get().append(Text.translatable("skyblocker.updaterepository.failed")), false);
                return;
            }
            loadRepository();
        });
    }

    @SuppressWarnings("ResultOfMethodCallIgnored")
    private static void recursiveDelete(Path dir) throws IOException {
        if (Files.isDirectory(dir) && !Files.isSymbolicLink(dir)) {
            Files.list(dir).forEach(child -> {
                try {
                    recursiveDelete(child);
                } catch (Exception e) {
                    LOGGER.error("[Skyblocker] Encountered an exception while deleting a file! Path: {}", child.toAbsolutePath(), e);
                }
            });
        }

        Files.delete(dir);
    }

    /**
     * Runs the given runnable after the NEU repo is initialized.
     * @param runnable the runnable to run
     * @return a completable future of the given runnable
     */
    public static CompletableFuture<Void> runAsyncAfterLoad(Runnable runnable) {
        return REPO_INITIALIZED.thenRunAsync(runnable);
    }
}