nix-community.nixvim/plugins/utils/specs.nix

{ pkgs, config, lib, ... }:
with lib;
let
  cfg = config.plugins.specs;
  helpers = import ../helpers.nix { inherit lib; };
in
{
  options.plugins.specs = {
    enable = mkEnableOption "Enable specs-nvim";

    package = mkOption {
      type = types.package;
      default = pkgs.vimPlugins.specs-nvim;
      description = "Plugin to use for specs-nvim";
    };

    show_jumps = mkOption {
      type = types.bool;
      default = true;
    };

    min_jump = mkOption {
      type = types.int;
      default = 30;
    };

    delay = mkOption {
      type = types.int;
      default = 0;
      description = "Delay in miliseconds";
    };

    increment = mkOption {
      type = types.int;
      default = 10;
      description = "Increment in miliseconds";
    };

    blend = mkOption {
      type = types.int;
      default = 10;
    };

    color = mkOption {
      type = types.nullOr types.str;
      default = null;
    };

    width = mkOption {
      type = types.int;
      default = 10;
    };

    fader = mkOption {
      type = types.submodule {
        options = {
          builtin = mkOption {
            type = types.nullOr (types.enum [
              "linear_fader"
              "exp_fader"
              "pulse_fader"
              "empty_fader"
            ]);
            default = "linear_fader";
          };

          custom = mkOption {
            type = types.lines;
            default = "";
            example = ''
              function(blend, cnt)
                if cnt > 100 then
                    return 80
                else return nil end
              end
            '';
          };
        };
      };
      default = { builtin = "linear_fader"; };
    };

    resizer = mkOption {
      type = types.submodule {
        options = {
          builtin = mkOption {
            type = types.nullOr
              (types.enum [ "shrink_resizer" "slide_resizer" "empty_resizer" ]);
            default = "shrink_resizer";
          };

          custom = mkOption {
            type = types.lines;
            default = "";
            example = ''
              function(width, ccol, cnt)
                  if width-cnt > 0 then
                      return {width+cnt, ccol}
                  else return nil end
              end
            '';
          };
        };
      };
      default = { builtin = "shrink_resizer"; };
    };

    ignored_filetypes = mkOption {
      type = with types; listOf string;
      default = [ ];
    };

    ignored_buffertypes = mkOption {
      type = with types; listOf string;
      default = [ "nofile" ];
    };

  };
  config =
    let
      setup = helpers.toLuaObject {
        inherit (cfg) show_jumps min_jump;
        ignore_filetypes = attrsets.listToAttrs
          (lib.lists.map (x: attrsets.nameValuePair x true)
            cfg.ignored_filetypes);
        ignore_buftypes = attrsets.listToAttrs
          (lib.lists.map (x: attrsets.nameValuePair x true)
            cfg.ignored_buffertypes);
        popup = {
          inherit (cfg) blend width;
          winhl = if (!isNull cfg.color) then "SpecsPopColor" else "PMenu";
          delay_ms = cfg.delay;
          inc_ms = cfg.increment;
          fader = helpers.mkRaw (if cfg.fader.builtin == null then
            cfg.fader.custom
          else
            ''require("specs").${cfg.fader.builtin}'');
          resizer = helpers.mkRaw (if cfg.resizer.builtin == null then
            cfg.resizer.custom
          else
            ''require("specs").${cfg.resizer.builtin}'');
        };
      };
    in
    mkIf cfg.enable {
      extraPlugins = [ cfg.package ];

      highlight.SpecsPopColor.bg = mkIf (!isNull cfg.color) cfg.color;

      extraConfigLua = ''
        require('specs').setup(${setup})
      '';
    };
}