feat: support id query for "list" apis (#908)

* feat: support id query for "list" apis

* gosec

model/common.go

@@ -1,9 +1,14 @@
 package model
 
 import (
+	"cmp"
+	"slices"
+	"strconv"
+	"strings"
 	"time"
 
 	"github.com/gin-gonic/gin"
+	"github.com/nezhahq/nezha/pkg/utils"
 )
 
 const (
@@ -61,6 +66,56 @@ func FindByUserID[S ~[]E, E CommonInterface](s S, uid uint64) []uint64 {
 	return list
 }
 
+func SearchByIDCtx[S ~[]E, E CommonInterface](c *gin.Context, x S) S {
+	switch any(x).(type) {
+	case []*Server:
+		l := searchByIDCtxServer(c, any(x).([]*Server))
+		return any(l).(S)
+	default:
+		var s S
+		for _, idStr := range strings.Split(c.Query("id"), ",") {
+			id, err := strconv.ParseUint(idStr, 10, 64)
+			if err != nil {
+				continue
+			}
+
+			if i, ok := slices.BinarySearchFunc(x, id, func(e E, t uint64) int {
+				return cmp.Compare(e.GetID(), t)
+			}); ok {
+				s = append(s, x[i])
+			}
+		}
+		return utils.IfOr(len(s) > 0, s, x)
+	}
+}
+
+func searchByIDCtxServer(c *gin.Context, x []*Server) []*Server {
+	list1, list2 := SplitList(x)
+
+	var clist1, clist2 []*Server
+	for _, idStr := range strings.Split(c.Query("id"), ",") {
+		id, err := strconv.ParseUint(idStr, 10, 64)
+		if err != nil {
+			continue
+		}
+
+		if i, ok := slices.BinarySearchFunc(list1, id, func(e *Server, t uint64) int {
+			return cmp.Compare(e.ID, t)
+		}); ok {
+			clist1 = append(clist1, list1[i])
+		}
+
+		if i, ok := slices.BinarySearchFunc(list2, id, func(e *Server, t uint64) int {
+			return cmp.Compare(e.ID, t)
+		}); ok {
+			clist2 = append(clist2, list2[i])
+		}
+	}
+
+	l := slices.Concat(clist1, clist2)
+	return utils.IfOr(len(l) > 0, l, x)
+}
+
 type Response struct {
 	Code    int         `json:"code,omitempty"`
 	Message string      `json:"message,omitempty"`

model/server.go

@@ -2,6 +2,7 @@ package model
 
 import (
 	"log"
+	"slices"
 	"time"
 
 	"gorm.io/gorm"
@@ -54,3 +55,20 @@ func (s *Server) AfterFind(tx *gorm.DB) error {
 	}
 	return nil
 }
+
+// Split a sorted server list into two separate lists:
+// The first list contains servers with a priority set (DisplayIndex != 0).
+// The second list contains servers without a priority set (DisplayIndex == 0).
+// The original slice is not modified. If no server without a priority is found, it returns nil.
+func SplitList(x []*Server) ([]*Server, []*Server) {
+	pri := func(s *Server) bool {
+		return s.DisplayIndex == 0
+	}
+
+	i := slices.IndexFunc(x, pri)
+	if i == -1 {
+		return nil, x
+	}
+
+	return x[:i], x[i:]
+}