package mindustry.ui.dialogs;

import arc.*;
import arc.Net.*;
import arc.files.*;
import arc.func.*;
import arc.graphics.*;
import arc.graphics.g2d.*;
import arc.input.*;
import arc.scene.style.*;
import arc.scene.ui.TextButton.*;
import arc.struct.*;
import arc.util.*;
import arc.util.io.*;
import arc.util.serialization.*;
import mindustry.*;
import mindustry.core.*;
import mindustry.ctype.*;
import mindustry.gen.*;
import mindustry.graphics.*;
import mindustry.mod.*;
import mindustry.mod.Mods.*;
import mindustry.ui.*;

import java.io.*;

import static mindustry.Vars.*;

public class ModsDialog extends BaseDialog{
    private String searchtxt = "";
    private @Nullable Seq<ModListing> modList;

    public ModsDialog(){
        super("@mods");
        addCloseButton();

        buttons.button("@mods.guide", Icon.link, () -> Core.app.openURI(modGuideURL)).size(210, 64f);

        shown(this::setup);

        hidden(() -> {
            if(mods.requiresReload()){
                reload();
            }
        });

        shown(() -> Core.app.post(() -> {
            Core.settings.getBoolOnce("modsalpha", () -> {
                ui.showText("@mods", "@mods.alphainfo");
            });
        }));
    }

    void modError(Throwable error){
        ui.loadfrag.hide();

        if(Strings.getCauses(error).contains(t -> t.getMessage() != null && (t.getMessage().contains("trust anchor") || t.getMessage().contains("SSL") || t.getMessage().contains("protocol")))){
            ui.showErrorMessage("@feature.unsupported");
        }else{
            ui.showException(error);
        }
    }

    void getModList(Cons<Seq<ModListing>> listener){
        if(modList == null){
            Core.net.httpGet("https://raw.githubusercontent.com/Anuken/MindustryMods/master/mods.json", response -> {
                String strResult = response.getResultAsString();
                var status = response.getStatus();

                Core.app.post(() -> {
                    if(status != HttpStatus.OK){
                        ui.showErrorMessage(Core.bundle.format("connectfail", status));
                    }else{
                        modList = new Json().fromJson(Seq.class, ModListing.class, strResult);

                        //potentially sort mods by game version compatibility, or other criteria
                        //modList.sort(Structs.comparingBool(m -> !Version.isAtLeast(m.minGameVersion)));

                        listener.get(modList);
                    }
                });
            }, error -> Core.app.post(() -> ui.showException(error)));
        }else{
            listener.get(modList);
        }
    }

