feat: bot add del command

Author: framist

CHANGELOG.md

@@ -22,4 +22,5 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - 2023-10-13 完成开始功能选择中的「快搜教师」和「快搜评论」功能
 - 2023-10-20 修正下标溢出 bug
 - 2024-09-23 优化使用时的帮助说明
+- 2025-10-12 增加了 /del 命令，可以删除自己的评论
 - 正在进行：web 端开发重构，更多功能加入

doc/usage.md

@@ -26,3 +26,11 @@
 - [@SAFC_bak_bot](https://t.me/SAFC_bak_bot)
 - ~~[@SAFC_bot](https://t.me/SAFC_bot)~~
 
+### 机器人常用命令
+
+- `/start`：启动会话并进入功能菜单。
+- `/find 客体 <关键字...>` 或 `/find 评价 <关键字...>`：快速检索导师或评价。
+- `/comment <客体ID>`：对指定客体提交新评价。
+- `/del <评价ID> <发布人 OTP>`：删除自己发布的评价；通过 OTP 签名校验后需回复“确认删除”完成操作。
+- `/downloaddb`：下载当前公开的数据库快照。
+- `/cancel`：随时终止当前会话。

src/db.rs

@@ -383,6 +383,28 @@ impl SAFCdb {
         Ok(rows.collect::<Result<Vec<_>, _>>()?)
     }
 
+    /// 根据评价 id 定位单条评价
+    pub fn find_comment_by_id(&self, comment_id: &String) -> HandlerResult<Option<ObjComment>> {
+        let conn = self.pool.clone().get()?;
+
+        let mut stmt = conn.prepare("SELECT * FROM comments WHERE id = ?")?;
+        let mut rows = stmt.query([comment_id])?;
+
+        if let Some(row) = rows.next()? {
+            Ok(Some(ObjComment {
+                object: row.get::<_, String>(0)?,
+                description: row.get::<_, String>(1)?,
+                date: row.get::<_, String>(2)?,
+                source_cate: SourceCate::from_str(row.get::<_, String>(3)?.as_str()).unwrap(),
+                comment_type: CommentType::from_str(row.get::<_, String>(4)?.as_str()).unwrap(),
+                author_sign: row.get::<_, String>(5).ok(),
+                id: row.get::<_, String>(6)?,
+            }))
+        } else {
+            Ok(None)
+        }
+    }
+
     /// 查找评价 - like 方式
     pub fn find_comment_like(&self, s: &String) -> HandlerResult<Vec<ObjComment>> {
         let conn = self.pool.clone().get()?;
@@ -425,6 +447,13 @@ impl SAFCdb {
         Ok(())
     }
 
+    /// 删除评价
+    pub fn delete_comment(&self, comment_id: &String) -> HandlerResult<bool> {
+        let conn = self.pool.clone().get()?;
+        let affected = conn.execute("DELETE FROM comments WHERE id = ?", [comment_id])?;
+        Ok(affected > 0)
+    }
+
     /// 增加评价
     pub fn add_comment(&self, obj_comment: &ObjComment) -> HandlerResult<()> {
         let conn = self.pool.clone().get()?;

src/main.rs

@@ -40,6 +40,8 @@ enum Command {
     Comment(String),
     #[command(description = "搜索")]
     Find(String),
+    #[command(description = "删除评价")]
+    Del(String),
 }
 
 #[tokio::main]
@@ -52,7 +54,7 @@ async fn main() {
     bot.set_my_commands(Command::bot_commands()) // 向 telegram 注册命令
         .await
         .expect("Failed to set bot commands to telegram");
-    
+
     log::info!("Bot commands have been set");
 
     Dispatcher::builder(bot, schema())
@@ -78,6 +80,7 @@ fn schema() -> UpdateHandler<Box<dyn std::error::Error + Send + Sync + 'static>>
         .branch(case![Command::DownloadDb].endpoint(download_db_command))
         .branch(case![Command::Find(arg)].endpoint(find_command))
         .branch(case![Command::Comment(arg)].endpoint(comment_command))
+        .branch(case![Command::Del(arg)].endpoint(delete_comment_command))
         .branch(dptree::endpoint(invalid_command));
 
     // 文本消息
@@ -90,7 +93,8 @@ fn schema() -> UpdateHandler<Box<dyn std::error::Error + Send + Sync + 'static>>
         .branch(case![State::Department { school_cate, university }].endpoint(choose_supervisor))
         .branch(case![State::Supervisor { school_cate, university, department }].endpoint(read_or_comment))
         .branch(case![State::Comment { object_id, comment_type }].endpoint(add_comment))
-        .branch(case![State::Publish { object_id, comment, comment_type }].endpoint(publish_comment));
+        .branch(case![State::Publish { object_id, comment, comment_type }].endpoint(publish_comment))
+        .branch(case![State::DeleteConfirm { comment_id }].endpoint(confirm_delete_comment));
 
     // 消息
     let message_handler = Update::filter_message()
@@ -197,6 +201,90 @@ async fn find_command(bot: Bot, dialogue: MyDialogue, arg: String, msg: Message)
     Ok(())
 }
 
+/// 删除评价命令
+async fn delete_comment_command(
+    bot: Bot,
+    dialogue: MyDialogue,
+    arg: String,
+    msg: Message,
+) -> HandlerResult {
+    let parts: Vec<&str> = arg.split_whitespace().collect();
+
+    if parts.len() != 2 {
+        bot.send_message(
+            msg.chat.id,
+            "使用方法：\n/del <评价 ID> <发布人 OTP>\n示例：/del ab12cd34 xxxxxxxx",
+        )
+        .reply_to_message_id(msg.id)
+        .await?;
+        return Ok(());
+    }
+
+    let comment_id = parts[0].to_string();
+    let otp = parts[1].to_string();
+
+    let comment = match SAFC_DB.find_comment_by_id(&comment_id)? {
+        Some(c) => c,
+        None => {
+            bot.send_message(
+                msg.chat.id,
+                format!("未找到评价 `{}`。请检查评价 ID。", comment_id),
+            )
+            .reply_to_message_id(msg.id)
+            .await?;
+            return Ok(());
+        }
+    };
+
+    let author_sign = match comment.author_sign.as_ref() {
+        Some(sign) => sign,
+        None => {
+            bot.send_message(
+                msg.chat.id,
+                "该评价未设置 OTP，暂不支持自行删除。",
+            )
+            .reply_to_message_id(msg.id)
+            .await?;
+            return Ok(());
+        }
+    };
+
+    let expected_sign = hash_author_sign(&comment_id, &otp);
+    if author_sign != &expected_sign {
+        bot.send_message(
+            msg.chat.id,
+            "签名验证失败。请确认评价 ID 与 OTP 是否正确。",
+        )
+        .reply_to_message_id(msg.id)
+        .await?;
+        return Ok(());
+    }
+
+    let object_info = match SAFC_DB.find_objteacher_with_id(comment.object.as_str())? {
+        Some(obj) => obj.display_path(),
+        None => format!("客体 ID {}", comment.object),
+    };
+
+    let comment_body = escape(comment.description.as_str());
+    let comment_type_label = comment.comment_type.to_string();
+    let confirm_message = format!(
+        "评价 ID `{}`\n对象：{}\n日期：{}\n类型：{}\n\n```text\n{}\n```\n\n签名验证成功。\n请回复「确认删除」以继续，或发送 /cancel 终止此操作。",
+        escape(comment_id.as_str()),
+        escape(object_info.as_str()),
+        escape(comment.date.as_str()),
+        escape(comment_type_label.as_str()),
+        comment_body
+    );
+
+    bot.send_message(msg.chat.id, confirm_message)
+        .reply_to_message_id(msg.id)
+        .parse_mode(MarkdownV2)
+        .await?;
+
+    dialogue.update(State::DeleteConfirm { comment_id }).await?;
+    Ok(())
+}
+
 /// 直接评价命令处理函数
 async fn comment_command(
     bot: Bot,
@@ -993,6 +1081,55 @@ async fn publish_comment(
     Ok(())
 }
 
+/// 删除评价确认处理
+async fn confirm_delete_comment(
+    bot: Bot,
+    dialogue: MyDialogue,
+    comment_id: String,
+    msg: Message,
+) -> HandlerResult {
+    let text = match msg.text() {
+        Some(t) => t.trim(),
+        None => {
+            bot.send_message(
+                msg.chat.id,
+                "未识别的输入。请回复「确认删除」，或使用 /cancel 取消操作。",
+            )
+            .await?;
+            return Ok(());
+        }
+    };
+
+    if text != "确认删除" {
+        bot.send_message(
+            msg.chat.id,
+            "如需删除，请回复「确认删除」。若不再需要删除，可使用 /cancel 结束操作。",
+        )
+        .await?;
+        return Ok(());
+    }
+
+    match SAFC_DB.delete_comment(&comment_id)? {
+        true => {
+            bot.send_message(
+                msg.chat.id,
+                format!("评价 `{}` 已删除。感谢您的更新！", comment_id),
+            )
+            .await?;
+        }
+        false => {
+            bot.send_message(
+                msg.chat.id,
+                "评价不存在或已被删除。",
+            )
+            .await?;
+        }
+    }
+
+    dialogue.update(State::Start).await?;
+    Ok(())
+}
+
 /// 一维向量转换为 n 列纵向键盘
 fn convert_to_n_columns_keyboard(data: Vec<String>, n: usize) -> Vec<Vec<KeyboardButton>> {
     data.chunks(n)

src/msg.rs

@@ -97,6 +97,9 @@ pub enum State {
         comment: String,
         comment_type: CommentType,
     },
+    DeleteConfirm {
+        comment_id: String,
+    },
     /// 分页显示回调状态
     PagingCb {
         data: PagingCbData,