first commit

2023-06-01 08:41:54 +08:00
parent 61b0e99833
commit d253ed8d5b
80 changed files with 14245 additions and 0 deletions
@@ -0,0 +1,91 @@
+package core
+
+import (
+	"encoding/json"
+	"fmt"
+	"strings"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+type Nickname struct {
+	ID       string   `json:"i"`
+	Group    bool     `json:"g"`
+	Unix     int      `json:"u"`
+	Value    string   `json:"v"`
+	Platform string   `json:"p"`
+	BotsID   []string `json:"bs"`
+}
+
+var nickname = MakeBucket("nickname")
+
+type NicklabeL struct {
+	Label    string `json:"label"`
+	Value    string `json:"value"`
+	Platform string `json:"platform"`
+	ChatName string `json:"chat_name"`
+}
+
+func CreateNickName(nick *Nickname) {
+	nick.Unix = int(time.Now().Unix())
+	nickname.Create(nick)
+}
+
+func init() {
+	GinApi(GET, "/api/nickname/labels", RequireAuth, func(ctx *gin.Context) {
+		group := true
+		keyword := ctx.Query("gkeyword")
+		if keyword == "" {
+			keyword = ctx.Query("ukeyword")
+			if keyword != "" {
+				group = false
+			}
+		}
+		platform := ctx.Query("platform")
+		data := []NicklabeL{}
+		if keyword != "" {
+			full := false
+			nickname.Foreach(func(b1, b2 []byte) error {
+				v := &Nickname{}
+				code := string(b1)
+				err := json.Unmarshal(b2, v)
+				if err == nil {
+					if v.Group != group {
+						return nil
+					}
+					if platform != "" && v.Platform != platform {
+						return nil
+					}
+					if strings.HasPrefix(code, keyword) || strings.Contains(v.Value, keyword) {
+						nl := NicklabeL{
+							ChatName: v.Value,
+							Value:    code,
+							Platform: v.Platform,
+						}
+						if !group {
+							nl.Label = fmt.Sprintf("%s(%s)", v.Value, code)
+						} else {
+							nl.Label = fmt.Sprintf("%s %s@%s", v.Value, code, v.Platform)
+						}
+						data = append(data, nl)
+						if code == keyword {
+							full = true
+						}
+					}
+				}
+				return nil
+			})
+			if !full {
+				data = append([]NicklabeL{{
+					Label: keyword,
+					Value: keyword,
+				}}, data...)
+			}
+		}
+		ctx.JSON(200, map[string]interface{}{
+			"success": true,
+			"data":    data,
+		})
+	})
+}