diff --git a/cmd/api.go b/cmd/api.go index 5bc7ebd538bc4c3b61c114f372073639df95fcf9..a024c19c5c6d6548af50b984a459394e5edcf37f 100644 --- a/cmd/api.go +++ b/cmd/api.go @@ -11,6 +11,7 @@ import ( "log" "os" "os/exec" + "regexp" "strconv" "strings" ) @@ -76,7 +77,20 @@ var apiCmd *cobra.Command = &cobra.Command{ } func replacePackage(content string) string { - return strings.ReplaceAll(content, apiCmdConfig.RemovePackageNames, "") + var packagegsReggex string + packageNameArray := strings.Split(apiCmdConfig.RemovePackageNames, ",") + for index, name := range packageNameArray { + packagegsReggex = packagegsReggex + strings.ReplaceAll(name, ".", "\\.") + if index != len(packageNameArray) - 1 { + packagegsReggex = packagegsReggex + "|" + } + } + + re, _ := regexp.Compile(packagegsReggex) + + + return re.ReplaceAllString(content, "") + //return strings.ReplaceAll(content, apiCmdConfig.RemovePackageNames, "") } func init() {