cargo-wasi-src 0.1.28

use crate::{Cache, ToolPath};
use anyhow::Result;
use std::path::PathBuf;
use termcolor::{Color, ColorChoice, ColorSpec, StandardStream, WriteColor};

pub struct Config {
    cache: Option<Cache>,
    verbose: bool,
    choice: ColorChoice,
}

impl Config {
    pub fn new() -> Config {
        Config {
            cache: None,
            verbose: false,
            choice: if atty::is(atty::Stream::Stderr) {
                ColorChoice::Auto
            } else {
                ColorChoice::Never
            },
        }
    }

    pub fn load_cache(&mut self) -> Result<()> {
        assert!(!self.cache.is_some());
        self.cache = Some(Cache::new()?);
        Ok(())
    }

    pub fn cache(&self) -> &Cache {
        self.cache.as_ref().expect("cache not loaded yet")
    }

    pub fn is_verbose(&self) -> bool {
        self.verbose
    }

    pub fn verbose(&self, f: impl FnOnce()) {
        if self.verbose {
            f();
        }
    }

    pub fn set_verbose(&mut self, verbose: bool) {
        self.verbose = verbose;
    }

    pub fn status(&self, name: &str, rest: &str) {
        let mut shell = StandardStream::stderr(self.choice);
        drop(shell.set_color(ColorSpec::new().set_fg(Some(Color::Green)).set_bold(true)));
        eprint!("{:>12}", name);
        drop(shell.reset());
        eprintln!(" {}", rest);
    }

    pub fn print_error(&self, err: &anyhow::Error) {
        if let Some(code) = crate::utils::normal_process_exit_code(err) {
            std::process::exit(code);
        }
        let mut shell = StandardStream::stderr(self.choice);
        drop(shell.set_color(ColorSpec::new().set_fg(Some(Color::Red)).set_bold(true)));
        eprint!("error");
        drop(shell.reset());
        eprintln!(": {}", err);
        for cause in err.chain().skip(1) {
            eprintln!("");
            drop(shell.set_color(ColorSpec::new().set_bold(true)));
            eprint!("Caused by");
            drop(shell.reset());
            eprintln!(":");
            eprintln!("    {}", cause.to_string().replace("\n", "\n    "));
        }
    }

    pub fn info(&self, msg: &str) {
        let mut shell = StandardStream::stderr(self.choice);
        drop(shell.set_color(ColorSpec::new().set_fg(Some(Color::Cyan)).set_bold(true)));
        eprint!("info");
        drop(shell.reset());
        eprintln!(": {}", msg);
    }

    /// Returns the path to execute a tool, which may be the cache path to
    /// download it to if unavailable, and whether the path has been
    /// overridden.
    ///
    /// To override the path used for a tool, set an env var of the tool's name
    /// in uppercase with hyphens replaced with underscores to the desired
    /// path. For example, `WASM_BINDGEN=path/to/wasm-bindgen` to override the
    /// `wasm-bindgen` used, or `WASM_OPT=path/to/wasm-opt` for `wasm-opt`.  or
    /// the `cache` as the fallback.
    fn get_tool(&self, tool: &str, version: Option<&str>) -> (PathBuf, bool) {
        if let Some(s) = std::env::var_os(tool.to_uppercase().replace("-", "_")) {
            (s.into(), true)
        } else {
            let mut cache_path = self.cache().root().join(tool);
            if let Some(v) = version {
                cache_path.push(v);
                cache_path.push(tool)
            }
            (cache_path, false)
        }
    }

    /// Get the path to our `wasm-bindgen` tool for the given version, which
    /// may be the path to download it to if missing, and whether the path has
    /// been overridden.
    ///
    /// Overridable via setting the `WASM_BINDGEN=path/to/wasm-bindgen` env var.
    pub fn get_wasm_bindgen(&self, version: &str) -> (PathBuf, bool) {
        let (mut path, is_overridden) = self.get_tool("wasm-bindgen", Some(version));
        if !is_overridden {
            path.set_extension(std::env::consts::EXE_EXTENSION);
        }
        (path, is_overridden)
    }

    /// Get the path to our `wasm-opt`, which may be the cache path where it
    /// should be download to if missing, and whether the path has been
    /// overridden.
    ///
    /// Overridable via setting the `WASM_OPT=path/to/wasm-opt` env var.
    pub fn get_wasm_opt(&self) -> ToolPath {
        let (path, is_overridden) = self.get_tool("wasm-opt", None);
        if !is_overridden {
            let mut bin = ["bin", "wasm-opt"].iter().collect::<PathBuf>();
            bin.set_extension(std::env::consts::EXE_EXTENSION);

            let bin_path = path.join(&bin);
            let mut sub_paths = vec![bin];

            // wasm-opt on MacOS requires a dylib to execute
            if cfg!(target_os = "macos") {
                let mut dylib = ["lib", "libbinaryen"].iter().collect::<PathBuf>();
                dylib.set_extension(std::env::consts::DLL_EXTENSION);
                sub_paths.push(dylib);
            }

            ToolPath::Cached {
                bin_path,
                base: path,
                sub_paths,
            }
        } else {
            ToolPath::Overridden(path)
        }
    }
}