chore: expose def_limit arg

README.md

@@ -163,14 +163,14 @@ While there is undoubtedly a trade-off in precision, the benefits are clear:
 > the areas where it is primarily used.
 
 The method we use to demonstrate accuracy is to compare the results with those of LSP/LSIF. It must be admitted that
-static inference is almost impossible to obtain all reference relationships like LSP, but in strict mode, our
-calculation accuracy is still quite considerable. In normal mode, you can decide whether to adopt the relationship based
-on the weight returned.
+static inference is almost impossible to obtain all reference relationships like LSP. 
 
-| Repo                                | Precision (Strict Mode) | Graph Generated Time |
-|-------------------------------------|-------------------------|----------------------|
-| https://github.com/williamfzc/srctx | 80/80 = 100 %           | 83.139791ms          |
-| https://github.com/gin-gonic/gin    | 160/167 = 95.80838 %    | 310.6805ms           |
+You can further combine your own needs and use other methods such as tfidf to process the results to meet more complex requirements.
+
+| Repo                             | Coverage of LSP Edges by Gossiphs |
+|----------------------------------|-----------------------------------|
+| https://github.com/go-gorm/gorm  | 442/499 = 88.5 %                  |
+| https://github.com/gin-gonic/gin | 238/252 = 94.4%                   |
 
 ## Contribution
 
@@ -183,6 +183,8 @@ You just need to:
 2. Test it in [src/extractor.rs](src/extractor.rs)
 3. Try it with your repo in [src/graph.rs](src/graph.rs)
 
+[Tree-sitter Playground](https://tree-sitter.github.io/tree-sitter/playground) is a good helper.
+
 ## License
 
 [Apache 2.0](LICENSE)

src/graph.rs

@@ -457,20 +457,23 @@ pub struct GraphConfig {
     pub project_path: String,
 
     // a ref can only belong to limit def
+    #[pyo3(get, set)]
     pub def_limit: usize,
 
     // commit size limit
     // reduce the impact of large commits
     // large commit: edit more than xx% files once
     // default to 1.0, do nothing
     // set to 0.3, means 30%
+    #[pyo3(get, set)]
     pub commit_size_limit_ratio: f32,
 
     // commit history search depth
     #[pyo3(get, set)]
     pub depth: u32,
 
-    // symbol limit of each file
+    // symbol limit of each file, for ignoring large files
+    #[pyo3(get, set)]
     pub symbol_limit: usize,
 
     #[pyo3(get, set)]
@@ -487,7 +490,7 @@ impl GraphConfig {
     pub fn default() -> GraphConfig {
         GraphConfig {
             project_path: String::from("."),
-            def_limit: 1,
+            def_limit: 4,
             commit_size_limit_ratio: 1.0,
             depth: 10240,
             symbol_limit: 4096,

src/main.rs

@@ -63,6 +63,9 @@ struct CommonOptions {
     #[clap(default_value = "false")]
     strict: bool,
 
+    #[clap(long)]
+    def_limit: Option<usize>,
+
     /// git commit history search depth
     #[clap(long)]
     depth: Option<u32>,
@@ -80,6 +83,7 @@ impl CommonOptions {
         CommonOptions {
             project_path: String::from("."),
             strict: false,
+            def_limit: None,
             depth: None,
             exclude_file_regex: None,
             exclude_author_regex: None,
@@ -247,6 +251,10 @@ fn handle_relation(relation_cmd: RelationCommand) {
     if relation_cmd.common_options.strict {
         config.def_limit = 1;
     }
+    if relation_cmd.common_options.def_limit.is_some() {
+        config.def_limit = relation_cmd.common_options.def_limit.unwrap();
+    }
+
     if let Some(depth) = relation_cmd.common_options.depth {
         config.depth = depth;
     }