chore: improve generate-protobufs script

.github/workflows/pre-commit.yml

@@ -21,6 +21,7 @@ jobs:
         uses: ./.github/actions/set-sha
 
       - uses: ./.github/actions/install-python-and-poetry
+      - uses: ./.github/actions/install-canvas
 
       - name: Run the pre-commit hooks
         uses: pre-commit/action@v3.0.0

.pre-commit-config.yaml

@@ -119,3 +119,11 @@ repos:
             && poetry install --no-ansi \
             && exec poetry run mypy --show-traceback $@
           "
+
+  - repo: local
+    hooks:
+      - id: check-protobufs-definitions
+        name: Check Protobufs Definitions
+        entry: poetry run ./bin/generate-protobufs
+        language: system
+        pass_filenames: false

bin/generate-protobufs

@@ -0,0 +1,28 @@
+#!/usr/bin/env bash
+set -e
+
+# Default input and output paths, run from root
+input_dir="protobufs/"
+output_dir="./"
+
+# Parse arguments for input and output paths
+while getopts i:o: flag
+do
+    case "${flag}" in
+        i) input_dir=${OPTARG};;
+        o) output_dir=${OPTARG};;
+        *) exit 1;;
+    esac
+done
+
+echo "Generating Python code from Protobufs in \"${input_dir}\" to \"${output_dir}\"..."
+
+python \
+  -m grpc_tools.protoc \
+  -I="${input_dir}" \
+  --python_out="${output_dir}" \
+  --pyi_out="${output_dir}" \
+  --grpc_python_out="${output_dir}" \
+  "${input_dir}"canvas_generated/**/*.proto || exit 1
+
+echo "Done!"