feat: suuport trimming enums

Author: DMwangnima

tool/trimmer/test_cases/test_enum/base.thrift

@@ -0,0 +1,34 @@
+// 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
+//
+//   http://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.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// This enum will be referenced and used
+enum AccountType {
+    FREE = 1,
+    PREMIUM = 2,
+    ENTERPRISE = 3
+}
+
+// This enum will not be used and should be trimmed
+enum UnusedLevel {
+    LEVEL1 = 1,
+    LEVEL2 = 2,
+    LEVEL3 = 3
+}
+
+// This enum will be used via typedef
+enum Country {
+    US = 1,
+    UK = 2,
+    CN = 3
+}

tool/trimmer/test_cases/test_enum/enum_test.thrift

@@ -0,0 +1,59 @@
+// 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
+//
+//   http://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.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// This enum is used by Person struct
+enum Status {
+    ACTIVE = 1,
+    INACTIVE = 2,
+    PENDING = 3
+}
+
+// This enum is NOT used anywhere and should be trimmed
+enum UnusedColor {
+    RED = 1,
+    GREEN = 2,
+    BLUE = 3
+}
+
+// This enum is used as a field type
+enum Gender {
+    MALE = 1,
+    FEMALE = 2,
+    OTHER = 3
+}
+
+// This enum is NOT used and should be trimmed
+enum UnusedPriority {
+    LOW = 1,
+    MEDIUM = 2,
+    HIGH = 3
+}
+
+// This enum is used in service method
+enum ResponseCode {
+    SUCCESS = 0,
+    ERROR = 1,
+    TIMEOUT = 2
+}
+
+struct Person {
+    1: string name
+    2: Status status
+    3: Gender gender
+}
+
+service TestService {
+    ResponseCode getStatus(1: string id)
+    Person getPerson(1: string id)
+}

tool/trimmer/test_cases/test_enum/include_test.thrift

@@ -0,0 +1,27 @@
+// 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
+//
+//   http://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.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+include "base.thrift"
+
+typedef base.Country Location
+
+struct Account {
+    1: string username
+    2: base.AccountType type
+    3: Location location
+}
+
+service AccountService {
+    Account getAccount(1: string id)
+}

tool/trimmer/trim/config.go

@@ -31,6 +31,7 @@ type YamlArguments struct {
 	DisablePreserveComment *bool    `yaml:"disable_preserve_comment,omitempty"`
 	MatchGoName            *bool    `yaml:"match_go_name,omitempty"`
 	PreservedFiles         []string `yaml:"preserved_files,omitempty"`
+	TrimEnums              *bool    `yaml:"trim_enums,omitempty"`
 }
 
 func ParseYamlConfig(path string) *YamlArguments {

tool/trimmer/trim/traversal.go

@@ -88,4 +88,15 @@ func (t *Trimmer) traversal(ast *parser.Thrift, filename string) {
 	}
 
 	t.structsTrimmed -= len(ast.Structs) + len(ast.Includes) + len(ast.Services) + len(ast.Unions) + len(ast.Exceptions)
+	if t.trimEnums {
+		listEnum := make([]*parser.Enum, 0, len(ast.Enums))
+		for i := range ast.Enums {
+			_, ok := currentMap[ast.Enums[i]]
+			if ok {
+				listEnum = append(listEnum, ast.Enums[i])
+			}
+		}
+		ast.Enums = listEnum
+		t.structsTrimmed -= len(ast.Enums)
+	}
 }

tool/trimmer/trim/trimmer.go

@@ -43,6 +43,7 @@ type Trimmer struct {
 	forceTrimming          bool
 	preservedStructsMap    map[string]struct{}
 	disablePreserveComment bool
+	trimEnums              bool
 	structsTrimmed         int
 	fieldsTrimmed          int
 	extServices            []*parser.Service
@@ -60,6 +61,7 @@ type TrimASTArg struct {
 	PreserveStructs        []string
 	PreservedFiles         []string
 	IncludeDirs            []string
+	TrimEnums              *bool
 }
 
 type TrimResultInfo struct {
@@ -106,6 +108,9 @@ func TrimAST(arg *TrimASTArg) (trimResultInfo *TrimResultInfo, err error) {
 			if arg.DisablePreserveComment == nil && cfg.DisablePreserveComment != nil {
 				arg.DisablePreserveComment = cfg.DisablePreserveComment
 			}
+			if arg.TrimEnums == nil && cfg.TrimEnums != nil {
+				arg.TrimEnums = cfg.TrimEnums
+			}
 			if len(preservedStructs) == 0 {
 				preservedStructs = cfg.PreservedStructs
 			}
@@ -127,11 +132,15 @@ func TrimAST(arg *TrimASTArg) (trimResultInfo *TrimResultInfo, err error) {
 	if arg.DisablePreserveComment != nil {
 		disablePreserveComment = *arg.DisablePreserveComment
 	}
-	return doTrimAST(arg.Ast, arg.TrimMethods, forceTrim, matchGoName, disablePreserveComment, preservedStructs, preservedFiles)
+	trimEnums := false
+	if arg.TrimEnums != nil {
+		trimEnums = *arg.TrimEnums
+	}
+	return doTrimAST(arg.Ast, arg.TrimMethods, forceTrim, matchGoName, disablePreserveComment, trimEnums, preservedStructs, preservedFiles)
 }
 
 // doTrimAST trim the single AST, pass method names if -m specified
-func doTrimAST(ast *parser.Thrift, trimMethods []string, forceTrimming, matchGoName, disablePreserveComment bool, preservedStructs, preserveFiles []string) (
+func doTrimAST(ast *parser.Thrift, trimMethods []string, forceTrimming, matchGoName, disablePreserveComment, trimEnums bool, preservedStructs, preserveFiles []string) (
 	trimResultInfo *TrimResultInfo, err error) {
 	trimmer, err := newTrimmer(nil, "")
 	if err != nil {
@@ -142,6 +151,7 @@ func doTrimAST(ast *parser.Thrift, trimMethods []string, forceTrimming, matchGoN
 	trimmer.trimMethodValid = make([]bool, len(trimMethods))
 	trimmer.forceTrimming = forceTrimming
 	trimmer.matchGoName = matchGoName
+	trimmer.trimEnums = trimEnums
 	if len(ast.Services) > 0 {
 		for i, method := range trimMethods {
 			parts := strings.Split(method, ".")
@@ -229,6 +239,9 @@ func Trim(files, includeDir []string, outDir string) error {
 
 func (t *Trimmer) countStructs(ast *parser.Thrift) {
 	t.structsTrimmed += len(ast.Structs) + len(ast.Includes) + len(ast.Services) + len(ast.Unions) + len(ast.Exceptions)
+	if t.trimEnums {
+		t.structsTrimmed += len(ast.Enums)
+	}
 	for _, v := range ast.Structs {
 		t.fieldsTrimmed += len(v.Fields)
 	}