mirror of
https://github.com/natsukagami/youmubot.git
synced 2025-05-24 17:20:49 +00:00
Implement Flags parser and handle choose --everyone
- A generic Flags parser that could handle any `--flag` argument. - `choose --everyone` overrides the Offline/DnD check.
This commit is contained in:
parent
2449b09cb2
commit
2c2092eb91
3 changed files with 46 additions and 0 deletions
38
youmubot-prelude/src/flags.rs
Normal file
38
youmubot-prelude/src/flags.rs
Normal file
|
@ -0,0 +1,38 @@
|
|||
use serenity::prelude::Args;
|
||||
use std::collections::HashSet as Set;
|
||||
|
||||
/// Handle flags parsing.
|
||||
pub struct Flags(Set<String>);
|
||||
|
||||
struct Flag(pub String);
|
||||
|
||||
impl std::str::FromStr for Flag {
|
||||
type Err = ();
|
||||
|
||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||
if s.starts_with("--") {
|
||||
Ok(Flag(s.trim_start_matches("--").to_owned()))
|
||||
} else {
|
||||
Err(())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Flags {
|
||||
/// Parses the set of flags from a given `Args` structure.
|
||||
pub fn collect_from(args: &mut Args) -> Flags {
|
||||
let mut set = Set::new();
|
||||
loop {
|
||||
if let Some(Flag(s)) = args.find().ok() {
|
||||
set.insert(s);
|
||||
} else {
|
||||
break Flags(set);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Checks whether `flag` exists in the flags set.
|
||||
pub fn contains(&self, flag: impl AsRef<str>) -> bool {
|
||||
self.0.contains(flag.as_ref())
|
||||
}
|
||||
}
|
|
@ -5,6 +5,7 @@ use std::sync::Arc;
|
|||
|
||||
pub mod announcer;
|
||||
pub mod args;
|
||||
pub mod flags;
|
||||
pub mod hook;
|
||||
pub mod member_cache;
|
||||
pub mod pagination;
|
||||
|
@ -13,6 +14,7 @@ pub mod setup;
|
|||
|
||||
pub use announcer::{Announcer, AnnouncerHandler};
|
||||
pub use args::{Duration, UsernameArg};
|
||||
pub use flags::Flags;
|
||||
pub use hook::Hook;
|
||||
pub use member_cache::MemberCache;
|
||||
pub use pagination::{paginate, paginate_fn, paginate_reply, paginate_reply_fn, Paginate};
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue