flake.nix

{
  inputs = {
    nixpkgs.url = "github:NixOS/nixpkgs/nixpkgs-unstable";

    flake-utils.url = "github:numtide/flake-utils";

    norg.url = "github:nvim-neorg/tree-sitter-norg/dev";
    norg-meta.url = "github:nvim-neorg/tree-sitter-norg-meta";

    neorg = {
      url = "github:nvim-neorg/neorg";
      flake = false;
    };
    neorg-telescope = {
      url = "github:nvim-neorg/neorg-telescope";
      flake = false;
    };
  };
  outputs = {
    self,
    nixpkgs,
    flake-utils,
    norg,
    norg-meta,
    neorg,
    neorg-telescope,
    ...
  }:
    {
      overlays.default = final: prev: let
        inherit (final.lib) attrValues elem filter filterAttrs isDerivation;
      in {
        vimPlugins = prev.vimPlugins.extend (f: p: {
          nvim-treesitter = let
            norgGrammars = {
              tree-sitter-norg = norg.defaultPackage.${final.system};
              tree-sitter-norg-meta = norg-meta.defaultPackage.${final.system};
            };
            builtGrammars = (filterAttrs (_: isDerivation) p.nvim-treesitter.builtGrammars) // norgGrammars;
            allGrammars = attrValues builtGrammars;
            withPlugins = grammarFn:
              p.nvim-treesitter.withPlugins (
                _: let
                  plugins = grammarFn builtGrammars;
                in
                  plugins ++ (filter (p: !(elem p plugins)) (attrValues norgGrammars))
              );
            withAllGrammars = withPlugins (_: allGrammars);
          in
            p.nvim-treesitter.overrideAttrs (a: {
              passthru =
                a.passthru
                // {
                  inherit builtGrammars allGrammars withPlugins withAllGrammars;
                  grammarPlugins =
                    a.passthru.grammarPlugins
                    // {
                      norg = p.nvim-treesitter.grammarToPlugin norgGrammars.tree-sitter-norg;
                      norg-meta = p.nvim-treesitter.grammarToPlugin norgGrammars.tree-sitter-norg-meta;
                    };
                };
            });

          lua-utils-nvim = final.vimUtils.buildVimPlugin {
            inherit (prev.luaPackages.lua-utils-nvim) pname version src;
          };

          pathlib-nvim = final.vimUtils.buildVimPlugin {
            inherit (prev.luaPackages.pathlib-nvim) pname version src;
            doCheck = false;
          };

          neorg = final.vimUtils.buildVimPlugin {
            pname = "neorg";
            version = neorg.rev;
            src = neorg;
            dependencies = [
              (f.nvim-treesitter.withPlugins (_: []))
              f.lua-utils-nvim
              f.nui-nvim
              f.nvim-nio
              f.pathlib-nvim
              f.plenary-nvim
            ];
            doCheck = false;
          };

          neorg-telescope = final.vimUtils.buildVimPlugin {
            pname = "neorg-telescope";
            version = neorg-telescope.rev;
            src = neorg-telescope;
            dependencies = [
              f.lua-utils-nvim
              f.neorg
              f.plenary-nvim
              f.telescope-nvim
            ];
          };
        });
      };
      # https://github.com/NixOS/nix/issues/5532
      overlay = nixpkgs.lib.warn "`neorg-overlay.overlay` is deprecated; use `neorg-overlay.overlays.default` instead" self.overlays.default;
    }
    // (flake-utils.lib.eachDefaultSystem (system: let
      pkgs = import nixpkgs {
        inherit system;
        overlays = [self.overlays.default];
      };
    in {
      checks = import ./tests.nix pkgs;
      formatter = pkgs.alejandra;
    }));
}