109 lines
2.7 KiB
Go
109 lines
2.7 KiB
Go
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}
|
|
}
|