runx-cli 0.6.19

use std::collections::BTreeMap;
use std::env;
use std::fmt;
use std::fs;
use std::io::{self, Read};
use std::path::{Path, PathBuf};
use std::process::ExitCode;

use runx_runtime::kernel_eval::{KernelEvalError, KernelEvalOutput, evaluate_kernel_document_str};
use serde::Serialize;

#[derive(Clone, Debug, Eq, PartialEq)]
pub struct KernelPlan {
    pub input: KernelInputSource,
    pub json: bool,
}

#[derive(Clone, Debug, Eq, PartialEq)]
pub enum KernelInputSource {
    Path(PathBuf),
    Stdin,
}

pub fn run_native_kernel(plan: KernelPlan) -> ExitCode {
    let cwd = match env::current_dir() {
        Ok(cwd) => cwd,
        Err(error) => {
            let error = KernelCliError::CurrentDirectory(error);
            return write_error(&error, plan.json);
        }
    };

    match run_kernel_command(&plan, &crate::cli_io::env_map(), &cwd) {
        Ok(output) => crate::cli_io::write_stdout_code(&output.stdout, output.exit_code),
        Err(error) => write_error(&error, plan.json),
    }
}

pub fn run_kernel_command(
    plan: &KernelPlan,
    env: &BTreeMap<String, String>,
    cwd: &Path,
) -> Result<KernelCliOutput, KernelCliError> {
    if !plan.json {
        return Err(KernelCliError::InvalidArgs(
            "runx kernel eval requires --json".to_owned(),
        ));
    }

    let raw = read_kernel_input(&plan.input, env, cwd)?;
    let result = evaluate_kernel_document_str(&raw)?;
    let stdout = serde_json::to_string_pretty(&KernelJsonEnvelope {
        status: "success",
        result: &result,
    })
    .map(|json| format!("{json}\n"))
    .map_err(KernelCliError::Serialize)?;
    Ok(KernelCliOutput {
        stdout,
        exit_code: 0,
    })
}

#[derive(Debug)]
pub struct KernelCliOutput {
    pub stdout: String,
    pub exit_code: u8,
}

#[derive(Debug)]
pub enum KernelCliError {
    CurrentDirectory(io::Error),
    InvalidArgs(String),
    Read(PathBuf, io::Error),
    ReadStdin(io::Error),
    Eval(KernelEvalError),
    Serialize(serde_json::Error),
}

impl KernelCliError {
    fn code(&self) -> &'static str {
        match self {
            Self::CurrentDirectory(_) => "current_directory",
            Self::InvalidArgs(_) => "invalid_args",
            Self::Read(_, _) => "read_input",
            Self::ReadStdin(_) => "read_stdin",
            Self::Eval(error) => error.code(),
            Self::Serialize(_) => "serialize_output",
        }
    }

    fn exit_code(&self) -> u8 {
        match self {
            Self::InvalidArgs(_) => 64,
            Self::CurrentDirectory(_)
            | Self::Read(_, _)
            | Self::ReadStdin(_)
            | Self::Eval(_)
            | Self::Serialize(_) => 1,
        }
    }
}

impl fmt::Display for KernelCliError {
    fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::CurrentDirectory(error) => write!(formatter, "failed to resolve cwd: {error}"),
            Self::InvalidArgs(message) => formatter.write_str(message),
            Self::Read(path, error) => {
                write!(
                    formatter,
                    "failed to read kernel input {}: {error}",
                    path.display()
                )
            }
            Self::ReadStdin(error) => {
                write!(formatter, "failed to read kernel input stdin: {error}")
            }
            Self::Eval(error) => write!(formatter, "{error}"),
            Self::Serialize(error) => {
                write!(formatter, "failed to serialize kernel result: {error}")
            }
        }
    }
}

impl std::error::Error for KernelCliError {}

impl From<KernelEvalError> for KernelCliError {
    fn from(error: KernelEvalError) -> Self {
        Self::Eval(error)
    }
}

#[derive(Serialize)]
struct KernelJsonEnvelope<'a> {
    status: &'static str,
    result: &'a KernelEvalOutput,
}

#[derive(Serialize)]
struct KernelJsonError<'a> {
    status: &'static str,
    code: &'static str,
    message: &'a str,
}

fn read_kernel_input(
    source: &KernelInputSource,
    env: &BTreeMap<String, String>,
    cwd: &Path,
) -> Result<String, KernelCliError> {
    match source {
        KernelInputSource::Path(path) => {
            let resolved = resolve_kernel_path(path, env, cwd);
            fs::read_to_string(&resolved).map_err(|error| KernelCliError::Read(resolved, error))
        }
        KernelInputSource::Stdin => {
            let mut raw = String::new();
            io::stdin()
                .read_to_string(&mut raw)
                .map_err(KernelCliError::ReadStdin)?;
            Ok(raw)
        }
    }
}

fn resolve_kernel_path(path: &Path, env: &BTreeMap<String, String>, cwd: &Path) -> PathBuf {
    if path.is_absolute() {
        return path.to_path_buf();
    }
    env.get("RUNX_CWD")
        .map(PathBuf::from)
        .or_else(|| env.get("INIT_CWD").map(PathBuf::from))
        .unwrap_or_else(|| cwd.to_path_buf())
        .join(path)
}

fn write_error(error: &KernelCliError, json: bool) -> ExitCode {
    if json {
        let message = error.to_string();
        match serde_json::to_string_pretty(&KernelJsonError {
            status: "error",
            code: error.code(),
            message: &message,
        }) {
            Ok(body) => {
                return crate::cli_io::write_stdout_code(&format!("{body}\n"), error.exit_code());
            }
            Err(serialize_error) => {
                let _ignored = crate::cli_io::write_stderr_code(&format!(
                    "runx: failed to serialize kernel error: {serialize_error}\n"
                ));
                return ExitCode::from(1);
            }
        }
    }

    let _ignored = crate::cli_io::write_stderr_code(&format!("runx: {error}\n"));
    ExitCode::from(error.exit_code())
}