From 38d2264c73518b0b55a66297249e547e1f51cca9 Mon Sep 17 00:00:00 2001
From: Phodal Huang <h@phodal.com>
Date: Tue, 17 Dec 2019 08:03:26 +0800
Subject: [PATCH] feat: add multiple API support

---
 cmd/api.go | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/cmd/api.go b/cmd/api.go
index 5bc7ebd..a024c19 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() {
-- 
GitLab