nix-community.nixvim/lib/vim-plugin.nix

{ lib }:
rec {
  mkSettingsOptionDescription =
    { name, globalPrefix }:
    ''
      The configuration options for **${name}** without the `${globalPrefix}` prefix.

      For example, the following settings are equivialent to these `:setglobal` commands:
      - `foo_bar = 1` -> `:setglobal ${globalPrefix}foo_bar=1`
      - `hello = "world"` -> `:setglobal ${globalPrefix}hello="world"`
      - `some_toggle = true` -> `:setglobal ${globalPrefix}some_toggle`
      - `other_toggle = false` -> `:setglobal no${globalPrefix}other_toggle`
    '';

  mkSettingsOption =
    {
      options ? { },
      example ? { },
      name,
      globalPrefix,
    }:
    lib.nixvim.mkSettingsOption {
      inherit options example;
      description = mkSettingsOptionDescription { inherit name globalPrefix; };
    };

  mkVimPlugin =
    {
      name,
      url ? throw "default",
      maintainers,
      imports ? [ ],
      description ? null,
      # deprecations
      deprecateExtraConfig ? false,
      optionsRenamedToSettings ? [ ],
      # colorscheme
      isColorscheme ? false,
      colorscheme ? name,
      # options
      packPathName ? name,
      # Can be a string, a list of strings, or a module option:
      # - A string will be intrpreted as `pkgs.vimPlugins.${package}`
      # - A list will be interpreted as a "pkgs path", e.g. `pkgs.${elem1}.${elem2}.${etc...}`
      # - An option will be used as-is, but should be built using `lib.mkPackageOption`
      # Defaults to `name`, i.e. `pkgs.vimPlugins.${name}`
      package ? name,
      settingsOptions ? { },
      settingsExample ? null,
      globalPrefix ? "",
      extraOptions ? { },
      # config
      extraConfig ? cfg: { },
      extraPlugins ? [ ],
      extraPackages ? [ ],
    }@args:
    let
      namespace = if isColorscheme then "colorschemes" else "plugins";

      createSettingsOption = (lib.isString globalPrefix) && (globalPrefix != "");

      settingsOption = lib.optionalAttrs createSettingsOption {
        settings = mkSettingsOption {
          options = settingsOptions;
          example = settingsExample;
          inherit name globalPrefix;
        };
      };

      module =
        {
          config,
          options,
          pkgs,
          ...
        }:
        let
          cfg = config.${namespace}.${name};
          opts = options.${namespace}.${name};
        in
        {
          meta = {
            inherit maintainers;
            nixvimInfo = {
              inherit description;
              url = args.url or opts.package.default.meta.homepage;
              path = [
                namespace
                name
              ];
            };
          };

          options.${namespace}.${name} =
            {
              enable = lib.mkEnableOption packPathName;
              package =
                if lib.isOption package then
                  package
                else
                  lib.mkPackageOption pkgs packPathName {
                    default =
                      if builtins.isList package then
                        package
                      else
                        [
                          "vimPlugins"
                          package
                        ];
                  };
              packageDecorator = lib.mkOption {
                type = lib.types.functionTo lib.types.package;
                default = lib.id;
                defaultText = lib.literalExpression "x: x";
                description = ''
                  Additional transformations to apply to the final installed package.
                  The result of these transformations is **not** visible in the `package` option's value.
                '';
                internal = true;
              };
            }
            // settingsOption
            // extraOptions;

          config = lib.mkIf cfg.enable (
            lib.mkMerge [
              {
                inherit extraPackages;
                extraPlugins = extraPlugins ++ [
                  (cfg.packageDecorator cfg.package)
                ];
                globals = lib.nixvim.applyPrefixToAttrs globalPrefix (cfg.settings or { });
              }
              (lib.optionalAttrs (isColorscheme && (colorscheme != null)) {
                colorscheme = lib.mkDefault colorscheme;
              })
              (lib.optionalAttrs (args ? extraConfig) (
                lib.nixvim.modules.applyExtraConfig {
                  inherit extraConfig cfg opts;
                }
              ))
            ]
          );
        };
    in
    {
      imports =
        let
          basePluginPath = [
            namespace
            name
          ];
          settingsPath = basePluginPath ++ [ "settings" ];
        in
        imports
        ++ [ module ]
        ++ (lib.optional (deprecateExtraConfig && createSettingsOption) (
          lib.mkRenamedOptionModule (basePluginPath ++ [ "extraConfig" ]) settingsPath
        ))
        ++ (lib.nixvim.mkSettingsRenamedOptionModules basePluginPath settingsPath optionsRenamedToSettings);
    };
}