token_bind/
token_bind.rs

1//! 跨会话的 token/bind:用户在群里发 `绑定` 拿到一个一次性 token,再私聊把它发给 bot 完成绑定。
2//! 两半是各自独立的分发事件,经一个共享的 `Rendezvous<String, Uin>` 关联(群里签发、私聊认领)——
3//! 没有挂起的任务,能熬过长时间间隔。默认的 `Rendezvous<String, Uin>` 由 `App::new` 自动提供,所以
4//! handler 的 `State<Rendezvous<..>>` 不用自己注册就能用。
5use nagisa::prelude::*;
6use std::sync::atomic::{AtomicU64, Ordering};
7use std::time::Duration;
8
9nagisa::plugin! { name = "账号绑定", category = User, key = "bind" }
10
11static COUNTER: AtomicU64 = AtomicU64::new(0);
12
13/// 群:签发一个映射到发送者的短 token,5 分钟内有效。
14#[command("绑定", id = "issue")]
15async fn issue(reply: Reply, s: Sender, pend: State<Rendezvous<String, Uin>>) -> HandlerResult {
16    let token = format!("BIND{:04}", COUNTER.fetch_add(1, Ordering::Relaxed) % 10000);
17    pend.issue_with_ttl(token.clone(), s.0, Duration::from_secs(300));
18    reply.text(format!("私聊我发送 {token} 完成绑定(5 分钟内有效)")).await?;
19    Ok(())
20}
21
22/// 私聊:发来一个 `BINDxxxx` token 认领待定的绑定。`PrivateMessage` 把它限定在私聊;`Command`
23/// 携带匹配到的 token。
24#[command(regex = r"^BIND\d{4}$", id = "claim")]
25async fn claim(_pm: PrivateMessage, cmd: Command, pend: State<Rendezvous<String, Uin>>, reply: Reply) -> HandlerResult {
26    match pend.claim(&cmd.0) {
27        Some(issuer) => {
28            // 真实 bot 里你会在这里持久化(issuer ↔ 这个私聊账号)的关联。
29            reply.text(format!("绑定成功:已关联群账号 {}", issuer.0)).await?;
30        }
31        None => {
32            reply.text("token 无效或已过期,请重新在群里发送「绑定」").await?;
33        }
34    }
35    Ok(())
36}
37
38#[cfg(feature = "onebot")]
39#[tokio::main]
40async fn main() -> Result<()> {
41    let shutdown = nagisa::ctrl_c_shutdown();
42    // `Rendezvous<String, Uin>` 由 `App::new` 自动提供(默认 TTL 5 分钟),所以
43    // token/bind handler 的 `State<Rendezvous<..>>` 无需手动 `.data(..)` 就能用。要覆盖就自己
44    // 注册一个。
45    App::new().run_onebot(OneBotConfig::new("ws://127.0.0.1:8080/onebot/v11/ws"), shutdown).await
46}
47
48#[cfg(not(feature = "onebot"))]
49fn main() {
50    eprintln!("enable the `onebot` feature to run this example");
51}
token_bind/token_bind.rs

token_bind/
token_bind.rs
