Add run functionality

This commit is contained in:
SkyfallWasTaken 2024-07-10 18:07:42 +01:00
parent f0a0a6a909
commit 9aedd1d496
7 changed files with 72 additions and 8 deletions

3
.gitignore vendored
View file

@ -1,4 +1,5 @@
/target /target
package.json package.json
package-lock.json package-lock.json
bun.lockb bun.lockb
test.js

10
Cargo.lock generated
View file

@ -1075,6 +1075,15 @@ dependencies = [
"lazy_static", "lazy_static",
] ]
[[package]]
name = "signal-hook-registry"
version = "1.4.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a9e9e0b4211b72e7b8b6e85c807d36c212bdb33ea8587f7569562a84df5465b1"
dependencies = [
"libc",
]
[[package]] [[package]]
name = "slab" name = "slab"
version = "0.4.9" version = "0.4.9"
@ -1225,6 +1234,7 @@ dependencies = [
"mio", "mio",
"num_cpus", "num_cpus",
"pin-project-lite", "pin-project-lite",
"signal-hook-registry",
"socket2", "socket2",
"tokio-macros", "tokio-macros",
"windows-sys 0.48.0", "windows-sys 0.48.0",

View file

@ -8,7 +8,11 @@ clap = { version = "4.5.9", features = ["derive"] }
color-eyre = "0.6.3" color-eyre = "0.6.3"
owo-colors = "4.0.0" owo-colors = "4.0.0"
reqwest = "0.12.5" reqwest = "0.12.5"
tokio = { version = "1.38.0", features = ["macros", "rt-multi-thread"] } tokio = { version = "1.38.0", features = [
"macros",
"rt-multi-thread",
"process",
] }
dinopkg-package-json = { path = "../dinopkg-package-json", features = [ dinopkg-package-json = { path = "../dinopkg-package-json", features = [
"tokio", "tokio",
] } ] }

View file

@ -2,8 +2,11 @@ use color_eyre::{eyre::eyre, Result};
use dinopkg_package_json::PackageJson; use dinopkg_package_json::PackageJson;
use owo_colors::OwoColorize; use owo_colors::OwoColorize;
use crate::run_script::{run_script, DEFAULT_SHELL, DEFAULT_SHELL_EXEC_ARG};
pub async fn run(script_name: Option<String>) -> Result<()> { pub async fn run(script_name: Option<String>) -> Result<()> {
let package_json = PackageJson::from_file(10).await?; let (package_json, package_json_path) = PackageJson::from_file(10).await?;
let root_path = package_json_path.parent().unwrap(); // Should never happen, `package.json` should always be there
match script_name { match script_name {
Some(script_name) => { Some(script_name) => {
@ -12,7 +15,26 @@ pub async fn run(script_name: Option<String>) -> Result<()> {
}; };
match scripts.get(&script_name) { match scripts.get(&script_name) {
Some(script) => { Some(script) => {
log::info!("{script_name}: {script}") println!("> {}", script.bold());
let status =
run_script(DEFAULT_SHELL, DEFAULT_SHELL_EXEC_ARG, &script, root_path)
.await?;
if cfg!(unix) {
use std::os::unix::process::ExitStatusExt;
if let Some(signal) = status.signal() {
return Err(eyre!(format!("process terminated by signal {signal}")));
}
}
// The only time the exit code isn't there is if the process was terminated by a signal.
// We check for that above (and on non-Unix systems, there will always be an exit code.)
let exit_code = status.code().unwrap();
if exit_code != exitcode::OK {
return Err(eyre!(format!("process exited with code {exit_code}")));
}
} }
_ => return Err(eyre!(format!("script `{script_name}` not found"))), _ => return Err(eyre!(format!("script `{script_name}` not found"))),
} }

View file

@ -1,8 +1,9 @@
use clap::Parser; use clap::Parser;
use color_eyre::Result;
use env_logger::Env; use env_logger::Env;
mod command; mod command;
use color_eyre::Result; mod run_script;
use command::{Cli, Command}; use command::{Cli, Command};
#[tokio::main] #[tokio::main]

View file

@ -0,0 +1,25 @@
use std::{path::Path, process::ExitStatus};
use color_eyre::eyre::Result;
use tokio::process::Command;
pub const DEFAULT_SHELL: &str = if cfg!(windows) { "cmd.exe" } else { "/bin/sh" };
pub const DEFAULT_SHELL_EXEC_ARG: &str = if cfg!(windows) { "/c" } else { "-c" };
pub async fn run_script(
shell: &str,
shell_exec_arg: &str,
command: &str,
cwd_path: &Path,
) -> Result<ExitStatus> {
// Scripts are run from the root of the package folder, regardless of what
// the current working directory is when npm run is called. As such, we
// do this too for compatibility (and also because it's ten times less annoying!)
let mut tokio_command = Command::new(shell);
debug_assert!(cwd_path.is_dir());
tokio_command
.arg(shell_exec_arg)
.arg(command)
.current_dir(cwd_path);
Ok(tokio_command.status().await?)
}

View file

@ -1,4 +1,5 @@
use std::collections::HashMap; use std::collections::HashMap;
use std::path::PathBuf;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
@ -43,13 +44,13 @@ impl PackageJson {
} }
#[cfg(feature = "tokio")] #[cfg(feature = "tokio")]
pub async fn from_file(max_attempts: usize) -> Result<Self, Error> { pub async fn from_file(max_attempts: usize) -> Result<(Self, PathBuf), Error> {
let path = util::find_package_json(max_attempts).await?; let path = util::find_package_json(max_attempts).await?;
let Some(path) = path else { let Some(path) = path else {
return Err(Error::NotFound); return Err(Error::NotFound);
}; };
let file = tokio::fs::read(path).await?; let file = tokio::fs::read(path.clone()).await?;
let file = String::from_utf8(file)?; let file = String::from_utf8(file)?;
Self::parse(&file) Ok((Self::parse(&file)?, path))
} }
} }