Jean
31d23cf6b3
Canonicalise a requested vhost the way the ircd displays it (a disallowed underscore becomes a hyphen) before storing and applying, so what we keep matches what shows on the network. A vhost can no longer be assigned to two accounts: prepare_vhost normalises, validates, and checks vhost_owner at every assignment point (SET/REQUEST/ACTIVATE/TAKE/DEFAULT), so two inputs that collapse to the same host are caught as one. Approach recommended by siniStar (the ns_nethost normalisation logic).
36 lines
1.5 KiB
Rust
36 lines
1.5 KiB
Rust
use fedserv_api::{parse_duration, NetView, Sender, ServiceCtx, Store};
|
|
|
|
// SET <account> <host> [duration]: assign a vhost to an account, applying it at
|
|
// once to online sessions. An optional duration (e.g. 30d) makes it temporary.
|
|
// Operators only.
|
|
pub fn handle(me: &str, from: &Sender, args: &[&str], ctx: &mut ServiceCtx, net: &dyn NetView, db: &mut dyn Store) {
|
|
if !super::require_oper(me, from, ctx) {
|
|
return;
|
|
}
|
|
let (Some(&account), Some(&host)) = (args.get(1), args.get(2)) else {
|
|
ctx.notice(me, from.uid, "Syntax: SET <account> <host> [duration]");
|
|
return;
|
|
};
|
|
if db.account(account).is_none() {
|
|
ctx.notice(me, from.uid, format!("\x02{account}\x02 isn't registered."));
|
|
return;
|
|
}
|
|
let host = match super::prepare_vhost(host, account, db) {
|
|
Ok(h) => h,
|
|
Err(msg) => {
|
|
ctx.notice(me, from.uid, msg);
|
|
return;
|
|
}
|
|
};
|
|
let ttl = args.get(3).and_then(|s| parse_duration(s));
|
|
match db.set_vhost(account, &host, from.nick, ttl) {
|
|
Ok(()) => {
|
|
for uid in net.uids_logged_into(account) {
|
|
ctx.apply_vhost(&uid, &host);
|
|
}
|
|
let when = args.get(3).filter(|_| ttl.is_some()).map(|d| format!(" (expires in {d})")).unwrap_or_default();
|
|
ctx.notice(me, from.uid, format!("Vhost \x02{host}\x02 assigned to \x02{account}\x02{when}."));
|
|
}
|
|
Err(_) => ctx.notice(me, from.uid, "Sorry, that didn't work. Please try again in a moment."),
|
|
}
|
|
}
|