use super::hook::{DefaultDungeonHook, DungeonHook};
use super::service::DungeonService;
use super::system::DungeonSystem;
use super::types::{DungeonConfig, DungeonState};
use crate::Plugin;
use std::sync::Arc;
#[derive(Plugin)]
#[plugin(name = "issun:dungeon")]
pub struct DungeonPlugin {
hook: Arc<dyn DungeonHook>,
#[resource]
config: DungeonConfig,
#[state]
state: DungeonState,
#[service]
service: DungeonService,
#[system]
system: DungeonSystem,
}
impl DungeonPlugin {
pub fn new() -> Self {
let hook = Arc::new(DefaultDungeonHook);
Self {
hook: hook.clone(),
config: DungeonConfig::default(),
state: DungeonState::default(),
service: DungeonService::new(),
system: DungeonSystem::new(hook),
}
}
pub fn with_hook<H: DungeonHook + 'static>(mut self, hook: H) -> Self {
let hook = Arc::new(hook);
self.hook = hook.clone();
self.system = DungeonSystem::new(hook);
self
}
pub fn with_config(mut self, config: DungeonConfig) -> Self {
self.config = config;
self
}
}
impl Default for DungeonPlugin {
fn default() -> Self {
Self::new()
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::plugin::Plugin;
#[test]
fn test_plugin_creation() {
let plugin = DungeonPlugin::new();
assert_eq!(plugin.name(), "issun:dungeon");
}
#[test]
fn test_plugin_with_custom_hook() {
struct CustomHook;
#[async_trait::async_trait]
impl DungeonHook for CustomHook {}
let plugin = DungeonPlugin::new().with_hook(CustomHook);
assert_eq!(plugin.name(), "issun:dungeon");
}
#[test]
fn test_plugin_with_custom_config() {
use super::super::types::ConnectionPattern;
let config = DungeonConfig {
total_floors: 10,
rooms_per_floor: 5,
connection_pattern: ConnectionPattern::Branching,
};
let plugin = DungeonPlugin::new().with_config(config);
assert_eq!(plugin.name(), "issun:dungeon");
}
}