commit 6013fe8f4198f1c2d6cb5e51a9ebd1400a843448 · pyrox.dev/nixpkgs

+34 -16
pkgs/applications/editors/vim/plugins/update.py
···

       31
       31
        
       log = logging.getLogger()

     

       32
       32
        
       

     

       33
       33
        
       sh = logging.StreamHandler()

     

       34
       34
       -
       formatter = logging.Formatter('%(name)s:%(levelname)s: %(message)s')

     

       34
       34
       +
       formatter = logging.Formatter("%(name)s:%(levelname)s: %(message)s")

     

       35
       35
        
       sh.setFormatter(formatter)

     

       36
       36
        
       log.addHandler(sh)

     

       37
       37
        
       

     

       38
       38
        
       # Import plugin update library from maintainers/scripts/pluginupdate.py

     

       39
       39
        
       ROOT = Path(os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe()))))

     

       40
       40
        
       # Ideally, ROOT.(parent^5) points to root of Nixpkgs official tree

     

       41
       41
       -
       sys.path.insert(0, os.path.join(ROOT.parent.parent.parent.parent.parent, "maintainers", "scripts"))

     

       41
       41
       +
       sys.path.insert(

     

       42
       42
       +
           0, os.path.join(ROOT.parent.parent.parent.parent.parent, "maintainers", "scripts")

     

       43
       43
       +
       )

     

       42
       44
        
       import pluginupdate

     

       43
       45
        
       from pluginupdate import run_nix_expr, PluginDesc

     

       46
       46
       +
       nvim_treesitter = importlib.import_module("treesitter")

     

       44
       47
        
       

     

       45
       48
        
       

     

       46
       46
       -
       

     

       47
       49
        
       GET_PLUGINS_LUA = """

     

       48
       50
        
       with import <localpkgs> {};

     

       49
       51
        
       lib.attrNames lua51Packages"""

     
···

       52
       54
        
           "# GENERATED by ./pkgs/applications/editors/vim/plugins/update.py. Do not edit!"

     

       53
       55
        
       )

     

       54
       56
        
       

     

       57
       57
       +
       

     

       55
       58
        
       def isNeovimPlugin(plug: pluginupdate.Plugin) -> bool:

     

       56
       56
       -
           '''

     

       59
       59
       +
           """

     

       57
       60
        
           Whether it's a neovim-only plugin

     

       58
       61
        
           We can check if it's available in lua packages

     

       59
       59
       -
           '''

     

       62
       62
       +
           """

     

       60
       63
        
           global luaPlugins

     

       61
       64
        
           if plug.normalized_name in luaPlugins:

     

       62
       65
        
               log.debug("%s is a neovim plugin", plug)

     
···

       67
       70
        
       class VimEditor(pluginupdate.Editor):

     

       68
       71
        
           nvim_treesitter_updated = False

     

       69
       72
        
       

     

       70
       70
       -
           def generate_nix(self, plugins: List[Tuple[PluginDesc, pluginupdate.Plugin]], outfile: str):

     

       73
       73
       +
           def generate_nix(

     

       74
       74
       +
               self, plugins: List[Tuple[PluginDesc, pluginupdate.Plugin]], outfile: str

     

       75
       75
       +
           ):

     

       71
       76
        
               sorted_plugins = sorted(plugins, key=lambda v: v[0].name.lower())

     

       72
       72
       -
               nvim_treesitter_rev = pluginupdate.run_nix_expr("(import <localpkgs> { }).vimPlugins.nvim-treesitter.src.rev")

     

       77
       77
       +
               nvim_treesitter_rev = pluginupdate.run_nix_expr(

     

       78
       78
       +
                   "(import <localpkgs> { }).vimPlugins.nvim-treesitter.src.rev"

     

       79
       79
       +
               )

     

       73
       80
        
       

     

       74
       81
        
               with open(outfile, "w+") as f:

     

       75
       82
        
                   f.write(HEADER)

     

       76
       76
       -
                   f.write(textwrap.dedent("""

     

       83
       83
       +
                   f.write(

     

       84
       84
       +
                       textwrap.dedent(

     

       85
       85
       +
                           """

     

       77
       86
        
                       { lib, buildVimPlugin, buildNeovimPlugin, fetchFromGitHub, fetchgit }:

     

       78
       87
        
       

     

       79
       88
        
                       final: prev:

     

       80
       89
        
                       {

     

       81
       90
        
                       """

     

       82
       82
       -
                   ))

     

       91
       91
       +
                       )

     

       92
       92
       +
                   )

     

       83
       93
        
                   for pdesc, plugin in sorted_plugins:

     

       84
       94
        
                       content = self.plugin2nix(pdesc, plugin)

     

       85
       95
        
                       f.write(content)

     

       86
       86
       -
                       if plugin.name == "nvim-treesitter" and plugin.commit != nvim_treesitter_rev:

     

       96
       96
       +
                       if (

     

       97
       97
       +
                           plugin.name == "nvim-treesitter"

     

       98
       98
       +
                           and plugin.commit != nvim_treesitter_rev

     

       99
       99
       +
                       ):

     

       87
       100
        
                           self.nvim_treesitter_updated = True

     

       88
       101
        
                   f.write("\n}\n")

     

       89
       102
        
               print(f"updated {outfile}")

     

       90
       103
        
       

     

       91
       104
        
           def plugin2nix(self, pdesc: PluginDesc, plugin: pluginupdate.Plugin) -> str:

     

       92
       92
       -
       

     

       93
       105
        
               repo = pdesc.repo

     

       94
       106
        
               isNeovim = isNeovimPlugin(plugin)

     

       95
       107
        
       

     
