dotfiles/.config/nixpkgs/neovim/treesitter/update-treesitter.sh

#!/usr/bin/env bash
# shellcheck disable=SC2016

set -euo pipefail

grammars=$(< grammars.json)

q() {
  jq -r "$@" <<<"$grammars"
}

u() {
  grammars=$(jq "$@" <<<"$grammars")
}

update_grammar() {
  name=$1
  repo=$(q --arg name "$name" '.[$name].repo // "tree-sitter/tree-sitter-\($name)"')
  branch=$(q -e --arg name "$name" '.[$name].branch // "HEAD"')

  echo "Updating $name" >&2
  revision=$(git ls-remote "https://github.com/$repo" "$branch" | cut -b -40)
  sha256=$(nix-prefetch-url --unpack "https://github.com/$repo/archive/$revision.tar.gz")

  u --arg name "$name" --arg rev "$revision" --arg sha256 "$sha256" '
    .[$name] = (.[$name] // {}) * {
      rev: $rev,
      sha256: $sha256,
    }
  '
}

update_all() {
  for grammar in $(q 'to_entries | .[] | .key'); do
    update_grammar "$grammar"
  done
}

if [ "$#" -eq 0 ]; then
  update_all
else
  while [ "$#" -gt 0 ]; do
    update_grammar "$1"
    shift
  done
fi

echo "$grammars" > grammars.json