From 04b689f957a0528cd98189168e064f1489237065 Mon Sep 17 00:00:00 2001
From: arunoruto <mirza.arnaut@tu-dortmund.de>
Date: Wed, 6 Nov 2024 00:08:15 +0100
Subject: [PATCH] vivid: add module

---
 modules/modules.nix        |   1 +
 modules/programs/vivid.nix | 140 +++++++++++++++++++++++++++++++++++++
 2 files changed, 141 insertions(+)
 create mode 100644 modules/programs/vivid.nix

diff --git a/modules/modules.nix b/modules/modules.nix
index 75f8ac461..3585ed2af 100644
--- a/modules/modules.nix
+++ b/modules/modules.nix
@@ -248,6 +248,7 @@ let
     ./programs/vifm.nix
     ./programs/vim-vint.nix
     ./programs/vim.nix
+    ./programs/vivid.nix
     ./programs/vscode.nix
     ./programs/vscode/haskell.nix
     ./programs/pywal.nix
diff --git a/modules/programs/vivid.nix b/modules/programs/vivid.nix
new file mode 100644
index 000000000..4294ff019
--- /dev/null
+++ b/modules/programs/vivid.nix
@@ -0,0 +1,140 @@
+{ config, lib, pkgs, ... }:
+let
+  yaml = pkgs.formats.yaml { };
+  bashLine = theme:
+    ''export LS_COLORS="$(${lib.getExe pkgs.vivid} generate ${theme})"'';
+  fishLine = theme:
+    "set -gx LS_COLORS (${lib.getExe pkgs.vivid} generate ${theme})";
+  nushellLine = theme: "${lib.getExe pkgs.vivid} generate ${theme}";
+  zshLine = bashLine;
+in with lib; {
+  meta.maintainers = with maintainers; [ arunoruto ];
+
+  options.programs.vivid = {
+    enable = mkEnableOption ''
+      vivid - A themeable LS_COLORS generator with a rich filetype datebase
+      <link xlink:href="https://github.com/sharkdp/vivid" />
+    '';
+
+    package = mkPackageOption pkgs "vivid" { };
+
+    enableBashIntegration = mkOption {
+      default = true;
+      type = types.bool;
+      description = ''
+        Whether to enable Bash integration.
+        Adds `${bashLine "<theme>"}` to Bash.
+      '';
+    };
+
+    enableFishIntegration = mkOption {
+      default = true;
+      type = types.bool;
+      description = ''
+        Whether to enable Fish integration.
+        Adds `${fishLine "<theme>"}` to Fish.
+      '';
+    };
+
+    enableNushellIntegration = mkOption {
+      default = true;
+      type = types.bool;
+      description = ''
+        Whether to enable Nushell integration.
+        Adds `${nushellLine "<theme>"}` to Nushell.
+      '';
+    };
+
+    enableZshIntegration = mkOption {
+      default = true;
+      type = types.bool;
+      description = ''
+        Whether to enable Zsh integration.
+        Adds `${zshLine "<theme>"}` to Zsh.
+      '';
+    };
+
+    theme = mkOption {
+      type = with types; nullOr str;
+      default = null;
+      example = "molokai";
+      description = ''
+        Color theme to enable.
+        Run `vivid themes` for a list of available themes.
+      '';
+    };
+
+    filetypes = mkOption {
+      type = yaml.type;
+      default = { };
+      example = literalExpression ''
+        {
+          core = {
+            regular_file = [ "$fi" ];
+            directory = [ "$di" ];
+          };
+          text = {
+            readme = [ "README.md" ];
+            licenses = [ "LICENSE" ];
+          };
+        }
+      '';
+      description = ''
+        Configuration written to
+        <filename>~/.config/vivid/filetypes.yml</filename>.
+        Visit <link xlink:href="https://github.com/sharkdp/vivid/tree/master/config/filetypes.yml" />
+        for a reference file.
+      '';
+    };
+
+    themes = mkOption {
+      type = types.attrsOf (yaml.type);
+      default = { };
+      example = literalExpression ''
+        {
+          mytheme = {
+            colors = {
+              blue = "0000ff";
+            };
+            core = {
+              directory = {
+                foreground = "blue";
+                font-style = "bold";
+              };
+            };
+          };
+        }
+      '';
+      description = ''
+        Theme files written to
+        <filename>~/.config/vivid/themes/<mytheme>.yml</filename>.
+        Visit <link xlink:href="https://github.com/sharkdp/vivid/tree/master/themes" />
+        for references.
+      '';
+    };
+  };
+
+  config = let cfg = config.programs.vivid;
+  in mkIf cfg.enable {
+    home.packages = [ cfg.package ];
+
+    programs.bash.initExtra =
+      mkIf cfg.enableBashIntegration (bashLine cfg.theme);
+    programs.fish.interactiveShellInit =
+      mkIf cfg.enableFishIntegration (fishLine cfg.theme);
+    programs.nushell.environmentVariables.LS_COLORS =
+      mkIf cfg.enableNushellIntegration
+      (hm.nushell.mkNushellInline (nushellLine cfg.theme));
+    programs.zsh.initExtra = mkIf cfg.enableZshIntegration (zshLine cfg.theme);
+
+    xdg.configFile = {
+      "vivid/filetypes.yml" =
+        mkIf (builtins.length (builtins.attrNames cfg.filetypes) > 0) {
+          source = yaml.generate "filetypes.yml" cfg.filetypes;
+        };
+    } // mapAttrs' (name: value:
+      nameValuePair "vivid/themes/${name}.yml" {
+        source = yaml.generate "${name}.yml" value;
+      }) cfg.themes;
+  };
+}