nixos/modules/services/vaultwarden.nix

inputs:
{
  options.nixos.services = let inherit (inputs.lib) mkOption types; in
  {
    vaultwarden =
    {
      enable = mkOption { type = types.bool; default = false; };
      autoStart = mkOption { type = types.bool; default = true; };
      port = mkOption { type = types.ints.unsigned; default = 8000; };
      websocketPort = mkOption { type = types.ints.unsigned; default = 3012; };
      hostname = mkOption { type = types.str; default = "vaultwarden.chn.moe"; };
    };
    vaultwarden-proxy =
    {
      enable = mkOption { type = types.bool; default = false; };
      hostname = mkOption { type = types.nonEmptyStr; default = "vaultwarden.chn.moe"; };
      upstream = mkOption
      {
        type = types.oneOf [ types.nonEmptyStr (types.submodule { options =
        {
          address = mkOption { type = types.nonEmptyStr; default = "127.0.0.1"; };
          port = mkOption { type = types.ints.unsigned; default = 8000; };
          websocketPort = mkOption { type = types.ints.unsigned; default = 3012; };
        };})];
        default = {};
      };
    };
  };
  config =
    let
      inherit (inputs.config.nixos.services) vaultwarden vaultwarden-proxy;
      inherit (builtins) listToAttrs;
      inherit (inputs.lib) mkIf mkMerge;
    in mkMerge
    [
      (
        mkIf vaultwarden.enable
        {
          services.vaultwarden =
          {
            enable = true;
            dbBackend = "postgresql";
            config =
            {
              DATA_FOLDER = "/var/lib/vaultwarden";
              WEB_VAULT_ENABLED = true;
              WEBSOCKET_ENABLED = true;
              ROCKET_PORT = vaultwarden.port;
              WEBSOCKET_PORT = toString vaultwarden.websocketPort;
              SIGNUPS_VERIFY = true;
              DOMAIN = "https://${vaultwarden.hostname}";
              SMTP_HOST = "mail.chn.moe";
              SMTP_FROM = "bot@chn.moe";
              SMTP_FROM_NAME = "vaultwarden";
              SMTP_SECURITY = "force_tls";
              SMTP_USERNAME = "bot@chn.moe";
            };
            environmentFile = inputs.config.sops.templates."vaultwarden.env".path;
          };
          sops =
          {
            templates."vaultwarden.env" =
              let
                serviceConfig = inputs.config.systemd.services.vaultwarden.serviceConfig;
                placeholder = inputs.config.sops.placeholder;
              in
              {
                owner = serviceConfig.User;
                group = serviceConfig.Group;
                content =
                ''
                  DATABASE_URL=postgresql://vaultwarden:${placeholder."postgresql/vaultwarden"}@localhost/vaultwarden
                  ADMIN_TOKEN=${placeholder."vaultwarden/admin_token"}
                  SMTP_PASSWORD=${placeholder."mail/bot"}
                '';
              };
            secrets = listToAttrs (map
              (secret: { name = secret; value = {}; })
              [ "vaultwarden/admin_token" "mail/bot" ]);
          };
          systemd.services.vaultwarden =
          {
            enable = vaultwarden.autoStart;
            after = [ "postgresql.service" ];
          };
          nixos.services.postgresql = { enable = true; instances.vaultwarden = {}; };
        }
      )
      (
        mkIf vaultwarden-proxy.enable
        {
          nixos.services.nginx =
          {
            enable = true;
            httpProxy."${vaultwarden-proxy.hostname}" =
            {
              rewriteHttps = true;
              locations = let upstream = vaultwarden-proxy.upstream; in (listToAttrs (map
                (location: { name = location; value =
                {
                  upstream = "http://${upstream.address or upstream}:${builtins.toString upstream.port or 8000}";
                  setHeaders = { Host = vaultwarden-proxy.hostname; Connection = ""; };
                };})
                [ "/" "/notifications/hub/negotiate" ]))
                // { "/notifications/hub" =
                {
                  upstream =
                    "http://${upstream.address or upstream}:${builtins.toString upstream.websocketPort or 3012}";
                  websocket = true;
                  setHeaders.Host = vaultwarden-proxy.hostname;
                };};
            };
          };
        }
      )
    ];
}