    void setup(){
        float h = 110f;
        float w = mobile ? 440f : 524f;

        cont.clear();
        cont.defaults().width(mobile ? 500 : 560f).pad(4);
        cont.add("@mod.reloadrequired").visible(mods::requiresReload).center().get().setAlignment(Align.center);
        cont.row();

        cont.table(buttons -> {
            buttons.left().defaults().growX().height(60f).uniformX();

            TextButtonStyle style = Styles.clearPartialt;
            float margin = 12f;

            buttons.button("@mod.import", Icon.add, style, () -> {
                BaseDialog dialog = new BaseDialog("@mod.import");

                TextButtonStyle bstyle = Styles.cleart;

                dialog.cont.table(Tex.button, t -> {
                    t.defaults().size(300f, 70f);
                    t.margin(12f);

                    t.button("@mod.import.file", Icon.file, bstyle, () -> {
                        dialog.hide();

                        platform.showMultiFileChooser(file -> {
                            Runnable go = () -> {
                                try{
                                    mods.importMod(file);
                                    setup();
                                }catch(IOException e){
                                    ui.showException(e);
                                    e.printStackTrace();
                                }
                            };

                            //show unsafe jar file warning
                            if(file.extEquals("jar")){
                                ui.showConfirm("@warning", "@mod.jarwarn", go);
                            }else{
                                go.run();
                            }
                        }, "zip", "jar");
                    }).margin(12f);

                    t.row();

                    t.button("@mod.import.github", Icon.github, bstyle, () -> {
                        dialog.hide();

                        ui.showTextInput("@mod.import.github", "", 64, Core.settings.getString("lastmod", ""), text -> {
                            Core.settings.put("lastmod", text);

                            ui.loadfrag.show();
                            githubImportMod(text);
                        });
                    }).margin(12f);

                    t.row();

                    t.button("@mod.featured.title", Icon.star, bstyle, () -> {
                        Runnable[] rebuildBrowser = {null};
                        dialog.hide();
                        BaseDialog browser = new BaseDialog("$mod.featured.dialog.title");
                        browser.cont.table(table -> {
                            table.left();
                            table.image(Icon.zoom);
                            table.field(searchtxt, res -> {
                                searchtxt = res;
                                rebuildBrowser[0].run();
                            }).growX().get();
                        }).fillX().padBottom(4);

                        browser.cont.row();

                        browser.cont.pane(tablebrow -> {
                            tablebrow.margin(10f).top();
                            rebuildBrowser[0] = () -> {
                                tablebrow.clear();
                                tablebrow.add("@loading");

                                getModList(listings -> {
                                    tablebrow.clear();

                                    for(ModListing mod : listings){
                                        if(mod.hasJava || !searchtxt.isEmpty() && !mod.repo.contains(searchtxt) || (Vars.ios && mod.hasScripts)) continue;

                                        tablebrow.button(btn -> {
                                            btn.top().left();
                                            btn.margin(12f);
                                            btn.table(con -> {
                                                con.left();
                                                con.add("[accent]" + mod.name + "[white]\n[lightgray]Author:[] " + mod.author + "\n[accent]\uE809 " + mod.stars +
                                                (Version.isAtLeast(mod.minGameVersion) ? "" : "\n" + Core.bundle.format("mod.requiresversion", mod.minGameVersion)))
                                                .width(388f).wrap().growX().pad(0f, 6f, 0f, 6f).left().labelAlign(Align.left);
                                                con.add().growX().pad(0f, 6f, 0f, 6f);
                                            }).fillY().growX().pad(0f, 6f, 0f, 6f);
                                        }, Styles.modsb, () -> {
                                            var sel = new BaseDialog(mod.name);
                                            sel.cont.add(mod.description).width(mobile ? 400f : 500f).wrap().pad(4f).labelAlign(Align.center, Align.left);
                                            sel.buttons.defaults().size(150f, 54f).pad(2f);
                                            sel.setFillParent(false);
                                            sel.buttons.button("@back", Icon.left, () -> {
                                                sel.clear();
                                                sel.hide();
                                            });
                                            sel.buttons.button("@mods.browser.add", Icon.download, () -> {
                                                sel.hide();
                                                githubImportMod(mod.repo);
                                            });
                                            sel.buttons.button("@mods.github.open", Icon.link, () -> {
                                                Core.app.openURI("https://github.com/" + mod.repo);
                                            });
                                            sel.keyDown(KeyCode.escape, sel::hide);
                                            sel.keyDown(KeyCode.back, sel::hide);
                                            sel.show();
                                        }).width(480f).growX().left().fillY();
                                        tablebrow.row();
                                    }
                                });
                            };
                            rebuildBrowser[0].run();
                        });
                        browser.addCloseButton();
                        browser.show();
                    }).margin(12f);
                });
                dialog.addCloseButton();

                dialog.show();

            }).margin(margin);

            if(!mobile){
                buttons.button("@mods.openfolder", Icon.link, style, () -> Core.app.openFolder(modDirectory.absolutePath())).margin(margin);
            }
        }).width(w);

        cont.row();

        if(!mods.list().isEmpty()){
            cont.pane(table -> {
                table.margin(10f).top();

                boolean anyDisabled = false;
                for(LoadedMod mod : mods.list()){

                    if(!mod.enabled() && !anyDisabled && mods.list().size > 0){
                        anyDisabled = true;
                        table.row();
                        table.image().growX().height(4f).pad(6f).color(Pal.gray);
                        table.row();
                    }

                    table.button(t -> {
                        t.top().left();
                        t.margin(12f);

                        t.defaults().left().top();
                        t.table(title -> {
                            title.left();

                            title.add(new BorderImage(){{
                                if(mod.iconTexture != null){
                                    setDrawable(new TextureRegion(mod.iconTexture));
                                }else{
                                    setDrawable(Tex.nomap);
                                }
                                border(Pal.accent);
                            }}).size(h - 8f).padTop(-8f).padLeft(-8f).padRight(8f);

                            title.table(text -> {
                                boolean hideDisabled = !mod.isSupported() || mod.hasUnmetDependencies() || mod.hasContentErrors();

                                text.add("" + mod.meta.displayName() + "\n[lightgray]v" + mod.meta.version + (mod.enabled() || hideDisabled ? "" : "\n" + Core.bundle.get("mod.disabled") + ""))
                                    .wrap().top().width(300f).growX().left();

                                text.row();

                                if(mod.isOutdated()){
                                    text.labelWrap("@mod.outdated").growX();
                                    text.row();
                                }else if(!mod.isSupported()){
                                    text.labelWrap(Core.bundle.format("mod.requiresversion", mod.meta.minGameVersion)).growX();
                                    text.row();
                                }else if(mod.hasUnmetDependencies()){
                                    text.labelWrap(Core.bundle.format("mod.missingdependencies", mod.missingDependencies.toString(", "))).growX();
                                    t.row();
                                }else if(mod.hasContentErrors()){
                                    text.labelWrap("@mod.erroredcontent").growX();
                                    text.row();
                                }
                            }).top().growX();

                            title.add().growX();
                        }).growX().growY().left();

                        t.table(right -> {
                            right.right();
                            right.button(mod.enabled() ? Icon.downOpen : Icon.upOpen, Styles.clearPartiali, () -> {
                                mods.setEnabled(mod, !mod.enabled());
                                setup();
                            }).size(50f).disabled(!mod.isSupported());

                            right.button(mod.hasSteamID() ? Icon.link : Icon.trash, Styles.clearPartiali, () -> {
                                if(!mod.hasSteamID()){
                                    ui.showConfirm("@confirm", "@mod.remove.confirm", () -> {
                                        mods.removeMod(mod);
                                        setup();
                                    });
                                }else{
                                    platform.viewListing(mod);
                                }
                            }).size(50f);

                            if(steam && !mod.hasSteamID()){
                                right.row();
                                right.button(Icon.download, Styles.clearTransi, () -> {
                                    platform.publish(mod);
                                }).size(50f);
                            }
                        }).growX().right().padRight(-8f).padTop(-8f);


                    }, Styles.clearPartialt, () -> showMod(mod)).size(w, h).growX().pad(4f);
                    table.row();
                }
            });

        }else{
            cont.table(Styles.black6, t -> t.add("@mods.none")).height(80f);
        }

        cont.row();


    }

