nix-config/modules/nixos/services/aiostreams.nix

{
  flake.modules.nixos.services = {
    config,
    lib,
    ...
  }:
    with lib; let
      cfg = config.tnix.services.aiostreams;
      port = toString cfg.port;
      acmeHost = config.tnix.services.nginx.domain;
    in {
      options.tnix.services.aiostreams = {
        enable = mkEnableOption "AIOStreams";

        host = mkOption {
          type = types.str;
          default = "127.0.0.1";
          description = "Host on which AIOStreams listens";
        };

        port = mkOption {
          type = types.port;
          default = 3000;
          description = "Port on which AIOStreams listens";
        };

        domain = mkOption {
          type = types.str;
          default = "";
          description = "Domain on which AIOStreams is available";
        };

        configureNginx = mkOption {
          type = types.bool;
          default = false;
          description = "Whether to configure Nginx as a reverse proxy for AIOStreams";
        };

        configurePangolin = mkOption {
          type = types.bool;
          default = false;
          description = "Whether to configure Pangolin as a reverse proxy for AIOStreams";
        };

        image = mkOption {
          type = types.str;
          default = "ghcr.io/viren070/aiostreams:latest";
          description = "Container image to use";
        };

        dataDir = mkOption {
          type = types.path;
          default = "/var/lib/docker/volumes/aiostreams/_data";
          description = "Directory to store persistent AIOStreams data";
        };

        environmentFile = mkOption {
          type = types.nullOr types.path;
          default = null;
          description = "Environment file with secrets passed to the container";
        };
      };

      config = mkIf cfg.enable {
        assertions = [
          {
            assertion = cfg.domain != "";
            message = "tnix.services.aiostreams.domain must be set when tnix.services.aiostreams.enable is true.";
          }
        ];

        virtualisation.oci-containers.containers.aiostreams = {
          image = cfg.image;
          ports = [
            "${cfg.host}:${port}:3000"
          ];
          environment = {
            ADDON_ID = cfg.domain;
            BASE_URL = "https://${cfg.domain}";
          };
          environmentFiles = optional (cfg.environmentFile != null) cfg.environmentFile;
          volumes = [
            "${cfg.dataDir}:/app/data"
          ];
        };

        services = {
          nginx.virtualHosts.${cfg.domain} = mkIf cfg.configureNginx {
            forceSSL = acmeHost != "";
            useACMEHost = mkIf (acmeHost != "") acmeHost;
            locations."/" = {
              proxyPass = "http://${cfg.host}:${port}";
              proxyWebsockets = true;
            };
          };

          newt.blueprint.proxy-resources = mkIf cfg.configurePangolin {
            aiostreams = {
              auth = {
                sso-enabled = false;
              };
              full-domain = cfg.domain;
              name = "aiostreams";
              protocol = "http";
              targets = [
                {
                  hostname = "localhost";
                  method = "http";
                  port = cfg.port;
                  healthcheck = {
                    hostname = "localhost";
                    port = cfg.port;
                    scheme = "http";
                    method = "GET";
                    path = "/";
                  };
                }
              ];
            };
          };
        };
      };
    };
}