xtask/extendrtests/
with_absolute_path.rs

use std::error::Error;
use std::path::Path;

use toml_edit::{DocumentMut, InlineTable, Value};
use xshell::Shell;

use crate::extendrtests::path_helper::RCompatiblePath;

pub(crate) const R_FOLDER_PATH: &str = "tests/extendrtests";

const RUST_FOLDER_PATH: &str = "tests/extendrtests/src/rust";
const CARGO_TOML: &str = "Cargo.toml";

#[derive(Debug)]
pub(crate) struct DocumentMutHandle<'a> {
    document: Vec<u8>,
    shell: &'a Shell,
}

impl<'a> Drop for DocumentMutHandle<'a> {
    fn drop(&mut self) {
        let _rust_folder = self.shell.push_dir(RUST_FOLDER_PATH);
        self.shell
            .write_file(CARGO_TOML, &self.document)
            .expect("Failed to restore Cargo.toml");
    }
}

pub(crate) fn swap_extendr_api_path(shell: &Shell) -> Result<DocumentMutHandle, Box<dyn Error>> {
    let current_path = shell.current_dir();
    let _rust_folder = shell.push_dir(RUST_FOLDER_PATH);

    let original_cargo_toml_bytes = read_file_with_line_ending(shell, CARGO_TOML)?;

    let original_cargo_toml: DocumentMut =
        std::str::from_utf8(&original_cargo_toml_bytes)?.parse()?;

    let mut cargo_toml = original_cargo_toml.clone();

    let extendr_api_entry =
        get_extendr_api_entry(&mut cargo_toml).ok_or("`extendr-api` not found in Cargo.toml")?;

    let mut replacement = InlineTable::new();

    let item = Value::from(get_replacement_path_extendr_api(&current_path));
    replacement.entry("path").or_insert(item);
    *extendr_api_entry = Value::InlineTable(replacement);

    #[allow(non_snake_case)]
    let libR_sys_entry = get_libR_sys_entry(&mut cargo_toml);
    #[allow(non_snake_case)]
    if let Some(libR_sys_entry) = libR_sys_entry {
        let mut replacement = InlineTable::new();
        if let Some(replacement_path) = get_replacement_path_libR_sys(&current_path) {
            let item = Value::from(replacement_path);
            replacement.entry("path").or_insert(item);
            *libR_sys_entry = Value::InlineTable(replacement);
        }
    }

    // save altered paths
    shell.write_file(CARGO_TOML, cargo_toml.to_string())?;
    Ok(DocumentMutHandle {
        document: original_cargo_toml_bytes,
        shell,
    })
}

fn get_replacement_path_extendr_api(path: &Path) -> String {
    let path = path.adjust_for_r();

    format!("{path}/extendr-api")
}

#[allow(non_snake_case)]
fn get_replacement_path_libR_sys(path: &Path) -> Option<String> {
    let path = path.adjust_for_r();
    #[allow(non_snake_case)]
    let libR_sys_path = format!("{path}/libR-sys");
    let valid_path = std::path::Path::new(&libR_sys_path);
    matches!(valid_path.try_exists(), Ok(true)).then_some(libR_sys_path)
}

fn get_extendr_api_entry(document: &mut DocumentMut) -> Option<&mut Value> {
    document
        .get_mut("patch")?
        .get_mut("crates-io")?
        .get_mut("extendr-api")?
        .as_value_mut()
}

#[allow(non_snake_case)]
fn get_libR_sys_entry(document: &mut DocumentMut) -> Option<&mut Value> {
    document
        .get_mut("patch")?
        .get_mut("crates-io")?
        .get_mut("libR-sys")?
        .as_value_mut()
}

fn read_file_with_line_ending<P: AsRef<Path>>(
    shell: &Shell,
    path: P,
) -> Result<Vec<u8>, Box<dyn Error>> {
    let file_contents = shell.read_binary_file(path)?;
    Ok(file_contents)
}
xtask/extendrtests/with_absolute_path.rs

xtask/extendrtests/
with_absolute_path.rs
