run-rs 0.6.18

//! Bridges for the extra crates a script may use.

use num_traits::AsPrimitive;
use std::sync::Arc;

use anyhow::{Result, bail};
use parking_lot::Mutex;

use super::bytecode::{BuiltinId, MethodName, PathId};
use super::json_bridge::{json_to_pvalue, pvalue_to_json};
use super::native::Native;
use super::native_methods::value_to_bytes;
use super::std_bridge::make_path;
use super::value::{StructData, Value};

fn opt_path(p: Option<std::path::PathBuf>) -> Value {
    match p {
        Some(p) => Value::some(make_path(p.display().to_string())),
        None => Value::none(),
    }
}

pub(super) fn bytes_to_vec(b: &[u8]) -> Value {
    Value::vec(b.iter().map(|x| Value::Int(i64::from(*x))).collect())
}

pub(super) fn crate_bridge(id: PathId, args: &[Value]) -> Result<Option<Value>> {
    if let Some(v) = super::ed25519_bridge::ed25519_call(id, args)? {
        return Ok(Some(v));
    }
    let s0 = || args.first().map(Value::display).unwrap_or_default();
    Ok(Some(match id {
        // dirs
        PathId::DirsHomeDir => opt_path(dirs::home_dir()),
        PathId::DirsCacheDir => opt_path(dirs::cache_dir()),
        PathId::DirsConfigDir => opt_path(dirs::config_dir()),
        PathId::DirsConfigLocalDir => opt_path(dirs::config_local_dir()),
        PathId::DirsDataDir => opt_path(dirs::data_dir()),
        PathId::DirsDataLocalDir => opt_path(dirs::data_local_dir()),
        PathId::DirsExecutableDir => opt_path(dirs::executable_dir()),
        PathId::DirsRuntimeDir => opt_path(dirs::runtime_dir()),
        PathId::DirsDesktopDir => opt_path(dirs::desktop_dir()),
        PathId::DirsDownloadDir => opt_path(dirs::download_dir()),
        PathId::DirsDocumentDir => opt_path(dirs::document_dir()),
        // which
        PathId::WhichWhich => match which::which(s0()) {
            Ok(p) => Value::ok(make_path(p.display().to_string())),
            Err(e) => Value::err(Value::str(e.to_string())),
        },
        // glob
        PathId::GlobGlob => match glob::glob(&s0()) {
            Ok(paths) => Value::ok(Value::vec(
                paths
                    .map(|r| match r {
                        Ok(p) => Value::ok(make_path(p.display().to_string())),
                        Err(e) => Value::err(Value::str(e.to_string())),
                    })
                    .collect(),
            )),
            Err(e) => Value::err(Value::str(e.to_string())),
        },
        // sha2
        PathId::Sha256New | PathId::Sha256Default => {
            use sha2::Digest;
            Native::Sha256(sha2::Sha256::new()).wrap()
        }
        PathId::Sha256Digest => {
            use sha2::Digest;
            bytes_to_vec(&sha2::Sha256::digest(value_to_bytes(args.first())))
        }
        // regex free functions
        PathId::RegexEscape => Value::str(regex::escape(&s0())),
        // hex
        PathId::HexEncode => Value::str(hex::encode(value_to_bytes(args.first()))),
        PathId::HexDecode => match hex::decode(s0()) {
            Ok(b) => Value::ok(bytes_to_vec(&b)),
            Err(e) => Value::err(Value::str(e.to_string())),
        },
        // toml
        PathId::TomlFromStr => match toml::from_str::<serde_json::Value>(&s0()) {
            Ok(j) => Value::ok(json_to_pvalue(j)),
            Err(e) => Value::err(Value::str(e.to_string())),
        },
        PathId::TomlToString | PathId::TomlToStringPretty => {
            match toml::to_string(&pvalue_to_json(args.first().unwrap_or(&Value::Unit))?) {
                Ok(s) => Value::ok(Value::str(s)),
                Err(e) => Value::err(Value::str(e.to_string())),
            }
        }
        // serde_yaml
        PathId::SerdeYamlFromStr => match serde_yaml::from_str::<serde_json::Value>(&s0()) {
            Ok(j) => Value::ok(json_to_pvalue(j)),
            Err(e) => Value::err(Value::str(e.to_string())),
        },
        PathId::SerdeYamlToString => {
            match serde_yaml::to_string(&pvalue_to_json(args.first().unwrap_or(&Value::Unit))?) {
                Ok(s) => Value::ok(Value::str(s)),
                Err(e) => Value::err(Value::str(e.to_string())),
            }
        }
        // rand
        PathId::RandRng | PathId::RandThreadRng => Value::struct_of("Rng", []),
        PathId::RandRandom => Value::Float(rand::random::<f64>()),
        // chrono is handled in `dispatch_call`
        // jsonwebtoken
        PathId::JsonwebtokenEncode => super::jwt_bridge::jwt_encode(args)?,
        // tempfile
        PathId::TempfileTempdir => match tempfile::tempdir() {
            Ok(d) => Value::ok(Native::TempDir(d).wrap()),
            Err(e) => Value::err(Value::str(e.to_string())),
        },
        PathId::TempfileTempfile => match tempfile::tempfile() {
            Ok(f) => Value::ok(Native::File(std::io::BufReader::new(f)).wrap()),
            Err(e) => Value::err(Value::str(e.to_string())),
        },
        PathId::NamedTempFileNew => match tempfile::NamedTempFile::new() {
            Ok(f) => Value::ok(Native::NamedTempFile(f).wrap()),
            Err(e) => Value::err(Value::str(e.to_string())),
        },
        // winreg
        PathId::RegKeyPredef => super::winreg_bridge::predef(args),
        // windows-service
        PathId::ServiceManagerLocalComputer => super::service_bridge::local_computer(args),
        // wmi
        PathId::WMIConnectionNew => super::wmi_bridge::connection(args, true),
        PathId::WMIConnectionWithNamespacePath => super::wmi_bridge::connection(args, false),
        // crossterm
        PathId::TerminalSize => terminal_size(),
        // terminal-light
        PathId::TerminalLightLuma => terminal_luma(),
        _ => return Ok(None),
    }))
}

