zombienet_orchestrator/shared/
types.rs

use std::{
    collections::HashMap,
    net::TcpListener,
    path::PathBuf,
    sync::{Arc, RwLock},
};

use configuration::shared::{
    resources::Resources,
    types::{Arg, AssetLocation, Command, Image, Port},
};
use serde::{Deserialize, Serialize};

pub type Accounts = HashMap<String, NodeAccount>;

#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct NodeAccount {
    pub address: String,
    pub public_key: String,
}

impl NodeAccount {
    pub fn new(addr: impl Into<String>, pk: impl Into<String>) -> Self {
        Self {
            address: addr.into(),
            public_key: pk.into(),
        }
    }
}

#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize)]
pub struct NodeAccounts {
    pub seed: String,
    pub accounts: Accounts,
}

#[derive(Clone, Default, Debug, Serialize, Deserialize)]
pub struct ParkedPort(
    pub(crate) Port,
    #[serde(skip)] pub(crate) Arc<RwLock<Option<TcpListener>>>,
);

impl ParkedPort {
    pub(crate) fn new(port: u16, listener: TcpListener) -> ParkedPort {
        let listener = Arc::new(RwLock::new(Some(listener)));
        ParkedPort(port, listener)
    }

    pub(crate) fn drop_listener(&self) {
        // drop the listener will allow the running node to start listenen connections
        let mut l = self.1.write().unwrap();
        *l = None;
    }
}

#[derive(Debug, Clone, Default)]
pub struct ChainDefaultContext<'a> {
    pub default_command: Option<&'a Command>,
    pub default_image: Option<&'a Image>,
    pub default_resources: Option<&'a Resources>,
    pub default_db_snapshot: Option<&'a AssetLocation>,
    pub default_args: Vec<&'a Arg>,
}

#[derive(Debug, Clone)]
pub struct RegisterParachainOptions {
    pub id: u32,
    pub wasm_path: PathBuf,
    pub state_path: PathBuf,
    pub node_ws_url: String,
    pub onboard_as_para: bool,
    pub seed: Option<[u8; 32]>,
    pub finalization: bool,
}

pub struct RuntimeUpgradeOptions {
    /// Location of the wasm file (could be either a local file or an url)
    pub wasm: AssetLocation,
    /// Name of the node to use as rpc endpoint
    pub node_name: Option<String>,
    /// Seed to use to sign and submit (default to //Alice)
    pub seed: Option<[u8; 32]>,
}

impl RuntimeUpgradeOptions {
    pub fn new(wasm: AssetLocation) -> Self {
        Self {
            wasm,
            node_name: None,
            seed: None,
        }
    }
}
#[derive(Debug, Clone)]
pub struct ParachainGenesisArgs {
    pub genesis_head: String,
    pub validation_code: String,
    pub parachain: bool,
}
zombienet_orchestrator/shared/types.rs

zombienet_orchestrator/shared/
types.rs
