From acb917fbf2bc9ce9c556516d8a1f257709b3cf1e Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Thu, 28 Mar 2024 08:24:29 +0100
Subject: [PATCH] plugins/zen-mode: init

---
 plugins/default.nix                        |   1 +
 plugins/ui/zen-mode.nix                    | 145 +++++++++++++++++++++
 tests/test-sources/plugins/ui/zen-mode.nix |  89 +++++++++++++
 3 files changed, 235 insertions(+)
 create mode 100644 plugins/ui/zen-mode.nix
 create mode 100644 tests/test-sources/plugins/ui/zen-mode.nix

diff --git a/plugins/default.nix b/plugins/default.nix
index 9673e54c..b86ab37f 100644
--- a/plugins/default.nix
+++ b/plugins/default.nix
@@ -115,6 +115,7 @@
     ./ui/transparent.nix
     ./ui/twilight.nix
     ./ui/virt-column.nix
+    ./ui/zen-mode.nix
 
     ./utils/alpha.nix
     ./utils/auto-save.nix
diff --git a/plugins/ui/zen-mode.nix b/plugins/ui/zen-mode.nix
new file mode 100644
index 00000000..0e941ae1
--- /dev/null
+++ b/plugins/ui/zen-mode.nix
@@ -0,0 +1,145 @@
+{
+  lib,
+  helpers,
+  config,
+  pkgs,
+  ...
+}:
+with lib;
+  helpers.neovim-plugin.mkNeovimPlugin config {
+    name = "zen-mode";
+    originalName = "zen-mode.nvim";
+    defaultPackage = pkgs.vimPlugins.zen-mode-nvim;
+
+    maintainers = [maintainers.GaetanLepage];
+
+    # Optionally, explicitly declare some options. You don't have to.
+    settingsOptions = {
+      window = {
+        backdrop = helpers.defaultNullOpts.mkNullable (types.numbers.between 0.0 1.0) "0.95" ''
+          Shade the backdrop of the Zen window.
+          Set to 1 to keep the same as Normal.
+        '';
+
+        width =
+          helpers.defaultNullOpts.mkNullable
+          (
+            with helpers.nixvimTypes;
+              oneOf [
+                ints.positive
+                (numbers.between 0.0 1.0)
+                rawLua
+              ]
+          )
+          "120"
+          ''
+            Width of the zen window.
+
+            Can be:
+            - an absolute number of cells when > 1
+            - a percentage of the width / height of the editor when <= 1
+            - a function that returns the width or the height
+          '';
+
+        height =
+          helpers.defaultNullOpts.mkNullable
+          (
+            with helpers.nixvimTypes;
+              oneOf [
+                ints.positive
+                (numbers.between 0.0 1.0)
+                rawLua
+              ]
+          )
+          "1"
+          ''
+            Height of the Zen window.
+
+            Can be:
+            - an absolute number of cells when > 1
+            - a percentage of the width / height of the editor when <= 1
+            - a function that returns the width or the height
+          '';
+
+        options =
+          helpers.defaultNullOpts.mkAttrsOf types.anything "{}"
+          ''
+            By default, no options are changed for the Zen window.
+            You can set any `vim.wo` option here.
+
+            Example:
+            ```nix
+              {
+                signcolumn = "no";
+                number = false;
+                relativenumber = false;
+                cursorline = false;
+                cursorcolumn = false;
+                foldcolumn = "0";
+                list = false;
+              }
+            ```
+          '';
+      };
+      plugins = {
+        options =
+          helpers.defaultNullOpts.mkAttrsOf types.anything
+          ''
+            {
+              enabled = true;
+              ruler = false;
+              showcmd = false;
+              laststatus = 0;
+            }
+          ''
+          ''
+            Disable some global vim options (`vim.o`...).
+          '';
+      };
+
+      on_open = helpers.defaultNullOpts.mkLuaFn "function(win) end" ''
+        Callback where you can add custom code when the Zen window opens.
+      '';
+
+      on_close = helpers.defaultNullOpts.mkLuaFn "function(win) end" ''
+        Callback where you can add custom code when the Zen window closes.
+      '';
+    };
+
+    settingsExample = {
+      window = {
+        backdrop = 0.95;
+        width = 0.8;
+        height = 1;
+        options.signcolumn = "no";
+      };
+      plugins = {
+        options = {
+          enabled = true;
+          ruler = false;
+          showcmd = false;
+        };
+        twilight.enabled = false;
+        gitsigns.enabled = true;
+        tmux.enabled = false;
+      };
+      on_open = ''
+        function()
+          require("gitsigns.actions").toggle_current_line_blame()
+          vim.cmd('IBLDisable')
+          vim.opt.relativenumber = false
+          vim.opt.signcolumn = "no"
+          require("gitsigns.actions").refresh()
+        end
+      '';
+      on_close = ''
+        function()
+          require("gitsigns.actions").toggle_current_line_blame()
+          vim.cmd('IBLEnable')
+          vim.opt.relativenumber = true
+          vim.opt.signcolumn = "yes:2"
+          require("gitsigns.actions").refresh()
+        end
+      '';
+    };
+  }
diff --git a/tests/test-sources/plugins/ui/zen-mode.nix b/tests/test-sources/plugins/ui/zen-mode.nix
new file mode 100644
index 00000000..fbbc0495
--- /dev/null
+++ b/tests/test-sources/plugins/ui/zen-mode.nix
@@ -0,0 +1,89 @@
+{
+  empty = {
+    plugins.zen-mode.enable = true;
+  };
+
+  defaults = {
+    plugins.zen-mode = {
+      enable = true;
+
+      settings = {
+        window = {
+          backdrop = 0.95;
+          width = 120;
+          height = 1;
+          options = {};
+        };
+        plugins = {
+          options = {
+            enabled = true;
+            ruler = false;
+            showcmd = false;
+            laststatus = 0;
+          };
+
+          twilight.enabled = true;
+          gitsigns.enabled = false;
+          tmux.enabled = false;
+          kitty = {
+            enabled = false;
+            font = "+4";
+          };
+          alacritty = {
+            enabled = false;
+            font = "14";
+          };
+          wezterm = {
+            enabled = false;
+            font = "+4";
+          };
+        };
+        on_open = "function(win) end";
+        on_close = "function(win) end";
+      };
+    };
+  };
+
+  example = {
+    plugins.zen-mode = {
+      enable = true;
+
+      settings = {
+        window = {
+          backdrop = 0.95;
+          width = 0.8;
+          height = 1;
+          options.signcolumn = "no";
+        };
+        plugins = {
+          options = {
+            enabled = true;
+            ruler = false;
+            showcmd = false;
+          };
+          twilight.enabled = false;
+          gitsigns.enabled = true;
+          tmux.enabled = false;
+        };
+        on_open = ''
+          function()
+            require("gitsigns.actions").toggle_current_line_blame()
+            vim.cmd('IBLDisable')
+            vim.opt.relativenumber = false
+            vim.opt.signcolumn = "no"
+            require("gitsigns.actions").refresh()
+          end
+        '';
+        on_close = ''
+          function()
+            require("gitsigns.actions").toggle_current_line_blame()
+            vim.cmd('IBLEnable')
+            vim.opt.relativenumber = true
+            vim.opt.signcolumn = "yes:2"
+            require("gitsigns.actions").refresh()
+          end
+        '';
+      };
+    };
+  };
+}