lib.rs

use std::{
    collections::HashMap,
    env::{self, join_paths},
    ffi::OsStr,
    iter,
    path::PathBuf,
    sync::Arc,
};

use clap::Subcommand;
use monostate::MustBe;
use vite_path::AbsolutePath;
use vite_str::Str;
use vite_task::{
    EnabledCacheConfig, SessionCallbacks, UserCacheConfig, UserTaskOptions, get_path_env,
    plan_request::SyntheticPlanRequest,
};

/// Theses are the custom subcommands that synthesize tasks for vite-task
#[derive(Debug, Subcommand)]
pub enum CustomTaskSubcommand {
    /// oxlint
    #[clap(disable_help_flag = true)]
    Lint {
        #[clap(allow_hyphen_values = true, trailing_var_arg = true)]
        args: Vec<Str>,
    },
    /// vitest
    #[clap(disable_help_flag = true)]
    Test {
        #[clap(allow_hyphen_values = true, trailing_var_arg = true)]
        args: Vec<Str>,
    },
    /// Test command for testing additional_envs feature
    EnvTest {
        /// Environment variable name
        name: Str,
        /// Environment variable value
        value: Str,
    },
}

// These are the subcommands that is not handled by vite-task
#[derive(Debug, Subcommand)]
pub enum NonTaskSubcommand {
    Version,
}

#[derive(Debug, Default)]
pub struct TaskSynthesizer(());

fn find_executable(
    path_env: Option<&Arc<OsStr>>,
    cwd: &AbsolutePath,
    executable: &str,
) -> anyhow::Result<Arc<OsStr>> {
    let mut paths: Vec<PathBuf> =
        if let Some(path_env) = path_env { env::split_paths(path_env).collect() } else { vec![] };
    let mut current_cwd_parent = cwd;
    loop {
        let node_modules_bin = current_cwd_parent.join("node_modules").join(".bin");
        paths.push(node_modules_bin.as_path().to_path_buf());
        if let Some(parent) = current_cwd_parent.parent() {
            current_cwd_parent = parent;
        } else {
            break;
        }
    }
    let executable_path = which::which_in(executable, Some(join_paths(paths)?), cwd)?;
    Ok(executable_path.into_os_string().into())
}

#[async_trait::async_trait(?Send)]
impl vite_task::TaskSynthesizer<CustomTaskSubcommand> for TaskSynthesizer {
    fn should_synthesize_for_program(&self, program: &str) -> bool {
        program == "vite"
    }

    async fn synthesize_task(
        &mut self,
        subcommand: CustomTaskSubcommand,
        envs: &Arc<HashMap<Arc<OsStr>, Arc<OsStr>>>,
        cwd: &Arc<AbsolutePath>,
    ) -> anyhow::Result<SyntheticPlanRequest> {
        let synthesize_node_modules_bin_task = |subcommand_name: &str,
                                                executable_name: &str,
                                                args: Vec<Str>|
         -> anyhow::Result<SyntheticPlanRequest> {
            let direct_execution_cache_key: Arc<[Str]> =
                iter::once(Str::from(subcommand_name)).chain(args.iter().cloned()).collect();
            Ok(SyntheticPlanRequest {
                program: find_executable(get_path_env(envs), &*cwd, executable_name)?,
                args: args.into(),
                task_options: Default::default(),
                direct_execution_cache_key,
                envs: Arc::clone(envs),
            })
        };

        match subcommand {
            CustomTaskSubcommand::Lint { args } => {
                synthesize_node_modules_bin_task("lint", "oxlint", args)
            }
            CustomTaskSubcommand::Test { args } => {
                synthesize_node_modules_bin_task("test", "vitest", args)
            }
            CustomTaskSubcommand::EnvTest { name, value } => {
                let direct_execution_cache_key: Arc<[Str]> =
                    [Str::from("env-test"), name.clone(), value.clone()].into();

                let mut envs = HashMap::clone(&envs);
                // Update the env var for testing
                envs.insert(
                    Arc::from(OsStr::new(name.as_str())),
                    Arc::from(OsStr::new(value.as_str())),
                );

                Ok(SyntheticPlanRequest {
                    program: find_executable(get_path_env(&envs), &*cwd, "print-env")?,
                    args: [name.clone()].into(),
                    task_options: UserTaskOptions {
                        cache_config: UserCacheConfig::Enabled {
                            cache: MustBe!(true),
                            enabled_cache_config: EnabledCacheConfig {
                                envs: Box::new([]),
                                pass_through_envs: vec![name],
                            },
                        },
                        ..Default::default()
                    },
                    direct_execution_cache_key,
                    envs: Arc::new(envs),
                })
            }
        }
    }
}

#[derive(Default)]
pub struct OwnedSessionCallbacks {
    task_synthesizer: TaskSynthesizer,
    user_config_loader: vite_task::loader::JsonUserConfigLoader,
}

impl OwnedSessionCallbacks {
    pub fn as_callbacks(&mut self) -> SessionCallbacks<'_, CustomTaskSubcommand> {
        SessionCallbacks {
            task_synthesizer: &mut self.task_synthesizer,
            user_config_loader: &mut self.user_config_loader,
        }
    }
}