TheaninovOS/modules/home-manager/programs/nixvim/presets/auto-format.nix

{ lib, config, ... }:
let
  cfg = config.presets.auto-format;
in
{
  options.presets.auto-format = {
    enable = lib.mkEnableOption "auto-format";
    varName = lib.mkOption {
      type = lib.types.str;
      default = "disable_autoformat";
    };
    commandName = lib.mkOption {
      type = lib.types.str;
      default = "AutoFormatToggle";
    };
  };

  config = lib.mkIf cfg.enable {
    userCommands.${cfg.commandName}.command.__raw = # lua
      ''
        function(args)
          if args.bang then
            vim.b.${cfg.varName} = not vim.b.${cfg.varName}
          else
            vim.g.${cfg.varName} = not vim.g.${cfg.varName}
          end
          local lualine, lib = pcall(require, 'lualine')
          if lualine then
            lib.refresh()
          end
        end
      '';

    keymaps = [
      {
        key = "<leader>af";
        mode = "n";
        options.silent = true;
        action = "<cmd>:${cfg.commandName}<CR>";
      }
      {
        key = "<leader>aF";
        mode = "n";
        options.silent = true;
        action = "<cmd>:${cfg.commandName}!<CR>";
      }
    ];

    plugins = {
      which-key.settings.spec = [
        {
          __unkeyed-1 = "<leader>a";
          group = "Auto Actions";
          icon = "󰁨";
        }
        {
          __unkeyed-1 = "<leader>af";
          desc = "Tggle auto-format";
          icon = "󱌓";
        }
        {
          __unkeyed-1 = "<leader>aF";
          desc = "Tggle auto-format (buffer)";
          icon = "󱌓";
        }
      ];

      conform-nvim.settings.format_after_save = # lua
        ''
          function(bufnr)
            if vim.g.${cfg.varName} or vim.b[bufnr].${cfg.varName} then
              return
            end
            return { timeout_ms = 500, lsp_fallback = true };
          end
        '';

      lualine.settings.sections.lualine_x = lib.mkOrder 600 [
        "(vim.g.${cfg.varName} or vim.b.${cfg.varName}) and '󱌓' or nil"
      ];
    };
  };
}