refactor: simplify config

Author: AsterDY

README.md

@@ -53,17 +53,14 @@ refactoring guidance.
 ## Running through Coze OpenAPI
 1. Set .env file for configuration on ABCoder's working directory. Taking Coze as an example:
 ```
-# Base directory for other XXX_DIRs, default to current directory
-WORK_DIR=./
+# cache for repo，AST and so on
+WORK_DIR=tmp_abcoder
 
-# DIRs below support both relative and absolute paths, and relative paths are relative to WORK_DIR
-REPO_DIR=tmp_repos
-CACHE_DIR=tmp_caches
-TOOLS_DIR=tools
-EXCLUDE_DIRS=target,gen-codes
+# exclude dirs for repo parsing, separated by comma
+EXCLUDE_DIRS=target,gen-codes 
 
-# coze|ollama 
-API_TYPE=coze 
+# LLM's api type
+API_TYPE=coze # coze|ollama 
 
 # LLM's output language
 LANGUAGE=zh 
@@ -89,6 +86,10 @@ cargo run --bin cmd compress https://xxx.git
    3. Call the LLM to compress the repository codes, and refresh the AST for each call.
 You can stop the process at anytime after step 2. You can restart the compressing by running the same command.
 
+5. Export the compressed results
+```
+cargo run --bin cmd export https://xxx.git --out-dir {OUTPUT_DIR}
+
 # Status Update
 
 The system is designed to automatically fetch the latest data from Github upon triggering relevant tasks, ensuring the

src/bin/cmd.rs

@@ -51,7 +51,8 @@ fn main() {
 }
 
 const USAGE: &str = "Usage: ABCoder <Action> <RepoPath> [Flags]
-Actions:
+RepoPath: the path of the repo to compress. Can be a local path or a git url.
+Actions: compress|export
 compress: compress the repo. Including flags:
     --parse-only: only parse the repo, not compress it
     --export-compress: export the compress result
@@ -60,7 +61,7 @@ compress: compress the repo. Including flags:
     --no-need-comment: not need comment in symbol content (only works for Go now)
 export: export the compress result to csv or markdown (default). Including flags:
     --csv: export the compress result to csv
-    --out-dir <path>: output directory path, default is current directory
+    --out-dir <path>: output directory path, default is $WORK_DIR
     --public-only: only export the public symbols
 ";
 

src/compress/llm/maas.rs

@@ -1,11 +1,11 @@
 // Copyright 2025 CloudWeGo Authors
-// 
+//
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
-// 
+//
 //     https://www.apache.org/licenses/LICENSE-2.0
