run.rs

use crazy_train::{
    step::{Plan, StepTrait},
    Result, StringDef,
};
use serde::{Deserialize, Serialize};

#[derive(Serialize, Deserialize)]
struct StepOne {}

impl StepTrait for StepOne {
    fn plan(&self, randomizer: &crazy_train::Randomizer) -> Result<crazy_train::step::Plan> {
        let eco_string = randomizer.string(StringDef::default()).to_string();
        Ok(Plan {
            id: std::any::type_name::<Self>().to_string(),
            command: format!("echo {eco_string}"),
        })
    }

    fn is_success(
        &self,
        execution_result: &crazy_train::executer::Output,
    ) -> Result<bool, &'static str> {
        if execution_result.status_code == Some(0) {
            Ok(true)
        } else {
            Err("status code should be 0")
        }
    }

    fn to_yaml(&self) -> serde_yaml::Value {
        serde_yaml::to_value(self).expect("serialize")
    }
}

#[derive(Serialize, Deserialize)]
struct StepTwo {}

impl StepTrait for StepTwo {
    fn plan(&self, randomizer: &crazy_train::Randomizer) -> Result<crazy_train::step::Plan> {
        let eco_string = randomizer.string(StringDef::default()).to_string();
        Ok(Plan {
            id: std::any::type_name::<Self>().to_string(),
            command: format!("unknown-command {eco_string}"),
        })
    }

    fn is_success(
        &self,
        execution_result: &crazy_train::executer::Output,
    ) -> Result<bool, &'static str> {
        if execution_result.status_code == Some(0) {
            Err("expected failure command")
        } else {
            Ok(true)
        }
    }

    fn to_yaml(&self) -> serde_yaml::Value {
        serde_yaml::to_value(self).expect("serialize")
    }
}

fn main() {
    let step_one = StepOne {};
    let step_two = StepTwo {};
    let runner = crazy_train::new(vec![Box::new(step_one), Box::new(step_two)]);

    let res = runner.run();
    println!("{res:#?}");
}