/// Columns then rows like the real call, the opposite of a `Rect`.
fn terminal_size() -> Value {
    match crossterm::terminal::size() {
        Ok((cols, rows)) => Value::ok(Value::tuple(vec![
            Value::Int(i64::from(cols)),
            Value::Int(i64::from(rows)),
        ])),
        Err(e) => Value::err(Value::str(e.to_string())),
    }
}

/// Background brightness from 0 to 1. An error means neither the terminal nor `$COLORFGBG` gave
/// anything.
fn terminal_luma() -> Value {
    match terminal_light::luma() {
        Ok(luma) => Value::ok(Value::F32(luma)),
        Err(e) => Value::err(Value::str(e.to_string())),
    }
}

/// A marker with the alphabet, so `.encode` picks the right engine.
pub(super) fn base64_engine(id: PathId) -> Option<Value> {
    let kind = match id {
        PathId::Standard | PathId::Base64Standard => "standard",
        PathId::StandardNoPad | PathId::Base64StandardNoPad => "standard_no_pad",
        PathId::UrlSafe | PathId::Base64UrlSafe => "url_safe",
        PathId::UrlSafeNoPad | PathId::Base64UrlSafeNoPad => "url_safe_no_pad",
        _ => return None,
    };
    Some(Value::struct_of(
        "Base64Engine",
        [("kind".into(), Value::str(kind))],
    ))
}

pub(super) fn base64_method(s: &StructData, method: &MethodName, args: &[Value]) -> Result<Value> {
    use base64::Engine;
    use base64::engine::general_purpose::{STANDARD, STANDARD_NO_PAD, URL_SAFE, URL_SAFE_NO_PAD};
    let kind = s.get("kind").map(|v| v.display()).unwrap_or_default();
    macro_rules! pick {
        ($m:ident, $($a:tt)*) => {
            match kind.as_str() {
                "standard_no_pad" => STANDARD_NO_PAD.$m($($a)*),
                "url_safe" => URL_SAFE.$m($($a)*),
                "url_safe_no_pad" => URL_SAFE_NO_PAD.$m($($a)*),
                _ => STANDARD.$m($($a)*),
            }
        };
    }
    Ok(match method.id {
        BuiltinId::Encode => Value::str(pick!(encode, value_to_bytes(args.first()))),
        BuiltinId::Decode => {
            let input = args.first().map(Value::display).unwrap_or_default();
            match pick!(decode, &input) {
                Ok(b) => Value::ok(bytes_to_vec(&b)),
                Err(e) => Value::err(Value::str(e.to_string())),
            }
        }
        _ => bail!("unknown method `{method}` on a base64 engine"),
    })
}

pub(super) fn rng_method(name: &MethodName, args: &[Value]) -> Result<Value> {
    use rand::RngExt;
    let mut rng = rand::rng();
    Ok(match name.id {
        BuiltinId::RandomRange | BuiltinId::GenRange => match args.first() {
            Some(Value::Range {
                start,
                end,
                inclusive,
            }) => {
                let hi = if *inclusive { end + 1 } else { *end };
                if hi > *start {
                    Value::Int(rng.random_range(*start..hi))
                } else {
                    Value::Int(*start)
                }
            }
            _ => bail!("random_range needs a range"),
        },
        BuiltinId::RandomBool | BuiltinId::GenBool => {
            let p = match args.first() {
                Some(Value::Float(f)) => *f,
                Some(Value::Int(i)) => AsPrimitive::<f64>::as_(*i),
                _ => 0.5,
            };
            Value::Bool(rng.random_bool(p.clamp(0.0, 1.0)))
        }
        BuiltinId::Random | BuiltinId::Gen => Value::Float(rng.random::<f64>()),
        BuiltinId::FillBytes | BuiltinId::Fill => {
            if let Some(Value::Vec(v)) = args.first() {
                let mut buf = v.lock();
                for slot in buf.iter_mut() {
                    *slot = Value::Int(i64::from(rng.random::<u8>()));
                }
            }
            Value::Unit
        }
        _ => bail!("unknown method `{name}` on Rng"),
    })
}

/// `finalize` clones the hasher instead of consuming it.
pub(super) fn sha256_method(
    handle: &Arc<Mutex<Native>>,
    method: &MethodName,
    args: &[Value],
) -> Result<Option<Value>> {
    use sha2::Digest;
    let mut h = handle.lock();
    let Native::Sha256(hasher) = &mut *h else {
        return Ok(None);
    };
    Ok(Some(match method.id {
        BuiltinId::Update => {
            hasher.update(value_to_bytes(args.first()));
            Value::Unit
        }
        BuiltinId::ChainUpdate => {
            hasher.update(value_to_bytes(args.first()));
            Value::Native(handle.clone())
        }
        BuiltinId::Finalize => bytes_to_vec(&hasher.clone().finalize()),
        _ => return Ok(None),
    }))
}