软件包名可配

src/download_wrapper.rs

@@ -1,9 +1,9 @@
-use std::fmt::Display;
 use crate::download_wrapper::DownloadError::JlinkNotFindError;
+use std::fmt::Display;
 use std::fs;
+use std::io::Write;
 use std::path::PathBuf;
 use std::process::Command;
-use std::io::Write;
 
 pub struct DownloadWrapper {
     commander_path: PathBuf,
@@ -44,13 +44,27 @@ pub enum DownloadError {
 
 impl Default for DownloadWrapper {
     fn default() -> Self {
+        let config_str = fs::read_to_string("config.json").unwrap();
+        let config: serde_json::Value = serde_json::from_str(&config_str).unwrap();
+        let bootloader_name = config["bootloader_name"]
+            .as_str()
+            .unwrap_or("RAWM00-2-0-0_silicon-wisun_bootloader_D20241008.s37")
+            .to_string();
+        let app_name = config["app_name"]
+            .as_str()
+            .unwrap_or("RAWM00-2-0-0_silicon-combine-V1-D20241230-BZ.hex")
+            .to_string();
+        let rail_name = config["rail_name"]
+            .as_str()
+            .unwrap_or("rail_soc_railtest_sisdk(silabs).hex")
+            .to_string();
         Self {
             commander_path: PathBuf::from(std::env::current_dir().unwrap())
                 .join("commander/Simplicity Commander/commander.exe"),
             bin_path: PathBuf::from(std::env::current_dir().unwrap()).join("bin/"),
-            bootloader_name: String::from("RAWM00-2-0-0_silicon-wisun_bootloader_D20241008.s37"),
+            bootloader_name,
-            app_name: String::from("RAWM00-2-0-0_silicon-combine-V1-D20241230-BZ.hex"),
+            app_name,
-            rail_name: String::from("rail_soc_railtest_sisdk(silabs).hex"),
+            rail_name
         }
     }
 }
@@ -141,19 +155,21 @@ impl DownloadWrapper {
         {
             Err(DownloadError::EraseError)
         } else {
-            let Some(label) = label else { return Err(DownloadError::NoLabel) };
+            let Some(label) = label else {
+                return Err(DownloadError::NoLabel);
+            };
             match self.save_log(label, result_str.clone()) {
-                Ok(_) => {
+                Ok(_) => Ok(result_str),
-                    Ok(result_str)
+                Err(_) => Err(DownloadError::DownloadBootloaderError),
-                }
-                Err(_) => {
-                    Err(DownloadError::DownloadBootloaderError)
-                }
             }
         }
     }
 
-    pub fn download(&self, download_type: DownloadType, label: Option<String>) -> Result<String, DownloadError> {
+    pub fn download(
+        &self,
+        download_type: DownloadType,
+        label: Option<String>,
+    ) -> Result<String, DownloadError> {
         let mut binding: PathBuf;
         match download_type {
             DownloadType::Bootloader => {
@@ -192,14 +208,12 @@ impl DownloadWrapper {
         {
             Err(DownloadError::DownloadBootloaderError)
         } else {
-            let Some(label) = label else { return Err(DownloadError::NoLabel) };
+            let Some(label) = label else {
+                return Err(DownloadError::NoLabel);
+            };
             match self.save_log(label, result_str.clone()) {
-                Ok(_) => {
+                Ok(_) => Ok(result_str),
-                    Ok(result_str)
+                Err(_) => Err(DownloadError::DownloadBootloaderError),
-                }
-                Err(_) => {
-                    Err(DownloadError::DownloadBootloaderError)
-                }
             }
         }
     }

src/main.rs

@@ -292,18 +292,16 @@ impl MainWindow {
 
     fn start(&mut self, sender: Sender<Message>) -> () {
         let mes_config = self.mysql_config.clone();
-        let label = self.label.clone();
+        let mut label = self.label.clone();
         let download_type: DownloadType = self.selection.unwrap();
         let online = self.is_online.clone();
         if label == "" {
             add_log("请输入Label".to_string());
             return;
         }
-        if label.contains(":"){
+        label = label.replace(":", "");
-            add_log("Label不合法".to_string());
+        label = label.to_uppercase();
-            self.label = "".to_string();
+        add_log("Label: ".to_string() + label.as_str());
-            return;
-        }
         std::thread::spawn(move || {
             sleep(std::time::Duration::from_secs(5));
             let mut download_wrapper = download_wrapper::DownloadWrapper::new();
@@ -326,6 +324,17 @@ impl MainWindow {
                     }
                 }
                 add_log("连接成功".to_string());
+                // match mes_service.get_order_number(mes_config.work_order.clone()){
+                //     Ok(batch) => {
+                //         label = batch + &label[label.len() - 6..];
+                //         add_log("获取订单号成功".to_string());
+                //     },
+                //     Err(e) =>{
+                //         add_log(e.to_string());
+                //         sender.send(Message::DownloadEnd((label, false))).unwrap();
+                //         return;
+                //     }
+                // }
                 add_log("正在过站检测".to_string());
                 let check_result = mes_service.check_station(
                     mes_config.work_order.clone(),

src/mes_service.rs

@@ -61,6 +61,18 @@ impl MesService {
         }
     }
 
+    pub fn get_order_number(&self, work_order: String)->Result<String, Error>{
+        let mut conn = self.pool.get_conn()?;
+        let order_number: Vec<String> = conn.query(format!(
+            "SELECT `Batch` FROM wisun_ordertables WHERE `OrderId` = '{work_order}'"
+        ))?;
+        if order_number.is_empty() {
+            Err(Error::msg("No order number found"))
+        }else{
+            Ok(order_number[0].clone())
+        }
+    }
+
     pub fn update_station(
         &self,
         work_order: String,