feat: provide branch info in stdin for push hooks

Author: gaojunran

src/commit.rs

@@ -13,7 +13,7 @@ pub(crate) fn command_commit(
     if let Some(pre_commit) = &config.hooks.pre_commit
         && !no_pre_hook
     {
-        run_hook(config, pre_commit.clone(), "pre-commit")?;
+        run_hook(config, pre_commit.clone(), "pre-commit", None)?;
     }
     if let Some(msg) = message {
         cmd!("jj", "commit", "--interactive", "--message", msg).run()?;
@@ -23,7 +23,7 @@ pub(crate) fn command_commit(
     if let Some(post_commit) = &config.hooks.post_commit
         && !no_post_hook
     {
-        run_hook(config, post_commit.clone(), "post-commit")?;
+        run_hook(config, post_commit.clone(), "post-commit", None)?;
     }
     // TODO: should we give more options here?
     if push {
@@ -57,7 +57,7 @@ pub(crate) fn command_amend(
     if let Some(pre_commit) = &config.hooks.pre_commit
         && !no_pre_hook
     {
-        run_hook(config, pre_commit.clone(), "pre-commit")?;
+        run_hook(config, pre_commit.clone(), "pre-commit", None)?;
     }
     let args: Vec<&str> = vec![
         "squash",
@@ -76,7 +76,7 @@ pub(crate) fn command_amend(
     if let Some(post_commit) = &config.hooks.post_commit
         && !no_post_hook
     {
-        run_hook(config, post_commit.clone(), "post-commit")?;
+        run_hook(config, post_commit.clone(), "post-commit", None)?;
     }
     // TODO: should we give more options here?
     if push {

src/hook.rs

@@ -12,6 +12,7 @@ pub(crate) fn run_hook(
     config: &AppConfig,
     script: String,
     hook_name: &str,
+    stdin_input: Option<String>,
 ) -> anyhow::Result<bool> {
     if config.hooks.use_just && !check_just_installed() {
         step("Install just to run hooks");
@@ -42,7 +43,14 @@ pub(crate) fn run_hook(
         .split_once(' ')
         .map(|(p, a)| (p, a.trim().split(' ').collect()))
         .unwrap_or((&script, Vec::new()));
-    if let Err(e) = cmd(program, &args).run()
+
+    let result = if let Some(input) = stdin_input {
+        cmd(program, &args).stdin_bytes(input.as_bytes()).run()
+    } else {
+        cmd(program, &args).run()
+    };
+
+    if let Err(e) = result
         && hook_name.starts_with("pre-")
     {
         anyhow::bail!("HJ {} hook failed: {}. Aborting.", hook_name, e.to_string());

src/push.rs

@@ -32,7 +32,12 @@ pub(crate) fn command_push(
     if let Some(pre_push) = &config.hooks.pre_push
         && !no_pre_hook
     {
-        run_hook(config, pre_push.clone(), "pre-push")?;
+        let stdin_input = if !branch.is_empty() {
+            Some(branch.join(" "))
+        } else {
+            None
+        };
+        run_hook(config, pre_push.clone(), "pre-push", stdin_input)?;
     }
 
     let mut args = vec!["git", "push", "--allow-new"];
@@ -55,7 +60,12 @@ pub(crate) fn command_push(
     if let Some(post_push) = &config.hooks.post_push
         && !no_post_hook
     {
-        run_hook(config, post_push.clone(), "post-push")?;
+        let stdin_input = if !branch.is_empty() {
+            Some(branch.join(" "))
+        } else {
+            None
+        };
+        run_hook(config, post_push.clone(), "post-push", stdin_input)?;
     }
 
     Ok(())