package io.xpipe.ext.base.service;

import io.xpipe.app.comp.BaseRegionBuilder;
import io.xpipe.app.core.AppI18n;
import io.xpipe.app.ext.*;
import io.xpipe.app.hub.comp.*;
import io.xpipe.app.storage.DataStorage;
import io.xpipe.app.storage.DataStoreEntry;
import io.xpipe.app.util.DocumentationLink;
import io.xpipe.app.util.StoreStateFormat;
import io.xpipe.app.util.FailableRunnable;
import io.xpipe.ext.base.host.HostAddressGatewayStore;

import javafx.beans.binding.Bindings;
import javafx.beans.value.ObservableValue;

import java.util.ArrayList;
import java.util.List;

public abstract class AbstractServiceStoreProvider implements SingletonSessionStoreProvider, DataStoreProvider {

    @Override
    public boolean showIncompleteInfo() {
        return true;
    }

    @Override
    public DocumentationLink getHelpLink() {
        return DocumentationLink.SERVICES;
    }

    @Override
    public boolean supportsSession(SingletonSessionStore<?> s) {
        var abs = (AbstractServiceStore) s;
        if (abs.getAddress() != null) {
            return abs.getGateway() != null
                    && abs.getGateway().getStore().isLocallyTunnelable()
                    && abs.getGateway().getStore().requiresTunnel();
        }

        if (abs.getHost() != null) {
            if (abs.getHost().getStore() instanceof HostAddressGatewayStore a) {
                if (a.getTunnelGateway() != null
                        && a.getTunnelGateway().getStore().requiresTunnel()
                        && a.getTunnelGateway().getStore().isLocallyTunnelable()
                        && abs.shouldTunnel()) {
                    return true;
                }
            }

            if (abs.getHost().getStore() instanceof NetworkTunnelStore t) {
                if (!t.requiresTunnel()) {
                    return false;
                }

                if (!abs.shouldTunnel()) {
                    return false;
                }

                if (t.isLocallyTunnelable()) {
                    return true;
                }

                var parent = t.getNetworkParent();
                if (!t.isLocallyTunnelable() && parent.getStore() instanceof NetworkTunnelStore nts) {
                    return nts.isLocallyTunnelable();
                }

                return false;
            }
        }

        return false;
    }

    @Override
    public FailableRunnable<Exception> launch(DataStoreEntry store) {
        return () -> {
            AbstractServiceStore serviceStore = store.getStore().asNeeded();
            serviceStore.startSessionIfNeeded();
            var full = serviceStore.getServiceProtocolType().formatAddress(serviceStore.getOpenTargetUrl());
            serviceStore.getServiceProtocolType().open(full);
        };
    }

    public String displayName(DataStoreEntry entry) {
        AbstractServiceStore s = entry.getStore().asNeeded();
        return DataStorage.get().getStoreEntryDisplayName(s.getHost().get()) + " - Port " + s.getRemotePort();
    }

    @Override
    public List<String> getSearchableTerms(DataStore store) {
        AbstractServiceStore s = store.asNeeded();
        var l = new ArrayList<String>();
        l.add("" + s.getRemotePort());
        if (s.getLocalPort() != null) {
            l.add("" + s.getLocalPort());
        }
        if (s.getAddress() != null) {
            l.add(s.getAddress());
        }
        return l;
    }

    @Override
    public DataStoreUsageCategory getUsageCategory() {
        return DataStoreUsageCategory.TUNNEL;
    }

    @Override
    public DataStoreEntry getSyntheticParent(DataStoreEntry store) {
        AbstractServiceStore s = store.getStore().asNeeded();
        return DataStorage.get()
                .getOrCreateNewSyntheticEntry(
                        s.getHost().get(),
                        "Services",
                        CustomServiceGroupStore.builder().parent(s.getHost()).build());
    }

    @Override
    public ObservableValue<String> informationString(StoreSection section) {
        return Bindings.createStringBinding(
                () -> {
                    AbstractServiceStore s =
                            section.getWrapper().getEntry().getStore().asNeeded();
                    var desc = formatService(s);
                    var type = s.getServiceProtocolType() != null
                                    && !(s.getServiceProtocolType() instanceof ServiceProtocolType.Undefined)
                            ? AppI18n.get(s.getServiceProtocolType().getTranslationKey())
                            : null;
                    var state = !s.requiresTunnel()
                            ? null
                            : s.isSessionRunning()
                                    ? AppI18n.get("active")
                                    : s.isSessionEnabled() ? AppI18n.get("starting") : AppI18n.get("inactive");
                    return new StoreStateFormat(List.of(), desc, type, state).format();
                },
                section.getWrapper().getCache(),
                AppI18n.activeLanguage());
    }

    @Override
    public String getDisplayIconFileName(DataStore store) {
        return "base:service_icon.svg";
    }

    @Override
    public BaseRegionBuilder<?, ?> stateDisplay(StoreSection section) {
        return new SystemStateComp(Bindings.createObjectBinding(
                () -> {
                    if (!section.getWrapper().getEntry().getValidity().isUsable()) {
                        return SystemStateComp.State.OTHER;
                    }

                    AbstractServiceStore s =
                            section.getWrapper().getEntry().getStore().asNeeded();

                    if (!s.requiresTunnel()) {
                        return SystemStateComp.State.SUCCESS;
                    }

                    if (!s.isSessionEnabled() || (s.isSessionEnabled() && !s.isSessionRunning())) {
                        return SystemStateComp.State.OTHER;
                    }

                    return s.isSessionRunning() ? SystemStateComp.State.SUCCESS : SystemStateComp.State.FAILURE;
                },
                section.getWrapper().getCache()));
    }

    protected String formatService(AbstractServiceStore s) {
        var desc = s.getLocalPort() != null
                ? "localhost:" + s.getLocalPort() + " <- :" + s.getRemotePort()
                : s.isSessionRunning()
                        ? "localhost:" + s.getSession().getLocalPort() + " <- :" + s.getRemotePort()
                        : AppI18n.get("servicePort", s.getRemotePort());
        return desc;
    }
}
