bitwarden/util/Setup/DockerComposeBuilder.cs

namespace Bit.Setup;

public class DockerComposeBuilder
{
    private readonly Context _context;

    public DockerComposeBuilder(Context context)
    {
        _context = context;
    }

    public void BuildForInstaller()
    {
        _context.Config.DatabaseDockerVolume = _context.HostOS == "mac";
        Build();
    }

    public void BuildForUpdater()
    {
        Build();
    }

    private void Build()
    {
        Directory.CreateDirectory("/bitwarden/docker/");
        Helpers.WriteLine(_context, "Building docker-compose.yml.");
        if (!_context.Config.GenerateComposeConfig)
        {
            Helpers.WriteLine(_context, "...skipped");
            return;
        }

        var template = Helpers.ReadTemplate("DockerCompose");
        var model = new TemplateModel(_context);
        using (var sw = File.CreateText("/bitwarden/docker/docker-compose.yml"))
        {
            sw.Write(template(model));
        }
    }

    public class TemplateModel
    {
        public TemplateModel(Context context)
        {
            EnableBuiltInMsSql = context.Config.EnableBuiltInMsSql;
            MssqlDataDockerVolume = context.Config.DatabaseDockerVolume;
            EnableKeyConnector = context.Config.EnableKeyConnector;
            EnableScim = context.Config.EnableScim;
            HttpPort = context.Config.HttpPort;
            HttpsPort = context.Config.HttpsPort;
            if (!string.IsNullOrWhiteSpace(context.CoreVersion))
            {
                CoreVersion = context.CoreVersion;
            }
            if (!string.IsNullOrWhiteSpace(context.WebVersion))
            {
                WebVersion = context.WebVersion;
            }
            if (!string.IsNullOrWhiteSpace(context.KeyConnectorVersion))
            {
                KeyConnectorVersion = context.KeyConnectorVersion;
            }
        }

        public bool EnableBuiltInMsSql { get; set; }
        public bool MssqlDataDockerVolume { get; set; }
        public bool EnableKeyConnector { get; set; }
        public bool EnableScim { get; set; }
        public string HttpPort { get; set; }
        public string HttpsPort { get; set; }
        public bool HasPort => !string.IsNullOrWhiteSpace(HttpPort) || !string.IsNullOrWhiteSpace(HttpsPort);
        public string CoreVersion { get; set; } = "latest";
        public string WebVersion { get; set; } = "latest";
        public string KeyConnectorVersion { get; set; } = "latest";
    }
}