Inceptum/launcher/src/main/java/io/gitlab/jfronny/inceptum/launcher/system/source/ModrinthModSource.java

package io.gitlab.jfronny.inceptum.launcher.system.source;

import io.gitlab.jfronny.commons.HashUtils;
import io.gitlab.jfronny.commons.cache.MemoryOperationResultCache;
import io.gitlab.jfronny.commons.tuple.Tuple;
import io.gitlab.jfronny.inceptum.common.Net;
import io.gitlab.jfronny.inceptum.common.Utils;
import io.gitlab.jfronny.inceptum.launcher.api.ModrinthApi;
import io.gitlab.jfronny.inceptum.launcher.model.modrinth.*;

import java.io.IOException;
import java.net.URISyntaxException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.*;

public final class ModrinthModSource implements ModSource {
    private static final MemoryOperationResultCache<Tuple<String, String>, Optional<ModSource>> UPDATE_CACHE = new MemoryOperationResultCache<>(Utils.CACHE_SIZE);
    private static final MemoryOperationResultCache<String, Set<ModSource>> DEPENDENCIES_CACHE = new MemoryOperationResultCache<>(Utils.CACHE_SIZE);

    private final String versionId;
    private final ModrinthVersion current;
    private final ModrinthProject mod;

    public ModrinthModSource(String versionId) throws IOException {
        this.versionId = versionId;
        this.current = ModrinthApi.getVersion(versionId);
        this.mod = ModrinthApi.getMod(getModId());
    }

    @Override
    public ModDownload download() throws IOException {
        ModrinthVersion.File file = current.files.get(0);
        Path path = getJarPath();
        try {
            Net.downloadFile(file.url, file.hashes.sha1, path);
        } catch (URISyntaxException e) {
            throw new IOException("Could not download file", e);
        }
        return new ModDownload(file.hashes.sha1, HashUtils.murmur2(Files.readAllBytes(path)), path);
    }

    @Override
    public Set<ModSource> getDependencies(String gameVersion) throws IOException {
        return DEPENDENCIES_CACHE.get(versionId, () -> {
            Set<ModSource> deps = new HashSet<>();
            for (ModrinthVersion.Dependency dependency : current.dependencies) {
                //TODO show optional dependencies
                if (dependency.dependency_type == ModrinthVersion.Dependency.DependencyType.required)
                    deps.add(new ModrinthModSource(dependency.version_id));
            }
            return Set.copyOf(deps);
        });
    }

    @Override
    public Optional<ModSource> getUpdate(String gameVersion) throws IOException {
        return UPDATE_CACHE.get(Tuple.of(versionId, gameVersion), () -> {
            ModrinthVersion stable = null;
            ModrinthVersion beta = null;
            ModrinthVersion latest = null;
            for (ModrinthVersion version : ModrinthApi.getVersions(getModId())) {
                if (version.game_versions.contains(gameVersion) && version.loaders.contains("fabric")) {
                    latest = version;
                    if (version.version_type == ModrinthVersion.VersionType.beta || version.version_type == ModrinthVersion.VersionType.release)
                        beta = version;
                    if (version.version_type == ModrinthVersion.VersionType.release)
                        stable = version;
                }
            }
            ModrinthVersion next = switch (current.version_type) {
                case alpha -> latest;
                case beta -> beta;
                case release -> stable;
            };
            if (next == null) return Optional.empty();
            if (next.version_number.equals(current.version_number)) return Optional.empty();
            return Optional.of(new ModrinthModSource(next.id));
        });
    }

    @Override
    public String getVersion() {
        return current.version_number;
    }

    @Override
    public String getName() {
        return "modrinth/" + getShortName() + '/' + current.version_number;
    }

    @Override
    public String getShortName() {
        return (mod.slug == null ? mod.id : mod.slug);
    }

    @Override
    public String getFileName() {
        return current.files.get(0).filename;
    }

    @Override
    public boolean equals(ModSource other) {
        return other instanceof ModrinthModSource ms && ms.getModId().equals(getModId()) && ms.versionId.equals(versionId);
    }

    public String getVersionId() {
        return versionId;
    }

    public String getModId() {
        return current.project_id;
    }

    public ModrinthModpackManifest.File toManifest() throws IOException {
        ModrinthVersion.File orig = current.files.get(0);
        ModrinthModpackManifest.File f = new ModrinthModpackManifest.File();
        f.path = "mods/" + orig.filename;
        f.hashes = new ModrinthHashes();
        f.hashes.sha1 = orig.hashes.sha1;
        f.hashes.sha512 = orig.hashes.sha512;
        f.downloads = List.of(orig.url);
        f.fileSize = Files.size(getJarPath());
        return f;
    }
}