implement more models and rhai examples

heromodels/src/models/calendar/rhai.rs

@@ -0,0 +1,79 @@
+use rhai::{Engine, EvalAltResult, NativeCallContext};
+use std::sync::Arc;
+
+use heromodels_core::BaseModelData;
+use crate::db::hero::OurDB;
+use super::calendar::{Calendar, Event, Attendee, AttendanceStatus};
+use adapter_macros::{adapt_rhai_i64_input_fn, adapt_rhai_i64_input_method};
+use rhai_wrapper::wrap_vec_return;
+
+// Helper function for get_all_calendars registration
+fn get_all_calendars_helper(_db: Arc<OurDB>) -> Vec<Calendar> { 
+    // In a real implementation, this would retrieve all calendars from the database
+    vec![Calendar::new(1 as u32), Calendar::new(2 as u32)] 
+}
+
+pub fn register_rhai_engine_functions(engine: &mut Engine, db: Arc<OurDB>) {
+    engine.register_fn("new_calendar", adapt_rhai_i64_input_fn!(Calendar::new, u32));
+    engine.register_fn("name", move |calendar: Calendar, name: String| Calendar::name(calendar, name));
+    engine.register_fn("description", move |calendar: Calendar, description: String| Calendar::description(calendar, description));
+    engine.register_fn("add_event", Calendar::add_event); // Corrected: expects i64, Rhai provides i64
+    
+    engine.register_fn("new_event", Event::new); // Corrected: expects i64, Rhai provides i64
+    engine.register_fn("title", move |event: Event, title: String| Event::title(event, title));
+    engine.register_fn("description", move |event: Event, description: String| Event::description(event, description));
+    engine.register_fn("add_attendee", move |event: Event, attendee: Attendee| Event::add_attendee(event, attendee));
+    engine.register_fn("remove_attendee", adapt_rhai_i64_input_method!(Event, remove_attendee, u32));
+    engine.register_fn("update_attendee_status", move |context: NativeCallContext, event: Event, contact_id_i64: i64, status: AttendanceStatus| -> Result<Event, Box<EvalAltResult>> {
+        let contact_id_u32: u32 = contact_id_i64.try_into().map_err(|_e| {
+            Box::new(EvalAltResult::ErrorArithmetic(
+                format!("Conversion error for contact_id in Event::update_attendee_status from i64 to u32"),
+                context.position(),
+            ))
+        })?;
+        Ok(event.update_attendee_status(contact_id_u32, status))
+    });
+
+    engine.register_fn("new_attendee", adapt_rhai_i64_input_fn!(Attendee::new, u32));
+
+    // Register a function to get the database instance
+    engine.register_fn("get_db", move || db.clone());
+    
+    // Register getters for Calendar
+    engine.register_get("id", |c: &mut Calendar| -> Result<i64, Box<EvalAltResult>> { Ok(c.base_data.id as i64) });
+    engine.register_get("name", |c: &mut Calendar| -> Result<String, Box<EvalAltResult>> {
+        // println!("Rhai attempting to get Calendar.name: {}", c.name); // Debug print
+        Ok(c.name.clone())
+    });
+    engine.register_get("description", |c: &mut Calendar| -> Result<Option<String>, Box<EvalAltResult>> { Ok(c.description.clone()) }); 
+
+    // Register getter for Calendar.base_data
+    engine.register_get("base_data", |c: &mut Calendar| -> Result<BaseModelData, Box<EvalAltResult>> { Ok(c.base_data.clone()) });
+
+    // Register getters for BaseModelData
+    engine.register_get("id", |bmd: &mut BaseModelData| -> Result<i64, Box<EvalAltResult>> { Ok(bmd.id.into()) }); 
+
+    // Mock database interaction functions from the example - these would typically interact with the `db` Arc
+    engine.register_fn("set_calendar", |_db: Arc<OurDB>, calendar: Calendar| { 
+        println!("Mock save: Calendar saved: {}", calendar.name);
+    });
+    
+    engine.register_fn("get_calendar_by_id", |_db: Arc<OurDB>, id_i64: i64| -> Calendar { 
+        Calendar::new(id_i64 as u32) 
+    });
+    
+    engine.register_fn("calendar_exists", |_db: Arc<OurDB>, id_i64: i64| -> bool { 
+        id_i64 == 1 || id_i64 == 2 // Mock check
+    });
+        
+    engine.register_fn("get_all_calendars", wrap_vec_return!(get_all_calendars_helper, Arc<OurDB> => Calendar)); 
+    
+    engine.register_fn("delete_calendar_by_id", |_db: Arc<OurDB>, id_i64: i64| { 
+        println!("Mock delete: Calendar deleted with ID: {}", id_i64);
+    });
+
+    // Getters for Event
+    engine.register_get("id", |e: &mut Event| -> Result<i64, Box<EvalAltResult>> { Ok(e.base_data.id as i64) });
+    engine.register_get("title", |e: &mut Event| -> Result<String, Box<EvalAltResult>> { Ok(e.title.clone()) });
+    // Add more getters for Event fields as needed
+}