sp-config/auth/auth-module.nix

{ config, lib, pkgs, ... }:
let
  inherit (lib)
    mkOption
    types
    ;
  auth-passthru = config.selfprivacy.passthru.auth;
  keys-path = auth-passthru.keys-path;
  # generate OAuth2 client secret
  mkKanidmExecStartPreScript = oauthClientID: linuxGroup:
    let
      secretFP = auth-passthru.mkOAuth2ClientSecretFP linuxGroup;
    in
    pkgs.writeShellScript
      "${oauthClientID}-kanidm-ExecStartPre-script.sh" ''
      set -o pipefail
      set -o errexit
      if ! [ -f "${secretFP}" ]
      then
        "${lib.getExe pkgs.openssl}" rand -base64 32 \
        | tr "\n:@/+=" "012345" > "${secretFP}"
        chmod 640 "${secretFP}"
      fi
    '';
  mkKanidmExecStartPostScript = oauthClientID: linuxGroup: isMailserver:
    let
      kanidmServiceAccountName = "sp.${oauthClientID}.service-account";
      kanidmServiceAccountTokenName = "${oauthClientID}-service-account-token";
      kanidmServiceAccountTokenFP =
        auth-passthru.mkServiceAccountTokenFP linuxGroup;
      isRW = oauthClientID == "selfprivacy-api";
    in
    pkgs.writeShellScript
      "${oauthClientID}-kanidm-ExecStartPost-script.sh"
      (''
        export HOME=$RUNTIME_DIRECTORY/client_home
        readonly KANIDM="${pkgs.kanidm}/bin/kanidm"

        # try to get existing Kanidm service account
        KANIDM_SERVICE_ACCOUNT="$($KANIDM service-account list --name idm_admin | grep -E "^name: ${kanidmServiceAccountName}$")"
        echo KANIDM_SERVICE_ACCOUNT: "$KANIDM_SERVICE_ACCOUNT"
        if [ -n "$KANIDM_SERVICE_ACCOUNT" ]
        then
            echo "kanidm service account \"${kanidmServiceAccountName}\" is found"
        else
            echo "kanidm service account \"${kanidmServiceAccountName}\" is not found"
            echo "creating new kanidm service account \"${kanidmServiceAccountName}\""
            if $KANIDM service-account create --name idm_admin "${kanidmServiceAccountName}" "${kanidmServiceAccountName}" idm_admin
            then
                echo "kanidm service account \"${kanidmServiceAccountName}\" created"
            else
                echo "error: cannot create kanidm service account \"${kanidmServiceAccountName}\""
                exit 1
            fi
        fi

        # create a new token for kanidm
        if ! KANIDM_SERVICE_ACCOUNT_TOKEN_JSON="$($KANIDM service-account api-token generate --name idm_admin "${kanidmServiceAccountName}" "${kanidmServiceAccountTokenName}" ${lib.strings.optionalString isRW "--rw"} --output json)"
        then
            echo "error: kanidm CLI returns an error when trying to generate service-account api-token"
            exit 1
        fi
        if ! KANIDM_SERVICE_ACCOUNT_TOKEN="$(echo "$KANIDM_SERVICE_ACCOUNT_TOKEN_JSON" | ${lib.getExe pkgs.jq} -r .result)"
        then
            echo "error: cannot get service-account API token from JSON"
            exit 1
        fi

        if ! install --mode=640 \
        <(printf "%s" "$KANIDM_SERVICE_ACCOUNT_TOKEN") \
        ${kanidmServiceAccountTokenFP}
        then
            echo "error: cannot write token to \"${kanidmServiceAccountTokenFP}\""
            exit 1
        fi

      ''
      + lib.strings.optionalString isMailserver ''
        # add Kanidm service account to `idm_mail_servers` group
        $KANIDM group add-members idm_mail_servers "${kanidmServiceAccountName}"
      ''
      + lib.strings.optionalString isRW ''
        $KANIDM group add-members idm_admins "${kanidmServiceAccountName}"
      ''
      );
