From ea61a9cff1a46ab0f4ab451da65120ba1bb5e0b1 Mon Sep 17 00:00:00 2001
From: Michael Vincent <michael.vincent@garmin.com>
Date: Mon, 8 Nov 2021 11:03:56 -0600
Subject: [PATCH] Expose hook stage to hook scripts as env var

This change makes it possible for one hook script to run in several
stages and change its behavior based on context.
---
 pre_commit/commands/run.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pre_commit/commands/run.py b/pre_commit/commands/run.py
index 2714faf4..e07d1f8a 100644
--- a/pre_commit/commands/run.py
+++ b/pre_commit/commands/run.py
@@ -391,6 +391,9 @@ def run(
     if args.rewrite_command:
         environ['PRE_COMMIT_REWRITE_COMMAND'] = args.rewrite_command
 
+    if args.hook_stage:
+        environ['PRE_COMMIT_HOOK_STAGE'] = args.hook_stage
+
     # Set pre_commit flag
     environ['PRE_COMMIT'] = '1'