feat(chatroom): 增加查看自动猜拳分布功能并保存统计数据

Author: KwdeTfpv

Cargo.toml

@@ -59,6 +59,7 @@ ratatui = "0.29.0"
 serde_repr = "0.1.12"
 serde_path_to_error = "0.1.17"
 lru = "0.12"
+once_cell = "1.21.3"
 
 [package.metadata.docs.rs]
 all-features = true

client/src/commands/handlers/chatroom.rs

@@ -82,6 +82,7 @@ impl Command for ChatroomCommand {
             :q             - 退出
             :rp            - 红包
             :bl            - 消息屏蔽/过滤
+            :rw            - 查看自动猜拳分布
         "#
     }
 }
@@ -154,6 +155,10 @@ impl ChatroomCommand {
                 name: ":bl",
                 desc: "消息屏蔽/过滤",
             },
+            CommandItem {
+                name: ":rw",
+                desc: "查看自动猜拳分布",
+            },
         ]);
 
         let prompt = format!("{}", "聊天室> ".green().bold());
@@ -188,6 +193,31 @@ impl ChatroomCommand {
                             println!("{}", self.help().green());
                             self.context.show_switch_help();
                         }
+                        ":rw" => {
+                            let stats = crate::utils::GESTURE_STATS.lock().unwrap();
+                            let total: u64 = stats.iter().sum();
+                            if total == 0 {
+                                println!("石头: {}, 剪刀: {}, 布: {}，总数: 0", stats[0], stats[1], stats[2]);
+                            } else {
+                                let rock_pct = stats[0] as f64 / total as f64 * 100.0;
+                                let scissors_pct = stats[1] as f64 / total as f64 * 100.0;
+                                let paper_pct = stats[2] as f64 / total as f64 * 100.0;
+                                println!(
+                                    "{}: {:>5} ({:>6.2}%)\n{}: {:>5} ({:>6.2}%)\n{}: {:>7} ({:>6.2}%)\n{}: {}",
+                                    "石头".yellow(),
+                                    stats[0],
+                                    rock_pct,
+                                    "剪刀".yellow(),
+                                    stats[1],
+                                    scissors_pct,
+                                    "布".yellow(),
+                                    stats[2],
+                                    paper_pct,
+                                    "总数".magenta(),
+                                    total
+                                );
+                            }
+                        }
                         cmd if cmd.starts_with(":history") || cmd.starts_with(":h") => {
                             let parts: Vec<&str> = cmd.split_whitespace().collect();
                             let page = if parts.len() > 1 {

client/src/commands/handlers/redpacket.rs

@@ -5,6 +5,7 @@ use colored::*;
 use fishpi_rust::{GestureType, RedPacketMessage, RedPacketType};
 use std::collections::HashMap;
 use std::sync::{Arc, Mutex};
+use crate::utils::random_gesture;
 
 pub struct RedpacketCommand {
     context: CommandContext,
@@ -66,8 +67,8 @@ impl RedpacketCommand {
 
         // 只提供了红包ID，则随机生成一个手势
         let gesture = if args.len() == 1 {
-            let rand_num = rand::random_range(0..=2);
-            match rand_num {
+            // let rand_num = random_gesture();
+            match random_gesture() {
                 0 => GestureType::Rock,
                 1 => GestureType::Scissors,
                 _ => GestureType::Paper,
@@ -383,7 +384,7 @@ impl RedpacketCommand {
         match args.len() {
             0 => {
                 // 不给参数， 积分32 手势随机
-                let gesture = match rand::random_range(0..=2) {
+                let gesture = match random_gesture() {
                     0 => GestureType::Rock,
                     1 => GestureType::Scissors,
                     _ => GestureType::Paper,
@@ -408,7 +409,7 @@ impl RedpacketCommand {
             1 => {
                 // 只给了一个参数，手势随机
                 let money: i32 = args[0].parse().unwrap_or(default_money);
-                let gesture = match rand::random_range(0..=2) {
+                let gesture = match random_gesture() {
                     0 => GestureType::Rock,
                     1 => GestureType::Scissors,
                     _ => GestureType::Paper,
@@ -433,7 +434,7 @@ impl RedpacketCommand {
             2 => {
                 // 两个参数，手势随机，第二个参数msg
                 let money: i32 = args[0].parse().unwrap_or(default_money);
-                let gesture = match rand::random_range(0..=2) {
+                let gesture = match random_gesture() {
                     0 => GestureType::Rock,
                     1 => GestureType::Scissors,
                     2 => GestureType::Paper,
@@ -508,13 +509,13 @@ impl RedpacketCommand {
     async fn handle_list_command(&self) -> Result<()> {
         let cache = self.redpacket_cache.lock().unwrap();
         if cache.is_empty() {
-            println!("{}", "当前没有可领取的红包".yellow());
+            println!("\r{}", "当前没有可领取的红包".yellow());
         } else {
-            println!("{}", "当前可领取的红包:".bold());
+            println!("\r{}", "当前可领取的红包:".bold());
             for (id, info) in cache.iter().enumerate() {
                 let type_name = RedPacketType::to_name(&info.1.type_);
                 println!(
-                    "  {}. {} [{}] {} 个, 共 {} 积分, 已领取 {}/{}",
+                    "\r  {}. {} [{}] {} 个, 共 {} 积分, 已领取 {}/{}",
                     id + 1,
                     info.0.bright_black(),
                     type_name.red(),
@@ -531,7 +532,7 @@ impl RedpacketCommand {
     /// 自动打开红包
     async fn handle_auto_open_command(&self) -> Result<()> {
         if self.redpacket_cache.lock().unwrap().is_empty() {
-            println!("{}", "当前没有可领取的红包".yellow());
+            println!("\r{}", "当前没有可领取的红包".yellow());
             return Ok(());
         }
         let oids: Vec<(String, RedPacketMessage)> = {
@@ -544,7 +545,7 @@ impl RedpacketCommand {
         for (id, msg) in oids {
             if msg.type_ == RedPacketType::ROCK_PAPER_SCISSORS {
                 // 随机生成一个手势
-                let gesture = match rand::random_range(0..=2) {
+                let gesture = match random_gesture() {
                     0 => GestureType::Rock,
                     1 => GestureType::Scissors,
                     _ => GestureType::Paper,
@@ -576,13 +577,13 @@ impl RedpacketCommand {
                     let user_name = self.context.auth.get_user_name().await?;
                     if let Some(got) = info.who.iter().find(|got| got.user_name == user_name) {
                         println!(
-                            "你领取了 {} 积分 {} / {}",
+                            "\r你领取了 {} 积分 {} / {}",
                             got.money.to_string().yellow().bold(),
                             info.info.got.to_string().cyan(),
                             info.info.count.to_string().cyan()
                         );
                     } else {
-                        println!("{}", "红包已领完".yellow());
+                        println!("\r{}", "红包已领完".yellow());
                     }
                 }
             }

client/src/utils/mod.rs

@@ -5,6 +5,53 @@ pub use auth::AuthService;
 use chrono::{Local, TimeZone};
 use colored::*;
 use regex::Regex;
+use std::sync::{Arc, Mutex};
+use once_cell::sync::Lazy;
+use std::fs;
+use std::path::Path;
+use serde::{Serialize, Deserialize};
+
+const GESTURE_STATS_FILE: &str = "gesture_stats.json";
+
+#[derive(Serialize, Deserialize)]
+struct GestureStats([u64; 3]);
+
+pub fn save_gesture_stats(stats: &[u64; 3]) {
+    let data = GestureStats(*stats);
+    if let Ok(json) = serde_json::to_string(&data) {
+        let _ = fs::write(GESTURE_STATS_FILE, json);
+    }
+}
+
+pub fn load_gesture_stats() -> [u64; 3] {
+    if Path::new(GESTURE_STATS_FILE).exists() {
+        if let Ok(json) = fs::read_to_string(GESTURE_STATS_FILE) {
+            if let Ok(data) = serde_json::from_str::<GestureStats>(&json) {
+                return data.0;
+            }
+        }
+    }
+    [0; 3]
+}
+
+pub static GESTURE_STATS: Lazy<Arc<Mutex<[u64; 3]>>> = Lazy::new(|| {
+    Arc::new(Mutex::new(load_gesture_stats()))
+});
+
+//随机猜拳
+pub fn random_gesture() -> u8 {
+    let rand_u32: u32 = rand::random();
+    let rand_f64 = rand_u32 as f64 / 4294967296.0;
+    let gesture = (rand_f64 * 3.0).floor() as u8;
+    {
+        let mut stats = GESTURE_STATS.lock().unwrap();
+        if (gesture as usize) < 3 {
+            stats[gesture as usize] += 1;
+            save_gesture_stats(&*stats);
+        }
+    }
+    gesture
+}
 
 pub fn strip_html_tags(html: &str) -> String {
     let re = Regex::new(r"<[^>]+>").unwrap();