Support multiple prefixes to group them together and/or a regex daixi…

…ang0#107 Signed-off-by: Author Name <luw2007@gmail.com>
luw2007 · Jan 16, 2023 · 8db20a1 · 8db20a1
1 parent 33a14c3
commit 8db20a1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/pkg/section/parser_test.go b/pkg/section/parser_test.go
@@ -40,6 +40,11 @@ func TestParse(t *testing.T) {
 			expectedSection: nil,
 			expectedError:   errors.New("invalid params: prefix("),
 		},
+		{
+			input:           []string{"prefix(domainA;domainB)"},
+			expectedSection: SectionList{Custom{"domainA;domainB"}},
+			expectedError:   nil,
+		},
 	}
 	for _, test := range testCases {
 		parsedSection, err := Parse(test.input)

diff --git a/pkg/section/prefix.go b/pkg/section/prefix.go
@@ -15,9 +15,12 @@ type Custom struct {
 const CustomType = "custom"
 
 func (c Custom) MatchSpecificity(spec *parse.GciImports) specificity.MatchSpecificity {
-	if strings.HasPrefix(spec.Path, c.Prefix) {
-		return specificity.Match{Length: len(c.Prefix)}
+	for _, prefix := range strings.Split(c.Prefix, ";") {
+		if strings.HasPrefix(spec.Path, prefix) {
+			return specificity.Match{Length: len(prefix)}
+		}
 	}
+
 	return specificity.MisMatch{}
 }