style: cargo fmt, bump to v0.1.6

LiXiang050789 · LiXiang050789 · commit 6b7781e8ea23 · 2026-07-25T19:58:56.000+08:00
diff --git a/src/cli/Cargo.toml b/src/cli/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "qtcloud-data-cli"
-version = "0.1.5"
+version = "0.1.6"
 edition = "2024"
 authors = ["QuantTide Inc. <opensource@quanttide.com>"]
 description = "量潮数据云 CLI"
diff --git a/src/cli/src/blueprint_core.rs b/src/cli/src/blueprint_core.rs
@@ -113,7 +113,10 @@ pub fn extract_cue(response: &str) -> String {
     for marker in &["```cue", "```CUE", "```"] {
         if let Some(start) = response.find(marker) {
             let s = start + marker.len();
-            let e = response[s..].find("```").map(|i| s + i).unwrap_or(response.len());
+            let e = response[s..]
+                .find("```")
+                .map(|i| s + i)
+                .unwrap_or(response.len());
             let code = response[s..e].trim();
             if code.contains("package") || code.contains("#Blueprint") {
                 return code.to_string();
@@ -285,7 +288,8 @@ mod tests {
 
     #[test]
     fn test_extract_cue_from_markdown_block() {
-        let response = "Here is the CUE:\n```cue\npackage blueprints\nx: {name: \"test\"}\n```\nDone.";
+        let response =
+            "Here is the CUE:\n```cue\npackage blueprints\nx: {name: \"test\"}\n```\nDone.";
         let cue = extract_cue(response);
         assert!(cue.contains("package blueprints"));
         assert!(!cue.contains("```"));
@@ -309,15 +313,33 @@ mod tests {
 
     #[test]
     fn test_render_html_contains_name() {
-        let html = render_html("test", None, "draft", "2026-01-01", "2026-01-01", "in", "out", &[]);
+        let html = render_html(
+            "test",
+            None,
+            "draft",
+            "2026-01-01",
+            "2026-01-01",
+            "in",
+            "out",
+            &[],
+        );
         assert!(html.contains("<title>test — Blueprint</title>"));
         assert!(html.contains("<h1>test</h1>"));
     }
 
     #[test]
     fn test_render_html_with_steps() {
         let steps = [("s1", "a", "b", "do it"), ("s2", "b", "c", "then this")];
-        let html = render_html("bp", Some("desc"), "confirmed", "t1", "t2", "in", "out", &steps);
+        let html = render_html(
+            "bp",
+            Some("desc"),
+            "confirmed",
+            "t1",
+            "t2",
+            "in",
+            "out",
+            &steps,
+        );
         assert!(html.contains("s1"));
         assert!(html.contains("s2"));
         assert!(html.contains("do it"));
@@ -344,7 +366,11 @@ mod tests {
         let tmp = std::env::temp_dir().join("bp-test-resolve");
         let _ = std::fs::remove_dir_all(&tmp);
         std::fs::create_dir_all(&tmp).unwrap();
-        std::fs::write(tmp.join("my-bp.cue"), "package blueprints\n{name: \"test\"}").unwrap();
+        std::fs::write(
+            tmp.join("my-bp.cue"),
+            "package blueprints\n{name: \"test\"}",
+        )
+        .unwrap();
 
         let result = resolve_cue_path("my-bp", tmp.to_str().unwrap());
         assert!(result.is_some());
@@ -482,7 +508,9 @@ pub fn contract_tables_to_yaml(md_tables: &str) -> (String, String) {
     let output_fields = parse_md_table(md_tables, "输出契约");
 
     if !input_fields.is_empty() && input_fields == output_fields {
-        eprintln!("错误: 输入契约和输出契约解析到相同字段。LLM 可能跳过了 section 标题，请在 prompt 中要求 LLM 输出 ## 输入契约 和 ## 输出契约 标题。");
+        eprintln!(
+            "错误: 输入契约和输出契约解析到相同字段。LLM 可能跳过了 section 标题，请在 prompt 中要求 LLM 输出 ## 输入契约 和 ## 输出契约 标题。"
+        );
         std::process::exit(1);
     }
 
@@ -519,8 +547,10 @@ fn parse_md_table(text: &str, section: &str) -> Vec<Vec<String>> {
             found_section = true;
             continue;
         }
-        if in_section && line.starts_with('|') && !line.contains("---") && !line.contains("字段名") {
-            let cells: Vec<String> = line.split('|')
+        if in_section && line.starts_with('|') && !line.contains("---") && !line.contains("字段名")
+        {
+            let cells: Vec<String> = line
+                .split('|')
                 .map(|c| c.trim().to_string())
                 .filter(|c| !c.is_empty())
                 .collect();
@@ -536,7 +566,8 @@ fn parse_md_table(text: &str, section: &str) -> Vec<Vec<String>> {
     if !found_section {
         for line in text.lines() {
             if line.starts_with('|') && !line.contains("---") && !line.contains("字段名") {
-                let cells: Vec<String> = line.split('|')
+                let cells: Vec<String> = line
+                    .split('|')
                     .map(|c| c.trim().to_string())
                     .filter(|c| !c.is_empty())
                     .collect();
@@ -550,15 +581,23 @@ fn parse_md_table(text: &str, section: &str) -> Vec<Vec<String>> {
 }
 
 fn fields_to_schema_desc(fields: &[Vec<String>]) -> String {
-    if fields.is_empty() { return "待定义".into(); }
-    let items: Vec<String> = fields.iter()
-        .filter_map(|f| f.first().map(|name| format!("{name}: {}", f.get(1).unwrap_or(&"string".into()))))
+    if fields.is_empty() {
+        return "待定义".into();
+    }
+    let items: Vec<String> = fields
+        .iter()
+        .filter_map(|f| {
+            f.first()
+                .map(|name| format!("{name}: {}", f.get(1).unwrap_or(&"string".into())))
+        })
         .collect();
     format!("{{\n    {}\n  }}", items.join(",\n    "))
 }
 
 fn render_contract_md(input: &[Vec<String>], output: &[Vec<String>]) -> String {
-    let mut md = String::from("## 输入契约\n\n| 字段名 | 类型 | 业务含义 | 约束条件 |\n|--------|------|----------|----------|\n");
+    let mut md = String::from(
+        "## 输入契约\n\n| 字段名 | 类型 | 业务含义 | 约束条件 |\n|--------|------|----------|----------|\n",
+    );
     for row in input {
         let cells: Vec<&str> = row.iter().map(|c| c.as_str()).collect();
         if cells.len() >= 4 {
@@ -613,7 +652,10 @@ pub fn blueprint_table_to_yaml(md_table: &str, project_name: &str) -> (String, S
         let deps_yaml = if deps.is_empty() || deps == "-" {
             String::new()
         } else {
-            let dep_list: String = deps.split(',').map(|d| format!("\n          - {}", d.trim())).collect();
+            let dep_list: String = deps
+                .split(',')
+                .map(|d| format!("\n          - {}", d.trim()))
+                .collect();
             format!("\n        depends:{}", dep_list)
         };
 
@@ -657,10 +699,14 @@ updated_at: "2026-07-24T00:00:00+00:00"
 }
 
 fn render_blueprint_md(name: &str, steps: &[Vec<String>]) -> String {
-    let mut md = format!("# {name}\n\n## 处理步骤\n\n| 步骤名 | 输入 | 输出 | 处理逻辑 | 依赖 |\n|--------|------|------|----------|------|\n");
+    let mut md = format!(
+        "# {name}\n\n## 处理步骤\n\n| 步骤名 | 输入 | 输出 | 处理逻辑 | 依赖 |\n|--------|------|------|----------|------|\n"
+    );
     for row in steps {
         let cells: Vec<&str> = row.iter().map(|c| c.as_str()).collect();
-        let padded: Vec<String> = (0..5).map(|i| cells.get(i).unwrap_or(&"").to_string()).collect();
+        let padded: Vec<String> = (0..5)
+            .map(|i| cells.get(i).unwrap_or(&"").to_string())
+            .collect();
         md.push_str(&format!("| {} |\n", padded.join(" | ")));
     }
     md
@@ -702,7 +748,11 @@ pub fn implement_step_prompt(
         from_desc = from_desc,
         to_desc = to_desc,
         step_desc = step_desc,
-        prev_section = if prev_functions.is_empty() { "无（这是第一步）" } else { prev_functions },
+        prev_section = if prev_functions.is_empty() {
+            "无（这是第一步）"
+        } else {
+            prev_functions
+        },
         func_name = to_snake(step_name),
     )
 }
diff --git a/src/cli/src/contract.rs b/src/cli/src/contract.rs
@@ -19,7 +19,8 @@ pub enum ContractAction {
 }
 
 pub fn run(args: &ContractArgs) {
-    let dir = std::env::var("CONTRACT_DIR").unwrap_or_else(|_| ".quanttide/data/contract".to_string());
+    let dir =
+        std::env::var("CONTRACT_DIR").unwrap_or_else(|_| ".quanttide/data/contract".to_string());
 
     match &args.action {
         ContractAction::List => {
@@ -37,14 +38,14 @@ pub fn run(args: &ContractArgs) {
                             && (name.ends_with(".yaml")
                                 || name.ends_with(".cue")
                                 || name.ends_with(".json"))
-                            {
-                                println!(
-                                    "  - {}",
-                                    name.trim_end_matches(".yaml")
-                                        .trim_end_matches(".cue")
-                                        .trim_end_matches(".json")
-                                );
-                            }
+                        {
+                            println!(
+                                "  - {}",
+                                name.trim_end_matches(".yaml")
+                                    .trim_end_matches(".cue")
+                                    .trim_end_matches(".json")
+                            );
+                        }
                     }
                 } else {
                     eprintln!("{stderr}");
@@ -68,10 +69,11 @@ pub fn run(args: &ContractArgs) {
                 .args(["export", "--out", "yaml", "--expression", &key, &dir])
                 .output();
             if let Ok(out) = output
-                && out.status.success() {
-                    println!("{}", String::from_utf8_lossy(&out.stdout));
-                    return;
-                }
+                && out.status.success()
+            {
+                println!("{}", String::from_utf8_lossy(&out.stdout));
+                return;
+            }
             // 回退：直接读取文件
             for ext in &["yaml", "yml", "cue", "json"] {
                 let path = format!("{dir}/{name}.{ext}");
diff --git a/src/cli/src/design.rs b/src/cli/src/design.rs
@@ -54,7 +54,10 @@ fn cmd_contract(input: &str) {
     let llm = quanttide_agent::LLM::default();
     let messages = vec![quanttide_agent::Message::new("user", &prompt)];
 
-    let stem = Path::new(input).file_stem().unwrap_or_default().to_string_lossy();
+    let stem = Path::new(input)
+        .file_stem()
+        .unwrap_or_default()
+        .to_string_lossy();
     println!("正在从 DRD 生成 Contract: {stem} ...");
 
     match llm.complete(&messages, quanttide_agent::llm::CompleteOptions::default()) {
@@ -77,12 +80,16 @@ fn cmd_blueprint(input: &str) {
     let llm = quanttide_agent::LLM::default();
     let messages = vec![quanttide_agent::Message::new("user", &prompt)];
 
-    let stem = Path::new(input).file_stem().unwrap_or_default().to_string_lossy();
+    let stem = Path::new(input)
+        .file_stem()
+        .unwrap_or_default()
+        .to_string_lossy();
     println!("正在从 DRD 生成 Blueprint: {stem} ...");
 
     match llm.complete(&messages, quanttide_agent::llm::CompleteOptions::default()) {
         Ok(resp) => {
-            let (yaml_content, md_content) = blueprint_core::blueprint_table_to_yaml(&resp.content, &stem);
+            let (yaml_content, md_content) =
+                blueprint_core::blueprint_table_to_yaml(&resp.content, &stem);
             write_spec_files(&stem, "blueprint", &yaml_content, &md_content);
 
             // Generate HTML preview from YAML
@@ -91,12 +98,28 @@ fn cmd_blueprint(input: &str) {
                     eprintln!("解析 YAML 失败: {e}");
                     std::process::exit(1);
                 });
-            let step_refs: Vec<(&str, &str, &str, &str)> = bp.pipeline.steps.iter()
-                .map(|s| (s.name.as_str(), s.from.as_str(), s.to.as_str(), s.desc.as_str()))
+            let step_refs: Vec<(&str, &str, &str, &str)> = bp
+                .pipeline
+                .steps
+                .iter()
+                .map(|s| {
+                    (
+                        s.name.as_str(),
+                        s.from.as_str(),
+                        s.to.as_str(),
+                        s.desc.as_str(),
+                    )
+                })
                 .collect();
             let html = blueprint_core::render_html(
-                &bp.name, bp.description.as_deref(), bp.status.as_str(),
-                &bp.created_at, &bp.updated_at, "", "", &step_refs,
+                &bp.name,
+                bp.description.as_deref(),
+                bp.status.as_str(),
+                &bp.created_at,
+                &bp.updated_at,
+                "",
+                "",
+                &step_refs,
             );
             let spec_dir = blueprint_core::spec_dir();
             let html_path = Path::new(&spec_dir).join(format!("{stem}-blueprint.html"));
@@ -168,18 +191,34 @@ fn cmd_preview(input: &str, output: &Option<String>) {
         std::process::exit(1);
     });
 
-    let bp: quanttide_data_core::Blueprint = serde_yaml::from_str(&yaml_content)
-        .unwrap_or_else(|e| {
+    let bp: quanttide_data_core::Blueprint =
+        serde_yaml::from_str(&yaml_content).unwrap_or_else(|e| {
             eprintln!("解析 YAML 失败: {e}");
             std::process::exit(1);
         });
 
-    let step_refs: Vec<(&str, &str, &str, &str)> = bp.pipeline.steps.iter()
-        .map(|s| (s.name.as_str(), s.from.as_str(), s.to.as_str(), s.desc.as_str()))
+    let step_refs: Vec<(&str, &str, &str, &str)> = bp
+        .pipeline
+        .steps
+        .iter()
+        .map(|s| {
+            (
+                s.name.as_str(),
+                s.from.as_str(),
+                s.to.as_str(),
+                s.desc.as_str(),
+            )
+        })
         .collect();
     let html = blueprint_core::render_html(
-        &bp.name, bp.description.as_deref(), bp.status.as_str(),
-        &bp.created_at, &bp.updated_at, "", "", &step_refs,
+        &bp.name,
+        bp.description.as_deref(),
+        bp.status.as_str(),
+        &bp.created_at,
+        &bp.updated_at,
+        "",
+        "",
+        &step_refs,
     );
     std::fs::write(&output_path, &html).unwrap_or_else(|e| {
         eprintln!("写入 .html 失败: {e}");
diff --git a/src/cli/src/implement.rs b/src/cli/src/implement.rs
@@ -34,8 +34,8 @@ fn cmd_implement_python(input: &str, output: &Option<String>) {
         std::process::exit(1);
     });
 
-    let bp: quanttide_data_core::Blueprint = serde_yaml::from_str(&yaml_content)
-        .unwrap_or_else(|e| {
+    let bp: quanttide_data_core::Blueprint =
+        serde_yaml::from_str(&yaml_content).unwrap_or_else(|e| {
             eprintln!("解析 YAML 失败: {e}");
             std::process::exit(1);
         });
@@ -52,7 +52,11 @@ fn cmd_implement_python(input: &str, output: &Option<String>) {
     let mut generated_functions = String::new();
     let mut prev_signatures = String::new();
 
-    println!("正在生成 {} 的 Python 实现 ({} 个步骤)...", bp.name, bp.pipeline.steps.len());
+    println!(
+        "正在生成 {} 的 Python 实现 ({} 个步骤)...",
+        bp.name,
+        bp.pipeline.steps.len()
+    );
 
     for (i, step) in bp.pipeline.steps.iter().enumerate() {
         let prompt = blueprint_core::implement_step_prompt(
@@ -63,7 +67,12 @@ fn cmd_implement_python(input: &str, output: &Option<String>) {
             &prev_signatures,
         );
 
-        println!("  [{}/{}] 正在生成: {} ...", i + 1, bp.pipeline.steps.len(), step.name);
+        println!(
+            "  [{}/{}] 正在生成: {} ...",
+            i + 1,
+            bp.pipeline.steps.len(),
+            step.name
+        );
 
         let messages = vec![quanttide_agent::Message::new("user", &prompt)];
         match llm.complete(&messages, quanttide_agent::llm::CompleteOptions::default()) {
@@ -118,7 +127,10 @@ fn extract_python_fn(response: &str) -> String {
     for marker in &["```python", "```py", "```"] {
         if let Some(start) = response.find(marker) {
             let s = start + marker.len();
-            let e = response[s..].find("```").map(|i| s + i).unwrap_or(response.len());
+            let e = response[s..]
+                .find("```")
+                .map(|i| s + i)
+                .unwrap_or(response.len());
             return response[s..e].trim().to_string();
         }
     }
diff --git a/src/cli/src/main.rs b/src/cli/src/main.rs
@@ -1,6 +1,7 @@
 use clap::{Parser, Subcommand};
 use qtcloud_data_cli::{
-    blueprint, catalog, clarify, contract, design, implement, pipeline, process, review, transfer, version,
+    blueprint, catalog, clarify, contract, design, implement, pipeline, process, review, transfer,
+    version,
 };
 
 #[derive(Parser)]
diff --git a/src/cli/src/review.rs b/src/cli/src/review.rs
diff --git a/src/cli/src/version.rs b/src/cli/src/version.rs
diff --git a/src/cli/tests/blueprint_test.rs b/src/cli/tests/blueprint_test.rs
