From 3ff88390c2bf63ff61010243b4f8726cf63f2ebc Mon Sep 17 00:00:00 2001
From: Martin Xu <15661672+martinxu9@users.noreply.github.com>
Date: Mon, 29 Jan 2024 20:32:36 -0800
Subject: [PATCH] Add git precommit hook that generates pyi files (#2447)

---
 .github/workflows/pre-commit.yml |  2 ++
 .pre-commit-config.yaml          | 27 ++++++++++++++++++---------
 2 files changed, 20 insertions(+), 9 deletions(-)

diff --git a/.github/workflows/pre-commit.yml b/.github/workflows/pre-commit.yml
index a785299bd..612092652 100644
--- a/.github/workflows/pre-commit.yml
+++ b/.github/workflows/pre-commit.yml
@@ -27,3 +27,5 @@ jobs:
       - run: |
           poetry run pip install pre-commit
           poetry run pre-commit run --all-files
+        env:
+          SKIP: update-pyi-files
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index a5fcf4b6b..3380683c9 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -2,24 +2,33 @@ repos:
   - repo: https://github.com/psf/black
     rev: 22.10.0
     hooks:
-    - id: black
-      args: [integration, reflex, tests]
+      - id: black
+        args: [integration, reflex, tests]
 
   - repo: https://github.com/charliermarsh/ruff-pre-commit
     rev: v0.0.244
     hooks:
-    - id: ruff
-      args: [--fix, --exit-non-zero-on-fix]
+      - id: ruff
+        args: [--fix, --exit-non-zero-on-fix]
 
   - repo: https://github.com/RobertCraigie/pyright-python
     rev: v1.1.313
     hooks:
-    - id: pyright
-      args: [integration, reflex, tests]
-      language: system
+      - id: pyright
+        args: [integration, reflex, tests]
+        language: system
 
   - repo: https://github.com/terrencepreilly/darglint
     rev: v1.8.1
     hooks:
-    - id: darglint
-      exclude: '^reflex/reflex.py'
+      - id: darglint
+        exclude: '^reflex/reflex.py'
+
+  - repo: local
+    hooks:
+      - id: update-pyi-files
+        name: update-pyi-files
+        always_run: true
+        language: system
+        description: 'Update pyi files as needed'
+        entry: python scripts/pyi_generator.py