From a31d055835cf1df3b0c34a335ce72ebed1b29918 Mon Sep 17 00:00:00 2001
From: Hatter Jiang <jht5945@gmail.com>
Date: Sun, 19 Jan 2025 15:44:29 +0800
Subject: [PATCH] feat: add publish date and update time

---
 script-meta-v2.json |  8 ++++++--
 update-meta-v2.rs   | 41 +++++++++++++++++++++++++++++++++++++++--
 2 files changed, 45 insertions(+), 4 deletions(-)

diff --git a/script-meta-v2.json b/script-meta-v2.json
index bb0eb8c..6600519 100644
--- a/script-meta-v2.json
+++ b/script-meta-v2.json
@@ -3,12 +3,16 @@
     "script_name": "helloworld.go",
     "script_length": 128,
     "script_sha256": "b807f5d29dd2dbd4dc8c665616c836b1e1c0274d377672109a3e2397bb1139bf",
-    "script_full_url": "https://git.hatter.ink/hatter/go-scripts/raw/branch/main/helloworld-go/main.go"
+    "script_full_url": "https://git.hatter.ink/hatter/go-scripts/raw/branch/main/helloworld-go/main.go",
+    "publish_time": 1737272661406,
+    "update_time": 1737272661406
   },
   "myip.go": {
     "script_name": "myip.go",
     "script_length": 798,
     "script_sha256": "2abb41d64288774c01c465508070bec5ba7350228ab0dc79a2fe679b5afb07b6",
-    "script_full_url": "https://git.hatter.ink/hatter/go-scripts/raw/branch/main/myip-go/main.go"
+    "script_full_url": "https://git.hatter.ink/hatter/go-scripts/raw/branch/main/myip-go/main.go",
+    "publish_time": 1737272661405,
+    "update_time": 1737272661405
   }
 }
\ No newline at end of file
diff --git a/update-meta-v2.rs b/update-meta-v2.rs
index f14586c..58b61c1 100755
--- a/update-meta-v2.rs
+++ b/update-meta-v2.rs
@@ -9,7 +9,8 @@
 //! ```
 
 use rust_util::{
-    debugging, failure_and_exit, iff, opt_result, opt_value_result, success, warning, XResult,
+    debugging, failure_and_exit, iff, opt_result, opt_value_result, success, util_time, warning,
+    XResult,
 };
 use serde::{Deserialize, Serialize};
 use std::collections::BTreeMap;
@@ -41,7 +42,7 @@ impl ScriptConfig {
     }
 }
 
-#[derive(Serialize)]
+#[derive(Serialize, Deserialize)]
 struct ScriptMeta {
     script_name: String,
     script_length: u64,
@@ -49,6 +50,10 @@ struct ScriptMeta {
     script_full_url: String,
     #[serde(skip_serializing_if = "Option::is_none")]
     single_script_file: Option<bool>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    publish_time: Option<u128>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    update_time: Option<u128>,
 }
 
 fn main() -> XResult<()> {
@@ -71,6 +76,7 @@ fn main() -> XResult<()> {
     );
     debugging!("Script config: {:#?}", script_config);
 
+    let former_script_meta_map = read_script_meta_map_from_file();
     let mut script_meta_map = BTreeMap::new();
 
     let current_read_dir = opt_result!(fs::read_dir("."), "Read dir '.' failed: {}");
@@ -105,6 +111,7 @@ fn main() -> XResult<()> {
                     &main_script,
                     false,
                     &script_config,
+                    &former_script_meta_map,
                 )?,
             );
         }
@@ -142,6 +149,7 @@ fn main() -> XResult<()> {
                         &abs_file_entry,
                         true,
                         &script_config,
+                        &former_script_meta_map,
                     )?,
                 );
             }
@@ -157,6 +165,14 @@ fn main() -> XResult<()> {
     Ok(())
 }
 
+fn read_script_meta_map_from_file() -> BTreeMap<String, ScriptMeta> {
+    let script_meta_content = match fs::read_to_string(SCRIPT_META_FILE) {
+        Ok(script_meta_content) => script_meta_content,
+        Err(_) => return BTreeMap::new(),
+    };
+    serde_json::from_str(&script_meta_content).unwrap_or_default()
+}
+
 // translate filename-ext to filename.ext
 fn translate_script_dir_to_script_name(
     script_dir: &str,
@@ -180,6 +196,7 @@ fn read_script_meta(
     script_path: &PathBuf,
     is_simple_script: bool,
     script_config: &ScriptConfig,
+    former_script_meta_map: &BTreeMap<String, ScriptMeta>,
 ) -> XResult<ScriptMeta> {
     let script_meta = opt_result!(
         script_path.metadata(),
@@ -198,11 +215,31 @@ fn read_script_meta(
             .replace("$NAME", script_dir)
     };
     let single_script_file = iff!(is_simple_script, Some(true), None);
+    let former_script_meta = former_script_meta_map.get(&script_name);
+    let publish_time = Some(
+        former_script_meta
+            .map(|m| m.publish_time)
+            .flatten()
+            .unwrap_or_else(|| util_time::get_current_millis()),
+    );
+    let is_file_same = former_script_meta
+        .map(|m| m.script_sha256 == script_sha256)
+        .unwrap_or(false);
+    let update_time = Some(if is_file_same {
+        former_script_meta
+            .map(|m| m.update_time)
+            .flatten()
+            .unwrap_or_else(|| util_time::get_current_millis())
+    } else {
+        util_time::get_current_millis()
+    });
     Ok(ScriptMeta {
         script_name: script_name.clone(),
         script_length: script_meta.len(),
         script_sha256,
         script_full_url,
         single_script_file,
+        publish_time,
+        update_time,
     })
 }