fix github search branch in code info api

Signed-off-by: Patrick Zhao <[email protected]>
koderover · Feb 11, 2025 · afd83b0 · afd83b0
1 parent 745ff8f
commit afd83b0
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/pkg/microservice/aslan/core/code/client/github/github.go b/pkg/microservice/aslan/core/code/client/github/github.go
@@ -18,6 +18,8 @@ package github
 
 import (
 	"context"
+	"fmt"
+	"regexp"
 
 	github2 "github.com/google/go-github/v35/github"
 	e "github.com/koderover/zadig/v2/pkg/tool/errors"
@@ -57,10 +59,13 @@ func (c *Client) ListBranches(opt client.ListOpt) ([]*client.Branch, error) {
 	}
 	var res []*client.Branch
 	for _, o := range bList {
-		res = append(res, &client.Branch{
-			Name:      o.GetName(),
-			Protected: o.GetProtected(),
-		})
+		matched, _ := regexp.MatchString(fmt.Sprintf(`%s`, opt.Key), o.GetName())
+		if matched {
+			res = append(res, &client.Branch{
+				Name:      o.GetName(),
+				Protected: o.GetProtected(),
+			})
+		}
 	}
 	return res, nil
 }