From 09c96dfb9640157cdfa47eeffc747064a1ebd5cf Mon Sep 17 00:00:00 2001
From: Serenity <1360359624@qq.com>
Date: Tue, 21 May 2024 22:58:09 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20cli=E5=AE=9E=E7=8E=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Cargo.toml          |  2 ++
 src/main.rs         | 30 +++++++++++++++++++++++++++---
 src/util/cmd.rs     | 20 ++++++++++++++++++++
 src/util/mod.rs     |  1 +
 src/util/service.rs | 16 ++++++++--------
 5 files changed, 58 insertions(+), 11 deletions(-)
 create mode 100644 src/util/cmd.rs

diff --git a/Cargo.toml b/Cargo.toml
index 9d8aae2..bd87e5c 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -21,3 +21,5 @@ fern = "0.6"
 #log4rs = "1.3.0"
 yaml-rust = "0.4.5"
 service-manager = "0.6.1"
+clap = { version = "4.5.4", features = ["derive"] }
+clap_derive = "4.5.4"
diff --git a/src/main.rs b/src/main.rs
index 620bdbf..772e755 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -13,23 +13,47 @@ use crate::core::sangfor::Sangfor;
 use std::io::{Error, Write};
 use std::path::PathBuf;
 use std::time::SystemTime;
+use clap::{Command, Parser};
 use serde::{Deserialize, Serialize};
 use service_manager::{ServiceInstallCtx, ServiceLabel, ServiceManager, ServiceStartCtx, ServiceStopCtx, ServiceUninstallCtx};
 use util::service::Service;
+use crate::util::cmd::Cli;
 
 mod core;
-mod test;
 mod util;
 
 #[tokio::main]
 async fn main() {
     util::logs::init("nal.log", log::LevelFilter::Debug).expect("初始化日志出错");
 
+    let cli = util::cmd::Cli::parse();
+    let service = Service::new("net-auto-login");
+    if cli.install {
+        service.install();
+        return;
+    }
+    if cli.uninstall {
+        service.uninstall();
+        return;
+    }
+    if cli.start {
+        service.start();
+        return;
+    }
+    if cli.stop {
+        service.stop();
+        return;
+    }
+    if cli.run {
+        handler().await;
+        return;
+    }
+}
+
+async fn handler(){
     let config = nal::init_config();
     info!("config: {config:#?}");
 
-    let service = Service::new("net-auto-login");
-
     /*let expression = "* 1 * * * * *";
     let schedule = Schedule::from_str(expression).unwrap();
     println!("All stars: Upcoming fire times for '{}':", expression);
diff --git a/src/util/cmd.rs b/src/util/cmd.rs
new file mode 100644
index 0000000..da23b45
--- /dev/null
+++ b/src/util/cmd.rs
@@ -0,0 +1,20 @@
+use clap_derive::Parser;
+
+#[derive(Parser)]
+#[command(version, about, long_about = None)]
+pub struct Cli {
+    #[arg(short = 'i')]
+    pub install: bool,
+
+    #[arg(short,long)]
+    pub uninstall: bool,
+
+    #[arg(long)]
+    pub start: bool,
+
+    #[arg(long)]
+    pub stop: bool,
+
+    #[arg(short,long)]
+    pub run: bool
+}
diff --git a/src/util/mod.rs b/src/util/mod.rs
index 253fbd5..c2b2970 100644
--- a/src/util/mod.rs
+++ b/src/util/mod.rs
@@ -1,3 +1,4 @@
 pub mod logs;
 pub mod rc4;
 pub mod service;
+pub mod cmd;
diff --git a/src/util/service.rs b/src/util/service.rs
index 1c8a93e..cb2a790 100644
--- a/src/util/service.rs
+++ b/src/util/service.rs
@@ -35,7 +35,7 @@ impl Service {
     }
 
     /// 安装到系统服务
-    fn install(&self) {
+    pub fn install(&self) {
         // 使用底层服务管理平台安装我们的服务
         self.service_manage.install(ServiceInstallCtx {
             label: self.name.clone(),
@@ -46,33 +46,33 @@ impl Service {
             working_directory: None, // 服务进程的工作目录的可选字符串。
             environment: None, // 用于提供服务进程的环境变量的可选列表。
         }).expect("Failed to install");
-        info!("{:?}服务安装完成。",self.path.clone())
+        info!("{:}服务安装完成。",self.name.clone().to_string())
     }
 
     /// 从系统服务卸载
-    fn uninstall(&self) {
+    pub fn uninstall(&self) {
         // 使用底层服务管理平台卸载我们的服务
         self.service_manage.uninstall(ServiceUninstallCtx {
             label: self.name.clone()
         }).expect("Failed to uninstall");
-        info!("{:?}服务卸载完成。",self.path.clone())
+        info!("{:}服务卸载完成。",self.name.clone().to_string())
     }
 
     /// 启动这个服务
-    fn start(&self) {
+    pub fn start(&self) {
         // 使用底层服务管理平台启动我们的服务
         self.service_manage.start(ServiceStartCtx {
             label: self.name.clone()
         }).expect("Failed to start");
-        info!("{:?}服务启动完成。",self.path.clone())
+        info!("{:}服务启动完成。",self.name.clone().to_string())
     }
 
     /// 停止这个服务
-    fn stop(&self) {
+    pub fn stop(&self) {
         // 使用底层服务管理平台停止我们的服务
         self.service_manage.stop(ServiceStopCtx {
             label: self.name.clone()
         }).expect("Failed to stop");
-        info!("{:?}服务停止完成。",self.path.clone())
+        info!("{:}服务停止完成。",self.name.clone().to_string())
     }
 }
\ No newline at end of file