dotfiles/systems/ceres/config/comfyui.nix

{
  config,
  pkgs,
  flake,
  ...
}:

let
  inherit (flake.config.services.instances) comfyui web;
  service = comfyui;
  localhost = web.localhost.address1;
  host = service.domains.url0;
in
{
  nixpkgs.overlays = [
    flake.inputs.nix-comfyui.overlays.default
  ];

  services = {
    comfyui = {
      enable = true;
      openFirewall = true;
      host = localhost;
      package = pkgs.comfyuiPackages.comfyui.override {
        extensions = with pkgs.comfyuiPackages.extensions; [
          acly-inpaint
          acly-tooling
          kosinkadink-advanced-controlnet
          kosinkadink-animatediff-evolved
          kosinkadink-video-helper-suite
          lev145-images-grid
          ssitu-ultimate-sd-upscale
        ];
        commandLineArgs = [
          "--preview-method"
          "auto"
        ];
      };
    };
    caddy = {
      virtualHosts = {
        "${host}" = {
          extraConfig = ''
            basicauth {
              {$CADDY_AUTH_USER} {$CADDY_AUTH_PASSWORD_HASH}
            }

            reverse_proxy ${localhost}:${toString service.ports.port0}

            tls ${service.ssl.cert} ${service.ssl.key}
          '';
        };
      };
    };
  };

  postInstall = ''
    rm -rf $out/custom_nodes
    ln -s /var/lib/comfyui/custom_nodes $out/custom_nodes
  '';

  systemd.tmpfiles.rules = [
    "Z ${service.sops.path0} 755 caddy caddy -"
    "d ${service.service.paths.path0}/custom_nodes 0755 ${service.name} ${service.name} -"
    "r ${config.services.comfyui.package}/custom_nodes"
    "L+ ${config.services.comfyui.package}/custom_nodes - - - - ${service.service.paths.path0}/custom_nodes"
  ];

  users.users.${service.name}.extraGroups = [
    "users"
    "caddy"
  ];

  networking = {
    firewall = {
      allowedTCPPorts = [
        service.ports.port0
      ];
    };
  };
}