forcebot_rs/src/core/identity.rs

820 lines
32 KiB
Rust
Raw Normal View History

2024-01-29 06:18:27 -05:00
2024-02-04 14:28:37 -05:00
use std::borrow::Borrow;
2024-01-29 06:18:27 -05:00
use std::collections::HashMap;
2024-01-29 12:41:26 -05:00
use std::error::Error;
2024-01-29 11:09:33 -05:00
use crate::core::botmodules::{ModulesManager,Listener,BotModule,BotActionTrait, BotCommand};
use crate::core::botmodules::bot_actions::actions_util;
2024-02-04 14:28:37 -05:00
use crate::core::botinstance::{self,BotInstance};
use futures::lock::Mutex;
2024-01-29 22:57:07 -05:00
use twitch_irc::message::{Badge, PrivmsgMessage};
2024-01-29 11:09:33 -05:00
2024-01-29 12:41:26 -05:00
use crate::core::botmodules::ChType;
2024-02-04 14:28:37 -05:00
use crate::core::botinstance::ArcBox;
use std::rc::Rc;
use std::cell::RefCell;
2024-02-12 05:25:38 -05:00
use std::sync::{Arc};
use tokio::sync::RwLock;
2024-02-04 14:28:37 -05:00
2024-02-12 01:25:12 -05:00
use super::botmodules::bot_actions::actions_util::BotAR;
2024-01-29 06:18:27 -05:00
fn adminvector() -> Vec<String> {
vec![String::from("ModulatingForce")]
}
2024-02-04 14:28:37 -05:00
// pub fn init(mgr:&mut ModulesManager)
2024-02-12 02:34:32 -05:00
pub async fn init(mgr:Arc<ModulesManager>)
2024-01-29 11:09:33 -05:00
{
2024-02-12 02:34:32 -05:00
println!("Went into Identiy Module init");
2024-02-12 01:25:12 -05:00
2024-02-12 02:34:32 -05:00
// let a = actions_util::asyncbox(cmd_promote) ;
let tempb = BotCommand {
2024-01-29 11:09:33 -05:00
module : BotModule(String::from("identity")),
command : String::from("promote"), // command call name
alias : vec![], // String of alternative names
exec_body : actions_util::asyncbox(cmd_promote) ,
help : String::from("promote"),
required_roles : vec![
UserRole::Mod(ChType::Channel(String::new())),
UserRole::SupMod(ChType::Channel(String::new())),
UserRole::Broadcaster,
UserRole::BotAdmin,
],
2024-02-12 02:34:32 -05:00
};
2024-02-12 05:25:38 -05:00
tempb.add_to_modmgr(Arc::clone(&mgr)).await;
2024-01-29 11:09:33 -05:00
2024-02-12 01:25:12 -05:00
async fn cmd_promote(bot:BotAR,msg:PrivmsgMessage) -> ()
2024-02-04 14:28:37 -05:00
{
2024-01-29 11:09:33 -05:00
//println!("(#{}) {}: {}", msg.channel_login, msg.sender.name, msg.message_text);
2024-01-29 22:57:07 -05:00
println!("Called cmd promote");
2024-01-29 12:41:26 -05:00
2024-01-31 21:30:08 -05:00
// -- If the BotCommand.command was called (e.g., promote) & required roles were validated OUTSIDE of this call
// , this is the current function body to execute
2024-01-30 15:07:40 -05:00
/*
- `promote` / `demote`
- [ ] `SupMod` & `Broadcaster` & `BotAdmin` can run
- [ ] `UserRole`s that can run, can
- [ ] run `promote` on a regular `Chatter` to make them a `Mod`
- [ ] run `demote` on a `Mod` to make them a `Chatter`
- [ ] Only `BotAdmin` can :
- [ ] target themselves to `promote` / `demote` , in the case that they want to make themselves either a `Mod` or `SupMod` for the channel temporarily
- [ ] `promote admin <Chatter>` to assign them `BotAdmin` role
- `[ ] Broadcaster` & `BotAdmin` can `demote` a `SupMod` to make them a `Mod`
*/
2024-01-31 21:30:08 -05:00
/*
Usage :
promote <user> <channel>
demote <user> <channel>
promote admin <user>
*/
2024-02-04 14:28:37 -05:00
2024-02-01 08:40:09 -05:00
println!("{}",msg.message_text);
let mut argv = msg.message_text.split(" ");
argv.next(); // Skip the command name
let arg1 = argv.next();
let arg2 = argv.next();
let mut sender_badge:Option<ChatBadge> = None;
for b in &msg.badges {
if b.name == "moderator" {
sender_badge = Some(ChatBadge::Mod);
} else if b.name == "broadcaster" {
sender_badge = Some(ChatBadge::Broadcaster);
}
}
match arg1 {
2024-02-04 14:28:37 -05:00
Some(a1) if a1 == String::from("admin") => {
2024-02-01 08:40:09 -05:00
// - BotAdmins can promote admin to give BotAdmin UserRole
2024-02-12 14:54:35 -05:00
2024-02-12 01:25:12 -05:00
let botlock = bot.read().await;
2024-02-12 05:25:38 -05:00
let ta = botlock.get_identity();
let ta = ta.read().await;
let ta = ta.getspecialuserroles(String::from("Hello"), Some(ChType::Channel(msg.channel_login.to_lowercase()))).await;
let ta = ta.unwrap();
let a = ta.read().await;
// let ta = *ta;
// let ta = *ta;
// if let Some(a) = *ta {
2024-02-01 08:40:09 -05:00
if a.contains(&UserRole::BotAdmin) {
println!("BotAdmin allowed to promote admin");
2024-02-04 14:28:37 -05:00
{
2024-02-12 14:54:35 -05:00
2024-02-12 01:25:12 -05:00
let botlock = Arc::clone(&bot.read().await.get_identity());
let idlock = botlock.write().await;
// let mut idlock = *idlock;
let ta = idlock.promote(msg.sender.name.to_lowercase(), None, Some(UserRole::BotAdmin)).await;
2024-02-04 14:28:37 -05:00
match ta {
ChangeResult::Success(a) => println!("Succesfully promoted : {a} ;"),
ChangeResult::Failed(a) => println!("Failed to promote : {a} ; "),
ChangeResult::NoChange(a) => println!("No Changes Made : {a} ; "),
}
2024-02-01 08:40:09 -05:00
}
}
}
2024-02-12 14:54:35 -05:00
2024-02-12 05:25:38 -05:00
// },
_ => (),
2024-02-01 08:40:09 -05:00
}
let arg2 = argv.next();
let targetchnl = arg2;
2024-01-29 11:09:33 -05:00
}
2024-02-12 02:34:32 -05:00
// BotCommand {
// module : BotModule(String::from("identity")),
// command : String::from("demote"), // command call name
// alias : vec![], // String of alternative names
// exec_body : actions_util::asyncbox(cmd_demote) ,
// help : String::from("demote"),
// required_roles : vec![
// UserRole::Mod(ChType::Channel(String::new())),
// UserRole::SupMod(ChType::Channel(String::new())),
// UserRole::Broadcaster,
// UserRole::BotAdmin,
// ],
// }.add_to_modmgr(Arc::clone(&mgr));
let tempb = BotCommand {
2024-01-29 11:09:33 -05:00
module : BotModule(String::from("identity")),
command : String::from("demote"), // command call name
alias : vec![], // String of alternative names
exec_body : actions_util::asyncbox(cmd_demote) ,
help : String::from("demote"),
required_roles : vec![
UserRole::Mod(ChType::Channel(String::new())),
UserRole::SupMod(ChType::Channel(String::new())),
UserRole::Broadcaster,
UserRole::BotAdmin,
],
2024-02-12 02:34:32 -05:00
};
tempb.add_to_modmgr(Arc::clone(&mgr)).await;
2024-01-29 11:09:33 -05:00
2024-02-12 01:25:12 -05:00
// async fn cmd_demote(mut _chat:Arc<Mutex<BotInstance>>,_msg:PrivmsgMessage) {
async fn cmd_demote(mut _chat:BotAR,_msg:PrivmsgMessage) {
2024-01-29 22:57:07 -05:00
println!("Called cmd demote");
2024-01-29 11:09:33 -05:00
}
2024-01-31 21:30:08 -05:00
2024-02-12 02:34:32 -05:00
// BotCommand {
// module : BotModule(String::from("identity")),
// command : String::from("getroles"), // command call name
// alias : vec![], // String of alternative names
// exec_body : actions_util::asyncbox(getroles) ,
// help : String::from("getroles"),
// required_roles : vec![
// UserRole::Mod(ChType::Channel(String::new())),
// UserRole::SupMod(ChType::Channel(String::new())),
// UserRole::Broadcaster,
// UserRole::BotAdmin,
// ],
// }.add_to_modmgr(Arc::clone(&mgr));
let tempcomm = BotCommand {
2024-01-31 21:30:08 -05:00
module : BotModule(String::from("identity")),
command : String::from("getroles"), // command call name
alias : vec![], // String of alternative names
exec_body : actions_util::asyncbox(getroles) ,
help : String::from("getroles"),
required_roles : vec![
UserRole::Mod(ChType::Channel(String::new())),
UserRole::SupMod(ChType::Channel(String::new())),
UserRole::Broadcaster,
UserRole::BotAdmin,
],
2024-02-12 02:34:32 -05:00
};
tempcomm.add_to_modmgr(Arc::clone(&mgr)).await;
2024-01-31 21:30:08 -05:00
2024-02-12 05:25:38 -05:00
2024-02-12 01:25:12 -05:00
// async fn getroles(bot:Arc<Mutex<BotInstance>>,msg:PrivmsgMessage) {
async fn getroles(bot:BotAR,msg:PrivmsgMessage) {
2024-01-31 21:30:08 -05:00
println!("Called cmd getroles");
/*
Usage
getroles <user> <Channel>
- If channel is provided, provide roles for that channel specifically
*/
// IN other code areas , I see this
// ServerMessage::Privmsg(msg) => {
// println!("(#{}) {}: {}", msg.channel_login, msg.sender.name, msg.message_text);
2024-02-01 08:40:09 -05:00
// println!("{}",msg.message_text);
2024-01-31 21:30:08 -05:00
let mut argv = msg.message_text.split(" ");
// for v in argv {
// println!("args : {v}");
// }
2024-02-01 08:40:09 -05:00
argv.next(); // Skip the command name
2024-01-31 21:30:08 -05:00
let arg1 = argv.next();
// if arg == None {
// return ; // Do nothing if no arguments
// }
let targetuser = match arg1 {
None => return , // exit if no arguments
Some(arg) => arg,
};
// match String::from(arg1) {
// a if a == String::from("admin") => (),
// _ => (),
// }
// match argv[1] {
// String::from("admin") => (),
// }
let arg2 = argv.next();
let targetchnl = arg2;
2024-02-04 14:28:37 -05:00
// // let a = bot.read().ok().unwrap().get_identity();
// let a = bot.lock().await;
// // let a = a.lock().await;
// // let a = a.get_identity().await;
// let a = a.botmgrs;
// // let a = *(*a).lock().await;
// let a = *a.lock().await;
// let a = a.identity;
// let a = *a.lock().await;
// let a = bot.clone();
// let a = a.into_inner();
// let a = a.botmgrs;
// let a = a.into_inner();
// let a = a.identity;
// let a = a.into_inner();
2024-02-12 01:25:12 -05:00
// let a = bot.clone();
2024-02-04 14:28:37 -05:00
// let a = a.lock().await;
2024-02-12 01:25:12 -05:00
// let a = a.get_identity();
// let a = a.lock().await;
// let a = bot.get_identity();
let botlock = bot.read().await;
2024-02-12 05:25:38 -05:00
println!("botlock read");
2024-02-12 01:25:12 -05:00
let idlock = botlock.get_identity();
2024-02-12 05:25:38 -05:00
println!("got identity");
let idlock = idlock.read().await; // <-- 02.12 - Latest where it gest stuck - before or at this point
println!("id lock");
2024-02-04 14:28:37 -05:00
let sproles = match targetchnl {
2024-01-31 21:30:08 -05:00
None => {
2024-02-04 14:28:37 -05:00
// let bot = Rc::clone(&bot);
//let bot = Arc::clone(&bot);
// let a = bot.botmgrs.identity.getspecialuserroles(String::from(targetuser),None);
// let a = Arc::try_unwrap(bot).ok().unwrap().into_inner().ok().unwrap();
// let a = Arc::try_unwrap(bot.clone()).ok().unwrap().into_inner();
// let a = a.botmgrs.identity.getspecialuserroles(String::from(targetuser),None);
// let a = a.ok().getspecialuserroles(String::from(targetuser),None);
// let a = bot.read().ok().unwrap().rIdentity().getspecialuserroles(String::from(targetuser),None);
// println!("Retrieved User Roles >> {:?}",a);
2024-02-12 01:25:12 -05:00
// let a = idlock.read().await;
idlock.getspecialuserroles(String::from(targetuser),None).await
2024-01-31 21:30:08 -05:00
},
Some(targetchnl) => {
2024-02-04 14:28:37 -05:00
// let bot = Rc::clone(&bot);
// let bot = Arc::clone(&bot);
// let a = bot.botmgrs.identity.getspecialuserroles(String::from(targetuser), Some(ChType::Channel(String::from(targetchnl))));
// Arc::try_unwrap(boxed_bot.clone()).ok().unwrap().into_inner()
// let a = Arc::try_unwrap(bot).ok().unwrap().into_inner().ok().unwrap();
// let a = Arc::try_unwrap(bot.clone()).ok().unwrap().into_inner();
// let a = a.botmgrs.identity.getspecialuserroles(String::from(targetuser), Some(ChType::Channel(String::from(targetchnl))));
// let a = bot.read().ok().unwrap().rIdentity().getspecialuserroles(String::from(targetuser),None);
// println!("Retrieved User Roles >> {:?}",a);
// bot.read().ok().unwrap().rIdentity().getspecialuserroles(String::from(targetuser),None)
2024-02-12 01:25:12 -05:00
// let a = a.read().await;
idlock.getspecialuserroles(String::from(targetuser),None).await
2024-02-04 14:28:37 -05:00
2024-01-31 21:30:08 -05:00
},
2024-02-04 14:28:37 -05:00
};
2024-01-31 21:30:08 -05:00
2024-02-04 14:28:37 -05:00
println!("Retrieved User Roles >> {:?}",sproles);
2024-01-31 21:30:08 -05:00
// let a = bot.identity.getuserroles(String::from("ModulatingForce"), Some(ChType::Channel(String::from("ModulatingForcebot"))));
// println!("{:?}",a);
}
2024-02-12 02:34:32 -05:00
println!("End of Init MOdule add");
2024-01-31 21:30:08 -05:00
2024-01-29 11:09:33 -05:00
}
2024-01-29 22:57:07 -05:00
// #[derive(Debug, PartialEq, Eq, Hash, Clone)]
// pub enum ChType {
// Channel(String),
// }
#[derive(Debug, PartialEq, Eq , Clone)]
2024-01-29 11:09:33 -05:00
pub enum UserRole {
2024-01-29 06:18:27 -05:00
Chatter,
Mod(ChType), // String specifies Channel
SupMod(ChType), // String specifies Channel
2024-01-29 06:18:27 -05:00
Broadcaster,
BotAdmin,
2024-01-29 12:41:26 -05:00
2024-01-29 06:18:27 -05:00
}
2024-01-29 22:57:07 -05:00
pub enum Permissible {
2024-01-29 12:41:26 -05:00
Allow,
Block
}
2024-01-31 18:36:23 -05:00
#[derive(Clone)]
2024-01-29 06:18:27 -05:00
pub struct IdentityManager {
2024-02-04 14:28:37 -05:00
// special_roles_users : HashMap<String,Vec<UserRole>>, // # <-- (!) This must be String instead of ChType because we're checking a User not a Channel
2024-02-12 05:25:38 -05:00
// special_roles_users : Arc<Mutex<HashMap<String,Vec<UserRole>>>>, // # <-- (!) This must be String instead of ChType because we're checking a User not a Channel
// special_roles_users : Arc<RwLock<HashMap<String,Vec<UserRole>>>>,
special_roles_users : Arc<RwLock<HashMap<String,Arc<RwLock<Vec<UserRole>>>>>>,
2024-01-31 09:31:14 -05:00
// parent_mgr : Box<crate::core::botinstance::BotManagers>,
2024-01-31 21:30:08 -05:00
//parent_mgr : Option<Box<crate::core::botinstance::BotManagers>>,
2024-01-29 06:18:27 -05:00
}
2024-01-29 22:57:07 -05:00
pub enum ChatBadge {
2024-01-29 12:41:26 -05:00
Broadcaster,
Mod,
}
2024-01-29 06:18:27 -05:00
2024-02-13 07:54:35 -05:00
pub enum ChangeResult {
2024-01-31 21:30:08 -05:00
Success(String),
Failed(String),
NoChange(String),
}
2024-01-29 06:18:27 -05:00
impl IdentityManager {
pub fn init() -> IdentityManager {
let mut a = HashMap::new();
for admn in adminvector() {
2024-02-12 05:25:38 -05:00
a.insert(admn.to_lowercase(),Arc::new(RwLock::new(vec![UserRole::BotAdmin])));
2024-01-29 06:18:27 -05:00
};
IdentityManager {
2024-02-12 05:25:38 -05:00
special_roles_users : Arc::new(RwLock::new(a)),
2024-01-31 21:30:08 -05:00
//parent_mgr : None,
2024-01-29 06:18:27 -05:00
}
}
2024-01-29 12:41:26 -05:00
2024-01-29 22:57:07 -05:00
// [ ] Maybe I should create a can_user_run version that simply takes PrvMsg, but then calls can_user_run directly
// pub fn can_user_run_PRVMSG(self,msg:&PrivmsgMessage,cmdreqroles:Vec<UserRole>) -> Result<Permissible,Box<dyn Error>>
2024-02-04 14:28:37 -05:00
// pub fn can_user_run_PRVMSG(&self,msg:&PrivmsgMessage,cmdreqroles:Vec<UserRole>) -> Permissible
2024-02-12 01:25:12 -05:00
// pub async fn can_user_run_PRVMSG(self,msg:&PrivmsgMessage,cmdreqroles:Vec<UserRole>) -> Permissible
2024-02-13 07:54:35 -05:00
pub async fn can_user_run_PRVMSG(&mut self,msg:&PrivmsgMessage,cmdreqroles:Vec<UserRole>) -> (Permissible,ChangeResult)
2024-01-29 22:57:07 -05:00
{
// println!("(#{}) {}: {}", msg.channel_login, msg.sender.name, msg.message_text);
// [ ] Check what Badges in PrivmsgMessage
2024-02-12 05:25:38 -05:00
println!{"Checking within PRVMSG"};
2024-01-29 22:57:07 -05:00
let mut sender_badge:Option<ChatBadge> = None;
for b in &msg.badges {
if b.name == "moderator" {
sender_badge = Some(ChatBadge::Mod);
} else if b.name == "broadcaster" {
sender_badge = Some(ChatBadge::Broadcaster);
}
}
// if &msg.badges.contains(Badge{}) {
// }
if let Some(sender_badge) = sender_badge {
2024-01-31 18:36:23 -05:00
// return &self.can_user_run(msg.sender.name.to_owned(),
// ChType::Channel(msg.channel_login.to_owned()),
// sender_badge,
// cmdreqroles
2024-02-04 14:28:37 -05:00
// return self.can_user_run(msg.sender.name.to_owned(),
// let a = Arc::new(Mutex::new(self));
// let mut a = a.lock().await;
// let a = **a;
// let a = a.can_user_run(msg.sender.name.to_owned(),
// ChType::Channel(msg.channel_login.to_owned()),
// sender_badge,
// cmdreqroles
// ) ;
// let a = *self;
// let a = Arc::new(Mutex::new(a));
// let a = a.lock().await.can_user_run(msg.sender.name.to_owned(),
// ChType::Channel(msg.channel_login.to_owned()),
// sender_badge,
// cmdreqroles
// ) ;
// return a;
2024-02-13 07:54:35 -05:00
// return self.can_user_run(msg.sender.name.to_owned(),
// ChType::Channel(msg.channel_login.to_owned()),
// sender_badge,
// cmdreqroles
// ).await
// * NOTE : We're preferring to pass the ChangeResult up , where we have access to Chat via BotInstance
// that have more strained chatting rules
2024-02-12 05:25:38 -05:00
return self.can_user_run(msg.sender.name.to_owned(),
2024-02-04 14:28:37 -05:00
ChType::Channel(msg.channel_login.to_owned()),
sender_badge,
cmdreqroles
2024-02-12 05:25:38 -05:00
).await
2024-01-29 22:57:07 -05:00
}
// [ ] Call can_user_run()
2024-02-12 01:25:12 -05:00
// (self,Permissible::Block)
2024-02-13 07:54:35 -05:00
(Permissible::Block,ChangeResult::NoChange("".to_string()))
2024-01-29 22:57:07 -05:00
}
2024-02-12 01:25:12 -05:00
2024-02-12 05:25:38 -05:00
pub async fn can_user_run(&mut self,
2024-01-29 12:41:26 -05:00
usr:String,
channelname:ChType,
2024-01-29 22:57:07 -05:00
chat_badge:ChatBadge,
2024-01-29 12:41:26 -05:00
cmdreqroles:Vec<UserRole>
2024-01-29 22:57:07 -05:00
// ) -> Result<Permissible,Box<dyn Error>> {
2024-02-13 07:54:35 -05:00
) -> (Permissible,ChangeResult) {
2024-02-12 05:25:38 -05:00
println!{"Checking within can_user_run()"};
2024-01-29 12:41:26 -05:00
/*
canUserRun -
Input :
usr:String,
channelname:ChType,
2024-01-29 22:57:07 -05:00
chat_badge:ChatBadge,
2024-01-29 12:41:26 -05:00
cmdreqroles:Vec<UserRole>
Output : Result<Permissible,Box<dyn Error>>
Some Possible outcomes : Ok(Permissible::Allow) , Ok(Permissible::Block)
Description
For a Given Chatter (with any special ChatBadge) who ran the Command at a Given Channel , check if that user can
run the command based on the given cmdreqroles required by the command
Inputs and business logic determine if the user can run the command based on the command's required roles
*/
// Requirements
/*
2024-01-29 22:57:07 -05:00
[x] If cmdreqroles is empty vector , automatically assume Ok(Permissible::Allow)
[x] If chatBadge::Broadcaster ...
[x] and cmdreqroles includes UserRole::Broadcaster , Ok(Permissible::Allow)
[x] and cmdreqroles includes UserRole::Mod("") OR UserRole::SupMod("") , Ok(Permissible::Allow)
[x] If chatBadge::Mod ...
[x] Check if they have either UserRole::Mod(channelname::ChType) or UserRole::SupMod(channelname::ChType)
[x] If not, assign them UserRole::Mod(channelname::ChType)
[x] If cmdreqroles includes UserRole::Mod("") , checks if chatter has UserRole::Mod(channelname::ChType) or UserRole::SupMod(channelname::ChType) to determine if Ok(Permissible::Allow)
[x] If cmdreqroles includes UserRole::SupMod("") , checks if chatter has UserRole::SupMod(channelname::ChType) to determine if Ok(Permissible::Allow)
[x] If cmdreqroles includes UserRole::BotAdmin and chatter has UserRole::BotAdmin , Ok(Permissible::Allow)
[x] Otherwise, Ok(Permissible::Block)
*/
2024-01-29 12:41:26 -05:00
2024-01-29 22:57:07 -05:00
// [x] If cmdreqroles is empty vector , automatically assume Ok(Permissible::Allow)
2024-01-29 12:41:26 -05:00
2024-02-12 05:25:38 -05:00
// let idar = Arc::new(RwLock::new(self));
2024-01-29 22:57:07 -05:00
if cmdreqroles.len() == 0 {
// return Ok(Permissible::Allow)
2024-02-13 07:54:35 -05:00
return (Permissible::Allow , ChangeResult::NoChange("Command has no required cmdreqroles".to_string()))
2024-01-29 22:57:07 -05:00
}
2024-02-13 07:54:35 -05:00
let mut modrolechange = ChangeResult::NoChange("".to_string());
2024-01-29 22:57:07 -05:00
match chat_badge {
// [x] If chatBadge::Broadcaster ...
// [x] and cmdreqroles includes UserRole::Broadcaster , Ok(Permissible::Allow)
// [x] and cmdreqroles includes UserRole::Mod("") OR UserRole::SupMod("") , Ok(Permissible::Allow)
ChatBadge::Broadcaster => {
if cmdreqroles.contains(&UserRole::Broadcaster) ||
cmdreqroles.contains(&UserRole::Mod(ChType::Channel(String::new()))) ||
cmdreqroles.contains(&UserRole::SupMod(ChType::Channel(String::new()))) {
// return Ok(Permissible::Allow)
2024-02-13 07:54:35 -05:00
return (Permissible::Allow , ChangeResult::NoChange("Broadcaster Role".to_string()))
2024-01-29 22:57:07 -05:00
}
},
// [x] If chatBadge::Mod ...
// [x] Check if they have either UserRole::Mod(channelname::ChType) or UserRole::SupMod(channelname::ChType)
// [x] If not, assign them UserRole::Mod(channelname::ChType)
ChatBadge::Mod => {
2024-02-12 05:25:38 -05:00
println!("Mod Chatbadge detected");
2024-01-29 22:57:07 -05:00
// println!("debug special roles : {:?}",self.special_roles_users);
// println!("debug usr : {}",&usr.to_lowercase());
2024-01-29 22:57:07 -05:00
// let Some((k,v)) = self.special_roles_users.get_key_value(usr);
2024-01-31 09:31:14 -05:00
// match self.special_roles_users.get_mut(&usr.to_lowercase()) {
2024-02-04 14:28:37 -05:00
// match self.special_roles_users.get(&usr.to_lowercase()) {
2024-02-12 05:25:38 -05:00
println!("Creating clone");
let roleslock = Arc::clone(&(*self).special_roles_users);
println!("Read lock on : Special_Roles_User"); // <-- after this is slightly different between working and problem
let mut roleslock = roleslock.write().await;
match (*roleslock).get(&usr.to_lowercase()) {
2024-02-12 15:32:53 -05:00
Some(usrroles) if
usrroles.read().await.contains(&UserRole::Mod(channelname.clone())) ||
usrroles.read().await.contains(&UserRole::SupMod(channelname.clone())) => { // <-- working got to this point
// println!("contains mod : {}", usrroles.read().await.contains(&UserRole::Mod(channelname.clone())));
// println!("contains supmod : {}", usrroles.read().await.contains(&UserRole::SupMod(channelname.clone())));
2024-02-13 07:54:35 -05:00
// Do nothing when theh have a mod badge and have either a supmod or mod badge for the channel
2024-02-12 05:25:38 -05:00
println!("Already a mod in roles");
2024-02-12 15:32:53 -05:00
}
2024-02-12 15:27:52 -05:00
_ => {
2024-02-13 07:54:35 -05:00
// In the event they have a mod badge , are running a bot command, but don't have a channel mod role yet...
2024-02-12 15:27:52 -05:00
println!("lock created > adding with a mod role o7");
2024-02-13 07:54:35 -05:00
let mut roleslock = roleslock;
let mut a = roleslock.get_mut(&usr.to_lowercase()).unwrap();
let mut alock = a.write().await;
alock.push(UserRole::Mod(channelname.clone()));
modrolechange = ChangeResult::Success("Auto Promoted Mod".to_string());
// alock.get_mut(&usr.to_lowercase())
// .get_or_insert_with(|| UserRole::Mod(channelname.clone()))
// // .expect("ERROR")
// .unwrap()
// .write().await
// // .get_mut()
// .push(UserRole::Mod(channelname.clone()));
2024-02-12 15:32:53 -05:00
} // <-- I'm assuming problem got to here
2024-01-29 22:57:07 -05:00
}
},
// _ => (),
}
// [x] If cmdreqroles includes UserRole::Mod("") , checks if chatter has UserRole::Mod(channelname::ChType) or UserRole::SupMod(channelname::ChType) to determine if Ok(Permissible::Allow)
println!("cmd required roles : {:?}",cmdreqroles);
if cmdreqroles.contains(&UserRole::Mod(ChType::Channel(String::new()))) {
// match self.special_roles_users.get(&channelname) {
// Some(usrroles) => {},
// None => (),
// }
println!("Mod Role required");
2024-02-12 05:25:38 -05:00
if let Some(a) = (&*self).special_roles_users.read().await.get(&usr.to_lowercase()) {
println!("Special roles found for user");
if a.read().await.contains(&UserRole::Mod(channelname.clone())) || a.read().await.contains(&UserRole::SupMod(channelname.clone())){
2024-01-29 22:57:07 -05:00
// return Ok(Permissible::Allow);
2024-02-12 05:25:38 -05:00
println!("Special roles found for user : A mod idenfified ");
2024-02-13 07:54:35 -05:00
return (Permissible::Allow , modrolechange)
2024-01-29 22:57:07 -05:00
}
}
}
// [x] If cmdreqroles includes UserRole::SupMod("") , checks if chatter has UserRole::SupMod(channelname::ChType) to determine if Ok(Permissible::Allow)
if cmdreqroles.contains(&UserRole::SupMod(ChType::Channel(String::new()))) {
2024-02-12 05:25:38 -05:00
if let Some(a) = (&*self).special_roles_users.read().await.get(&usr.to_lowercase()) {
if a.read().await.contains(&UserRole::SupMod(channelname.clone())) {
2024-01-29 22:57:07 -05:00
// return Ok(Permissible::Allow);
2024-02-13 07:54:35 -05:00
return (Permissible::Allow,modrolechange)
2024-01-29 22:57:07 -05:00
}
}
}
// [x] If cmdreqroles includes UserRole::BotAdmin and chatter has UserRole::BotAdmin , Ok(Permissible::Allow)
println!("Eval cmdreqroles with botadmin : {}",cmdreqroles.contains(&UserRole::BotAdmin));
if cmdreqroles.contains(&UserRole::BotAdmin) {
2024-02-12 05:25:38 -05:00
println!("special roles get : {:?}",(&*self).special_roles_users.read().await.get(&usr.to_lowercase()));
if let Some(a) = (&*self).special_roles_users.read().await.get(&usr.to_lowercase()) {
println!("special roles contains BotAdmin: {}",a.read().await.contains(&UserRole::BotAdmin));
if a.read().await.contains(&UserRole::BotAdmin) {
2024-01-29 22:57:07 -05:00
// return Ok(Permissible::Allow);
2024-02-13 07:54:35 -05:00
return (Permissible::Allow,modrolechange)
2024-01-29 22:57:07 -05:00
}
}
}
2024-02-13 07:54:35 -05:00
(Permissible::Block , ChangeResult::NoChange("Not any permissiable condition".to_string()))
2024-01-29 12:41:26 -05:00
}
2024-01-31 21:30:08 -05:00
2024-02-12 01:25:12 -05:00
// pub async fn promote(&mut self,trgchatter:String,channel:Option<ChType>,trg_role:Option<UserRole>) -> ChangeResult {
pub async fn promote(&self,trgchatter:String,channel:Option<ChType>,trg_role:Option<UserRole>) -> ChangeResult {
2024-02-01 08:40:09 -05:00
// Note : If channel is none, getspecialuserroles() returns all roles for the user
// let chatterroles = self.getspecialuserroles(trgchatter, channel);
2024-02-04 14:28:37 -05:00
// let chatterroles = self.getspecialuserroles(trgchatter.clone(), channel.clone());
// let chatterroles = *self.getspecialuserroles(trgchatter.clone(), channel.clone()).await;
// let chatterroles = chatterroles.lock().await;
// // let chatterroles = *chatterroles;
// let chatterroles = chatterroles.unwrap();
2024-02-01 08:40:09 -05:00
2024-02-04 14:28:37 -05:00
let chatterroles = self.getspecialuserroles(trgchatter.clone(), channel.clone()).await;
// let chatterroles = chatterroles.lock().await;
// let chatterroles = *chatterroles;
2024-02-12 05:25:38 -05:00
// let chatterroles = chatterroles.unwrap();
2024-02-04 14:28:37 -05:00
// let emptyvec = vec![];
// let chatterroles = match chatterroles {
// Some(a) => a,
// _ => &(emptyvec),
// };
2024-02-01 08:40:09 -05:00
2024-02-12 05:25:38 -05:00
let chatterroles = chatterroles.unwrap();
let chatterroles = chatterroles.read().await;
let rolemap = &(*chatterroles);
2024-02-01 08:40:09 -05:00
match trg_role {
Some(UserRole::Mod(a)) => {
if let Some(trg_chnl) = channel {
2024-02-12 05:25:38 -05:00
// let rolemap = rolemap.unwrap();
if rolemap.contains(&UserRole::Mod(trg_chnl.clone())) {
2024-02-01 08:40:09 -05:00
return ChangeResult::NoChange(String::from("Target User already has Target Role"));
}
// # otherwise, trg_role for the given chnl is not assigned to the trgchatter
// chatterroles.push(UserRole::Mod(trg_chnl.clone()));
self.special_roles_users
2024-02-12 05:25:38 -05:00
.write().await
2024-02-01 08:40:09 -05:00
.get_mut(&trgchatter)
.expect("Error getting roles")
2024-02-12 05:25:38 -05:00
.write().await
2024-02-01 08:40:09 -05:00
.push(UserRole::Mod(trg_chnl));
return ChangeResult::Success(String::from("Promotion Successful"));
}
},
Some(UserRole::SupMod(a)) => (),
Some(UserRole::BotAdmin) => (),
Some(_) => (),
None => (),
}
// match chatterroles {
// Some(chatterroles) => {
// // [x] chatter already has the target role
// if chatterroles.contains(&trg_role) {
// return ChangeResult::NoChange(String::from("Target User already has Target Role"));
// }
// // By this point, chatteroles does not contain target role
// // match trgRole {
// // Some(trgRole) => {
// // match trgRole {
// // UserRole::Mod(a) => {
// // },
// // UserRole::SupMod(a) => (),
// // UserRole::BotAdmin => (),
// // _ => (), // <-- do nothing with al other options
// // }
// // },
// // None => {
// // /*
// // - If trgRole is None , then promote by implicit rules . For example,
// // - For UserRoles without Mod or SupMod & Caller is SupMod | Broadcaster | BotAdmin > To Mod
// // - For Mod & Caller is SupMod | Broadcaster | BotAdmin > To SupMod
// // - For UserRoles without BotAdmin & Caller is BotAdmin > To BotAdmin
// // */
// // },
// // }
// // let trgRole = match trgRole {
// // Some(UserRole::Mod(a)) => a,
// // Some(UserRole::SupMod(a)) => a,
// // Some(UserRole::BotAdmin) => UserRole::BotAdmin,
// // None => {
// // /*
// // - If trgRole is None , then promote by implicit rules . For example,
// // - For UserRoles without Mod or SupMod & Caller is SupMod | Broadcaster | BotAdmin > To Mod
// // - For Mod & Caller is SupMod | Broadcaster | BotAdmin > To SupMod
// // - For UserRoles without BotAdmin & Caller is BotAdmin > To BotAdmin
// // */
// // },
// // };
// // if let Some(trgRole) = trgRole {
// // // [x] chatter already has the target role
// // if chatterroles.contains(&trgRole) {
// // return ChangeResult::NoChange(String::from("Target User already has Target Role"));
// // }
// // // [ ] trgRole should be assigned based on the input channel
// // let roletoassign = UserRole::
// // }
// },
// _ => (),
// }
2024-01-31 21:30:08 -05:00
2024-02-01 08:40:09 -05:00
ChangeResult::Success(String::from("TEST > Promotion Successful"))
2024-01-31 21:30:08 -05:00
}
2024-02-01 08:40:09 -05:00
pub fn demote(self,trgchatter:String,channel:Option<ChType>,trgRole:Option<UserRole>) -> ChangeResult {
2024-01-31 21:30:08 -05:00
2024-02-01 08:40:09 -05:00
ChangeResult::Success(String::from("TEST > Promotion Successful"))
2024-01-31 21:30:08 -05:00
}
2024-02-12 05:25:38 -05:00
pub async fn getspecialuserroles(&self,chattername:String,channel:Option<ChType>) -> Option<Arc<RwLock<Vec<UserRole>>>> {
/*
Note : Ideally this be called for a given chatter name ?
*/
2024-01-31 21:30:08 -05:00
2024-02-12 05:25:38 -05:00
let chattername = chattername.to_lowercase();
2024-01-31 21:30:08 -05:00
2024-02-12 05:25:38 -05:00
let rolesa = Arc::clone(&self.special_roles_users);
let a = rolesa.read().await;
// let a = Arc::clone(a)
let a = a;
let outr = &(*a);
let outr = outr.get(&chattername);
match outr {
Some(a) => Some(Arc::clone(a)),
None => None,
2024-02-04 14:28:37 -05:00
2024-02-12 05:25:38 -05:00
}
2024-02-04 14:28:37 -05:00
2024-01-31 21:30:08 -05:00
}
2024-01-29 06:18:27 -05:00
}