feat: support multi company, split using ,

internal/imports/formatter.go

@@ -52,8 +52,8 @@ type Formatter struct {
 	stdPackages      map[string]struct{}
 	moduleNames      map[string]string
 	formattedPaths   map[string]struct{}
+	companyPrefixes  map[string]struct{}
 	eg               errgroup.Group
-	companyPrefix    string
 	muModuleNames    sync.RWMutex
 	muFormattedPaths sync.RWMutex
 	isList           bool
@@ -316,7 +316,7 @@ func (ft *Formatter) groupDSTImportSpecs(importSpecs []*dst.ImportSpec, moduleNa
 	result := make(map[string][]*dst.ImportSpec)
 	result[stdImport] = make([]*dst.ImportSpec, 0, 8)
 	result[thirdPartyImport] = make([]*dst.ImportSpec, 0, 8)
-	if ft.companyPrefix != "" {
+	if len(ft.companyPrefixes) != 0 {
 		result[companyImport] = make([]*dst.ImportSpec, 0, 8)
 	}
 	result[localImport] = make([]*dst.ImportSpec, 0, 8)
@@ -335,18 +335,27 @@ func (ft *Formatter) groupDSTImportSpecs(importSpecs []*dst.ImportSpec, moduleNa
 			continue
 		}
 
-		if ft.companyPrefix != "" &&
-			strings.HasPrefix(importPath, ft.companyPrefix) {
+		if len(ft.companyPrefixes) != 0 {
+			existImport := false
+			for companyPrefix := range ft.companyPrefixes {
+				if strings.HasPrefix(importPath, companyPrefix) {
 					result[companyImport] = append(result[companyImport], importSpec)
+					existImport = true
+					break
+				}
+			}
+
+			if existImport {
 				continue
 			}
+		}
 
 		result[thirdPartyImport] = append(result[thirdPartyImport], importSpec)
 	}
 
 	ft.logDSTImportSpecs("groupDSTImportSpecs: stdImport", result[stdImport])
 	ft.logDSTImportSpecs("groupDSTImportSpecs: thirdPartyImport", result[thirdPartyImport])
-	if ft.companyPrefix != "" {
+	if len(ft.companyPrefixes) != 0 {
 		ft.logDSTImportSpecs("groupDSTImportSpecs: companyImport", result[companyImport])
 	}
 	ft.logDSTImportSpecs("groupDSTImportSpecs: localImport", result[localImport])

internal/imports/formatter_option.go

@@ -1,5 +1,7 @@
 package imports
 
+import "strings"
+
 type FormatterOptionFn func(*Formatter)
 
 func FormatterWithList(isList bool) FormatterOptionFn {
@@ -28,6 +30,13 @@ func FormatterWithVerbose(isVerbose bool) FormatterOptionFn {
 
 func FormatterWithCompanyPrefix(companyPrefix string) FormatterOptionFn {
 	return func(ft *Formatter) {
-		ft.companyPrefix = companyPrefix
+		ft.companyPrefixes = make(map[string]struct{})
+		for _, prefix := range strings.Split(companyPrefix, ",") {
+			prefix = strings.TrimSpace(prefix)
+			if prefix == "" {
+				continue
+			}
+			ft.companyPrefixes[prefix] = struct{}{}
+		}
 	}
 }