using CodexDistTestCore.Marketplace;

namespace CodexDistTestCore
{
    public class CodexNodeContainer
    {
        public CodexNodeContainer(string name, int servicePort, string servicePortName, int apiPort, string containerPortName, int discoveryPort, int listenPort, string dataDir, int metricsPort, GethCompanionNodeContainer? gethCompanionNodeContainer)
        {
            Name = name;
            ServicePort = servicePort;
            ServicePortName = servicePortName;
            ApiPort = apiPort;
            ContainerPortName = containerPortName;
            DiscoveryPort = discoveryPort;
            ListenPort = listenPort;
            DataDir = dataDir;
            MetricsPort = metricsPort;
            GethCompanionNodeContainer = gethCompanionNodeContainer;
        }

        public string Name { get; }
        public int ServicePort { get; }
        public string ServicePortName { get; }
        public int ApiPort { get; }
        public string ContainerPortName { get; }
        public int DiscoveryPort { get; }
        public int ListenPort { get; }
        public string DataDir { get; }
        public int MetricsPort { get; }

        public GethCompanionNodeContainer? GethCompanionNodeContainer { get; }
    }

    public class CodexGroupNumberSource
    {
        private readonly NumberSource codexNodeGroupNumberSource = new NumberSource(0);
        private readonly NumberSource groupContainerNameSource = new NumberSource(1);
        private readonly NumberSource servicePortSource = new NumberSource(30001);

        public int GetNextCodexNodeGroupNumber()
        {
            return codexNodeGroupNumberSource.GetNextNumber();
        }

        public string GetNextServicePortName()
        {
            return $"node{groupContainerNameSource.GetNextNumber()}";
        }

        public int GetNextServicePort()
        {
            return servicePortSource.GetNextNumber();
        }
    }

    public class CodexNodeContainerFactory
    {
        private readonly NumberSource containerNameSource = new NumberSource(1);
        private readonly NumberSource codexPortSource = new NumberSource(8080);
        private readonly CodexGroupNumberSource groupContainerFactory;

        public CodexNodeContainerFactory(CodexGroupNumberSource groupContainerFactory)
        {
            this.groupContainerFactory = groupContainerFactory;
        }

        public CodexNodeContainer CreateNext(OfflineCodexNodes offline)
        {
            var n = containerNameSource.GetNextNumber();
            return new CodexNodeContainer(
                name: $"codex-node{n}",
                servicePort: groupContainerFactory.GetNextServicePort(),
                servicePortName: groupContainerFactory.GetNextServicePortName(),
                apiPort: codexPortSource.GetNextNumber(),
                containerPortName: $"api-{n}",
                discoveryPort: codexPortSource.GetNextNumber(),
                listenPort: codexPortSource.GetNextNumber(),
                dataDir: $"datadir{n}",
                metricsPort: GetMetricsPort(offline),
                CreateGethNodeContainer(offline, n)
            );
        }

        private int GetMetricsPort(OfflineCodexNodes offline)
        {
            if (offline.MetricsEnabled) return codexPortSource.GetNextNumber();
            return 0;
        }

        private GethCompanionNodeContainer? CreateGethNodeContainer(OfflineCodexNodes offline, int n)
        {
            if (offline.MarketplaceConfig == null) return null;

            return new GethCompanionNodeContainer(
                name: $"geth-node{n}",
                servicePort: groupContainerFactory.GetNextServicePort(),
                servicePortName: groupContainerFactory.GetNextServicePortName(),
                apiPort: codexPortSource.GetNextNumber(),
                rpcPort: codexPortSource.GetNextNumber(),
                containerPortName: $"geth-{n}"
            );
        }
    }
}