-// 
+//
 // Unless required by applicable law or agreed to in writing, software
 // distributed under the License is distributed on an "AS IS" BASIS,
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@@ -24,7 +24,7 @@ pub fn maas_compress_py(to_compress: &ToCompress, model_name: &str) -> String {
     use std::process::{Command, Stdio};
 
     let command = "python3";
-    let path = format!("{}/maas/model_call.py", &CONFIG.tools_dir);
+    let path = format!("{}/maas/model_call.py", &CONFIG.work_dir);
     println!("executing command: {} {} {}", command, path, model_name);
 
     // Create a new command process with piped stdin

src/config/mod.rs

@@ -28,13 +28,10 @@ pub enum Language {
 #[derive(Debug)]
 pub struct Config {
     pub work_dir: String,
-    pub repo_dir: String,
-    pub cache_dir: String,
     pub parser_dir: String,
     pub api_type: String,
     pub maas_model_name: String,
     pub mass_http_url: String,
-    pub tools_dir: String,
 
     pub coze_api_token: Option<String>,
     pub coze_bot_id: Option<String>,
@@ -46,23 +43,7 @@ pub struct Config {
 }
 
 fn default_work_dir() -> String {
-    std::env::current_dir()
-        .unwrap()
-        .to_str()
-        .unwrap()
-        .to_string()
-}
-
-fn default_repo_dir() -> String {
-    "tmp_repos".to_string()
-}
-
-fn default_tools_dir() -> String {
-    "tools".to_string()
-}
-
-fn default_cache_dir() -> String {
-    "tmp_caches".to_string()
+    "tmp_abcoder".to_string()
 }
 
 fn default_parser_dir() -> String {
@@ -81,11 +62,8 @@ impl Config {
     pub fn new() -> Self {
         Self {
             work_dir: default_work_dir(),
-            repo_dir: default_repo_dir(),
-            cache_dir: default_cache_dir(),
             parser_dir: default_parser_dir(),
             api_type: default_api_type(),
-            tools_dir: default_tools_dir(),
             maas_model_name: default_maas_model_name(),
             mass_http_url: "".to_string(),
             coze_api_token: None,
@@ -97,13 +75,10 @@ impl Config {
     }
 
     pub fn parse_from_env() -> Self {
-        let mut s = Self {
+        let s = Self {
             work_dir: std::env::var("WORK_DIR").unwrap_or_else(|_| default_work_dir()),
-            repo_dir: std::env::var("REPO_DIR").unwrap_or_else(|_| default_repo_dir()),
-            cache_dir: std::env::var("CACHE_DIR").unwrap_or_else(|_| default_cache_dir()),
             parser_dir: std::env::var("PARSER_DIR").unwrap_or_else(|_| default_parser_dir()),
             api_type: std::env::var("API_TYPE").unwrap_or_else(|_| default_api_type()),
-            tools_dir: std::env::var("TOOLS_DIR").unwrap_or_else(|_| default_tools_dir()),
             maas_model_name: std::env::var("MAAS_MODEL_NAME")
                 .unwrap_or_else(|_| default_maas_model_name()),
             mass_http_url: std::env::var("MASS_HTTP_URL").unwrap_or_else(|_| "".to_string()),
@@ -121,35 +96,6 @@ impl Config {
                 })
                 .unwrap_or(Language::Chinese),
         };
-
-        if !s.repo_dir.starts_with("/") {
-            s.repo_dir = Path::new(&s.work_dir)
-                .join(s.repo_dir)
-                .to_str()
-                .unwrap()
-                .to_string();
-        }
-        if !s.cache_dir.starts_with("/") {
-            s.cache_dir = Path::new(&s.work_dir)
-                .join(s.cache_dir)
-                .to_str()
-                .unwrap()
-                .to_string();
-        }
-        if !s.parser_dir.starts_with("/") {
-            s.parser_dir = Path::new(&s.work_dir)
-                .join(s.parser_dir)
-                .to_str()
-                .unwrap()
-                .to_string();
-        }
-        if !s.tools_dir.starts_with("/") {
-            s.tools_dir = Path::new(&s.work_dir)
-                .join(s.tools_dir)
-                .to_str()
-                .unwrap()
-                .to_string();
-        }
         s
     }
 }
@@ -161,18 +107,8 @@ lazy_static! {
     };
 }
 
-pub fn go_ast_path() -> String {
-    Path::new(&CONFIG.tools_dir)
-        .join("parser")
-        .join("go_ast")
-        .to_str()
-        .unwrap()
-        .to_string()
-}
-
-pub fn rust_ast_path() -> String {
-    Path::new(&CONFIG.tools_dir)
-        .join("parser")
+pub fn parser_path() -> String {
+    Path::new(&CONFIG.parser_dir)
         .join("lang")
         .to_str()
         .unwrap()
@@ -223,7 +159,8 @@ pub fn parser_and_args<'a>(
     opts: &parse::CompressOptions,
 ) -> (String, Vec<String>) {
     let lang = decide_language(repo_path);
-    let path = rust_ast_path();
+    let path = parser_path();
+    println!("parser path: {:?}", path);
     let mut args = vec![
         "collect".to_string(),
         lang.to_string(),

src/export.rs

@@ -256,7 +256,7 @@ pub fn export_repo(repo: &Repository, opts: &ExportOptions) {
         Path::new(path)
     } else {
         // pwd
-        Path::new(".")
+        Path::new(&CONFIG.work_dir)
     };
 
     if opts.csv {

src/parse.rs

@@ -41,7 +41,7 @@ pub fn force_parse_repo(repo_path: &String, opts: &CompressOptions) -> Result<Re
 }
 
 fn parse_repo_path(repo_path: &String) -> Result<PathBuf, Error> {
-    let git_dir = Path::new(CONFIG.repo_dir.as_str());
+    let git_dir = Path::new(CONFIG.work_dir.as_str());
 
     let ps: Vec<&str> = repo_path.split('/').collect();
     let path = if repo_path.ends_with(".git") || repo_path.starts_with("https://") {

src/storage/cache.rs

@@ -1,11 +1,11 @@
 // Copyright 2025 CloudWeGo Authors
-// 
+//
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
-// 
+//
 //     https://www.apache.org/licenses/LICENSE-2.0
-// 
+//
 // Unless required by applicable law or agreed to in writing, software
 // distributed under the License is distributed on an "AS IS" BASIS,
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@@ -59,7 +59,7 @@ impl<C: StorageEngine, B: StorageEngine> StorageEngine for CachingStorageEngine<
 
 pub fn get_cache() -> Box<dyn StorageEngine> {
     let mem = MemoryCache::new(12);
-    let fs = FileStorage::new(Path::new(CONFIG.cache_dir.as_str())).unwrap();
+    let fs = FileStorage::new(Path::new(CONFIG.work_dir.as_str())).unwrap();
 
     let mut cache = CachingStorageEngine::new(mem, fs);
     return Box::new(cache);

src/utils/files.rs

@@ -1,11 +1,11 @@
 // Copyright 2025 CloudWeGo Authors
-// 
+//
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
-// 
+//
 //     https://www.apache.org/licenses/LICENSE-2.0
-// 
+//
 // Unless required by applicable law or agreed to in writing, software
 // distributed under the License is distributed on an "AS IS" BASIS,
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@@ -69,7 +69,7 @@ impl fmt::Display for Node {
             for _ in 0..depth {
                 write!(f, "   ")?;
             }
-            let display_name = node.name.trim_start_matches(&CONFIG.repo_dir);
+            let display_name = node.name.trim_start_matches(&CONFIG.work_dir);
             writeln!(f, "└──{}", display_name)?;
 
             for child in &node.children {