feat(fs): 新增预压缩gzip文件参数

Author: kotiln

src/mods/fs/mod.rs

@@ -21,6 +21,14 @@ pub struct Param {
     /// 指定目录
     #[arg(long, short, default_value_t = String::from("."))]
     pub dir: String,
+
+    /// 指定预压缩的 gzip 文件后缀名，如：.gz .terrain
+    #[arg(long, short, default_values_t = vec![String::from(".gz"), String::from(".terrain")])]
+    pub gzip: Vec<String>,
+
+    /// 是否即使命中文件后缀名，也要读取文件判断是否为 gzip 文件
+    #[arg(long = "gf")]
+    pub gzip_file: bool,
 }
 
 pub async fn main(param: Param) -> tokio::io::Result<()> {

src/mods/fs/serve_dir.rs

@@ -18,11 +18,18 @@ pub async fn add_gzip_encoding(
     assert!(uri.starts_with('/'));
 
     let mut response = next.run(request).await;
-    if response.status() == StatusCode::OK {
-        let path = format!("{}{}", param.dir, normalize_path(&uri));
-        macro_log::d!("path: {}", path);
 
-        if is_gzip_file(&path).await {
+    if response.status() == StatusCode::OK && param.gzip.iter().any(|it| uri.ends_with(it)) {
+        let mut is_gzip = true;
+
+        // 即使命中文件后缀名，也要读取文件判断是否为 gzip 文件
+        if param.gzip_file {
+            let path = format!("{}{}", param.dir, normalize_path(&uri));
+            macro_log::d!("path: {}", path);
+            is_gzip = is_gzip_file(&path).await;
+        }
+
+        if is_gzip {
             response
                 .headers_mut()
                 .insert(header::CONTENT_ENCODING, HeaderValue::from_static("gzip"));