feat: enrich search with msglib display entities

2026-07-10 03:09:37 +08:00
parent 8be4a6a6c5
commit 1277316663
11 changed files with 400 additions and 30 deletions
--- a/internal/tools/display_entities.go
+++ b/internal/tools/display_entities.go
@@ -0,0 +1,108 @@
+package tools
+
+import (
+	"context"
+	"strings"
+
+	"isphere-ai-bridge/internal/isphere"
+	"isphere-ai-bridge/internal/msglib"
+)
+
+type DisplayEntitySource interface {
+	DisplayEntities(ctx context.Context, opts msglib.DisplayEntitiesOptions) ([]msglib.DisplayEntity, error)
+}
+
+func displayEntityQueryLimit(limit int) int {
+	if limit <= 0 {
+		return msglib.DefaultDisplayEntityLimit
+	}
+	if limit > msglib.MaxDisplayEntityLimit {
+		return msglib.MaxDisplayEntityLimit
+	}
+	return limit
+}
+
+func contactsFromDisplayEntities(entities []msglib.DisplayEntity) []isphere.Contact {
+	contacts := make([]isphere.Contact, 0, len(entities))
+	for _, entity := range entities {
+		id := strings.TrimSpace(entity.JID)
+		name := strings.TrimSpace(entity.DisplayName)
+		if id == "" {
+			id = name
+		}
+		if name == "" {
+			name = id
+		}
+		if id == "" {
+			continue
+		}
+		contacts = append(contacts, isphere.Contact{
+			ContactID:   id,
+			DisplayName: name,
+			Account:     strings.TrimSpace(entity.JID),
+			Source:      "msglib_readonly",
+			Confidence:  entity.Confidence,
+			RawRef:      "msglib:" + entity.SourceTable,
+		})
+	}
+	return contacts
+}
+
+func groupsFromDisplayEntities(entities []msglib.DisplayEntity) []isphere.Group {
+	groups := make([]isphere.Group, 0, len(entities))
+	for _, entity := range entities {
+		id := strings.TrimSpace(entity.JID)
+		name := strings.TrimSpace(entity.DisplayName)
+		if id == "" {
+			id = name
+		}
+		if name == "" {
+			name = id
+		}
+		if id == "" {
+			continue
+		}
+		groups = append(groups, isphere.Group{
+			GroupID:     id,
+			DisplayName: name,
+			Source:      "msglib_readonly",
+			Confidence:  entity.Confidence,
+			RawRef:      "msglib:" + entity.SourceTable,
+		})
+	}
+	return groups
+}
+
+func mergeContacts(primary []isphere.Contact, fallback []isphere.Contact, limit int) isphere.SearchContactsResult {
+	seen := map[string]bool{}
+	contacts := make([]isphere.Contact, 0, len(primary)+len(fallback))
+	for _, contact := range append(primary, fallback...) {
+		id := strings.TrimSpace(contact.ContactID)
+		if id == "" || seen[strings.ToLower(id)] {
+			continue
+		}
+		seen[strings.ToLower(id)] = true
+		contacts = append(contacts, contact)
+		if limit > 0 && len(contacts) >= limit {
+			break
+		}
+	}
+	return isphere.SearchContactsResult{Contacts: contacts}
+}
+
+func mergeGroups(primary []isphere.Group, fallback []isphere.Group, limit int) isphere.SearchGroupsResult {
+	seen := map[string]bool{}
+	groups := make([]isphere.Group, 0, len(primary)+len(fallback))
+	for _, group := range append(primary, fallback...) {
+		id := strings.TrimSpace(group.GroupID)
+		if id == "" || seen[strings.ToLower(id)] {
+			continue
+		}
+		seen[strings.ToLower(id)] = true
+		groups = append(groups, group)
+		if limit > 0 && len(groups) >= limit {
+			break
+		}
+	}
+	return isphere.SearchGroupsResult{Groups: groups}
+}