tezos-smart-rollup-installer 0.2.2

// SPDX-FileCopyrightText: 2023 TriliTech <contact@trili.tech>
//
// SPDX-License-Identifier: MIT

use std::{ffi::OsString, path::Path};

use std::fs::File;
use tezos_smart_rollup_encoding::dac::PreimageHash;
use tezos_smart_rollup_host::path::{OwnedPath, RefPath};
use tezos_smart_rollup_installer_config::binary::owned::{
    OwnedConfigInstruction, OwnedConfigProgram,
};
use tezos_smart_rollup_installer_config::yaml::{ConfigConversionError, YamlConfig};
use thiserror::Error;

#[derive(Debug, Error)]
pub enum ConfigurationError {
    #[error("Unable to read config file: {0}.")]
    FileNotFound(std::io::Error),
    #[error("Unable to parse config file: {0}.")]
    ParseError(serde_yaml::Error),
    #[error("Unable to convert config to a valid program: {0}.")]
    YamlConfigInvalid(#[from] ConfigConversionError),
}

// Path that we write the kernel to, before upgrading.
const PREPARE_KERNEL_PATH: RefPath = RefPath::assert_from(b"/installer/kernel/boot.wasm");

// Path of currently running kernel.
const KERNEL_BOOT_PATH: RefPath = RefPath::assert_from(b"/kernel/boot.wasm");

pub fn create_installer_config(
    root_hash: PreimageHash,
    setup_file: Option<OsString>,
) -> Result<OwnedConfigProgram, ConfigurationError> {
    let mut reveal_instructions = vec![
        OwnedConfigInstruction::reveal_instr(
            root_hash,
            OwnedPath::from(PREPARE_KERNEL_PATH),
        ),
        OwnedConfigInstruction::move_instr(
            OwnedPath::from(PREPARE_KERNEL_PATH),
            OwnedPath::from(KERNEL_BOOT_PATH),
        ),
    ];

    let setup_program: OwnedConfigProgram = match setup_file {
        None => OwnedConfigProgram(vec![]),
        Some(setup_file) => {
            let setup_file = File::open(Path::new(&setup_file))
                .map_err(ConfigurationError::FileNotFound)?;
            let yaml_config: YamlConfig = YamlConfig::from_reader(setup_file)
                .map_err(ConfigurationError::ParseError)?;
            yaml_config.try_into()?
        }
    };

    reveal_instructions.extend(setup_program.0);

    Ok(OwnedConfigProgram(reveal_instructions))
}