From 26e2eddd1efeb0a5fff8ffabebefdae48c4a65dc Mon Sep 17 00:00:00 2001
From: Silas Bartha <silas@exvacuum.dev>
Date: Thu, 21 Nov 2024 12:17:44 -0500
Subject: Thu Nov 21 12:17:44 PM EST 2024

---
 src/actor/components.rs | 52 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 src/actor/components.rs

(limited to 'src/actor/components.rs')

diff --git a/src/actor/components.rs b/src/actor/components.rs
new file mode 100644
index 0000000..43987c0
--- /dev/null
+++ b/src/actor/components.rs
@@ -0,0 +1,52 @@
+//! Components related to actors
+
+use bevy::{prelude::*, utils::HashMap};
+use yarnspinner::{compiler::{Compiler, File}, core::{Library, LineId}, runtime::{Dialogue, MemoryVariableStorage, StringTableTextProvider}};
+
+/// Main actor component, holds state about dialogue along with the dialogue runner itself
+#[derive(Component)]
+pub struct Actor {
+    /// Whether this actor is currently conversing
+    pub active: bool,
+    /// Yarnspinner dialogue runner
+    pub dialogue: Dialogue,
+    /// Yarnspinner dialogue metadata
+    pub metadata: HashMap<LineId, Vec<String>>,
+}
+
+impl Actor {
+    /// Create a new actor from the given source code, starting on the given start node, and with
+    /// the given function library
+    pub fn new(file_name: &str, source: &[u8], start_node: &str, function_library: &Library) -> Self {
+        let compilation = Compiler::new()
+            .add_file(File {
+                source: String::from_utf8_lossy(source).into(),
+                file_name: file_name.into(),
+            })
+            .compile()
+            .unwrap();
+
+        let mut base_language_string_table = std::collections::HashMap::new();
+        let mut metadata = HashMap::new();
+
+        for (k, v) in compilation.string_table {
+            base_language_string_table.insert(k.clone(), v.text);
+            metadata.insert(k, v.metadata);
+        }
+
+        let mut text_provider = StringTableTextProvider::new();
+        text_provider.extend_base_language(base_language_string_table);
+
+        let mut dialogue = Dialogue::new(Box::new(MemoryVariableStorage::new()), Box::new(text_provider));
+        dialogue.library_mut().extend(function_library.clone());
+        dialogue.add_program(compilation.program.unwrap());
+        dialogue.set_node(start_node).unwrap();
+
+        Self {
+            active: false,
+            dialogue,
+            metadata,
+        }
+    }
+}
+
-- 
cgit v1.2.3