buddy_list_manager.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646
  1. package webapi
  2. import (
  3. "context"
  4. "errors"
  5. "fmt"
  6. "log/slog"
  7. "math/rand"
  8. "slices"
  9. "strings"
  10. "github.com/mk6i/open-oscar-server/state"
  11. "github.com/mk6i/open-oscar-server/wire"
  12. )
  13. // BuddyListManager handles the conversion of OSCAR feedbag data
  14. // to WebAPI buddy list format for web clients.
  15. type BuddyListManager struct {
  16. feedbagService FeedbagService
  17. locateService LocateService
  18. iconSource BuddyIconSource
  19. logger *slog.Logger
  20. }
  21. // NewBuddyListManager creates a new instance of the buddy list manager.
  22. func NewBuddyListManager(feedbagService FeedbagService, locateService LocateService, iconSource BuddyIconSource, logger *slog.Logger) *BuddyListManager {
  23. return &BuddyListManager{
  24. feedbagService: feedbagService,
  25. locateService: locateService,
  26. iconSource: iconSource,
  27. logger: logger,
  28. }
  29. }
  30. // BuddyGroup represents a group in the WebAPI buddy list format.
  31. type BuddyGroup struct {
  32. Name string `json:"name" xml:"name"`
  33. // ID is the feedbag group id, read strictly by clients. It follows Name to keep
  34. // the rendered element order unchanged.
  35. ID int `json:"id" xml:"id"`
  36. Buddies []BuddyInfo `json:"buddies" xml:"buddies>buddy"`
  37. Recent bool `json:"recent,omitempty" xml:"recent,omitempty"`
  38. // Smart is null or a number. It is a pointer rather than an interface so
  39. // encoding/xml can render it: a non-nil interface holding a map cannot be
  40. // marshalled, while a nil pointer is simply omitted.
  41. Smart *int `json:"smart,omitempty" xml:"smart,omitempty"`
  42. }
  43. // BuddyInfo represents a buddy in the WebAPI format.
  44. type BuddyInfo struct {
  45. AimID string `json:"aimId" xml:"aimId"`
  46. DisplayID string `json:"displayId" xml:"displayId"`
  47. Friendly string `json:"friendly,omitempty" xml:"friendly,omitempty"` // Viewer's private alias, rendered in preference to DisplayID
  48. State string `json:"state" xml:"state"` // "online", "offline", "away", "idle"
  49. StatusMsg string `json:"statusMsg,omitempty" xml:"statusMsg,omitempty"`
  50. AwayMsg string `json:"awayMsg,omitempty" xml:"awayMsg,omitempty"`
  51. OnlineTime int64 `json:"onlineTime,omitempty" xml:"onlineTime,omitempty"`
  52. IdleTime int `json:"idleTime,omitempty" xml:"idleTime,omitempty"` // Minutes idle
  53. UserType string `json:"userType" xml:"userType"` // "aim", "icq"
  54. Bot bool `json:"bot" xml:"bot"`
  55. Service string `json:"service,omitempty" xml:"service,omitempty"` // Non-native network; omitted for AIM
  56. PresenceIcon string `json:"presenceIcon,omitempty" xml:"presenceIcon,omitempty"`
  57. BuddyIcon string `json:"buddyIcon,omitempty" xml:"buddyIcon,omitempty"`
  58. MoodIcon string `json:"moodIcon,omitempty" xml:"moodIcon,omitempty"`
  59. Capabilities []string `json:"capabilities,omitempty" xml:"capabilities>capability,omitempty"`
  60. MemberSince int64 `json:"memberSince,omitempty" xml:"memberSince,omitempty"`
  61. }
  62. // GetBuddyListForUser retrieves and converts the buddy list for a user.
  63. func (m *BuddyListManager) GetBuddyListForUser(ctx context.Context, sess *Session) ([]BuddyGroup, error) {
  64. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagQuery}
  65. snac, err := m.feedbagService.Query(ctx, sess.OSCARSession, frame)
  66. if err != nil {
  67. return nil, fmt.Errorf("failed to retrieve feedbag: %w", err)
  68. }
  69. reply, ok := snac.Body.(wire.SNAC_0x13_0x06_FeedbagReply)
  70. if !ok {
  71. return nil, fmt.Errorf("failed to retrieve feedbag: unexpected reply type")
  72. }
  73. items := reply.Items
  74. type buddy struct {
  75. name string
  76. alias string
  77. }
  78. type group struct {
  79. name string
  80. buddies map[uint16]buddy
  81. order []uint16
  82. }
  83. type feedbagBL struct {
  84. order []uint16
  85. groups map[uint16]group
  86. }
  87. bl := feedbagBL{groups: make(map[uint16]group)}
  88. for _, item := range items {
  89. if item.ClassID != wire.FeedbagClassIdGroup {
  90. continue
  91. }
  92. if item.GroupID == 0 {
  93. val, hasVal := item.Uint16SliceBE(wire.FeedbagAttributesOrder)
  94. if hasVal {
  95. bl.order = val
  96. }
  97. continue
  98. }
  99. name := item.Name
  100. if name == "" {
  101. name = "Buddies"
  102. }
  103. g := group{
  104. name: name,
  105. buddies: make(map[uint16]buddy),
  106. }
  107. val, _ := item.Uint16SliceBE(wire.FeedbagAttributesOrder)
  108. g.order = val
  109. bl.groups[item.GroupID] = g
  110. }
  111. for _, item := range items {
  112. if item.ClassID != wire.FeedbagClassIdBuddy || item.Name == "" {
  113. continue
  114. }
  115. if _, exists := bl.groups[item.GroupID]; !exists {
  116. bl.groups[item.GroupID] = group{
  117. name: "Buddies",
  118. buddies: make(map[uint16]buddy),
  119. order: nil,
  120. }
  121. }
  122. b := buddy{name: item.Name}
  123. if val, hasVal := item.String(wire.FeedbagAttributesAlias); hasVal {
  124. b.alias = val
  125. }
  126. g := bl.groups[item.GroupID]
  127. g.buddies[item.ItemID] = b
  128. bl.groups[item.GroupID] = g
  129. }
  130. groupOrder := bl.order
  131. if len(groupOrder) == 0 && len(bl.groups) > 0 {
  132. groupOrder = make([]uint16, 0, len(bl.groups))
  133. for gid := range bl.groups {
  134. groupOrder = append(groupOrder, gid)
  135. }
  136. slices.Sort(groupOrder)
  137. }
  138. var out []BuddyGroup
  139. for _, gid := range groupOrder {
  140. g, ok := bl.groups[gid]
  141. if !ok {
  142. continue
  143. }
  144. groupName := g.name
  145. if groupName == "" {
  146. groupName = "Buddies"
  147. }
  148. wg := BuddyGroup{ID: int(gid), Name: groupName, Buddies: []BuddyInfo{}}
  149. for _, bid := range g.order {
  150. b, ok := g.buddies[bid]
  151. if !ok {
  152. continue
  153. }
  154. info := m.getBuddyInfo(ctx, sess.OSCARSession, sess.BaseURL, b.name)
  155. // The alias belongs in friendly, not displayId: the client renders
  156. // friendly in preference to displayId but still shows displayId as
  157. // the buddy's actual screen name.
  158. info.Friendly = b.alias
  159. wg.Buddies = append(wg.Buddies, info)
  160. }
  161. out = append(out, wg)
  162. }
  163. return out, nil
  164. }
  165. // getBuddyInfo retrieves a buddy's current presence by issuing a locate
  166. // UserInfoQuery on behalf of the requesting session's OSCAR instance.
  167. func (m *BuddyListManager) getBuddyInfo(ctx context.Context, instance *state.SessionInstance, baseURL string, buddyName string) BuddyInfo {
  168. // Default to offline. The web client keys users by the normalized aimId and
  169. // shallow-merges each buddy map onto the shared user object, so a display-form
  170. // aimId here overwrites the id every other event is keyed by.
  171. //
  172. // Feedbag buddy names are stored normalized, so they are not a source of
  173. // display names. DisplayID is filled in from the locate reply below when the
  174. // buddy is online, or overridden by the caller's alias when one is set.
  175. ident := state.NewIdentScreenName(buddyName)
  176. info := BuddyInfo{
  177. AimID: ident.String(),
  178. DisplayID: buddyName,
  179. State: "offline",
  180. UserType: userTypeFor(ident),
  181. Service: serviceFor(ident),
  182. Bot: false,
  183. }
  184. reply, err := m.locateService.UserInfoQuery(ctx, instance, wire.SNACFrame{},
  185. wire.SNAC_0x02_0x05_LocateUserInfoQuery{
  186. Type: uint16(wire.LocateTypeUnavailable), // away message
  187. ScreenName: ident.String(),
  188. })
  189. if err != nil {
  190. m.logger.WarnContext(ctx, "failed to query buddy info", "screenName", buddyName, "error", err)
  191. return info
  192. }
  193. userInfo, ok := reply.Body.(wire.SNAC_0x02_0x06_LocateUserInfoReply)
  194. if !ok {
  195. // Locate error => buddy is blocked or offline.
  196. return info
  197. }
  198. info.State = "online"
  199. info.Capabilities = []string{}
  200. // Publish the icon only now that locate has confirmed the buddy is online and
  201. // has not blocked the caller. Offline and blocking buddies return above without
  202. // an icon, so neither their icon nor its activity-revealing hash leaks.
  203. info.BuddyIcon = m.iconSource.PublishedURL(ctx, baseURL, ident)
  204. // The locate reply carries the screen name as the buddy formatted it.
  205. if userInfo.ScreenName != "" {
  206. info.DisplayID = userInfo.ScreenName
  207. }
  208. if tod, ok := userInfo.Uint32BE(wire.OServiceUserInfoSignonTOD); ok {
  209. info.OnlineTime = int64(tod)
  210. }
  211. if st := statusBitState(userInfo.TLVUserInfo, instance.IdentScreenName().UIN() == 0); st != "" {
  212. info.State = st
  213. } else if userInfo.IsAway() {
  214. info.State = "away"
  215. if msg, ok := userInfo.LocateInfo.String(wire.LocateTLVTagsInfoUnavailableData); ok {
  216. info.AwayMsg = msg
  217. }
  218. }
  219. if idle, ok := userInfo.Uint16BE(wire.OServiceUserInfoIdleTime); ok && idle > 0 {
  220. info.IdleTime = int(idle)
  221. if info.State == "online" {
  222. info.State = "idle"
  223. }
  224. }
  225. info.MoodIcon = moodIconURL(baseURL, info.State, userInfoCaps(userInfo.TLVUserInfo))
  226. return info
  227. }
  228. // RemoveBuddyFromFeedbag removes a buddy from a group (or all groups if allGroups is true) using feedbag delete/update SNACs.
  229. func (m *BuddyListManager) RemoveBuddyFromFeedbag(ctx context.Context, sess *Session, buddyName, groupName string, allGroups bool) (resultCode string, err error) {
  230. // Buddy items carry the owner's alias for the buddy, and the feedbag service
  231. // relays a session's own writes only to the owner's other instances, so every
  232. // method here that rewrites buddy items has to drop the alias cache itself.
  233. defer sess.InvalidateAliases()
  234. buddyName = strings.TrimSpace(buddyName)
  235. if buddyName == "" {
  236. return "error", fmt.Errorf("empty buddy")
  237. }
  238. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagQuery}
  239. snac, err := m.feedbagService.Query(ctx, sess.OSCARSession, frame)
  240. if err != nil {
  241. m.logger.ErrorContext(ctx, "remove buddy: feedbag query failed", "err", err.Error())
  242. return "error", err
  243. }
  244. reply, ok := snac.Body.(wire.SNAC_0x13_0x06_FeedbagReply)
  245. if !ok {
  246. return "error", fmt.Errorf("unexpected feedbag reply type")
  247. }
  248. fl := state.NewFeedbagList(reply.Items, rand.Intn)
  249. target := groupName
  250. if allGroups {
  251. target = "*"
  252. }
  253. if err := fl.DeleteBuddy(target, buddyName); err != nil {
  254. if errors.Is(err, state.ErrGroupNotFound) {
  255. return "notFound", nil
  256. }
  257. m.logger.ErrorContext(ctx, "remove buddy: DeleteBuddy failed", "err", err.Error())
  258. return "error", err
  259. }
  260. if pending := fl.PendingDeletes(); len(pending) > 0 {
  261. delFrame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagDeleteItem}
  262. delBody := wire.SNAC_0x13_0x0A_FeedbagDeleteItem{Items: pending}
  263. if _, err := m.feedbagService.DeleteItem(ctx, sess.OSCARSession, delFrame, delBody); err != nil {
  264. m.logger.ErrorContext(ctx, "remove buddy: Feedbag DeleteItem failed", "err", err.Error())
  265. return "error", err
  266. }
  267. } else {
  268. return "notFound", nil
  269. }
  270. if pending := fl.PendingUpdates(); len(pending) > 0 {
  271. upFrame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem}
  272. if _, err := m.feedbagService.UpsertItem(ctx, sess.OSCARSession, upFrame, pending); err != nil {
  273. m.logger.ErrorContext(ctx, "remove buddy: Feedbag UpsertItem failed", "err", err.Error())
  274. return "error", err
  275. }
  276. }
  277. return "success", nil
  278. }
  279. // RemoveGroupFromFeedbag deletes a buddy group and updates the root order (TOC DelGroup).
  280. func (m *BuddyListManager) RemoveGroupFromFeedbag(ctx context.Context, sess *Session, requestedGroup string) (resultCode string, err error) {
  281. defer sess.InvalidateAliases()
  282. req := strings.TrimSpace(requestedGroup)
  283. if req == "" {
  284. return "error", fmt.Errorf("empty group")
  285. }
  286. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagQuery}
  287. snac, err := m.feedbagService.Query(ctx, sess.OSCARSession, frame)
  288. if err != nil {
  289. m.logger.ErrorContext(ctx, "remove group: feedbag query failed", "err", err.Error())
  290. return "error", err
  291. }
  292. reply, ok := snac.Body.(wire.SNAC_0x13_0x06_FeedbagReply)
  293. if !ok {
  294. return "error", fmt.Errorf("unexpected feedbag reply type")
  295. }
  296. storedName, found := storedGroupNameForRequest(reply.Items, req)
  297. if !found {
  298. return "notFound", nil
  299. }
  300. fl := state.NewFeedbagList(reply.Items, rand.Intn)
  301. fl.DeleteGroup(storedName)
  302. if pending := fl.PendingDeletes(); len(pending) > 0 {
  303. delFrame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagDeleteItem}
  304. delBody := wire.SNAC_0x13_0x0A_FeedbagDeleteItem{Items: pending}
  305. if _, err := m.feedbagService.DeleteItem(ctx, sess.OSCARSession, delFrame, delBody); err != nil {
  306. m.logger.ErrorContext(ctx, "remove group: Feedbag DeleteItem failed", "err", err.Error())
  307. return "error", err
  308. }
  309. } else {
  310. return "notFound", nil
  311. }
  312. if pending := fl.PendingUpdates(); len(pending) > 0 {
  313. upFrame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem}
  314. if _, err := m.feedbagService.UpsertItem(ctx, sess.OSCARSession, upFrame, pending); err != nil {
  315. m.logger.ErrorContext(ctx, "remove group: Feedbag UpsertItem failed", "err", err.Error())
  316. return "error", err
  317. }
  318. }
  319. return "success", nil
  320. }
  321. // RenameGroupInFeedbag renames a buddy group, updating the group item in place.
  322. func (m *BuddyListManager) RenameGroupInFeedbag(ctx context.Context, sess *Session, oldGroup, newGroup string) (resultCode string, err error) {
  323. defer sess.InvalidateAliases()
  324. oldGroup = strings.TrimSpace(oldGroup)
  325. newGroup = strings.TrimSpace(newGroup)
  326. if oldGroup == "" || newGroup == "" {
  327. return "error", fmt.Errorf("empty group name")
  328. }
  329. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagQuery}
  330. snac, err := m.feedbagService.Query(ctx, sess.OSCARSession, frame)
  331. if err != nil {
  332. m.logger.ErrorContext(ctx, "rename group: feedbag query failed", "err", err.Error())
  333. return "error", err
  334. }
  335. reply, ok := snac.Body.(wire.SNAC_0x13_0x06_FeedbagReply)
  336. if !ok {
  337. return "error", fmt.Errorf("unexpected feedbag reply type")
  338. }
  339. storedName, found := storedGroupNameForRequest(reply.Items, oldGroup)
  340. if !found {
  341. return "notFound", nil
  342. }
  343. fl := state.NewFeedbagList(reply.Items, rand.Intn)
  344. if err := fl.RenameGroup(storedName, newGroup); err != nil {
  345. switch {
  346. case errors.Is(err, state.ErrGroupNotFound):
  347. return "notFound", nil
  348. case errors.Is(err, state.ErrGroupExists):
  349. return "alreadyExists", nil
  350. default:
  351. m.logger.ErrorContext(ctx, "rename group: RenameGroup failed", "err", err.Error())
  352. return "error", err
  353. }
  354. }
  355. if pending := fl.PendingUpdates(); len(pending) > 0 {
  356. upFrame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem}
  357. if _, err := m.feedbagService.UpsertItem(ctx, sess.OSCARSession, upFrame, pending); err != nil {
  358. m.logger.ErrorContext(ctx, "rename group: Feedbag UpsertItem failed", "err", err.Error())
  359. return "error", err
  360. }
  361. }
  362. return "success", nil
  363. }
  364. // MoveBuddyInFeedbag moves a buddy to a different group and/or repositions it
  365. // within a group's order.
  366. func (m *BuddyListManager) MoveBuddyInFeedbag(ctx context.Context, sess *Session, buddyName, fromGroup, toGroup, beforeBuddy string) (resultCode string, err error) {
  367. defer sess.InvalidateAliases()
  368. buddyName = strings.TrimSpace(buddyName)
  369. fromGroup = strings.TrimSpace(fromGroup)
  370. toGroup = strings.TrimSpace(toGroup)
  371. beforeBuddy = strings.TrimSpace(beforeBuddy)
  372. if buddyName == "" || fromGroup == "" {
  373. return "error", fmt.Errorf("empty buddy or group")
  374. }
  375. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagQuery}
  376. snac, err := m.feedbagService.Query(ctx, sess.OSCARSession, frame)
  377. if err != nil {
  378. m.logger.ErrorContext(ctx, "move buddy: feedbag query failed", "err", err.Error())
  379. return "error", err
  380. }
  381. reply, ok := snac.Body.(wire.SNAC_0x13_0x06_FeedbagReply)
  382. if !ok {
  383. return "error", fmt.Errorf("unexpected feedbag reply type")
  384. }
  385. storedFrom, found := storedGroupNameForRequest(reply.Items, fromGroup)
  386. if !found {
  387. return "notFound", nil
  388. }
  389. storedTo := ""
  390. if toGroup != "" {
  391. storedTo, found = storedGroupNameForRequest(reply.Items, toGroup)
  392. if !found {
  393. return "notFound", nil
  394. }
  395. }
  396. fl := state.NewFeedbagList(reply.Items, rand.Intn)
  397. if err := fl.MoveBuddy(storedFrom, storedTo, buddyName, beforeBuddy); err != nil {
  398. if errors.Is(err, state.ErrGroupNotFound) || errors.Is(err, state.ErrBuddyNotFound) {
  399. return "notFound", nil
  400. }
  401. m.logger.ErrorContext(ctx, "move buddy: MoveBuddy failed", "err", err.Error())
  402. return "error", err
  403. }
  404. if pending := fl.PendingDeletes(); len(pending) > 0 {
  405. delFrame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagDeleteItem}
  406. delBody := wire.SNAC_0x13_0x0A_FeedbagDeleteItem{Items: pending}
  407. if _, err := m.feedbagService.DeleteItem(ctx, sess.OSCARSession, delFrame, delBody); err != nil {
  408. m.logger.ErrorContext(ctx, "move buddy: Feedbag DeleteItem failed", "err", err.Error())
  409. return "error", err
  410. }
  411. }
  412. if pending := fl.PendingUpdates(); len(pending) > 0 {
  413. var inserts, updates []wire.FeedbagItem
  414. for _, item := range pending {
  415. if item.ClassID == wire.FeedbagClassIdBuddy {
  416. inserts = append(inserts, item)
  417. } else {
  418. updates = append(updates, item)
  419. }
  420. }
  421. if len(inserts) > 0 {
  422. insFrame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem}
  423. if _, err := m.feedbagService.UpsertItem(ctx, sess.OSCARSession, insFrame, inserts); err != nil {
  424. m.logger.ErrorContext(ctx, "move buddy: Feedbag InsertItem failed", "err", err.Error())
  425. return "error", err
  426. }
  427. }
  428. if len(updates) > 0 {
  429. upFrame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem}
  430. if _, err := m.feedbagService.UpsertItem(ctx, sess.OSCARSession, upFrame, updates); err != nil {
  431. m.logger.ErrorContext(ctx, "move buddy: Feedbag UpdateItem failed", "err", err.Error())
  432. return "error", err
  433. }
  434. }
  435. }
  436. return "success", nil
  437. }
  438. // SetBuddyAttributeInFeedbag sets a buddy's friendly (alias) name across all
  439. // groups it belongs to. An empty friendly clears the alias.
  440. func (m *BuddyListManager) SetBuddyAttributeInFeedbag(ctx context.Context, sess *Session, buddyName, friendly string) (resultCode string, err error) {
  441. defer sess.InvalidateAliases()
  442. buddyName = strings.TrimSpace(buddyName)
  443. if buddyName == "" {
  444. return "error", fmt.Errorf("empty buddy")
  445. }
  446. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagQuery}
  447. snac, err := m.feedbagService.Query(ctx, sess.OSCARSession, frame)
  448. if err != nil {
  449. m.logger.ErrorContext(ctx, "set buddy attribute: feedbag query failed", "err", err.Error())
  450. return "error", err
  451. }
  452. reply, ok := snac.Body.(wire.SNAC_0x13_0x06_FeedbagReply)
  453. if !ok {
  454. return "error", fmt.Errorf("unexpected feedbag reply type")
  455. }
  456. fl := state.NewFeedbagList(reply.Items, rand.Intn)
  457. found, err := fl.SetBuddyAlias(buddyName, friendly)
  458. if err != nil {
  459. m.logger.ErrorContext(ctx, "set buddy attribute: SetBuddyAlias failed", "err", err.Error())
  460. return "error", err
  461. }
  462. if !found {
  463. return "notFound", nil
  464. }
  465. if pending := fl.PendingUpdates(); len(pending) > 0 {
  466. upFrame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem}
  467. if _, err := m.feedbagService.UpsertItem(ctx, sess.OSCARSession, upFrame, pending); err != nil {
  468. m.logger.ErrorContext(ctx, "set buddy attribute: Feedbag UpsertItem failed", "err", err.Error())
  469. return "error", err
  470. }
  471. }
  472. return "success", nil
  473. }
  474. // SetGroupAttributeInFeedbag sets a group's collapsed state. An empty group
  475. // targets the unnamed default group.
  476. func (m *BuddyListManager) SetGroupAttributeInFeedbag(ctx context.Context, sess *Session, groupName string, collapsed bool) (resultCode string, err error) {
  477. defer sess.InvalidateAliases()
  478. groupName = strings.TrimSpace(groupName)
  479. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagQuery}
  480. snac, err := m.feedbagService.Query(ctx, sess.OSCARSession, frame)
  481. if err != nil {
  482. m.logger.ErrorContext(ctx, "set group attribute: feedbag query failed", "err", err.Error())
  483. return "error", err
  484. }
  485. reply, ok := snac.Body.(wire.SNAC_0x13_0x06_FeedbagReply)
  486. if !ok {
  487. return "error", fmt.Errorf("unexpected feedbag reply type")
  488. }
  489. // An empty group targets the unnamed default group (stored name ""); a named
  490. // group is resolved through the "Buddies" default-label mapping.
  491. storedName := ""
  492. if groupName != "" {
  493. var found bool
  494. storedName, found = storedGroupNameForRequest(reply.Items, groupName)
  495. if !found {
  496. return "notFound", nil
  497. }
  498. }
  499. fl := state.NewFeedbagList(reply.Items, rand.Intn)
  500. if err := fl.SetGroupCollapsed(storedName, collapsed); err != nil {
  501. if errors.Is(err, state.ErrGroupNotFound) {
  502. return "notFound", nil
  503. }
  504. m.logger.ErrorContext(ctx, "set group attribute: SetGroupCollapsed failed", "err", err.Error())
  505. return "error", err
  506. }
  507. if pending := fl.PendingUpdates(); len(pending) > 0 {
  508. upFrame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem}
  509. if _, err := m.feedbagService.UpsertItem(ctx, sess.OSCARSession, upFrame, pending); err != nil {
  510. m.logger.ErrorContext(ctx, "set group attribute: Feedbag UpsertItem failed", "err", err.Error())
  511. return "error", err
  512. }
  513. }
  514. return "success", nil
  515. }
  516. // feedbagGroupMatchesRequested returns true if a feedbag group row matches the
  517. // group the Web client asked for. OSCAR often stores the default group with an
  518. // empty name; GetBuddyListForUser labels that as "Buddies", so addBuddy must
  519. // treat "" and "Buddies" as the same bucket when the client sends group=Buddies.
  520. func feedbagGroupMatchesRequested(storedName, requested string) bool {
  521. req := strings.TrimSpace(requested)
  522. st := strings.TrimSpace(storedName)
  523. if strings.EqualFold(st, req) {
  524. return true
  525. }
  526. if strings.EqualFold(strings.TrimSpace(req), "Buddies") && st == "" {
  527. return true
  528. }
  529. return false
  530. }
  531. // storedGroupNameForRequest returns the feedbag group row Name for a Web client group label.
  532. // Rows with GroupID 0 are the root order record, not a named buddy group.
  533. func storedGroupNameForRequest(items []wire.FeedbagItem, requested string) (string, bool) {
  534. for _, item := range items {
  535. if item.ClassID != wire.FeedbagClassIdGroup {
  536. continue
  537. }
  538. if item.GroupID == 0 {
  539. continue
  540. }
  541. if feedbagGroupMatchesRequested(item.Name, requested) {
  542. return item.Name, true
  543. }
  544. }
  545. return "", false
  546. }
  547. // FeedbagAliases collects the aliases the feedbag owner has assigned to their
  548. // buddies, keyed by normalized screen name. Buddies without an alias are absent.
  549. func FeedbagAliases(items []wire.FeedbagItem) map[string]string {
  550. aliases := make(map[string]string)
  551. for _, item := range items {
  552. if item.ClassID != wire.FeedbagClassIdBuddy || item.Name == "" {
  553. continue
  554. }
  555. alias, ok := item.String(wire.FeedbagAttributesAlias)
  556. if !ok || alias == "" {
  557. continue
  558. }
  559. aliases[state.NewIdentScreenName(item.Name).String()] = alias
  560. }
  561. return aliases
  562. }
  563. // LookupBuddyAliases returns the aliases the session owner has assigned to their
  564. // buddies, keyed by normalized screen name.
  565. //
  566. // Aliases are private to the viewer and live only in their feedbag, so they cannot
  567. // be derived from a locate reply the way display names are.
  568. func LookupBuddyAliases(ctx context.Context, feedbagService FeedbagService, instance *state.SessionInstance) (map[string]string, error) {
  569. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagQuery}
  570. snac, err := feedbagService.Query(ctx, instance, frame)
  571. if err != nil {
  572. return nil, err
  573. }
  574. reply, ok := snac.Body.(wire.SNAC_0x13_0x06_FeedbagReply)
  575. if !ok {
  576. return nil, fmt.Errorf("unexpected feedbag reply type")
  577. }
  578. return FeedbagAliases(reply.Items), nil
  579. }