2023-12-21 20:11:32 -05:00
|
|
|
use core::{panic};
|
2023-12-21 00:48:09 -05:00
|
|
|
use std::error::Error;
|
|
|
|
|
|
|
|
use std::collections::HashMap;
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
|
|
|
ModulesManager is made of modulesdb , a HashMap of BotModules with a Vector representing their enabled/disabled status based on channel and instance
|
|
|
|
|
|
|
|
Example
|
|
|
|
{
|
|
|
|
BotModule("Experiments") , [Enabled(Channel("modulatingforce")) , Disabled(Channel("modulatingforce")), Enabled(Instance)]
|
|
|
|
}
|
|
|
|
|
|
|
|
*/
|
|
|
|
|
2023-12-21 17:22:40 -05:00
|
|
|
#[derive(Debug, PartialEq, Eq, Hash, Clone)]
|
2023-12-21 00:48:09 -05:00
|
|
|
pub enum ModType {
|
|
|
|
BotModule(String),
|
|
|
|
}
|
|
|
|
|
|
|
|
pub use ModType::BotModule;
|
|
|
|
|
|
|
|
#[derive(Debug, PartialEq, Eq, Hash)]
|
|
|
|
pub enum ChType {
|
|
|
|
Channel(String),
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
pub use ChType::Channel;
|
|
|
|
|
2023-12-21 20:11:32 -05:00
|
|
|
use crate::core::botinstance::{self, BotInstance};
|
|
|
|
|
2023-12-21 00:48:09 -05:00
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
enum StatusLvl {
|
|
|
|
Instance,
|
|
|
|
Ch(ChType),
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
pub enum ModStatusType {
|
|
|
|
Enabled(StatusLvl),
|
|
|
|
Disabled(StatusLvl),
|
|
|
|
}
|
|
|
|
|
|
|
|
// pub use EnType::Enabled;
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
2023-12-21 17:22:40 -05:00
|
|
|
pub enum BotAction {
|
2023-12-21 00:48:09 -05:00
|
|
|
C(BotCommand),
|
|
|
|
L(Listener),
|
|
|
|
R(Routine),
|
|
|
|
}
|
|
|
|
#[derive(Debug)]
|
2023-12-21 20:11:32 -05:00
|
|
|
struct BotCommand {
|
|
|
|
module : ModType,
|
|
|
|
command : String, // command call name
|
|
|
|
alias : Vec<String>, // String of alternative names
|
|
|
|
// bot_prefix : char, // although should be global?
|
|
|
|
// exec_body : fn,
|
|
|
|
help : String,
|
|
|
|
}
|
2023-12-21 00:48:09 -05:00
|
|
|
|
2023-12-21 20:43:10 -05:00
|
|
|
impl BotCommand {
|
|
|
|
pub fn add_to_bot(self, bot:BotInstance) {
|
|
|
|
// let mut mgr = bot.botmodules;
|
|
|
|
// let nmod = self.module.clone();
|
|
|
|
// mgr.add_botaction(nmod, BotAction::C(self));
|
|
|
|
let mut mgr = bot.botmodules;
|
|
|
|
//let nmod = self.module.clone();
|
|
|
|
// mgr.add_botaction(self.module.clone(), BotAction::C(self));
|
|
|
|
|
|
|
|
self.add_to_modmgr(&mut mgr);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
pub fn add_to_modmgr(self, modmgr:&mut ModulesManager) {
|
|
|
|
// // let mut mgr = bot.botmodules;
|
|
|
|
// // let nmod = self.module.clone();
|
|
|
|
// // mgr.add_botaction(nmod, BotAction::C(self));
|
|
|
|
// let mut mgr = modmgr;
|
|
|
|
// //let nmod = self.module.clone();
|
|
|
|
// mgr.add_botaction(self.module.clone(), BotAction::C(self));
|
|
|
|
// let mut mgr = modmgr;
|
|
|
|
// //let nmod = self.module.clone();
|
|
|
|
modmgr.add_botaction(self.module.clone(), BotAction::C(self));
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2023-12-21 00:48:09 -05:00
|
|
|
#[derive(Debug)]
|
|
|
|
struct Listener {
|
|
|
|
module : ModType,
|
|
|
|
name : String,
|
|
|
|
// exec_body : fn,
|
|
|
|
help : String
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
struct Routine {}
|
|
|
|
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
pub struct ModulesManager {
|
2023-12-21 12:10:50 -05:00
|
|
|
statusdb: HashMap<ModType,Vec<ModStatusType>>,
|
2023-12-21 00:48:09 -05:00
|
|
|
botactions: HashMap<ModType,Vec<BotAction>>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl ModulesManager {
|
|
|
|
|
|
|
|
pub fn init() -> ModulesManager {
|
|
|
|
|
|
|
|
// initializes the modulers manager
|
|
|
|
// Ideally, this should have added known modules based on
|
|
|
|
// directory structure and API user recommendations
|
|
|
|
|
|
|
|
let mut m = HashMap::new();
|
|
|
|
let mut act = HashMap::new();
|
|
|
|
|
|
|
|
// -- some processing including adding into the hashmap
|
|
|
|
|
|
|
|
// let newmodule = BotModule(String::from("GambaCore"));
|
|
|
|
|
2023-12-21 17:22:40 -05:00
|
|
|
// let newlistener = Listener {
|
|
|
|
// module : BotModule(String::from("experiments").to_owned()),
|
|
|
|
// name : String::from("socklistener"),
|
|
|
|
// help : String::from("This will listen and react to sock randomly"),
|
|
|
|
// };
|
2023-12-21 00:48:09 -05:00
|
|
|
|
|
|
|
|
2023-12-21 17:22:40 -05:00
|
|
|
// // As a Demonstration, the listener's Module is added and Enabled at Instance level
|
|
|
|
// let statusvector = m
|
|
|
|
// .entry(BotModule(String::from("experiments")))
|
|
|
|
// .or_insert(Vec::new());
|
2023-12-21 00:48:09 -05:00
|
|
|
|
2023-12-21 17:22:40 -05:00
|
|
|
// statusvector.push(ModStatusType::Enabled(StatusLvl::Instance));
|
2023-12-21 00:48:09 -05:00
|
|
|
|
2023-12-21 17:22:40 -05:00
|
|
|
// let modactions = act
|
|
|
|
// .entry( BotModule(String::from("experiments")))
|
|
|
|
// .or_insert(Vec::new());
|
2023-12-21 00:48:09 -05:00
|
|
|
|
2023-12-21 17:22:40 -05:00
|
|
|
// modactions.push(BotAction::L(newlistener));
|
2023-12-21 00:48:09 -05:00
|
|
|
|
|
|
|
|
2023-12-21 20:11:32 -05:00
|
|
|
let mut mgr = ModulesManager {
|
2023-12-21 12:10:50 -05:00
|
|
|
statusdb : m,
|
2023-12-21 00:48:09 -05:00
|
|
|
botactions : act,
|
|
|
|
};
|
|
|
|
|
2023-12-21 20:11:32 -05:00
|
|
|
// // -- This area can be where bot actions are imported to the bot's module manager
|
|
|
|
|
|
|
|
// // -- Below can be used to validate what occurs when dup BotCommands are added
|
|
|
|
|
|
|
|
// let bcmd = BotCommand {
|
|
|
|
// module : BotModule(String::from("experiments 001")),
|
|
|
|
// command : String::from("DUPCMD1"), // command call name
|
|
|
|
// alias : vec![String::from("DUPALIAS1A"),String::from("DUPALIAS1B")], // String of alternative names
|
|
|
|
// // bot_prefix : char, // although should be global?
|
|
|
|
// // exec_body : fn,
|
|
|
|
// help : String::from("DUPCMD1 tester"),
|
|
|
|
// };
|
|
|
|
|
|
|
|
// mgr.add_botaction(BotModule(String::from("experiments 001")), BotAction::C(bcmd));
|
|
|
|
|
|
|
|
|
|
|
|
// let bcmd = BotCommand {
|
|
|
|
// module : BotModule(String::from("experiments 002")),
|
|
|
|
// command : String::from("DUPCMD2"), // command call name
|
|
|
|
// alias : vec![String::from("DUPALIAS2A"),String::from("DUPALIAS2B")], // String of alternative names
|
|
|
|
// // bot_prefix : char, // although should be global?
|
|
|
|
// // exec_body : fn,
|
|
|
|
// help : String::from("DUPCMD2 tester"),
|
|
|
|
// };
|
|
|
|
|
|
|
|
// mgr.add_botaction(BotModule(String::from("experiments 002")), BotAction::C(bcmd));
|
2023-12-21 17:22:40 -05:00
|
|
|
|
|
|
|
|
2023-12-21 20:43:10 -05:00
|
|
|
BotCommand {
|
|
|
|
module : BotModule(String::from("experiments 002")),
|
|
|
|
command : String::from("DUPCMD2"), // command call name
|
|
|
|
alias : vec![String::from("DUPALIAS2A"),String::from("DUPALIAS2B")], // String of alternative names
|
|
|
|
// bot_prefix : char, // although should be global?
|
|
|
|
// exec_body : fn,
|
|
|
|
help : String::from("DUPCMD2 tester"),
|
|
|
|
}.add_to_modmgr(&mut mgr);
|
|
|
|
|
|
|
|
|
2023-12-21 00:48:09 -05:00
|
|
|
println!(">> Modules Manager : {:?}",mgr);
|
|
|
|
|
|
|
|
mgr
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn modstatus(&self, _:ModType, _:ChType) -> ModStatusType {
|
|
|
|
// Example usage : botmanager.modstatus(
|
|
|
|
// BotModule("GambaCore"),
|
|
|
|
// Channel("modulatingforce")
|
|
|
|
// )
|
|
|
|
// - The ModStatusType checks in the context of the given channel ,
|
|
|
|
// but also validates based on wheher the module is disabled at a bot instance
|
|
|
|
// level as well
|
|
|
|
ModStatusType::Enabled(StatusLvl::Instance)
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
pub fn togglestatus(&self, _:ModType, _:ChType) -> ModStatusType {
|
|
|
|
// enables or disables based on current status
|
|
|
|
ModStatusType::Enabled(StatusLvl::Instance)
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
pub fn setstatus(&self, _:ModType, _:ModStatusType) -> Result<&str,Box<dyn Error>> {
|
|
|
|
// sets the status based given ModSatusType
|
|
|
|
// e.g., b.setstatus(BodModule("GambaCore"), Enabled(Channel("modulatingforce"))).expect("ERROR")
|
|
|
|
Ok("")
|
|
|
|
}
|
|
|
|
|
2023-12-21 17:22:40 -05:00
|
|
|
|
2023-12-21 20:11:32 -05:00
|
|
|
pub fn add_botaction(&mut self, in_module:ModType, in_action:BotAction ) -> () {
|
|
|
|
/*
|
|
|
|
adds a BotAction to the Modules Manager - This will require a BotModule passed as well
|
|
|
|
This will including the logic of a valid add
|
|
|
|
If it fails to add, either a PANIC or some default coded business rules that handles the botaction add
|
|
|
|
For example, this Should PANIC (ideally Panic?) if it does not successfully add a bot module
|
|
|
|
-- Being unable to indicates a Programming/Developer code logic issue : They cannot add botactions that already exists (?)
|
|
|
|
-- In particular to BotCommands, which must have Unique command call names and aliases that to not conflict with any other
|
|
|
|
already BotCommand added name or alias
|
|
|
|
Other types might be fine? For example, if 2 modules have their own listeners but each have the name "targetchatter" ,
|
|
|
|
both would be called separately, even if they both have the same or different logic
|
|
|
|
*/
|
|
|
|
|
|
|
|
// let newlistener = Listener {
|
|
|
|
// // module : BotModule(String::from("experiments").to_owned()),
|
|
|
|
// module : in_module.clone(),
|
|
|
|
// name : String::from("socklistener"),
|
|
|
|
// help : String::from("This will listen and react to sock randomly"),
|
|
|
|
// };
|
2023-12-21 17:22:40 -05:00
|
|
|
|
|
|
|
|
|
|
|
// As a Demonstration, the listener's Module is added and Enabled at Instance level
|
2023-12-21 20:11:32 -05:00
|
|
|
|
|
|
|
|
|
|
|
// [x] Before Adding, validate the following :
|
|
|
|
// - If BotAction to Add is a BotCommand , In Module Manager DB (botactions),
|
|
|
|
// Check All Other BotAction Command Names & Aliases to ensure they don't conflict
|
|
|
|
|
|
|
|
fn find_conflict_module(mgr:& ModulesManager, act:& BotAction) -> Option<ModType> {
|
|
|
|
|
|
|
|
// Some(BotModule(String::from("GambaCore")))
|
|
|
|
|
|
|
|
|
|
|
|
// match act {
|
|
|
|
// BotAction::C(c) => {
|
|
|
|
// Some(BotModule(String::from("GambaCore")))
|
|
|
|
// },
|
|
|
|
// BotAction::L(l) => None,
|
|
|
|
// BotAction::R(r) => None,
|
|
|
|
// }
|
|
|
|
|
|
|
|
if let BotAction::C(incmd) = act {
|
|
|
|
|
|
|
|
// let n = & mgr.botactions;
|
|
|
|
|
|
|
|
let d = &mgr.botactions;
|
|
|
|
|
|
|
|
for (module,moduleactions) in d {
|
|
|
|
|
|
|
|
|
|
|
|
for modact in moduleactions.iter() {
|
|
|
|
if let BotAction::C(dbcmd) = &modact {
|
|
|
|
// At this point, there is an command incmd and looked up dbcmd
|
|
|
|
|
|
|
|
// [x] check if given botcommand c.command:String conflicts with any in botactions
|
|
|
|
|
|
|
|
if incmd.command.to_lowercase() == dbcmd.command.to_lowercase() {
|
|
|
|
// Returning State - with the identified module
|
|
|
|
// return Some((module.clone(),BotAction::C(*dbcmd.clone())));
|
|
|
|
// return Some(incmd); // for some reason I keep getting issues
|
|
|
|
//return Some(BotModule(String::from("GambaCore"))); // works
|
|
|
|
return Some(module.clone()); // works
|
|
|
|
// return Some(dbcmd.clone());
|
|
|
|
}
|
|
|
|
|
|
|
|
for a in &dbcmd.alias {
|
|
|
|
if incmd.command.to_lowercase() == a.to_lowercase() {
|
|
|
|
// Returning State - with the identified module
|
|
|
|
// return Some((module.clone(),BotAction::C(dbcmd)));
|
|
|
|
return Some(module.clone()); // works
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// [x] Then do the same check except for each c.alias
|
|
|
|
|
|
|
|
for inalias in &incmd.alias {
|
|
|
|
|
|
|
|
if inalias.to_lowercase() == dbcmd.command.to_lowercase() {
|
|
|
|
// Returning State - with the identified module
|
|
|
|
// return Some((module.clone(),BotAction::C(dbcmd)));
|
|
|
|
return Some(module.clone()); // works
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
for a in &dbcmd.alias {
|
|
|
|
if inalias.to_lowercase() == a.to_lowercase() {
|
|
|
|
// Returning State - with the identified module
|
|
|
|
// return Some((module.clone(),BotAction::C(dbcmd)));
|
|
|
|
return Some(module.clone()); // works
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
// return Some(BotModule(String::from("GambaCore")))
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// for all other scenarios (e.g., Listener, Routine), find no conflicts
|
|
|
|
None
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
// if let probmod = find_conflict_module(&self, &in_action) {
|
|
|
|
// // () // return because there was a conflict?
|
|
|
|
// panic!("ERROR: Could not add {:?} ; there was a conflict with existing module {:?}", in_action , probmod );
|
|
|
|
// }
|
|
|
|
match find_conflict_module(&self, &in_action) {
|
|
|
|
Some(c) => panic!("ERROR: Could not add {:?} ; there was a conflict with existing module {:?}", in_action , c ),
|
|
|
|
None => (),
|
|
|
|
}
|
|
|
|
|
2023-12-21 17:22:40 -05:00
|
|
|
let statusvector = self.statusdb
|
|
|
|
// .entry(BotModule(String::from("experiments")))
|
|
|
|
.entry(in_module.clone())
|
|
|
|
.or_insert(Vec::new());
|
|
|
|
|
2023-12-21 20:11:32 -05:00
|
|
|
statusvector.push(ModStatusType::Enabled(StatusLvl::Instance)); // Pushes the Module as Enabled at Instance Level
|
2023-12-21 17:22:40 -05:00
|
|
|
|
|
|
|
let modactions = self.botactions
|
|
|
|
//.entry( BotModule(String::from("experiments")))
|
|
|
|
.entry( in_module.clone())
|
|
|
|
.or_insert(Vec::new());
|
|
|
|
|
2023-12-21 20:11:32 -05:00
|
|
|
// modactions.push(BotAction::L(newlistener));
|
|
|
|
modactions.push(in_action);
|
2023-12-21 17:22:40 -05:00
|
|
|
|
|
|
|
();
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
2023-12-21 00:48:09 -05:00
|
|
|
|
2023-12-21 17:22:40 -05:00
|
|
|
fn statuscleanup(&self,_:Option<ChType>) -> () {
|
2023-12-21 12:10:50 -05:00
|
|
|
// internal cleans up statusdb . For example :
|
|
|
|
// - remove redudancies . If we see several Enabled("m"), only keep 1x
|
|
|
|
// - Clarify Conflict. If we see Enabled("m") and Disabled("m") , we remove Enabled("m") and keep Disabled("m")
|
|
|
|
// the IDEAL is that this is ran before every read/update operation to ensure quality
|
|
|
|
// Option<ChType> can pass Some(Channel("m")) (as an example) so statuscleanup only works on the given channel
|
|
|
|
// Passing None to chnl may be a heavy operation, as this will review and look at the whole table
|
|
|
|
()
|
|
|
|
}
|
2023-12-21 00:48:09 -05:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|