rngo-cli 0.20.0

Data simulation CLI using api.rngo.dev
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147

use anyhow::{Context, Result, anyhow, bail};
use serde_json::{Map, Value};
use std::fs;
use std::path::Path;

use crate::util::config::Config;

pub fn load_spec_from_file(spec_path: String) -> Result<Value> {
    let path = Path::new(&spec_path);

    if !path.exists() {
        bail!("Could not find file '{}'", spec_path)
    }

    let file_content = fs::read_to_string(path)?;
    serde_yaml::from_str(&file_content)
        .with_context(|| format!("Failed to parse spec file at {}", path.to_string_lossy()))
}

pub fn load_spec_from_project_directory(config: &Config) -> Result<Value> {
    let rngo_path = Path::new(".rngo");
    let effects_path = rngo_path.join("effects");

    let effect_files = fs::read_dir(effects_path.clone()).with_context(|| {
        format!(
            "Failed to read from effects directory at '{}'",
            effects_path.to_string_lossy()
        )
    })?;

    let mut effects_map = Map::new();

    for entry in effect_files {
        let entry = entry?;
        let path = entry.path();

        let content = fs::read_to_string(&path)?;
        let yaml_value: serde_yaml::Value = serde_yaml::from_str(&content).with_context(|| {
            format!("Failed to parse effect file at {}", path.to_string_lossy())
        })?;
        let mut json_value: serde_json::Value = serde_json::to_value(yaml_value)?;

        if let Some(obj) = json_value.as_object_mut() {
            obj.entry("type").or_insert_with(|| "state.create".into());
        }

        if let Some(filename) = path.file_stem().and_then(|s| s.to_str()) {
            effects_map.insert(filename.to_string(), json_value);
        }
    }

    if effects_map.is_empty() {
        bail!(
            "No effects found under {}",
            effects_path.to_string_lossy()
        )
    }

    let systems_map = load_systems_from_project_directory()?;

    let mut spec = Map::new();
    spec.insert("seed".into(), config.seed.into());

    if let Some(key) = &config.key {
        spec.insert("key".into(), key.clone().into());
    } else {
        let dir_name = std::env::current_dir()
            .ok()
            .and_then(|dir| {
                dir.file_name()
                    .and_then(|s| s.to_str())
                    .map(|s| s.to_string())
            })
            .ok_or_else(|| anyhow!("Failed to get current directory"))?;

        spec.insert("key".into(), dir_name.into());
    }

    if let Some(start) = &config.start {
        spec.insert("start".into(), start.clone().into());
    }

    if let Some(end) = &config.end {
        spec.insert("end".into(), end.clone().into());
    }

    if !systems_map.is_empty() {
        spec.insert("systems".into(), serde_json::Value::Object(systems_map));
    }
    spec.insert("effects".into(), serde_json::Value::Object(effects_map));

    Ok(serde_json::Value::Object(spec))
}

pub fn load_systems_from_project_directory() -> Result<Map<String, Value>> {
    let rngo_path = Path::new(".rngo");
    let systems_path = rngo_path.join("systems");
    let mut systems_map = Map::new();

    if systems_path.is_dir() {
        let system_files = fs::read_dir(systems_path.clone()).with_context(|| {
            format!(
                "Failed to read from systems directory at '{}'",
                systems_path.to_string_lossy()
            )
        })?;

        for system in system_files {
            let system = system?;
            let path = system.path();

            let content = fs::read_to_string(&path)?;
            let yaml_value: serde_yaml::Value = serde_yaml::from_str(&content)
                .with_context(|| format!("Failed to parse file at {}", path.to_string_lossy()))?;
            let json_value: serde_json::Value = serde_json::to_value(yaml_value)?;

            if let Some(filename) = path.file_stem().and_then(|s| s.to_str()) {
                systems_map.insert(filename.to_string(), json_value);
            }
        }
    }

    Ok(systems_map)
}

pub fn ensure_spec_output_is_stream(mut spec: Value) -> Value {
    match spec {
        Value::Object(ref mut map) => {
            map.insert("output".into(), "stream".into());
            spec
        }
        _ => spec,
    }
}

pub fn get_spec_key(spec: &mut Value) -> Option<String> {
    match spec {
        Value::Object(map) => {
            if let Some(key) = map.remove("key") {
                key.as_str().map(|s| s.to_string())
            } else {
                None
            }
        }
        _ => None,
    }
}