Added minimal lines length for documentation generation

Author: codemakerai-dev

cli/cli.go

@@ -113,6 +113,7 @@ func (c *Cli) parseGenerateArgs() {
 		lang := generateDocsCmd.String("language", "", "Programming language: JavaScript, Java, Kotlin")
 		replace := generateDocsCmd.Bool("replace", false, "Determines if the existing documentations are replaced")
 		codePath := generateDocsCmd.String("codepath", "", "The codepath to match.")
+		minimalLinesLength := generateDocsCmd.Int("minimal-lines-length", 0, "The minimal lines complexity.")
 		endpoint := generateDocsCmd.String("endpoint", "", "The endpoint name.")
 		failFast := generateDocsCmd.Bool("fail-fast", true, "Whether to stop on error.")
 
@@ -142,7 +143,7 @@ func (c *Cli) parseGenerateArgs() {
 
 		files := generateDocsCmd.Args()[0:]
 
-		if err := c.generateDocumentation(cl, lang, replace, codePath, failFast, files); err != nil {
+		if err := c.generateDocumentation(cl, lang, replace, codePath, minimalLinesLength, failFast, files); err != nil {
 			c.logger.Errorf("Could not generate the documentation %v", err)
 		}
 		break
@@ -224,7 +225,7 @@ func (c *Cli) generateCode(cl client.Client, lang *string, replace *bool, codePa
 			return err
 		}
 
-		output, err := c.process(cl, client.ModeCode, *lang, *replace, codePath, model, source)
+		output, err := c.process(cl, client.ModeCode, *lang, *replace, codePath, nil, model, source)
 		if err != nil {
 			if !c.isFailFast(failFast) {
 				c.logger.Errorf("Failed to process file %s %v", file, err)
@@ -245,7 +246,7 @@ func (c *Cli) generateCode(cl client.Client, lang *string, replace *bool, codePa
 	})
 }
 
-func (c *Cli) generateDocumentation(cl client.Client, lang *string, replace *bool, codePath *string, failFast *bool, files []string) error {
+func (c *Cli) generateDocumentation(cl client.Client, lang *string, replace *bool, codePath *string, minimalLinesLength *int, failFast *bool, files []string) error {
 	return c.walkPath(files, func(file string) error {
 		if lang == nil || len(*lang) == 0 {
 			actLang, err := languageFromExtension(filepath.Ext(file))
@@ -269,7 +270,7 @@ func (c *Cli) generateDocumentation(cl client.Client, lang *string, replace *boo
 			return err
 		}
 
-		output, err := c.process(cl, client.ModeDocument, *lang, *replace, codePath, nil, source)
+		output, err := c.process(cl, client.ModeDocument, *lang, *replace, codePath, minimalLinesLength, nil, source)
 		if err != nil {
 			if !c.isFailFast(failFast) {
 				c.logger.Errorf("Failed to process file %s %v", file, err)
@@ -314,7 +315,7 @@ func (c *Cli) fixSyntax(cl client.Client, lang *string, failFast *bool, files []
 			return nil
 		}
 
-		output, err := c.process(cl, client.ModeFixSyntax, *lang, false, nil, nil, source)
+		output, err := c.process(cl, client.ModeFixSyntax, *lang, false, nil, nil, nil, source)
 		if err != nil {
 			if !c.isFailFast(failFast) {
 				c.logger.Errorf("Failed to process file %s %v", file, err)
@@ -334,7 +335,7 @@ func (c *Cli) fixSyntax(cl client.Client, lang *string, failFast *bool, files []
 	})
 }
 
-func (c *Cli) process(cl client.Client, mode string, lang string, replace bool, codePath *string, model *string, source string) (*string, error) {
+func (c *Cli) process(cl client.Client, mode string, lang string, replace bool, codePath *string, minimalLinesLength *int, model *string, source string) (*string, error) {
 	ctx := context.Background()
 
 	modify := client.ModifyNone
@@ -346,16 +347,23 @@ func (c *Cli) process(cl client.Client, mode string, lang string, replace bool,
 		codePath = nil
 	}
 
+	var optMinimalLinesLength *int32 = nil
+	if minimalLinesLength != nil && *minimalLinesLength != 0 {
+		val := int32(*minimalLinesLength)
+		optMinimalLinesLength = &val
+	}
+
 	process, err := cl.Process(ctx, &client.ProcessRequest{
 		Mode:     mode,
 		Language: lang,
 		Input: client.Input{
 			Source: source,
 		},
 		Options: &client.Options{
-			Modify:   &modify,
-			CodePath: codePath,
-			Model:    model,
+			Modify:             &modify,
+			CodePath:           codePath,
+			Model:              model,
+			MinimalLinesLength: optMinimalLinesLength,
 		},
 	})
 	if err != nil {