功能：增加自动更新命令，支持检查和下载最新版本

Author: KwdeTfpv

Cargo.toml

@@ -9,6 +9,7 @@ repository = "https://github.com/FishPiOffical/fishpi-rust"
 documentation = "https://docs.rs/fishpi-rust"
 readme = "README.md"
 keywords = ["fishpi"]
+build = "build.rs"
 
 [lib]
 name = "fishpi_rust"

client/src/app.rs

@@ -48,7 +48,7 @@ impl App {
     }
 
     fn show_welcome(&self) {
-        println!("{}", "欢迎使用摸鱼派 Rust 客户端".bold().cyan());
+        println!("{} {}", "欢迎使用摸鱼派 Rust 客户端".bold().cyan(), env!("GIT_TAG").bold().cyan());
         println!("{}", "=====================================".cyan());
     }
 

client/src/commands/handlers/mod.rs

@@ -5,6 +5,7 @@ pub mod chatroom;
 pub mod filter;
 pub mod notice;
 pub mod redpacket;
+pub mod update;
 
 pub use article::ArticleCommand;
 pub use breezemoon::BreezemoonCommand;
@@ -13,3 +14,4 @@ pub use chatroom::ChatroomCommand;
 pub use filter::FilterCommand;
 pub use notice::NoticeCommand;
 pub use redpacket::RedpacketCommand;
+pub use update::UpdateCommand;

client/src/commands/handlers/update.rs

@@ -0,0 +1,119 @@
+use anyhow::Result;
+use reqwest::Client;
+use std::env;
+use std::fs;
+use std::io::Write;
+use std::process::Command;
+use async_trait::async_trait;
+use crate::commands::{Command as CCommand, CommandContext, CommandResult};
+
+#[allow(dead_code)]
+pub struct UpdateCommand {
+    context: CommandContext,
+}
+
+impl UpdateCommand {
+    pub fn new(context: CommandContext) -> Self {
+        Self { context }
+    }
+
+    async fn get_latest_release(&self) -> Result<(String, String)> {
+        let api = "https://api.github.com/repos/FishPiOffical/fishpi-rust/releases/latest";
+        let client = Client::new();
+        let resp = client
+            .get(api)
+            .header("User-Agent", "fishpi-rust-chatroom")
+            .send()
+            .await?
+            .json::<serde_json::Value>()
+            .await?;
+
+        let tag = resp["tag_name"].as_str().unwrap_or("").to_string();
+        let asset_url = resp["assets"][0]["browser_download_url"]
+            .as_str()
+            .unwrap_or("")
+            .to_string();
+        Ok((tag, asset_url))
+    }
+
+    async fn download_and_replace(&self, url: &str) -> Result<()> {
+        let exe_path = env::current_exe()?;
+        let tmp_path = exe_path.with_extension("new");
+
+        let resp = reqwest::get(url).await?;
+        let bytes = resp.bytes().await?;
+        let mut file = fs::File::create(&tmp_path)?;
+        file.write_all(&bytes)?;
+
+        self.launch_update_script(exe_path.to_str().unwrap(), tmp_path.to_str().unwrap())?;
+        println!("已启动自动更新，程序即将退出并自动替换为新版本。");
+        std::process::exit(0);
+    }
+
+    fn launch_update_script(&self, exe_path: &str, new_exe_path: &str) -> std::io::Result<()> {
+        #[cfg(target_os = "windows")]
+        {
+            let script = format!(
+                r#"
+@echo off
+ping 127.0.0.1 -n 2 > nul
+move /Y "{new}" "{old}"
+start "" "{old}"
+"#,
+                new = new_exe_path,
+                old = exe_path
+            );
+            let script_path = format!("{}.update.bat", exe_path);
+            let mut file = fs::File::create(&script_path)?;
+            file.write_all(script.as_bytes())?;
+            Command::new("cmd")
+                .args(&["/C", &script_path])
+                .spawn()?;
+        }
+        #[cfg(any(target_os = "linux", target_os = "macos"))]
+        {
+            let script = format!(
+                r#"
+#!/bin/sh
+sleep 1
+mv "{new}" "{old}"
+chmod +x "{old}"
+"{old}" &
+"#,
+                new = new_exe_path,
+                old = exe_path
+            );
+            let script_path = format!("{}.update.sh", exe_path);
+            let mut file = fs::File::create(&script_path)?;
+            file.write_all(script.as_bytes())?;
+            Command::new("sh")
+                .arg(&script_path)
+                .spawn()?;
+        }
+        Ok(())
+    }
+}
+
+#[async_trait]
+impl CCommand for UpdateCommand {
+    async fn execute(&mut self, _args: &[&str]) -> Result<CommandResult> {
+        println!("正在检查新版本...");
+        let (latest_tag, asset_url) = self.get_latest_release().await?;
+        let latest_tag_clean = latest_tag.trim().trim_start_matches('v');
+        let current_version = env!("GIT_TAG").trim().trim_start_matches('v');
+        if latest_tag_clean == current_version {
+            println!("已是最新版本。");
+            return Ok(CommandResult::Success);
+        }
+        println!(
+            "检测到新版本: v{}，当前版本: v{}",
+            latest_tag_clean, current_version
+        );
+        self.download_and_replace(&asset_url).await?;
+        Ok(CommandResult::Success)
+    }
+
+    fn help(&self) -> &'static str {
+        "检查并自动更新到最新版本"
+    }
+}

client/src/commands/mod.rs

@@ -10,6 +10,7 @@ use crate::commands::handlers::{
     // BreezemoonCommand,
     ChatCommand,
     ChatroomCommand,
+    UpdateCommand,
 };
 use colored::*;
 pub use registry::CommandRegistry;
@@ -66,6 +67,10 @@ impl CommandContext {
                 // command.execute(&[]).await?;
                 println!("清风明月模式暂未实现");
             }
+            "update" => {
+                let mut command = UpdateCommand::new(self.clone());
+                command.execute(&[]).await?;
+            }
             _ => {
                 println!("未知模式: {}", mode);
             }

client/src/commands/registry.rs

@@ -3,7 +3,7 @@ use colored::*;
 use std::collections::HashMap;
 
 use crate::commands::handlers::{
-    ArticleCommand, BreezemoonCommand, ChatCommand, ChatroomCommand, NoticeCommand,
+    ArticleCommand, BreezemoonCommand, ChatCommand, ChatroomCommand, NoticeCommand, UpdateCommand
 };
 use crate::commands::{Command, CommandContext, CommandFactory, CommandResult};
 pub struct CommandRegistry {
@@ -143,5 +143,12 @@ impl CommandRegistry {
             "清风明月",
             vec!["bm", "moon"],
         );
+
+        self.register(
+            "update",
+            |context| Box::new(UpdateCommand::new(context.clone())),
+            "检查并自动更新到最新版本",
+            vec!["upgrade"],
+        );
     }
 }