···

       103
       115
        
         }};

     

       104
       116
        
       

     

       105
       117
        
       """.format(

     

       106
       106
       -
               buildFn="buildNeovimPlugin" if isNeovim else "buildVimPlugin", plugin=plugin, src_nix=src_nix, repo=repo)

     

       118
       118
       +
                   buildFn="buildNeovimPlugin" if isNeovim else "buildVimPlugin",

     

       119
       119
       +
                   plugin=plugin,

     

       120
       120
       +
                   src_nix=src_nix,

     

       121
       121
       +
                   repo=repo,

     

       122
       122
       +
               )

     

       107
       123
        
               log.debug(content)

     

       108
       124
        
               return content

     

       109
       109
       -
       

     

       110
       125
        
       

     

       111
       126
        
           def update(self, args):

     

       112
       127
        
               pluginupdate.update_plugins(self, args)

     
···

       114
       129
        
               if self.nvim_treesitter_updated:

     

       115
       130
        
                   print("updating nvim-treesitter grammars")

     

       116
       131
        
                   nvim_treesitter_dir = ROOT.joinpath("nvim-treesitter")

     

       117
       117
       -
                   subprocess.check_call([nvim_treesitter_dir.joinpath("update.py")])

     

       132
       132
       +
                   # subprocess.check_call([nvim_treesitter_dir.joinpath("update.py")])

     

       133
       133
       +
                   nvim_treesitter.update_grammars()

     

       118
       134
        
       

     

       119
       135
        
                   if self.nixpkgs_repo:

     

       120
       136
        
                       index = self.nixpkgs_repo.index

     

       121
       137
        
                       for diff in index.diff(None):

     

       122
       122
       -
                           if diff.a_path == "pkgs/applications/editors/vim/plugins/nvim-treesitter/generated.nix":

     

       138
       138
       +
                           if (

     

       139
       139
       +
                               diff.a_path

     

       140
       140
       +
                               == "pkgs/applications/editors/vim/plugins/nvim-treesitter/generated.nix"

     

       141
       141
       +
                           ):

     

       123
       142
        
                               msg = "vimPlugins.nvim-treesitter: update grammars"

     

       124
       143
        
                               print(f"committing to nixpkgs: {msg}")

     

       125
       144
        
                               index.add([str(nvim_treesitter_dir.joinpath("generated.nix"))])

     
···

       129
       148
        
       

     

       130
       149
        
       

     

       131
       150
        
       def main():

     

       132
       132
       -
       

     

       133
       151
        
           global luaPlugins

     

       134
       152
        
           luaPlugins = run_nix_expr(GET_PLUGINS_LUA)

     

       135
       153