cs-codex-dist-tests/Framework/Core/PluginTools.cs

using FileUtils;
using KubernetesWorkflow;
using Logging;

namespace Core
{
    public interface IPluginTools : IWorkflowTool, ILogTool, IHttpFactoryTool, IFileTool
    {
        ITimeSet TimeSet { get; }

        /// <summary>
        /// Deletes kubernetes and tracked file resources.
        /// when `waitTillDone` is true, this function will block until resources are deleted.
        /// </summary>
        void Decommission(bool deleteKubernetesResources, bool deleteTrackedFiles, bool waitTillDone);
    }

    public interface IWorkflowTool
    {
        IStartupWorkflow CreateWorkflow(string? namespaceOverride = null);
    }

    public interface ILogTool
    {
        ILog GetLog();
    }

    public interface IHttpFactoryTool
    {
        IHttp CreateHttp(string id, Action<HttpClient> onClientCreated);
        IHttp CreateHttp(string id, Action<HttpClient> onClientCreated, ITimeSet timeSet);
        IHttp CreateHttp(string id);
    }

    public interface IFileTool
    {
        IFileManager GetFileManager();
    }

    internal class PluginTools : IPluginTools
    {
        private readonly WorkflowCreator workflowCreator;
        private readonly IFileManager fileManager;
        private readonly LogPrefixer log;

        internal PluginTools(ILog log, WorkflowCreator workflowCreator, string fileManagerRootFolder, ITimeSet timeSet)
        {
            this.log = new LogPrefixer(log);
            this.workflowCreator = workflowCreator;
            TimeSet = timeSet;
            fileManager = new FileManager(log, fileManagerRootFolder);
        }

        public ITimeSet TimeSet { get; }

        public void ApplyLogPrefix(string prefix)
        {
            log.Prefix = prefix;
        }

        public IHttp CreateHttp(string id, Action<HttpClient> onClientCreated)
        {
            return CreateHttp(id, onClientCreated, TimeSet);
        }

        public IHttp CreateHttp(string id, Action<HttpClient> onClientCreated, ITimeSet ts)
        {
            return new Http(id, log, ts, onClientCreated);
        }

        public IHttp CreateHttp(string id)
        {
            return new Http(id, log, TimeSet);
        }

        public IStartupWorkflow CreateWorkflow(string? namespaceOverride = null)
        {
            return workflowCreator.CreateWorkflow(namespaceOverride);
        }

        public void Decommission(bool deleteKubernetesResources, bool deleteTrackedFiles, bool waitTillDone)
        {
            if (deleteKubernetesResources) CreateWorkflow().DeleteNamespace(waitTillDone);
            if (deleteTrackedFiles) fileManager.DeleteAllFiles();
        }

        public IFileManager GetFileManager()
        {
            return fileManager;
        }

        public ILog GetLog()
        {
            return log;
        }
    }
}