1 files changed, 35 insertions, 28 deletions

diff --git a/server/src/bin/graph.rs b/server/src/bin/graph.rs
index aa65e91a..94dddbc4 100644
--- a/server/src/bin/graph.rs
+++ b/server/src/bin/graph.rs
@@ -1,41 +1,46 @@
 /*
     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::Result;
 use undercooked::{
+    data::DataIndex,
     interaction::Recipe,
-    load_gamedata,
     protocol::{ItemIndex, RecipeIndex},
 };
 
-fn main() {
-    let data = load_gamedata();
+fn main() -> Result<()> {
+    let mut index = DataIndex::default();
+    index.reload()?;
 
     println!("digraph {{");
 
-    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!(
+    for rn in &index.recipes {
+        let data = index.generate(format!("default-none-{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)
@@ -43,23 +48,25 @@ fn main() {
                 "anything"
             }
         );
-        for ItemIndex(input) in recipe.inputs() {
-            println!("i{input} -> r{ri}")
-        }
-        for ItemIndex(output) in recipe.outputs() {
-            println!("r{ri} -> i{output}")
+            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!(
+        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!("i{} -> d{di}", d.from.0);
+            println!("d{di} -> i{}", d.to.0);
+        }
     }
 
     println!("}}");
+    Ok(())
 }