mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-12-24 22:53:42 +00:00
153 lines
5.1 KiB
Python
Executable File
153 lines
5.1 KiB
Python
Executable File
#!/usr/bin/env nix-shell
|
|
#!nix-shell update-shell.nix -i python3
|
|
|
|
|
|
# format:
|
|
# $ nix run nixpkgs.python3Packages.black -c black update.py
|
|
# type-check:
|
|
# $ nix run nixpkgs.python3Packages.mypy -c mypy update.py
|
|
# linted:
|
|
# $ nix run nixpkgs.python3Packages.flake8 -c flake8 --ignore E501,E265,E402 update.py
|
|
|
|
# If you see `HTTP Error 429: too many requests` errors while running this script,
|
|
# refer to:
|
|
#
|
|
# https://github.com/NixOS/nixpkgs/blob/master/doc/languages-frameworks/vim.section.md#updating-plugins-in-nixpkgs-updating-plugins-in-nixpkgs
|
|
#
|
|
# (or the equivalent file /doc/languages-frameworks/vim.section.md from Nixpkgs master tree).
|
|
#
|
|
|
|
import inspect
|
|
import os
|
|
import sys
|
|
import logging
|
|
import subprocess
|
|
import textwrap
|
|
from typing import List, Tuple
|
|
from pathlib import Path
|
|
|
|
import git
|
|
|
|
log = logging.getLogger()
|
|
|
|
sh = logging.StreamHandler()
|
|
formatter = logging.Formatter('%(name)s:%(levelname)s: %(message)s')
|
|
sh.setFormatter(formatter)
|
|
log.addHandler(sh)
|
|
|
|
# Import plugin update library from maintainers/scripts/pluginupdate.py
|
|
ROOT = Path(os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe()))))
|
|
# Ideally, ROOT.(parent^5) points to root of Nixpkgs official tree
|
|
sys.path.insert(0, os.path.join(ROOT.parent.parent.parent.parent.parent, "maintainers", "scripts"))
|
|
import pluginupdate
|
|
from pluginupdate import run_nix_expr, PluginDesc
|
|
|
|
GET_PLUGINS = f"""(with import <localpkgs> {{}};
|
|
let
|
|
inherit (vimUtils.override {{inherit vim;}}) buildNeovimPluginFrom2Nix buildVimPluginFrom2Nix;
|
|
generated = callPackage {ROOT}/generated.nix {{
|
|
inherit buildNeovimPluginFrom2Nix buildVimPluginFrom2Nix;
|
|
}};
|
|
hasChecksum = value: lib.isAttrs value && lib.hasAttrByPath ["src" "outputHash"] value;
|
|
getChecksum = name: value:
|
|
if hasChecksum value then {{
|
|
submodules = value.src.fetchSubmodules or false;
|
|
sha256 = value.src.outputHash;
|
|
rev = value.src.rev;
|
|
}} else null;
|
|
checksums = lib.mapAttrs getChecksum generated;
|
|
in lib.filterAttrs (n: v: v != null) checksums)"""
|
|
|
|
GET_PLUGINS_LUA = """
|
|
with import <localpkgs> {};
|
|
lib.attrNames lua51Packages"""
|
|
|
|
HEADER = (
|
|
"# This file has been generated by ./pkgs/applications/editors/vim/plugins/update.py. Do not edit!"
|
|
)
|
|
|
|
def isNeovimPlugin(plug: pluginupdate.Plugin) -> bool:
|
|
'''
|
|
Whether it's a neovim-only plugin
|
|
We can check if it's available in lua packages
|
|
'''
|
|
global luaPlugins
|
|
if plug.normalized_name in luaPlugins:
|
|
log.debug("%s is a neovim plugin", plug)
|
|
return True
|
|
return False
|
|
|
|
|
|
class VimEditor(pluginupdate.Editor):
|
|
nvim_treesitter_updated = False
|
|
|
|
def generate_nix(self, plugins: List[Tuple[PluginDesc, pluginupdate.Plugin]], outfile: str):
|
|
sorted_plugins = sorted(plugins, key=lambda v: v[0].name.lower())
|
|
nvim_treesitter_rev = pluginupdate.run_nix_expr("(import <localpkgs> { }).vimPlugins.nvim-treesitter.src.rev")
|
|
|
|
with open(outfile, "w+") as f:
|
|
f.write(HEADER)
|
|
f.write(textwrap.dedent("""
|
|
{ lib, buildVimPluginFrom2Nix, buildNeovimPluginFrom2Nix, fetchFromGitHub, fetchgit }:
|
|
|
|
final: prev:
|
|
{
|
|
"""
|
|
))
|
|
for pdesc, plugin in sorted_plugins:
|
|
content = self.plugin2nix(pdesc, plugin)
|
|
f.write(content)
|
|
if plugin.name == "nvim-treesitter" and plugin.commit != nvim_treesitter_rev:
|
|
self.nvim_treesitter_updated = True
|
|
f.write("\n}\n")
|
|
print(f"updated {outfile}")
|
|
|
|
def plugin2nix(self, pdesc: PluginDesc, plugin: pluginupdate.Plugin) -> str:
|
|
|
|
repo = pdesc.repo
|
|
isNeovim = isNeovimPlugin(plugin)
|
|
|
|
content = f" {plugin.normalized_name} = "
|
|
src_nix = repo.as_nix(plugin)
|
|
content += """{buildFn} {{
|
|
pname = "{plugin.name}";
|
|
version = "{plugin.version}";
|
|
src = {src_nix};
|
|
meta.homepage = "{repo.uri}";
|
|
}};
|
|
|
|
""".format(
|
|
buildFn="buildNeovimPluginFrom2Nix" if isNeovim else "buildVimPluginFrom2Nix", plugin=plugin, src_nix=src_nix, repo=repo)
|
|
print(content)
|
|
return content
|
|
|
|
def main():
|
|
|
|
global luaPlugins
|
|
luaPlugins = run_nix_expr(GET_PLUGINS_LUA)
|
|
|
|
editor = VimEditor("vim", ROOT, GET_PLUGINS)
|
|
parser = editor.create_parser()
|
|
args = parser.parse_args()
|
|
pluginupdate.update_plugins(editor, args)
|
|
|
|
if editor.nvim_treesitter_updated:
|
|
print("updating nvim-treesitter grammars")
|
|
nvim_treesitter_dir = ROOT.joinpath("nvim-treesitter")
|
|
subprocess.check_call([nvim_treesitter_dir.joinpath("update.py")])
|
|
|
|
if editor.nixpkgs_repo:
|
|
index = editor.nixpkgs_repo.index
|
|
for diff in index.diff(None):
|
|
if diff.a_path == "pkgs/applications/editors/vim/plugins/nvim-treesitter/generated.nix":
|
|
msg = "vimPlugins.nvim-treesitter: update grammars"
|
|
print(f"committing to nixpkgs: {msg}")
|
|
index.add([str(nvim_treesitter_dir.joinpath("generated.nix"))])
|
|
index.commit(msg)
|
|
return
|
|
print("no updates to nvim-treesitter grammars")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|