    private void reload(){
        ui.showInfo("@mods.reloadexit", () -> Core.app.exit());
    }

    private void showMod(LoadedMod mod){
        BaseDialog dialog = new BaseDialog(mod.meta.displayName());

        dialog.addCloseButton();

        if(!mobile){
            dialog.buttons.button("@mods.openfolder", Icon.link, () -> Core.app.openFolder(mod.file.absolutePath()));
        }

        //TODO improve this menu later
        dialog.cont.pane(desc -> {
            desc.center();
            desc.defaults().padTop(10).left();

            desc.add("@editor.name").padRight(10).color(Color.gray).padTop(0);
            desc.row();
            desc.add(mod.meta.displayName()).growX().wrap().padTop(2);
            desc.row();
            if(mod.meta.author != null){
                desc.add("@editor.author").padRight(10).color(Color.gray);
                desc.row();
                desc.add(mod.meta.author).growX().wrap().padTop(2);
                desc.row();
            }
            if(mod.meta.description != null){
                desc.add("@editor.description").padRight(10).color(Color.gray).top();
                desc.row();
                desc.add(mod.meta.description).growX().wrap().padTop(2);
                desc.row();
            }

        }).width(400f);

        //TODO maybe enable later
        if(false){
            Seq<UnlockableContent> all = Seq.with(content.getContentMap()).<Content>flatten().select(c -> c.minfo.mod == mod && c instanceof UnlockableContent).as();
            if(all.any()){
                dialog.cont.row();
                dialog.cont.pane(cs -> {
                    int i = 0;
                    for(UnlockableContent c : all){
                        cs.button(new TextureRegionDrawable(c.icon(Cicon.medium)), Styles.cleari, Cicon.medium.size, () -> {
                            ui.content.show(c);
                        }).size(50f).with(im -> {
                            var click = im.getClickListener();
                            im.update(() -> im.getImage().color.lerp(!click.isOver() ? Color.lightGray : Color.white, 0.4f * Time.delta));
                        });

                        if(++i % 8 == 0) cs.row();
                    }
                }).growX().minHeight(60f);
            }
        }



        dialog.show();
    }

    private void handleMod(String repo, HttpResponse result){
        try{
            Fi file = tmpDirectory.child(repo.replace("/", "") + ".zip");
            Streams.copy(result.getResultAsStream(), file.write(false));
            mods.importMod(file);
            file.delete();
            Core.app.post(() -> {
                try{
                    setup();
                    ui.loadfrag.hide();
                }catch(Throwable e){
                    ui.showException(e);
                }
            });
        }catch(Throwable e){
            modError(e);
        }
    }

    private void githubImportMod(String name){
        //try several branches
        //TODO use only the main branch as specified in meta
        githubImportBranch("6.0", name, e1 -> {
            githubImportBranch("master", name, e2 -> {
                githubImportBranch("main", name, e3 -> {
                    ui.showErrorMessage(Core.bundle.format("connectfail", e2));
                    ui.loadfrag.hide();
                });
            });
        });
    }

    private void githubImportBranch(String branch, String repo, Cons<HttpStatus> err){
        Core.net.httpGet("https://api.github.com/repos/" + repo + "/zipball/" + branch, loc -> {
            if(loc.getStatus() == HttpStatus.OK){
                if(loc.getHeader("Location") != null){
                    Core.net.httpGet(loc.getHeader("Location"), result -> {
                        if(result.getStatus() != HttpStatus.OK){
                            err.get(result.getStatus());
                        }else{
                            handleMod(repo, result);
                        }
                    }, t2 -> Core.app.post(() -> modError(t2)));
                }else{
                    handleMod(repo, loc);
                }
            }else{
                err.get(loc.getStatus());
            }
         }, t2 -> Core.app.post(() -> modError(t2)));
    }
}