in
{
  options.selfprivacy.auth = {
    clients = mkOption {
      description =
        "Configurations for OAuth2 & LDAP servers clients services. Corresponding Kanidm provisioning configuration and systemd scripts are generated.";
      default = { };
      type = types.attrsOf (
        types.submodule {
          options = {
            clientID = mkOption {
              type = types.nullOr types.str;
              description = ''
                Name of this client service. Used as OAuth2 client ID and to form Kanidm sp.$\{clientID}.* group names. Defaults to attribute name in virtualHosts;
              '';
              default = null;
            };
            displayName = mkOption {
              type = types.nullOr types.str;
              description = "Display name showed in Kanidm Web GUI. Defaults to clientID.";
              default = null;
            };
            enablePkce = mkOption {
              type = lib.types.bool;
              description =
                "Whether PKCE must be used between client and Kanidm.";
              default = false;
            };
            adminsGroup = mkOption {
              type =
                types.nullOr (lib.types.strMatching "sp\.[A-Za-z0-9]+\.admins");
              description =
                "Name of admins group in Kanidm, whose members have admin level access to resources (service) associated with OAuth2 client authorization.";
              default = null;
            };
            usersGroup = mkOption {
              type =
                types.nullOr (lib.types.strMatching "sp\.[A-Za-z0-9]+\.users");
              description =
                "Name of users group in Kanidm, whose members have user level access to resources (service) associated with OAuth2 client authorization.";
              default = null;
            };
            originLanding = mkOption {
              type = types.nullOr lib.types.str;
              description =
                "The origin landing of the service for OAuth2 redirects.";
            };
            originUrl = mkOption {
              type = types.nullOr lib.types.str;
              description =
                "The origin URL of the service for OAuth2 redirects.";
            };
            subdomain = lib.mkOption {
              type =
                lib.types.strMatching "[A-Za-z0-9][A-Za-z0-9\-]{0,61}[A-Za-z0-9]";
              description = "Subdomain of the service.";
            };
            # when true, "name" is passed to a service instead of "name@domain"
            useShortPreferredUsername = mkOption {
              description =
                "Use 'name' instead of 'spn' in the preferred_username claim.";
              type = types.bool;
              default = true;
            };
            linuxUserOfClient = mkOption {
              type = types.nullOr lib.types.str;
              description =
                "Name of a Linux OAuth2 client user, under which it should get access through a folder with keys.";
              default = null;
            };
            linuxGroupOfClient = mkOption {
              type = types.nullOr lib.types.str;
              description =
                "Name of Linux OAuth2 client group, under which it should read an OAuth2 client secret file.";
              default = null;
            };
            isTokenNeeded = mkOption {
              description =
                "Whether a read-only needs to be generated for LDAP access.";
              type = types.bool;
              default = false;
            };
            clientSystemdUnits = mkOption {
              description = "A list of systemd services, which depend on OAuth service";
              # taken from nixos/lib/systemd-lib.nix: unitNameType
              type = types.listOf
                (types.strMatching "[a-zA-Z0-9@%:_.\\-]+[.](service|socket|device|mount|automount|swap|target|path|timer|scope|slice)");
            };
            scopeMaps = mkOption {
              description = ''
                Maps kanidm groups to returned oauth scopes.
                See [Scope Relations](https://kanidm.github.io/kanidm/stable/integrations/oauth2.html#scope-relationships) for more information.
              '';
              type = types.nullOr (types.attrsOf (types.listOf types.str));
              default = null;
            };
            claimMaps = mkOption {
              description = ''
                Adds additional claims (and values) based on which kanidm groups an authenticating party belongs to.
                See [Claim Maps](https://kanidm.github.io/kanidm/master/integrations/oauth2.html#custom-claim-maps) for more information.
              '';
              default = { };
              type = types.attrsOf (
                types.submodule {
                  options = {
                    joinType = mkOption {
                      description = ''
                        Determines how multiple values are joined to create the claim value.
                        See [Claim Maps](https://kanidm.github.io/kanidm/master/integrations/oauth2.html#custom-claim-maps) for more information.
                      '';
                      type = types.enum [
                        "array"
                        "csv"
                        "ssv"
                      ];
                      default = "array";
                    };

                    valuesByGroup = mkOption {
                      description = "Maps kanidm groups to values for the claim.";
                      default = { };
                      type = types.attrsOf (types.listOf types.str);
                    };
                  };
                }
              );
            };
            imageFile = mkOption {
              type = types.nullOr lib.types.path;
              description = ''
                Filepath of an image which is displayed in Kanidm web GUI for a service.
              '';
              default = null;
            };
            isMailserver = mkOption {
              type = types.bool;
              description = ''
                Whether client is a mailserver.
              '';
              default = false;
            };
          };
        }
      );
    };
  };
  config = lib.mkIf config.selfprivacy.sso.enable (
    let
      clientsAttrsList = lib.attrsets.mapAttrsToList
        (name: attrs: attrs // rec {
          clientID =
            if attrs.clientID == null
            then name
            else attrs.clientID;
          displayName =
            if attrs.displayName == null
            then clientID
            else attrs.displayName;
          adminsGroup =
            if attrs.adminsGroup == null
            then "sp.${clientID}.admins"
            else attrs.adminsGroup;
          usersGroup =
            if attrs.usersGroup == null
            then "sp.${clientID}.users"
            else attrs.usersGroup;
          basicSecretFile =
            "${keys-path}/${linuxGroupOfClient}/kanidm-oauth-client-secret";
          linuxUserOfClient =
            if attrs.linuxUserOfClient == null
            then clientID
            else attrs.linuxUserOfClient;
          linuxGroupOfClient =
            if attrs.linuxGroupOfClient == null
            then clientID
            else attrs.linuxGroupOfClient;
          originLanding =
            if attrs.originLanding == null
            then "https://${attrs.subdomain}.${config.selfprivacy.domain}/"
            else attrs.originLanding;
          scopeMaps =
            if attrs.scopeMaps == null
            then { "${usersGroup}" = [ "email" "openid" "profile" ]; }
            else attrs.scopeMaps;
        })
        config.selfprivacy.auth.clients;
    in
    {
      # for each OAuth2 client: member of the `keys` group for directory access
      users.groups.keys.members = lib.mkMerge (lib.forEach
        clientsAttrsList
        ({ linuxUserOfClient, ... }: [ linuxUserOfClient ])
      );

      systemd.tmpfiles.settings."kanidm-secrets" = lib.mkMerge (lib.forEach
        clientsAttrsList
        ({ linuxGroupOfClient, ... }: {
          "${keys-path}/${linuxGroupOfClient}".d = {
            user = "kanidm";
            group = linuxGroupOfClient;
            mode = "2750";
          };
        })
      );

      # for each OAuth2 client: scripts with Kanidm CLI commands
      systemd.services.kanidm = {
        before =
          lib.lists.concatMap
            ({ clientSystemdUnits, ... }: clientSystemdUnits)
            clientsAttrsList;
        serviceConfig =
          lib.mkMerge (lib.forEach
            clientsAttrsList
            ({ clientID, isTokenNeeded, linuxGroupOfClient, isMailserver, ... }:
              {
                ExecStartPre = [
                  # "-" prefix means to ignore exit code of prefixed script
                  ("-" + mkKanidmExecStartPreScript clientID linuxGroupOfClient)
                ];
                ExecStartPost = lib.mkIf isTokenNeeded
                  (lib.mkAfter [
                    ("-" +
                      mkKanidmExecStartPostScript
                        clientID
                        linuxGroupOfClient
                        isMailserver)
                  ]);
              }));
      };

      # for each OAuth2 client: Kanidm provisioning options
      services.kanidm.provision = lib.mkMerge (lib.forEach
        clientsAttrsList
        ({ adminsGroup
         , basicSecretFile
         , claimMaps
         , clientID
         , displayName
         , enablePkce
         , imageFile
         , originLanding
         , originUrl
         , scopeMaps
         , useShortPreferredUsername
         , usersGroup
         , ...
         }: {
          groups = lib.mkIf (clientID != "selfprivacy-api") {
            "${adminsGroup}".members =
              [ auth-passthru.admins-group ];
            "${usersGroup}".members =
              [ adminsGroup auth-passthru.full-users-group ];
          };
          systems.oauth2.${clientID} = {
            inherit
              basicSecretFile
              claimMaps
              displayName
              imageFile
              originLanding
              originUrl
              scopeMaps
              ;
            preferShortUsername = useShortPreferredUsername;
            allowInsecureClientDisablePkce = ! enablePkce;
            removeOrphanedClaimMaps = true;

            # NOTE https://github.com/oddlama/kanidm-provision/issues/15
            # add more scopes when a user is a member of specific group
            # currently not possible due to https://github.com/kanidm/kanidm/issues/2882#issuecomment-2564490144
            # supplementaryScopeMaps."${admins-group}" =
            #   [ "read:admin" "write:admin" ];
          };
        }));
    }
  );
}