From 8f7aff558f1a62a43852be5091302dc5572553ca Mon Sep 17 00:00:00 2001
From: Kevin Wan <wanjunfeng@gmail.com>
Date: Mon, 22 Jul 2024 21:17:59 +0800
Subject: [PATCH] chore: refactor BuildTypes in tsgen. (#4266)

---
 tools/goctl/api/tsgen/gencomponents.go | 41 ++++++++++++--------------
 1 file changed, 19 insertions(+), 22 deletions(-)

diff --git a/tools/goctl/api/tsgen/gencomponents.go b/tools/goctl/api/tsgen/gencomponents.go
index ceac13ee0..fab8ebd0f 100644
--- a/tools/goctl/api/tsgen/gencomponents.go
+++ b/tools/goctl/api/tsgen/gencomponents.go
@@ -14,13 +14,31 @@ import (
 //go:embed components.tpl
 var componentsTemplate string
 
+// BuildTypes generates the typescript code for the types.
+func BuildTypes(types []spec.Type) (string, error) {
+	var builder strings.Builder
+	first := true
+	for _, tp := range types {
+		if first {
+			first = false
+		} else {
+			builder.WriteString("\n")
+		}
+		if err := writeType(&builder, tp); err != nil {
+			return "", apiutil.WrapErr(err, "Type "+tp.Name()+" generate error")
+		}
+	}
+
+	return builder.String(), nil
+}
+
 func genComponents(dir string, api *spec.ApiSpec) error {
 	types := api.Types
 	if len(types) == 0 {
 		return nil
 	}
 
-	val, err := buildTypes(types)
+	val, err := BuildTypes(types)
 	if err != nil {
 		return err
 	}
@@ -45,24 +63,3 @@ func genComponents(dir string, api *spec.ApiSpec) error {
 		"componentTypes": val,
 	})
 }
-
-func buildTypes(types []spec.Type) (string, error) {
-	var builder strings.Builder
-	first := true
-	for _, tp := range types {
-		if first {
-			first = false
-		} else {
-			builder.WriteString("\n")
-		}
-		if err := writeType(&builder, tp); err != nil {
-			return "", apiutil.WrapErr(err, "Type "+tp.Name()+" generate error")
-		}
-	}
-
-	return builder.String(), nil
-}
-
-func BuildTypes(types []spec.Type) (string, error) {
-	return buildTypes(types)
-}