添加proxy与用户绑定机制，添加proxy有效性检测

README.md

@@ -82,7 +82,7 @@ Options:
 $ biliup help
 Upload video to bilibili.
 
-Usage: biliup [OPTIONS] <COMMAND>
+Usage: biliup.exe [OPTIONS] <COMMAND>
 
 Commands:
   login     登录B站并保存登录信息
@@ -93,6 +93,7 @@ Commands:
   dump-flv  输出flv元数据
   download  下载视频
   list      列出所有已上传的视频
+  bind      绑定用户与代理(用户通过user_cookie参数指定)
   help      Print this message or the help of the given subcommand(s)
 
 Options:
@@ -119,7 +120,20 @@ $biliup renew  # ./cookies.json
 ```powershell
 .\biliup.exe -p http://username:password@proxy.example.com:8080 upload
 ```
-
+您也可以通过以下方式为用户绑定代理
+```powershell
+.\biliup.exe -p http://username:password@proxy.example.com:8080 -u myname.json login
+```
+通过这样登录的账号会**自动**生成myname-proxy.json文件记录您使用的代理，后续您可以直接使用其他功能
+```powershell
+.\biliup.exe -u myname.json upoload
+```
+如果代理文件发生变更您可以直接编辑myname-proxy.json文件
+如果您的账号已经登录也可以使用以下命令自动创建代理配置文件
+```powershell
+.\biliup.exe -p http://username:password@proxy.example.com:8080 -u myname2.json bind
+```
+或者您可以手动建立yourname-proxy.json文件并填入"代理地址"
 ### Windows 演示
 
 登录：

crates/biliup-cli/src/cli.rs

@@ -134,6 +134,8 @@ pub enum Commands {
         #[arg(long)]
         not_pubed: bool,
     },
+    /// 绑定用户与代理(用户通过user_cookie参数指定)
+    Bind,
 }
 
 #[derive(Clone, PartialEq, Eq, PartialOrd, Ord, ValueEnum)]

crates/biliup-cli/src/main.rs

@@ -6,6 +6,7 @@ mod uploader;
 
 use anyhow::Result;
 use time::macros::format_description;
+use tracing::info;
 
 use crate::cli::{Cli, Commands};
 use crate::downloader::{download, generate_json};
@@ -16,6 +17,22 @@ use clap::Parser;
 use tracing_subscriber::layer::SubscriberExt;
 use tracing_subscriber::util::SubscriberInitExt;
 
+// 判断代理是否有效
+async fn check_proxy(proxy: &str) -> Result<()> {
+    let client = reqwest::Client::builder()
+        .proxy(reqwest::Proxy::all(proxy)?)
+        .build()?;
+    let resp = client.get("https://www.bilibili.com/").send().await?;
+    if resp.status().is_success() {
+        Ok(())
+    } else {
+        Err(anyhow::anyhow!("代理无效"))
+    }
+}
+
+
+
+
 #[tokio::main]
 async fn main() -> Result<()> {
     // a builder for `FmtSubscriber`.
@@ -27,7 +44,7 @@ async fn main() -> Result<()> {
     //     .finish();
 
     // tracing::subscriber::set_global_default(subscriber).expect("setting default subscriber failed");
-    let cli = Cli::parse();
+    let mut cli = Cli::parse(); //下面需要修改cli
 
     unsafe {
         time::util::local_offset::set_soundness(time::util::local_offset::Soundness::Unsound);
@@ -36,7 +53,39 @@ async fn main() -> Result<()> {
     let timer = tracing_subscriber::fmt::time::LocalTime::new(format_description!(
         "[year]-[month]-[day] [hour]:[minute]:[second]"
     ));
-
+
+    let proxy = cli.proxy.as_deref();
+    // 判断proxy是否为空 如果为空则尝试从user_cookie的文件名-proxy.json中读取
+    if proxy.is_none() {
+        let file_name = cli.user_cookie.file_name().unwrap().to_str().unwrap();
+        // 去除文件名后缀
+        let file_name = file_name.split('.').collect::<Vec<&str>>()[0];
+        let proxy_file = format!("{}-proxy.json", file_name);
+        let proxy_file = std::path::Path::new(&proxy_file);
+        if proxy_file.exists() {
+            // 如果文件存在则读取文件
+            let proxy_str = std::fs::read_to_string(proxy_file)?;
+            let proxy_str = proxy_str.trim();
+            // 去除双引号
+            let proxy_str = proxy_str.trim_matches(|c| c == '\"');
+            // 去除中文引号
+            let proxy_str = proxy_str.trim_matches(|c| c == '“' || c == '”');
+            if !proxy_str.is_empty() {
+                info!("读取到代理配置: {}", proxy_str);
+                // 检查代理是否有效
+                let is_available = check_proxy(proxy_str).await;
+                if is_available.is_err() {
+                    return Err(anyhow::anyhow!("代理无效"));
+                }
+                cli.proxy = Some(proxy_str.to_string());
+            }
+        }
+    }else {
+        // 检查代理是否有效
+        check_proxy(proxy.unwrap()).await?;
+    }
+    // 重新使cli不可变
+    let cli = cli;
     tracing_subscriber::registry()
         .with(tracing_subscriber::EnvFilter::new(&cli.rust_log))
         .with(tracing_subscriber::fmt::layer().with_timer(timer))
@@ -111,6 +160,25 @@ async fn main() -> Result<()> {
                 cli.proxy.as_deref(),
             )
             .await?
+        },
+        Commands::Bind  => {
+            // 判断proxy是否为空
+            if cli.proxy.is_none() {
+                return Err(anyhow::anyhow!("请指定代理"));
+            }
+            // 判断user_cookie是否存在
+            if cli.user_cookie.exists() {
+                return Err(anyhow::anyhow!("指定的user_cookie文件不存在"));
+            }
+            let proxy = cli.proxy.as_deref().unwrap();
+             // 判断代理是否有效
+             let is_available = check_proxy(proxy).await;
+             if is_available.is_err() {
+                 return Err(anyhow::anyhow!("代理无效"));
+             }
+            let user = cli.user_cookie.file_name().unwrap().to_str().unwrap().split('.').collect::<Vec<&str>>()[0];
+            let file_name = format!("{}-proxy.json", user);
+            std::fs::write(file_name, proxy)?;
         }
     };
     Ok(())

crates/biliup-cli/src/uploader.rs

@@ -46,9 +46,19 @@ pub async fn login(user_cookie: PathBuf, proxy: Option<&str>) -> Result<()> {
         5 => login_by_webqr_cookies(client).await?,
         _ => panic!(),
     };
-    let file = std::fs::File::create(user_cookie)?;
+    let file = std::fs::File::create(user_cookie.clone())?;
     serde_json::to_writer_pretty(&file, &info)?;
     info!("登录成功，数据保存在{:?}", file);
+    // 判断proxy是否为空
+    if let Some(proxy) = proxy {
+        let user_cookie = user_cookie.clone();
+        let file_name = user_cookie.file_name().unwrap().to_str().unwrap();
+        let file_name = file_name.split('.').collect::<Vec<&str>>()[0];
+        let file_name = format!("{}-proxy.json", file_name);
+        let file = std::fs::File::create(file_name)?;
+        serde_json::to_writer_pretty(&file, proxy)?;
+        info!("代理数据保存在{:?}", file);
+    }
     Ok(())
 }