buddylist.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628
  1. package handlers
  2. import (
  3. "context"
  4. "log/slog"
  5. "math/rand"
  6. "net/http"
  7. "strings"
  8. "github.com/mk6i/open-oscar-server/server/webapi/types"
  9. "github.com/mk6i/open-oscar-server/state"
  10. "github.com/mk6i/open-oscar-server/wire"
  11. )
  12. // BuddyListHandler handles Web AIM API buddy list management endpoints.
  13. type BuddyListHandler struct {
  14. BuddyListManager *BuddyListManager
  15. Logger *slog.Logger
  16. FeedbagService FeedbagService
  17. }
  18. type FeedbagService interface {
  19. DeleteItem(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, inBody wire.SNAC_0x13_0x0A_FeedbagDeleteItem) (*wire.SNACMessage, error)
  20. Query(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame) (wire.SNACMessage, error)
  21. QueryIfModified(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, inBody wire.SNAC_0x13_0x05_FeedbagQueryIfModified) (wire.SNACMessage, error)
  22. RespondAuthorizeToHost(ctx context.Context, instance state.IdentScreenName, inFrame wire.SNACFrame, inBody wire.SNAC_0x13_0x1A_FeedbagRespondAuthorizeToHost) error
  23. RightsQuery(ctx context.Context, inFrame wire.SNACFrame) wire.SNACMessage
  24. StartCluster(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, inBody wire.SNAC_0x13_0x11_FeedbagStartCluster)
  25. EndCluster(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame) error
  26. UpsertItem(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, items []wire.FeedbagItem) (*wire.SNACMessage, error)
  27. Use(ctx context.Context, instance *state.SessionInstance) error
  28. }
  29. // AddBuddy handles GET /buddylist/addBuddy requests.
  30. func (h *BuddyListHandler) AddBuddy(w http.ResponseWriter, r *http.Request, session *state.WebAPISession) {
  31. ctx := r.Context()
  32. aimsid := r.URL.Query().Get("aimsid")
  33. buddyName := strings.TrimSpace(r.URL.Query().Get("buddy"))
  34. groupName := strings.TrimSpace(r.URL.Query().Get("group"))
  35. if buddyName == "" {
  36. h.sendError(w, r, http.StatusBadRequest, "missing buddy parameter")
  37. return
  38. }
  39. if groupName == "" {
  40. groupName = "Buddies" // Default group
  41. }
  42. // Add buddy to feedbag
  43. resultCode, buddyInfo := h.addBuddyToFeedbag(ctx, session, buddyName, groupName)
  44. // Prepare response
  45. responseData := &ResultCodeData{ResultCode: resultCode}
  46. if resultCode == "success" {
  47. responseData.BuddyInfo = buddyInfo
  48. }
  49. resp := BaseResponse{}
  50. resp.Response.StatusCode = 200
  51. resp.Response.StatusText = "OK"
  52. resp.Response.Data = responseData
  53. SendResponse(w, r, resp, h.Logger)
  54. if resultCode == "success" {
  55. groups, err := h.BuddyListManager.GetBuddyListForUser(ctx, session)
  56. if err != nil {
  57. h.Logger.ErrorContext(ctx, "failed to get buddy list for event", "err", err.Error())
  58. } else {
  59. blPayload := &BuddyListData{Groups: groups}
  60. session.EventQueue.Push(types.EventTypeBuddyList, blPayload)
  61. }
  62. }
  63. h.Logger.InfoContext(ctx, "buddy added",
  64. "aimsid", aimsid,
  65. "buddy", buddyName,
  66. "group", groupName,
  67. "result", resultCode,
  68. )
  69. }
  70. // AddGroup handles GET /buddylist/addGroup requests.
  71. func (h *BuddyListHandler) AddGroup(w http.ResponseWriter, r *http.Request, session *state.WebAPISession) {
  72. ctx := r.Context()
  73. aimsid := r.URL.Query().Get("aimsid")
  74. groupName := strings.TrimSpace(r.URL.Query().Get("group"))
  75. if groupName == "" {
  76. h.sendError(w, r, http.StatusBadRequest, "missing group parameter")
  77. return
  78. }
  79. resultCode := h.addGroupToFeedbag(ctx, session, groupName)
  80. resp := BaseResponse{}
  81. resp.Response.StatusCode = 200
  82. resp.Response.StatusText = "OK"
  83. resp.Response.Data = &ResultCodeData{ResultCode: resultCode}
  84. SendResponse(w, r, resp, h.Logger)
  85. if resultCode == "success" {
  86. groups, err := h.BuddyListManager.GetBuddyListForUser(ctx, session)
  87. if err != nil {
  88. h.Logger.ErrorContext(ctx, "failed to get buddy list for event", "err", err.Error())
  89. } else {
  90. blPayload := &BuddyListData{Groups: groups}
  91. session.EventQueue.Push(types.EventTypeBuddyList, blPayload)
  92. }
  93. }
  94. h.Logger.InfoContext(ctx, "buddy list group added",
  95. "aimsid", aimsid,
  96. "group", groupName,
  97. "result", resultCode,
  98. )
  99. }
  100. func (h *BuddyListHandler) addGroupToFeedbag(ctx context.Context, sess *state.WebAPISession, groupName string) string {
  101. // A session sees no SNAC for its own feedbag writes, so it drops the alias
  102. // cache itself. See WebAPISession.InvalidateAliases.
  103. defer sess.InvalidateAliases()
  104. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagQuery}
  105. snac, err := h.FeedbagService.Query(ctx, sess.OSCARSession, frame)
  106. if err != nil {
  107. h.Logger.ErrorContext(ctx, "failed to retrieve feedbag", "err", err.Error())
  108. return "error"
  109. }
  110. reply, ok := snac.Body.(wire.SNAC_0x13_0x06_FeedbagReply)
  111. if !ok {
  112. return "error"
  113. }
  114. fl := state.NewFeedbagList(reply.Items, rand.Intn)
  115. fl.AddGroup(groupName)
  116. pending := fl.PendingUpdates()
  117. if len(pending) == 0 {
  118. return "alreadyExists"
  119. }
  120. insertFrame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem}
  121. if _, err := h.FeedbagService.UpsertItem(ctx, sess.OSCARSession, insertFrame, pending); err != nil {
  122. h.Logger.ErrorContext(ctx, "failed to add group", "err", err.Error())
  123. return "error"
  124. }
  125. return "success"
  126. }
  127. // feedbagGroupMatchesRequested returns true if a feedbag group row matches the
  128. // group the Web client asked for. OSCAR often stores the default group with an
  129. // empty name; GetBuddyListForUser labels that as "Buddies", so addBuddy must
  130. // treat "" and "Buddies" as the same bucket when the client sends group=Buddies.
  131. func feedbagGroupMatchesRequested(storedName, requested string) bool {
  132. req := strings.TrimSpace(requested)
  133. st := strings.TrimSpace(storedName)
  134. if strings.EqualFold(st, req) {
  135. return true
  136. }
  137. if strings.EqualFold(strings.TrimSpace(req), "Buddies") && st == "" {
  138. return true
  139. }
  140. return false
  141. }
  142. // storedGroupNameForRequest returns the feedbag group row Name for a Web client group label.
  143. // Rows with GroupID 0 are the root order record, not a named buddy group.
  144. func storedGroupNameForRequest(items []wire.FeedbagItem, requested string) (string, bool) {
  145. for _, item := range items {
  146. if item.ClassID != wire.FeedbagClassIdGroup {
  147. continue
  148. }
  149. if item.GroupID == 0 {
  150. continue
  151. }
  152. if feedbagGroupMatchesRequested(item.Name, requested) {
  153. return item.Name, true
  154. }
  155. }
  156. return "", false
  157. }
  158. // RemoveBuddy handles GET /buddylist/removeBuddy requests.
  159. func (h *BuddyListHandler) RemoveBuddy(w http.ResponseWriter, r *http.Request, session *state.WebAPISession) {
  160. ctx := r.Context()
  161. aimsid := r.URL.Query().Get("aimsid")
  162. buddyName := strings.TrimSpace(r.URL.Query().Get("buddy"))
  163. groupName := strings.TrimSpace(r.URL.Query().Get("group"))
  164. allGroupsParam := r.URL.Query().Get("allGroups")
  165. allGroups := allGroupsParam == "true" || allGroupsParam == "1"
  166. if buddyName == "" {
  167. h.sendError(w, r, http.StatusBadRequest, "missing buddy parameter")
  168. return
  169. }
  170. resultCode, rmErr := h.BuddyListManager.RemoveBuddyFromFeedbag(ctx, session, buddyName, groupName, allGroups)
  171. if rmErr != nil {
  172. h.Logger.ErrorContext(ctx, "remove buddy failed", "err", rmErr.Error())
  173. }
  174. resp := BaseResponse{}
  175. resp.Response.StatusCode = 200
  176. resp.Response.StatusText = "OK"
  177. resp.Response.Data = &ResultCodeData{ResultCode: resultCode}
  178. SendResponse(w, r, resp, h.Logger)
  179. if resultCode == "success" {
  180. groups, err := h.BuddyListManager.GetBuddyListForUser(ctx, session)
  181. if err != nil {
  182. h.Logger.ErrorContext(ctx, "failed to get buddy list for event", "err", err.Error())
  183. } else {
  184. blPayload := &BuddyListData{Groups: groups}
  185. session.EventQueue.Push(types.EventTypeBuddyList, blPayload)
  186. }
  187. }
  188. h.Logger.InfoContext(ctx, "buddy removed",
  189. "aimsid", aimsid,
  190. "buddy", buddyName,
  191. "group", groupName,
  192. "result", resultCode,
  193. )
  194. }
  195. // todo don't remove empty group?
  196. // RemoveGroup handles GET /buddylist/removeGroup requests.
  197. func (h *BuddyListHandler) RemoveGroup(w http.ResponseWriter, r *http.Request, session *state.WebAPISession) {
  198. ctx := r.Context()
  199. aimsid := r.URL.Query().Get("aimsid")
  200. groupName := strings.TrimSpace(r.URL.Query().Get("group"))
  201. if groupName == "" {
  202. h.sendError(w, r, http.StatusBadRequest, "missing group parameter")
  203. return
  204. }
  205. resultCode, rmErr := h.BuddyListManager.RemoveGroupFromFeedbag(ctx, session, groupName)
  206. if rmErr != nil {
  207. h.Logger.ErrorContext(ctx, "remove group failed", "err", rmErr.Error())
  208. }
  209. resp := BaseResponse{}
  210. resp.Response.StatusCode = 200
  211. resp.Response.StatusText = "OK"
  212. resp.Response.Data = &ResultCodeData{ResultCode: resultCode}
  213. SendResponse(w, r, resp, h.Logger)
  214. if resultCode == "success" {
  215. groups, err := h.BuddyListManager.GetBuddyListForUser(ctx, session)
  216. if err != nil {
  217. h.Logger.ErrorContext(ctx, "failed to get buddy list for event", "err", err.Error())
  218. } else {
  219. blPayload := &BuddyListData{Groups: groups}
  220. session.EventQueue.Push(types.EventTypeBuddyList, blPayload)
  221. }
  222. }
  223. h.Logger.InfoContext(ctx, "buddy list group removed",
  224. "aimsid", aimsid,
  225. "group", groupName,
  226. "result", resultCode,
  227. )
  228. }
  229. // addBuddyToFeedbag adds a buddy to the user's feedbag.
  230. func (h *BuddyListHandler) addBuddyToFeedbag(ctx context.Context, sess *state.WebAPISession, buddyName, groupName string) (string, *BuddyPresenceInfo) {
  231. defer sess.InvalidateAliases()
  232. // Retrieve current feedbag
  233. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagQuery}
  234. snac, err := h.FeedbagService.Query(ctx, sess.OSCARSession, frame)
  235. if err != nil {
  236. h.Logger.ErrorContext(ctx, "failed to retrieve feedbag", "err", err.Error())
  237. return "error", nil
  238. }
  239. reply, ok := snac.Body.(wire.SNAC_0x13_0x06_FeedbagReply)
  240. if !ok {
  241. // todo what
  242. return "error", nil
  243. }
  244. fl := state.NewFeedbagList(reply.Items, rand.Intn)
  245. fl.AddGroup(groupName)
  246. if pending := fl.PendingUpdates(); len(pending) > 0 {
  247. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem}
  248. if _, err := h.FeedbagService.UpsertItem(ctx, sess.OSCARSession, frame, pending); err != nil {
  249. h.Logger.ErrorContext(ctx, "failed to add buddy", "err", err.Error())
  250. return "error", nil
  251. }
  252. }
  253. added, err := fl.AddBuddy(groupName, buddyName, "", "")
  254. if err != nil {
  255. h.Logger.ErrorContext(ctx, "failed to add buddy to feedbag", "err", err.Error())
  256. return "error", nil
  257. }
  258. if !added {
  259. return "alreadyExists", nil
  260. }
  261. if pending := fl.PendingUpdates(); len(pending) > 0 {
  262. buddyItems := make(map[uint16][]wire.FeedbagItem)
  263. for _, item := range pending {
  264. if item.ClassID == wire.FeedbagClassIdBuddy {
  265. if _, ok := buddyItems[item.GroupID]; !ok {
  266. buddyItems[item.GroupID] = nil
  267. }
  268. buddyItems[item.GroupID] = append(buddyItems[item.GroupID], item)
  269. }
  270. }
  271. for _, buddies := range buddyItems {
  272. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem}
  273. if _, err := h.FeedbagService.UpsertItem(ctx, sess.OSCARSession, frame, buddies); err != nil {
  274. h.Logger.ErrorContext(ctx, "failed to add buddy", "err", err.Error())
  275. return "error", nil
  276. }
  277. }
  278. for _, item := range pending { // todo why not filter buddies out of pending?
  279. if item.ClassID == wire.FeedbagClassIdGroup {
  280. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem}
  281. if _, err := h.FeedbagService.UpsertItem(ctx, sess.OSCARSession, frame, []wire.FeedbagItem{item}); err != nil {
  282. h.Logger.ErrorContext(ctx, "failed to add buddy", "err", err.Error())
  283. return "error", nil
  284. }
  285. }
  286. }
  287. }
  288. // Get current presence for the buddy
  289. buddyInfo := &BuddyPresenceInfo{
  290. AimID: state.NewIdentScreenName(buddyName).String(),
  291. DisplayID: buddyName,
  292. State: "offline", // Default to offline
  293. UserType: "aim",
  294. }
  295. // TODO: Check actual presence status and update buddyInfo accordingly
  296. return "success", buddyInfo
  297. }
  298. // AddTempBuddy handles GET /aim/addTempBuddy requests.
  299. // This adds temporary buddies to the session without persisting them to the feedbag.
  300. // The temporary buddies are only visible for the duration of the session.
  301. func (h *BuddyListHandler) AddTempBuddy(w http.ResponseWriter, r *http.Request, session *state.WebAPISession) {
  302. ctx := r.Context()
  303. aimsid := r.URL.Query().Get("aimsid")
  304. buddyNames := r.URL.Query()["t"]
  305. if len(buddyNames) == 0 {
  306. h.sendError(w, r, http.StatusBadRequest, "missing buddy names (t parameter)")
  307. return
  308. }
  309. // Store temporary buddies in the session
  310. // Note: These are not persisted to the feedbag database
  311. if session.TempBuddies == nil {
  312. session.TempBuddies = make(map[string]bool)
  313. }
  314. for _, buddyName := range buddyNames {
  315. buddyName = strings.TrimSpace(buddyName)
  316. if buddyName != "" {
  317. session.TempBuddies[buddyName] = true
  318. }
  319. }
  320. // Prepare response
  321. responseData := &ResultCodeData{ResultCode: "success", BuddyNames: buddyNames}
  322. resp := BaseResponse{}
  323. resp.Response.StatusCode = 200
  324. resp.Response.StatusText = "OK"
  325. resp.Response.Data = responseData
  326. SendResponse(w, r, resp, h.Logger)
  327. // Do not push buddylist events for temp buddies. The Web AIM client handles
  328. // addTempBuddy via the API response; a buddylist event without "groups" causes
  329. // the client to clear the entire contact list (zC always calls clear() first).
  330. h.Logger.InfoContext(ctx, "temporary buddies added",
  331. "aimsid", aimsid,
  332. "buddies", buddyNames,
  333. "count", len(buddyNames),
  334. )
  335. }
  336. // RemoveTempBuddy handles GET /aim/removeTempBuddy requests.
  337. // This removes temporary session buddies added via addTempBuddy.
  338. func (h *BuddyListHandler) RemoveTempBuddy(w http.ResponseWriter, r *http.Request, session *state.WebAPISession) {
  339. ctx := r.Context()
  340. aimsid := r.URL.Query().Get("aimsid")
  341. buddyNames := r.URL.Query()["t"]
  342. if len(buddyNames) == 0 {
  343. h.sendError(w, r, http.StatusBadRequest, "missing buddy names (t parameter)")
  344. return
  345. }
  346. removed := make([]string, 0, len(buddyNames))
  347. for _, buddyName := range buddyNames {
  348. buddyName = strings.TrimSpace(buddyName)
  349. if buddyName == "" {
  350. continue
  351. }
  352. if session.TempBuddies != nil {
  353. delete(session.TempBuddies, buddyName)
  354. }
  355. removed = append(removed, buddyName)
  356. }
  357. resp := BaseResponse{}
  358. resp.Response.StatusCode = 200
  359. resp.Response.StatusText = "OK"
  360. resp.Response.Data = &ResultCodeData{ResultCode: "success", BuddyNames: removed}
  361. SendResponse(w, r, resp, h.Logger)
  362. h.Logger.InfoContext(ctx, "temporary buddies removed",
  363. "aimsid", aimsid,
  364. "buddies", removed,
  365. "count", len(removed),
  366. )
  367. }
  368. // RenameGroup handles GET /buddylist/renameGroup requests.
  369. //
  370. // The Web AIM client calls this with oldGroup (current group name) and newGroup
  371. // (the requested new name). This is a stub; it does not yet rename the group in
  372. // the feedbag.
  373. func (h *BuddyListHandler) RenameGroup(w http.ResponseWriter, r *http.Request, session *state.WebAPISession) {
  374. ctx := r.Context()
  375. aimsid := r.URL.Query().Get("aimsid")
  376. oldGroup := strings.TrimSpace(r.URL.Query().Get("oldGroup"))
  377. newGroup := strings.TrimSpace(r.URL.Query().Get("newGroup"))
  378. if oldGroup == "" || newGroup == "" {
  379. h.sendError(w, r, http.StatusBadRequest, "missing oldGroup or newGroup parameter")
  380. return
  381. }
  382. resultCode, rnErr := h.BuddyListManager.RenameGroupInFeedbag(ctx, session, oldGroup, newGroup)
  383. if rnErr != nil {
  384. h.Logger.ErrorContext(ctx, "rename group failed", "err", rnErr.Error())
  385. }
  386. resp := BaseResponse{}
  387. resp.Response.StatusCode = 200
  388. resp.Response.StatusText = "OK"
  389. resp.Response.Data = &ResultCodeData{ResultCode: resultCode}
  390. SendResponse(w, r, resp, h.Logger)
  391. if resultCode == "success" {
  392. h.pushBuddyListEvent(ctx, session)
  393. }
  394. h.Logger.InfoContext(ctx, "buddy list group renamed",
  395. "aimsid", aimsid,
  396. "oldGroup", oldGroup,
  397. "newGroup", newGroup,
  398. "result", resultCode,
  399. )
  400. }
  401. // MoveBuddy handles GET /buddylist/moveBuddy requests.
  402. //
  403. // The Web AIM client calls this with buddy (the buddy to move), group (the
  404. // current group), and optionally newGroup (destination group) and beforeBuddy
  405. // (buddy to position it before). This is a stub; it does not yet move the buddy
  406. // in the feedbag.
  407. func (h *BuddyListHandler) MoveBuddy(w http.ResponseWriter, r *http.Request, session *state.WebAPISession) {
  408. ctx := r.Context()
  409. aimsid := r.URL.Query().Get("aimsid")
  410. buddyName := strings.TrimSpace(r.URL.Query().Get("buddy"))
  411. groupName := strings.TrimSpace(r.URL.Query().Get("group"))
  412. newGroup := strings.TrimSpace(r.URL.Query().Get("newGroup"))
  413. beforeBuddy := strings.TrimSpace(r.URL.Query().Get("beforeBuddy"))
  414. if buddyName == "" {
  415. h.sendError(w, r, http.StatusBadRequest, "missing buddy parameter")
  416. return
  417. }
  418. if groupName == "" {
  419. h.sendError(w, r, http.StatusBadRequest, "missing group parameter")
  420. return
  421. }
  422. resultCode, mvErr := h.BuddyListManager.MoveBuddyInFeedbag(ctx, session, buddyName, groupName, newGroup, beforeBuddy)
  423. if mvErr != nil {
  424. h.Logger.ErrorContext(ctx, "move buddy failed", "err", mvErr.Error())
  425. }
  426. resp := BaseResponse{}
  427. resp.Response.StatusCode = 200
  428. resp.Response.StatusText = "OK"
  429. resp.Response.Data = &ResultCodeData{ResultCode: resultCode}
  430. SendResponse(w, r, resp, h.Logger)
  431. if resultCode == "success" {
  432. h.pushBuddyListEvent(ctx, session)
  433. }
  434. h.Logger.InfoContext(ctx, "buddy moved",
  435. "aimsid", aimsid,
  436. "buddy", buddyName,
  437. "group", groupName,
  438. "newGroup", newGroup,
  439. "beforeBuddy", beforeBuddy,
  440. "result", resultCode,
  441. )
  442. }
  443. // SetBuddyAttribute handles GET /buddylist/setBuddyAttribute requests.
  444. //
  445. // The Web AIM client calls this with t (the buddy) and friendly (the display
  446. // name / alias). This is a stub; it does not yet persist the attribute to the
  447. // feedbag.
  448. func (h *BuddyListHandler) SetBuddyAttribute(w http.ResponseWriter, r *http.Request, session *state.WebAPISession) {
  449. ctx := r.Context()
  450. aimsid := r.URL.Query().Get("aimsid")
  451. buddyName := strings.TrimSpace(r.URL.Query().Get("t"))
  452. friendly := strings.TrimSpace(r.URL.Query().Get("friendly"))
  453. if buddyName == "" {
  454. h.sendError(w, r, http.StatusBadRequest, "missing t parameter")
  455. return
  456. }
  457. resultCode, saErr := h.BuddyListManager.SetBuddyAttributeInFeedbag(ctx, session, buddyName, friendly)
  458. if saErr != nil {
  459. h.Logger.ErrorContext(ctx, "set buddy attribute failed", "err", saErr.Error())
  460. }
  461. resp := BaseResponse{}
  462. resp.Response.StatusCode = 200
  463. resp.Response.StatusText = "OK"
  464. resp.Response.Data = &ResultCodeData{ResultCode: resultCode}
  465. SendResponse(w, r, resp, h.Logger)
  466. if resultCode == "success" {
  467. h.pushBuddyListEvent(ctx, session)
  468. }
  469. h.Logger.InfoContext(ctx, "buddy attribute set",
  470. "aimsid", aimsid,
  471. "buddy", buddyName,
  472. "friendly", friendly,
  473. "result", resultCode,
  474. )
  475. }
  476. // SetGroupAttribute handles GET /buddylist/setGroupAttribute requests.
  477. //
  478. // The Web AIM client calls this with collapsed (the group's collapsed state)
  479. // and, for named groups, group. The unnamed default group omits group. This is
  480. // a stub; it does not yet persist the attribute to the feedbag.
  481. func (h *BuddyListHandler) SetGroupAttribute(w http.ResponseWriter, r *http.Request, session *state.WebAPISession) {
  482. ctx := r.Context()
  483. aimsid := r.URL.Query().Get("aimsid")
  484. query := r.URL.Query()
  485. groupName := strings.TrimSpace(query.Get("group"))
  486. collapsedParam := strings.TrimSpace(query.Get("collapsed"))
  487. if collapsedParam == "" {
  488. h.sendError(w, r, http.StatusBadRequest, "missing collapsed parameter")
  489. return
  490. }
  491. collapsed := collapsedParam == "true" || collapsedParam == "1"
  492. resultCode, saErr := h.BuddyListManager.SetGroupAttributeInFeedbag(ctx, session, groupName, collapsed)
  493. if saErr != nil {
  494. h.Logger.ErrorContext(ctx, "set group attribute failed", "err", saErr.Error())
  495. }
  496. resp := BaseResponse{}
  497. resp.Response.StatusCode = 200
  498. resp.Response.StatusText = "OK"
  499. resp.Response.Data = &ResultCodeData{ResultCode: resultCode}
  500. SendResponse(w, r, resp, h.Logger)
  501. if resultCode == "success" {
  502. h.pushBuddyListEvent(ctx, session)
  503. }
  504. h.Logger.InfoContext(ctx, "buddy list group attribute set",
  505. "aimsid", aimsid,
  506. "group", groupName,
  507. "collapsed", collapsed,
  508. "result", resultCode,
  509. )
  510. }
  511. // pushBuddyListEvent refreshes the buddy list and pushes it to the session's
  512. // event queue so the Web client re-renders after a mutation.
  513. func (h *BuddyListHandler) pushBuddyListEvent(ctx context.Context, session *state.WebAPISession) {
  514. groups, err := h.BuddyListManager.GetBuddyListForUser(ctx, session)
  515. if err != nil {
  516. h.Logger.ErrorContext(ctx, "failed to get buddy list for event", "err", err.Error())
  517. return
  518. }
  519. session.EventQueue.Push(types.EventTypeBuddyList, &BuddyListData{Groups: groups})
  520. }
  521. // ResultCodeData is the payload the buddy list editing methods answer with.
  522. //
  523. // The spec shows these methods returning an empty data; the Web AIM client
  524. // reads resultCode from it, so the server sends one.
  525. type ResultCodeData struct {
  526. ResultCode string `json:"resultCode" xml:"resultCode"`
  527. // BuddyInfo accompanies a successful addBuddy only.
  528. BuddyInfo *BuddyPresenceInfo `json:"buddyInfo,omitempty" xml:"buddyInfo,omitempty"`
  529. // BuddyNames accompanies the temp-buddy methods only.
  530. BuddyNames []string `json:"buddyNames,omitempty" xml:"buddyNames>buddyName,omitempty"`
  531. }
  532. // sendError is a convenience method that wraps the common SendError function.
  533. func (h *BuddyListHandler) sendError(w http.ResponseWriter, r *http.Request, statusCode int, message string) {
  534. SendError(w, r, statusCode, message)
  535. }