nki/youmubot
1
0
Fork 0
mirror of https://github.com/natsukagami/youmubot.git synced 2025-05-24 01:00:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Implement roles management (#18)

Browse source 
Fix formatting

Implement `roles` management

Add a 'roles' mod and DB
This commit is contained in:
Natsu Kagami 2020-03-27 21:22:15 +00:00
parent 99c7888dfb
commit 5dd26d7474
4 changed files with 250 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

236
youmubot-core/src/community/roles.rs Normal file
View file

@ -0,0 +1,236 @@
use crate::db::Roles as DB;
use serenity::{
framework::standard::{macros::command, Args, CommandError as Error, CommandResult},
model::{channel::Message, id::RoleId},
utils::MessageBuilder,
};
use youmubot_prelude::*;
#[command("listroles")]
#[description = "List all available roles in the server."]
#[num_args(0)]
#[only_in(guilds)]
fn list(ctx: &mut Context, m: &Message, _: Args) -> CommandResult {
let guild_id = m.guild_id.unwrap(); // only_in(guilds)
let db = DB::open(&*ctx.data.read());
let db = db.borrow()?;
let roles = db.get(&guild_id).filter(|v| !v.is_empty());
match roles {
None => {
m.reply(&ctx, "No roles available for assigning.")?;
}
Some(v) => {
let roles = guild_id.to_partial_guild(&ctx)?.roles;
let roles: Vec<_> = v
.iter()
.filter_map(|(_, role)| roles.get(&role.id).map(|r| (r, &role.description)))
.collect();
const ROLES_PER_PAGE: usize = 5;
let pages = (roles.len() + ROLES_PER_PAGE - 1) / ROLES_PER_PAGE;
let watcher = ctx.data.get_cloned::<ReactionWatcher>();
watcher.paginate_fn(
ctx.clone(),
m.channel_id,
|page, e| {
let page = page as usize;
let start = page * ROLES_PER_PAGE;
let end = roles.len().min(start + ROLES_PER_PAGE);
if end <= start {
return (e, Err(Error::from("No more roles to display")));
}
let roles = &roles[start..end];
let nw = roles // name width
.iter()
.map(|(r, _)| r.name.len())
.max()
.unwrap()
.max(6);
let idw = roles[0].0.id.to_string().len();
let dw = roles
.iter()
.map(|v| v.1.len())
.max()
.unwrap()
.max(" Description ".len());
let mut m = MessageBuilder::new();
m.push_line("```");
// Table header
m.push_line(format!(
"{:nw$} | {:idw$} | {:dw$}",
"Name",
"ID",
"Description",
nw = nw,
idw = idw,
dw = dw,
));
m.push_line(format!(
"{:->nw$}---{:->idw$}---{:->dw$}",
"",
"",
"",
nw = nw,
idw = idw,
dw = dw,
));
for (role, description) in roles.iter() {
m.push_line(format!(
"{:nw$} | {:idw$} | {:dw$}",
role.name,
role.id,
description,
nw = nw,
idw = idw,
dw = dw,
));
}
m.push_line("```");
m.push(format!("Page **{}/{}**", page + 1, pages));
(e.content(m.build()), Ok(()))
},
std::time::Duration::from_secs(60 * 10),
)?;
}
};
Ok(())
}
#[command("role")]
#[description = "Toggle a role by its name or ID."]
#[num_args(1)]
#[only_in(guilds)]
fn toggle(ctx: &mut Context, m: &Message, mut args: Args) -> CommandResult {
let role = args.single::<String>()?;
let guild_id = m.guild_id.unwrap();
let roles = guild_id.to_partial_guild(&ctx)?.roles;
let role = match role.parse::<u64>() {
Ok(id) => roles.get(&RoleId(id)).cloned(),
Err(_) => roles
.iter()
.find_map(|(_, r)| if r.name == role { Some(r) } else { None })
.cloned(),
};
match role {
None => {
m.reply(&ctx, "No such role exists")?;
}
Some(role)
if !DB::open(&*ctx.data.read())
.borrow()?
.get(&guild_id)
.map(|g| g.contains_key(&role.id))
.unwrap_or(false) =>
{
m.reply(&ctx, "This role is not self-assignable. Check the `listroles` command to see which role can be assigned.")?;
}
Some(role) => {
let mut member = m.member(&ctx).ok_or(Error::from("Cannot find member"))?;
if member.roles.contains(&role.id) {
member.remove_role(&ctx, &role)?;
m.reply(&ctx, format!("Role `{}` has been removed.", role.name))?;
} else {
member.add_role(&ctx, &role)?;
m.reply(&ctx, format!("Role `{}` has been assigned.", role.name))?;
}
}
};
Ok(())
}
#[command("addrole")]
#[description = "Add a role as the assignable role"]
#[usage = "{role-name-or-id} / {description}"]
#[example = "hd820 / Headphones role"]
#[num_args(2)]
#[required_permissions(MANAGE_ROLES)]
#[only_in(guilds)]
fn add(ctx: &mut Context, m: &Message, mut args: Args) -> CommandResult {
let role = args.single::<String>()?;
let description = args.single::<String>()?;
let guild_id = m.guild_id.unwrap();
let roles = guild_id.to_partial_guild(&ctx)?.roles;
let role = match role.parse::<u64>() {
Ok(id) => roles.get(&RoleId(id)).cloned(),
Err(_) => roles
.iter()
.find_map(|(_, r)| if r.name == role { Some(r) } else { None })
.cloned(),
};
match role {
None => {
m.reply(&ctx, "No such role exists")?;
}
Some(role)
if DB::open(&*ctx.data.read())
.borrow()?
.get(&guild_id)
.map(|g| g.contains_key(&role.id))
.unwrap_or(false) =>
{
m.reply(&ctx, "This role already exists in the database.")?;
}
Some(role) => {
DB::open(&*ctx.data.read())
.borrow_mut()?
.entry(guild_id)
.or_default()
.insert(
role.id,
crate::db::Role {
id: role.id,
description,
},
);
m.react(&ctx, "👌🏼")?;
}
};
Ok(())
}
#[command("removerole")]
#[description = "Remove a role from the assignable roles list."]
#[usage = "{role-name-or-id}"]
#[example = "hd820"]
#[num_args(1)]
#[required_permissions(MANAGE_ROLES)]
#[only_in(guilds)]
fn remove(ctx: &mut Context, m: &Message, mut args: Args) -> CommandResult {
let role = args.single::<String>()?;
let guild_id = m.guild_id.unwrap();
let roles = guild_id.to_partial_guild(&ctx)?.roles;
let role = match role.parse::<u64>() {
Ok(id) => roles.get(&RoleId(id)).cloned(),
Err(_) => roles
.iter()
.find_map(|(_, r)| if r.name == role { Some(r) } else { None })
.cloned(),
};
match role {
None => {
m.reply(&ctx, "No such role exists")?;
}
Some(role)
if !DB::open(&*ctx.data.read())
.borrow()?
.get(&guild_id)
.map(|g| g.contains_key(&role.id))
.unwrap_or(false) =>
{
m.reply(&ctx, "This role does not exist in the assignable list.")?;
}
Some(role) => {
DB::open(&*ctx.data.read())
.borrow_mut()?
.entry(guild_id)
.or_default()
.remove(&role.id);
m.react(&ctx, "👌🏼")?;
}
};
Ok(())
}