aboutsummaryrefslogtreecommitdiff
path: root/server/src/bin/graph.rs
blob: 0aac546878b8f2cbe363a99e9cf2cbf1a8f14740 (plain)
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
/*
    Undercooked - a game about cooking
    Copyright 2024 metamuffin

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, version 3 of the License only.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.

*/
use anyhow::{anyhow, Result};
use undercooked::{
    data::DataIndex,
    interaction::Recipe,
    protocol::{ItemIndex, RecipeIndex},
};

fn main() -> Result<()> {
    let mut index = DataIndex::default();
    index.reload()?;

    println!("digraph {{");

    let rn = std::env::args()
        .nth(1)
        .ok_or(anyhow!("first arg should be recipe set name"))?;

    let data = index.generate(format!("none-lobby-{rn}"))?;

    for i in 0..data.item_names.len() {
        println!("i{i} [label=\"{}\"]", data.item_name(ItemIndex(i)))
    }
    for (RecipeIndex(ri), recipe) in data.recipes() {
        let (kind, color) = match recipe {
            Recipe::Passive { .. } => ("Passive", "#2bc493"),
            Recipe::Active { .. } => ("Active", "#47c42b"),
            Recipe::Instant { .. } => ("Instant", "#5452d8"),
        };
        println!(
            "r{ri} [label=\"{kind}\\non {}\" shape=box color={color:?} fillcolor={color:?} style=filled]",
            if let Some(tile) = recipe.tile() {
                data.tile_name(tile)
            } else {
                "anything"
            }
        );
        for ItemIndex(input) in recipe.inputs() {
            println!("i{input} -> r{ri}")
        }
        for ItemIndex(output) in recipe.outputs() {
            println!("r{ri} -> i{output}")
        }
    }

    for (di, d) in data.demands.iter().enumerate() {
        let color = "#c4422b";
        println!(
            "d{di} [label=\"Demand\\ntakes {}s\" shape=box color={color:?} fillcolor={color:?} style=filled]",
            d.duration
        );
        println!("i{} -> d{di}", d.from.0);
        println!("d{di} -> i{}", d.to.0);
    }

    println!("}}");
    Ok(())
}