arion/nixos-module.nix

{ config, lib, pkgs, ... }:
let
  inherit (lib)
    any
    attrValues
    mkIf
    mkOption
    mkEnableOption
    mkMerge
    types
    ;

  cfg = config.virtualisation.arion;

  projectType = types.submoduleWith {
    modules = [ projectModule ];
  };

  projectModule = { config, name, ... }: {
    options = {
      settings = mkOption {
        description = ''
          Arion project definition, otherwise known as arion-compose.nix contents.

          See <link xlink:href="https://docs.hercules-ci.com/arion/options/">https://docs.hercules-ci.com/arion/options/</link>.
        '';
        type = arionSettingsType name;
        visible = "shallow";
      };
      rootless = mkEnableOption "Run this project in rootless mode";
      _systemd = mkOption { internal = true; };
    };
    config =
      let

        service = {
          wantedBy = [ "multi-user.target" ];
          after = [ "sockets.target" ];

          path = [
            cfg.package
            cfg.docker.client.package
          ];
          environment.ARION_PREBUILT = config.settings.out.dockerComposeYaml;
          environment.DOCKER_HOST = mkIf config.rootless "unix:///run/user/1000/docker.sock"; # TODO: Do not hardcode path
          script = ''
            echo 1>&2 "docker compose file: $ARION_PREBUILT"
            arion --prebuilt-file "$ARION_PREBUILT" up
          '';
        };

    in
      if false then
      # if false then
        { _systemd.user.services."arion-${name}" = service; }
      else
        { _systemd.services."arion-${name}" = service; };
  };

  arionSettingsType = name:
    (cfg.package.eval { modules = [ { project.name = lib.mkDefault name; } ]; }).type or (
      throw "lib.evalModules did not produce a type. Please upgrade Nixpkgs to nixos-unstable or >=nixos-21.11"
    );

in
{
  disabledModules = [ "virtualisation/arion.nix" ];

  options = {
    virtualisation.arion = {
      backend = mkOption {
        type = types.enum [ "podman-socket" "docker" ];
        description = ''
          Which container implementation to use.
        '';
      };
      package = mkOption {
        type = types.package;

        default = (import ./. { inherit pkgs; }).arion;
        description = ''
          Arion package to use. This will provide <literal>arion</literal>
          executable that starts the project.

          It also must provide the arion <literal>eval</literal> function as
          an attribute.
        '';
      };
      docker.client.package = mkOption {
        type = types.package;
        internal = true;
      };
      projects = mkOption {
        type = types.attrsOf projectType;
        default = { };
        description = ''
          Arion projects to be run as a service.
        '';
      };
    };
  };

  config = mkIf (cfg.projects != { }) (
    mkMerge [
      {
        systemd = mkMerge (map (p: p._systemd) (attrValues cfg.projects));
      }
      (mkIf (cfg.backend == "podman-socket") {
        virtualisation.docker.enable = false;
        virtualisation.podman.enable = true;
        virtualisation.podman.dockerSocket.enable = true;
        virtualisation.podman.defaultNetwork.dnsname.enable = true;

        virtualisation.arion.docker.client.package = pkgs.docker-client;
      })
      (mkIf (cfg.backend == "docker") {
        virtualisation.docker.enable = true;
        virtualisation.arion.docker.client.package = pkgs.docker;
      })
      (mkIf (any (project: project.rootless) (attrValues cfg.projects)) {
        virtualisation.docker.rootless.enable = true;
      })
    ]
  );
}