hooked/hooked-cli/source/cli/run.rs

//! The `run` subcommand.

use std::{io::Read, process::exit};

use {
  color_eyre::{eyre::eyre, Result},
  hooked_config::{Config, ExitAction},
  owo_colors::OwoColorize,
  subprocess::{Exec, Redirection},
};

use crate::{
  printer::{print, PrintType, PRINT_STYLE},
  utilities::{globset_from_strings, plural},
};

/// The `run` subcommand.
pub fn hooked_run(config: Config, hook_type: String) -> Result<()> {
  let global_noise_level = &config.general.noise_level;

  if hook_type == "pre-commit" {
    let hook_count = config.pre_commit.len();
    print(
      format!(
        "Hooked: Running {} pre-commit {}.",
        hook_count,
        plural(hook_count, "hook", None)
      ),
      &config.general.noise_level,
      PrintType::Info,
    );

    'hook_loop: for hook in config.pre_commit {
      let hook_name = hook.name.unwrap_or_else(|| "Unnamed Hook".to_string());

      if !hook.staged.is_empty() {
        let globs = globset_from_strings(&hook.staged)?;

        let staged_files = Exec::cmd("git")
          .args(&["diff", "--name-only", "--cached"])
          .capture()?
          .stdout_str();
        if !staged_files.lines().any(|line| globs.is_match(line)) {
          print(
            format!(
              "\t{} {}",
              "≫".style(PRINT_STYLE.skipped),
              hook_name.style(PRINT_STYLE.skipped)
            ),
            global_noise_level,
            PrintType::Info,
          );
          continue 'hook_loop;
        }
      }

      let command = match (hook.task.command, hook.task.script) {
        (Some(command), _) => Ok(Exec::shell(command)),

        (None, Some(script_file)) => {
          let script_path = config.general.directory.join(script_file);
          let script_path_str = script_path
            .to_str()
            .ok_or_else(|| eyre!("Failed to convert path to str"))?;
          Ok(Exec::shell(script_path_str))
        }

        (None, None) => Err(eyre!(
          "No command or script provided for hook: {}",
          hook_name
        )),
      }?;

      let mut process = command
        .stderr(Redirection::Merge)
        .stdout(Redirection::Pipe)
        .popen()?;
      let exit_status = process.wait()?;
      let output = {
        let mut output = String::new();
        let mut stdout_file = process.stdout.take().unwrap();
        stdout_file.read_to_string(&mut output)?;
        output
      };

      let (stop, print_output, prefix, style, print_type) =
        match (exit_status.success(), hook.on_failure) {
          (true, _) => {
            (false, false, "✓", PRINT_STYLE.success, PrintType::Info)
          }
          (false, ExitAction::Continue) => {
            (false, true, "⚠", PRINT_STYLE.warn, PrintType::Warn)
          }
          (false, ExitAction::Stop) => {
            (true, true, "✗", PRINT_STYLE.error, PrintType::Error)
          }
        };

      let hook_noise_level =
        hook.noise_level.as_ref().unwrap_or(global_noise_level);
      print(
        format!("\t{} {}", prefix.style(style), hook_name.style(style)),
        hook_noise_level,
        print_type,
      );
      if !output.is_empty() && print_output {
        print(output, hook_noise_level, PrintType::Info);
      }

      if stop {
        exit(1);
      }
    }
  }

  Ok(())
}