nyx/modules/flake/lib/helpers.nix

{lib}: let
  inherit (lib.lists) forEach filter;
  inherit (lib.attrsets) filterAttrs mapAttrsToList;
  inherit (lib.filesystem) listFilesRecursive;
  inherit (lib.strings) hasSuffix;

  /*
  *
  filter files for the .nix suffix

  # Arguments

  - [k] they key, which is the file name
  - [v] the value, which is the type of the file

  # Type

  ```
  filterNixFiles :: String -> String -> Bool
  ```

  # Example

  ```nix
  filterNixFiles "default.nix" "regular"
  => true
  ```
  */
  filterNixFiles = k: v: v == "regular" && hasSuffix ".nix" k;

  /*
  *
  Import all file that filterNixFiles allows for

  # Arguments

  - [path] the path to the directory

  # Type

  ```
  importNixFiles :: String -> List
  ```

  # Example

  ```nix
  importNixFiles ./.
  => [ {...} ]
  ```
  */
  importNixFiles = path:
    (forEach (
      mapAttrsToList (name: _: path + ("/" + name)) (filterAttrs filterNixFiles (builtins.readDir path))
    ))
    import;

  /*
  *
  import all nix files and directories

  # Arguments

  - [dir] the directory to search for nix files

  # Type

  ```
  importNixFilesAndDirs :: String -> List
  ```

  # Example

  ```nix
  importNixFilesAndDirs ./.
  => [ "flake.nix" ]
  ```
  */
  importNixFilesAndDirs = dir: filter (f: f != "default.nix") (listFilesRecursive dir);

  /*
  *
  return an int based on boolean value

  # Arguments

  - [bool] the boolean value

  # Type

  ```
  boolToNum :: Bool -> Int
  ```

  # Example

  ```nix
  boolToNum true
  => 1
  ```
  */
  boolToNum = bool:
    if bool
    then 1
    else 0;

  /*
  *
  convert a list of integers to a list of string

  # Arguments

  - [list] the list of integers

  # Type

  ```
  intListToStringList :: List -> List
  ```

  # Example

  ```nix
  intListToStringList [1 2 3]
  => ["1" "2" "3"]
  ```
  */
  intListToStringList = list: map (toString list);

  /*
  *
  a function that returns the index of an element in a list

  # Arguments

  - [list] the list to search in
  - [elem] the element to search for

  # Type

  ```
  indexOf :: List -> Any -> Int
  ```

  # Example

  ```nix
  indexOf [1 2 3] 2
  => 1
  ```
  */
  indexOf = list: elem: let
    f = f: i:
      if i == (builtins.length list)
      then null
      else if (builtins.elemAt list i) == elem
      then i
      else f f (i + 1);
  in
    f f 0;

  /*
  *
  a function that checks if a list contains a list of given strings

  # Arguments

  - [list] the list to search in
  - [targetStrings] the list of strings to search for

  # Type

  ```
  containsStrings :: List -> List -> Bool
  ```

  # Example

  ```nix
  containsStrings ["a" "b" "c"] ["a" "b"]
  => true
  ```
  */
  containsStrings = list: targetStrings: builtins.all (s: builtins.any (x: x == s) list) targetStrings;

  /*
  *
  Create git url aliases for a given domain

  # Arguments

  - [domain] the domain to create the alias for
  - [alias] the alias to use
  - [user] the user to use, this defaults to "git"
  - [port] the port to use, this is optional

  # Type

  ```
  giturl :: (String -> String -> String -> Int) -> AttrSet
  ```

  # Example

  ```nix
  giturl { domain = "github.com"; alias = "gh"; }
  => {
    "https://github.com/".insteadOf = "gh:";
    "ssh://git@github.com/".pushInsteadOf = "gh:";
  }
  ```
  */
  giturl = {
    domain,
    alias,
    user ? "git",
    port ? null,
    ...
  }: {
    "https://${domain}/".insteadOf = "${alias}:";
    "ssh://${user}@${domain}${
      if (builtins.isNull port)
      then ""
      else if (builtins.isInt port)
      then ":" + (builtins.toString port)
      else ":" + port
    }/".pushInsteadOf = "${alias}:";
  };

  /*
  *
  Create a public key for a given host

  # Arguments

  - [host] the host to create the public key for
  - [key] this is a attrset with the key type and key

  # Type

  ```
  mkPub :: (String -> AttrSet -> AttrSet) -> String -> AttrSet -> AttrSet
  ```

  # Example

  ```nix
  mkPub "github.com" {
    type = "rsa";
    key = "AAAAB3NzaC1yc2EAAAABIwAAAQEAq2A7hRGmdnm9tUDbO9IDSwBK6TbQa+PXYPCPy6rbTrTtw7PHkccKrpp0yVhp5HdEIcKr6pLlVDBfOLX9QUsyCOV0wzfjIJNlGEYsdlLJizHhbn2mUjvSAHQqZETYP81eFzLQNnPHt4EVVUh7VfDESU84KezmD5QlWpXLmvU31/yMf+Se8xhHTvKSCZIFImWwoG6mbUoWf9nzpIoaSjB+weqqUUmpaaasXVal72J+UX2B+2RPW3RcT0eOzQgqlJL3RKrTJvdsjE3JEAvGq3lGHSZXy28G3skua2SmVi/w4yCE6gbODqnTWlg7+wC604ydGXA8VJiS5ap43JXiUFFAaQ==";
  }
  => {
    "github.com-rsa" = {
      hostNames = [ "github.com" ];
      publicKey = "ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEAq2A7hRGmdnm9tUDbO9IDSwBK6TbQa+PXYPCPy6rbTrTtw7PHkccKrpp0yVhp5HdEIcKr6pLlVDBfOLX9QUsyCOV0wzfjIJNlGEYsdlLJizHhbn2mUjvSAHQqZETYP81eFzLQNnPHt4EVVUh7VfDESU84KezmD5QlWpXLmvU31/yMf+Se8xhHTvKSCZIFImWwoG6mbUoWf9nzpIoaSjB+weqqUUmpaaasXVal72J+UX2B+2RPW3RcT0eOzQgqlJL3RKrTJvdsjE3JEAvGq3lGHSZXy28G3skua2SmVi/w4yCE6gbODqnTWlg7+wC604ydGXA8VJiS5ap43JXiUFFAaQ==";
    };
  }
  ```
  */
  mkPub = host: key: {
    "${host}-${key.type}" = {
      hostNames = [host];
      publicKey = "ssh-${key.type} ${key.key}";
    };
  };

  /*
  *
  Create public keys for a given host

  # Arguments

  - [host] the host to create the public keys for
  - [keys] the list of keys to create

  # Type

  ```
  mkPubs :: (String -> List) -> String -> List -> AttrSet
  ```

  # Example

  ```nix
  mkPubs "github.com" [
    {
      type = "rsa";
      key = "AAAAB3NzaC1yc2EAAAABIwAAAQEAq2A7hRGmdnm9tUDbO9IDSwBK6TbQa+PXYPCPy6rbTrTtw7PHkccKrpp0yVhp5HdEIcKr6pLlVDBfOLX9QUsyCOV0wzfjIJNlGEYsdlLJizHhbn2mUjvSAHQqZETYP81eFzLQNnPHt4EVVUh7VfDESU84KezmD5QlWpXLmvU31/yMf+Se8xhHTvKSCZIFImWwoG6mbUoWf9nzpIoaSjB+weqqUUmpaaasXVal72J+UX2B+2RPW3RcT0eOzQgqlJL3RKrTJvdsjE3JEAvGq3lGHSZXy28G3skua2SmVi/w4yCE6gbODqnTWlg7+wC604ydGXA8VJiS5ap43JXiUFFAaQ==";
    }
    {
      type = "ed25519";
      key = "AAAAC3NzaC1lZDI1NTE5AAAAIOMqqnkVzrm0SdG6UOoqKLsabgH5C9okWi0dh2l9GKJl";
    }
  ]
  => {
    "github.com-ed25519" = {
      hostNames = [ "github.com" ];
      publicKey = "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIOMqqnkVzrm0SdG6UOoqKLsabgH5C9okWi0dh2l9GKJl";
    };
    "github.com-rsa" = {
      hostNames = [ "github.com" ];
      publicKey = "ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEAq2A7hRGmdnm9tUDbO9IDSwBK6TbQa+PXYPCPy6rbTrTtw7PHkccKrpp0yVhp5HdEIcKr6pLlVDBfOLX9QUsyCOV0wzfjIJNlGEYsdlLJizHhbn2mUjvSAHQqZETYP81eFzLQNnPHt4EVVUh7VfDESU84KezmD5QlWpXLmvU31/yMf+Se8xhHTvKSCZIFImWwoG6mbUoWf9nzpIoaSjB+weqqUUmpaaasXVal72J+UX2B+2RPW3RcT0eOzQgqlJL3RKrTJvdsjE3JEAvGq3lGHSZXy28G3skua2SmVi/w4yCE6gbODqnTWlg7+wC604ydGXA8VJiS5ap43JXiUFFAaQ==";
    };
  }
  ```
  */
  mkPubs = host: keys: lib.foldl' (acc: key: acc // mkPub host key) {} keys;
in {
  inherit
    mkPub
    mkPubs
    giturl
    filterNixFiles
    importNixFiles
    importNixFilesAndDirs
    boolToNum
    containsStrings
    indexOf
    intListToStringList
    ;
}