session.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700
  1. package handlers
  2. import (
  3. "context"
  4. "encoding/base64"
  5. "encoding/xml"
  6. "fmt"
  7. "log/slog"
  8. "net/http"
  9. "slices"
  10. "strconv"
  11. "strings"
  12. "time"
  13. "github.com/google/uuid"
  14. "github.com/mk6i/open-oscar-server/server/webapi/middleware"
  15. "github.com/mk6i/open-oscar-server/server/webapi/types"
  16. "github.com/mk6i/open-oscar-server/state"
  17. "github.com/mk6i/open-oscar-server/wire"
  18. )
  19. // SessionHandler handles Web AIM API session management endpoints.
  20. type SessionHandler struct {
  21. SessionManager *state.WebAPISessionManager
  22. OSCARSessionManager SessionManager
  23. OSCARAuthService AuthService
  24. BuddyListRegistry BuddyListRegistry
  25. BuddyBroadcaster BuddyBroadcaster
  26. FeedbagService FeedbagService
  27. BuddyListManager *BuddyListManager
  28. Logger *slog.Logger
  29. OServiceService OServiceService
  30. RecalcWarning func(ctx context.Context, instance *state.SessionInstance) error
  31. LowerWarnLevel func(ctx context.Context, instance *state.SessionInstance)
  32. ChatSessionManager ChatSessionManager
  33. }
  34. // AuthService defines methods needed for authentication.
  35. type AuthService interface {
  36. BUCPChallenge(ctx context.Context, bodyIn wire.SNAC_0x17_0x06_BUCPChallengeRequest, newUUID func() uuid.UUID) (wire.SNACMessage, error)
  37. BUCPLogin(ctx context.Context, bodyIn wire.SNAC_0x17_0x02_BUCPLoginRequest, advertisedHost string) (wire.SNACMessage, error)
  38. CrackCookie(authCookie []byte) (state.ServerCookie, error)
  39. RegisterBOSSession(ctx context.Context, authCookie state.ServerCookie, conf func(sess *state.Session)) (*state.SessionInstance, error)
  40. FLAPLogin(ctx context.Context, inFrame wire.FLAPSignonFrame, advertisedHost string) (wire.TLVRestBlock, error)
  41. Signout(ctx context.Context, session *state.Session)
  42. SignoutChat(ctx context.Context, sess *state.Session)
  43. }
  44. // SessionManager defines methods for OSCAR session management.
  45. type SessionManager interface {
  46. AddSession(ctx context.Context, screenName state.DisplayScreenName, doMultiSess bool, cfg ...func(sess *state.Session)) (*state.SessionInstance, error)
  47. RemoveSession(session *state.Session)
  48. RelayToScreenName(ctx context.Context, screenName state.IdentScreenName, msg wire.SNACMessage)
  49. }
  50. // BuddyListRegistry defines methods for buddy list management.
  51. type BuddyListRegistry interface {
  52. RegisterBuddyList(ctx context.Context, screenName state.IdentScreenName) error
  53. UnregisterBuddyList(ctx context.Context, screenName state.IdentScreenName) error
  54. }
  55. type ChatSessionManager interface {
  56. RemoveUserFromAllChats(user state.IdentScreenName)
  57. }
  58. // BuddyGroup represents a group of buddies.
  59. type BuddyGroup struct {
  60. Name string `json:"name"`
  61. Buddies []Buddy `json:"buddies"`
  62. }
  63. // Buddy represents a buddy in the buddy list.
  64. type Buddy struct {
  65. AimID string `json:"aimId"`
  66. State string `json:"state"`
  67. StatusMsg string `json:"statusMsg,omitempty"`
  68. AwayMsg string `json:"awayMsg,omitempty"`
  69. UserType string `json:"userType"`
  70. }
  71. // StartSessionResponse represents the response for startSession endpoint.
  72. type StartSessionResponse struct {
  73. Response struct {
  74. StatusCode int `json:"statusCode"`
  75. StatusText string `json:"statusText"`
  76. Data struct {
  77. AimSID string `json:"aimsid"`
  78. Ts int64 `json:"ts"`
  79. FetchTimeout int `json:"fetchTimeout"`
  80. TimeToNextFetch int `json:"timeToNextFetch"`
  81. FetchBaseURL string `json:"fetchBaseURL"` // Gromit expects this directly in data!
  82. MyInfo map[string]interface{} `json:"myInfo,omitempty"`
  83. Events map[string]interface{} `json:"events,omitempty"`
  84. WellKnownUrls map[string]string `json:"wellKnownUrls,omitempty"`
  85. } `json:"data"`
  86. } `json:"response"`
  87. }
  88. // StartSessionXMLResponse represents the XML response for startSession endpoint.
  89. type StartSessionXMLResponse struct {
  90. XMLName xml.Name `xml:"response"`
  91. StatusCode int `xml:"statusCode"`
  92. StatusText string `xml:"statusText"`
  93. Data struct {
  94. AimSID string `xml:"aimsid"`
  95. FetchTimeout int `xml:"fetchTimeout"`
  96. TimeToNextFetch int `xml:"timeToNextFetch"`
  97. FetchBaseURL string `xml:"fetchBaseURL"` // Gromit expects this directly!
  98. WellKnownUrls *struct {
  99. WebApiBase string `xml:"webApiBase"`
  100. FetchBaseURL string `xml:"fetchBaseURL"`
  101. LifestreamApiBase string `xml:"lifestreamApiBase"`
  102. } `xml:"wellKnownUrls,omitempty"`
  103. MyInfo *struct {
  104. AimID string `xml:"aimId"`
  105. DisplayID string `xml:"displayId"`
  106. Buddylist struct {
  107. Groups *[]BuddyGroup `xml:"group,omitempty"`
  108. } `xml:"buddylist,omitempty"`
  109. } `xml:"myInfo,omitempty"`
  110. Events *struct {
  111. BuddyList struct {
  112. Groups *[]BuddyGroup `xml:"group,omitempty"`
  113. } `xml:"buddylist"`
  114. } `xml:"events,omitempty"`
  115. } `xml:"data"`
  116. }
  117. // EndSessionResponse represents the response for endSession endpoint.
  118. type EndSessionResponse struct {
  119. Response struct {
  120. StatusCode int `json:"statusCode"`
  121. StatusText string `json:"statusText"`
  122. } `json:"response"`
  123. }
  124. // StartSession handles GET /aim/startSession requests.
  125. func (h *SessionHandler) StartSession(w http.ResponseWriter, r *http.Request) {
  126. ctx := r.Context()
  127. // Get API key info from context (set by auth middleware)
  128. apiKey, ok := ctx.Value(middleware.ContextKeyAPIKey).(*state.WebAPIKey)
  129. if !ok {
  130. h.sendError(w, r, http.StatusInternalServerError, "internal server error")
  131. return
  132. }
  133. // Parse parameters
  134. params := r.URL.Query()
  135. // Get authentication token if provided
  136. authToken := params.Get("a")
  137. // Get client info
  138. clientName := params.Get("clientName")
  139. if clientName == "" {
  140. clientName = "WebAIM"
  141. }
  142. clientVersion := params.Get("clientVersion")
  143. if clientVersion == "" {
  144. clientVersion = "1.0"
  145. }
  146. // Get events to subscribe to
  147. eventsParam := params.Get("events")
  148. var events []string
  149. if eventsParam != "" {
  150. events = strings.Split(eventsParam, ",")
  151. h.Logger.DebugContext(ctx, "parsing events from request",
  152. "eventsParam", eventsParam,
  153. "parsedEvents", events,
  154. )
  155. } else {
  156. // Default events if none specified
  157. events = []string{"buddylist", "presence", "im", "sentIM"}
  158. h.Logger.DebugContext(ctx, "using default events",
  159. "events", events,
  160. )
  161. }
  162. // Get timeout settings
  163. timeout := 60000 // Default 60 seconds for better stability with Gromit
  164. if t := params.Get("timeout"); t != "" {
  165. if val, err := strconv.Atoi(t); err == nil && val > 0 {
  166. timeout = val * 1000 // Convert to milliseconds
  167. }
  168. }
  169. // Determine screen name from auth token or anonymous
  170. var screenName state.DisplayScreenName
  171. var cookie state.ServerCookie
  172. if authToken != "" {
  173. rawCookie, err := base64.URLEncoding.DecodeString(strings.TrimSpace(authToken))
  174. if err != nil {
  175. h.Logger.Warn("invalid authentication token (base64)", "error", err)
  176. h.sendError(w, r, http.StatusUnauthorized, "invalid or expired token")
  177. return
  178. }
  179. cookie, err = h.OSCARAuthService.CrackCookie(rawCookie)
  180. if err != nil {
  181. h.Logger.Warn("invalid authentication token",
  182. "error", err)
  183. h.sendError(w, r, http.StatusUnauthorized, "invalid or expired token")
  184. return
  185. }
  186. screenName = cookie.ScreenName
  187. tokenPreview := authToken
  188. if len(tokenPreview) > 8 {
  189. tokenPreview = tokenPreview[:8] + "..."
  190. }
  191. h.Logger.Info("authenticated session requested",
  192. "token", tokenPreview,
  193. "screenName", screenName)
  194. } else {
  195. // Anonymous session - generate guest name
  196. screenName = state.DisplayScreenName("Guest_" + strconv.FormatInt(time.Now().Unix(), 36))
  197. h.Logger.Info("anonymous session requested",
  198. "screenName", screenName)
  199. }
  200. // Create OSCAR session for authenticated users
  201. var oscarInstance *state.SessionInstance
  202. var err error
  203. if authToken != "" && h.OSCARSessionManager != nil {
  204. fnCfg := func(sess *state.Session) {
  205. sess.OnSessionClose(func() {
  206. if !shuttingDown(ctx) {
  207. if err := h.BuddyBroadcaster.BroadcastBuddyDeparted(ctx, sess.IdentScreenName()); err != nil {
  208. h.Logger.ErrorContext(ctx, "error sending buddy departure notifications", "err", err.Error())
  209. }
  210. }
  211. ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
  212. defer cancel()
  213. // buddy list must be cleared before session is closed, otherwise
  214. // there will be a race condition that could cause the buddy list
  215. // be prematurely deleted.
  216. if err := h.BuddyListRegistry.UnregisterBuddyList(ctx, sess.IdentScreenName()); err != nil {
  217. h.Logger.ErrorContext(ctx, "error removing buddy list entry", "err", err.Error())
  218. }
  219. h.ChatSessionManager.RemoveUserFromAllChats(sess.IdentScreenName())
  220. h.OSCARAuthService.Signout(ctx, sess)
  221. })
  222. }
  223. // Create OSCAR session
  224. oscarInstance, err = h.OSCARAuthService.RegisterBOSSession(ctx, cookie, fnCfg)
  225. if err != nil {
  226. // A failed BOS registration (e.g. the per-user session cap) leaves no
  227. // OSCAR session. Downstream steps (buddy list, feedbag, presence) all
  228. // dereference it, so fail the request instead of continuing half-set-up.
  229. h.Logger.ErrorContext(ctx, "failed to create OSCAR session", "err", err.Error())
  230. h.sendError(w, r, http.StatusServiceUnavailable, "unable to establish session")
  231. return
  232. }
  233. if err = oscarInstance.Session().RunOnce(func() error {
  234. // make buddy list visible to other users
  235. if err := h.BuddyListRegistry.RegisterBuddyList(ctx, oscarInstance.IdentScreenName()); err != nil {
  236. return fmt.Errorf("unable to init buddy list: %w", err)
  237. }
  238. // restore warning level from last session
  239. if err := h.RecalcWarning(ctx, oscarInstance); err != nil {
  240. return fmt.Errorf("failed to recalculate warning level: %w", err)
  241. }
  242. // periodically decay warning level
  243. go h.LowerWarnLevel(ctx, oscarInstance)
  244. return nil
  245. }); err != nil {
  246. h.Logger.ErrorContext(ctx, "failed to init session", "err", err.Error())
  247. h.sendError(w, r, http.StatusInternalServerError, "internal server error")
  248. return
  249. }
  250. // Update user visibility when an instance closes, as the user's overall status may change.
  251. // Example: With 1 away and 1 non-away instance, the user appears available. If the non-away
  252. // instance closes, the user should appear away.
  253. oscarInstance.OnClose(func() {
  254. if shuttingDown(ctx) {
  255. return
  256. }
  257. if oscarInstance.Session().Invisible() {
  258. if err := h.BuddyBroadcaster.BroadcastBuddyDeparted(ctx, oscarInstance.IdentScreenName()); err != nil {
  259. h.Logger.ErrorContext(ctx, "error sending buddy departure notifications", "err", err.Error())
  260. }
  261. } else {
  262. if err := h.BuddyBroadcaster.BroadcastBuddyArrived(ctx, oscarInstance.IdentScreenName(), oscarInstance.Session().TLVUserInfo()); err != nil {
  263. h.Logger.ErrorContext(ctx, "error sending buddy arrival notifications", "err", err.Error())
  264. }
  265. }
  266. })
  267. if err := h.FeedbagService.Use(ctx, oscarInstance); err != nil {
  268. h.Logger.ErrorContext(ctx, "failed to use feedbag", "err", err.Error())
  269. }
  270. // A web client signals that it wants typing events through its event
  271. // subscription, not through a stored feedbag buddy pref. Reflect that
  272. // on the OSCAR session so ICBMService attaches the WantEvents TLV to
  273. // outgoing IMs, prompting recipients to send typing notifications
  274. // back. This must run after FeedbagService.Use, which otherwise
  275. // overwrites the flag from stored prefs the web user may not have set.
  276. oscarInstance.Session().SetTypingEventsEnabled(slices.Contains(events, "typing"))
  277. oscarInstance.SetSignonComplete()
  278. if err := h.OServiceService.ClientOnline(ctx, wire.BOS, wire.SNAC_0x01_0x02_OServiceClientOnline{}, oscarInstance); err != nil {
  279. h.Logger.ErrorContext(ctx, "failed to set client online", "err", err.Error())
  280. h.sendError(w, r, http.StatusInternalServerError, "internal server error")
  281. return
  282. }
  283. }
  284. // Create WebAPI session
  285. session, err := h.SessionManager.CreateSession(r.Context(), screenName, apiKey.DevID, events, oscarInstance, h.Logger)
  286. if err != nil {
  287. h.Logger.ErrorContext(ctx, "failed to create session", "err", err.Error())
  288. h.sendError(w, r, http.StatusInternalServerError, "failed to create session")
  289. return
  290. }
  291. h.Logger.DebugContext(ctx, "session created with event subscriptions",
  292. "aimsid", session.AimSID,
  293. "events", events,
  294. )
  295. // Wire buddy list refresher so feedbag SNACs from the OSCAR bridge trigger a buddylist event.
  296. session.BuddyListRefresher = func(ctx context.Context) (interface{}, error) {
  297. return h.BuddyListManager.GetBuddyListForUser(ctx, session)
  298. }
  299. // Wire permit/deny refresher so FeedbagUpdateItem SNACs trigger a permitDeny event.
  300. session.PermitDenyRefresher = func(ctx context.Context) (interface{}, error) {
  301. frame := wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagQuery}
  302. fb, err := h.FeedbagService.Query(ctx, session.OSCARSession, frame)
  303. if err != nil {
  304. return nil, err
  305. }
  306. reply, ok := fb.Body.(wire.SNAC_0x13_0x06_FeedbagReply)
  307. if !ok {
  308. return nil, fmt.Errorf("unexpected feedbag reply type")
  309. }
  310. return permitDenyData(reply.Items), nil
  311. }
  312. // Store client info
  313. session.ClientName = clientName
  314. session.ClientVersion = clientVersion
  315. session.FetchTimeout = timeout
  316. session.RemoteAddr = r.RemoteAddr
  317. // Queue myInfo event for authenticated users
  318. if authToken != "" {
  319. for _, event := range events {
  320. if event == "myInfo" || event == "presence" {
  321. myInfoData := map[string]interface{}{
  322. "aimId": screenName.String(),
  323. "displayId": screenName.String(),
  324. "friendly": screenName.String(),
  325. "state": "online",
  326. "onlineTime": time.Now().Unix(),
  327. "memberSince": time.Now().Unix() - 86400*30, // 30 days ago
  328. "capabilities": []string{},
  329. "bot": false,
  330. "service": "AIM",
  331. }
  332. session.EventQueue.Push(types.EventType("myInfo"), myInfoData)
  333. break
  334. }
  335. }
  336. for _, event := range events {
  337. if event == "conversation" {
  338. session.EventQueue.Push(types.EventTypeConversation,
  339. types.ConversationEventData("list", nil))
  340. break
  341. }
  342. }
  343. }
  344. now := time.Now().Unix()
  345. scheme := requestScheme(r)
  346. baseURL := fmt.Sprintf("%s://%s", scheme, r.Host)
  347. // Prepare response
  348. resp := StartSessionResponse{}
  349. resp.Response.StatusCode = 200
  350. resp.Response.StatusText = "OK"
  351. resp.Response.Data.AimSID = session.AimSID
  352. resp.Response.Data.Ts = now
  353. resp.Response.Data.FetchTimeout = session.FetchTimeout
  354. resp.Response.Data.TimeToNextFetch = session.TimeToNextFetch
  355. // Gromit expects fetchBaseURL directly in data, not in wellKnownUrls
  356. resp.Response.Data.FetchBaseURL = fmt.Sprintf("%s/aim/fetchEvents?aimsid=%s&seqNum=0", baseURL, session.AimSID)
  357. // Add wellKnownUrls for other clients that might use it.
  358. resp.Response.Data.WellKnownUrls = map[string]string{
  359. "webApiBase": baseURL + "/",
  360. "fetchBaseURL": baseURL + "/aim/fetchEvents",
  361. "lifestreamApiBase": baseURL + "/",
  362. }
  363. if authToken != "" {
  364. myInfoPayload := map[string]interface{}{
  365. "aimId": screenName.String(),
  366. "displayId": screenName.String(),
  367. "friendly": screenName.String(),
  368. "state": "online",
  369. "onlineTime": time.Now().Unix(),
  370. "memberSince": time.Now().Unix() - 86400*30, // 30 days ago
  371. "capabilities": []string{},
  372. "bot": false,
  373. "service": "AIM",
  374. "self": map[string]interface{}{
  375. "instNum": 1,
  376. "loginTime": time.Now().Unix(),
  377. "sessionTimeout": 30,
  378. "events": events,
  379. "assertCaps": []string{},
  380. "rightsInfo": map[string]interface{}{
  381. "maxDenies": 500,
  382. "maxPermits": 500,
  383. "maxWatchers": 3000,
  384. "maxBuddies": 500,
  385. "maxTempBuddies": 160,
  386. "maxIMSize": 3987,
  387. "minInterIcbmInterval": 1000,
  388. "maxSourceEvil": 900,
  389. "maxDstEvil": 999,
  390. "maxSigLen": 4096,
  391. },
  392. },
  393. }
  394. resp.Response.Data.MyInfo = myInfoPayload
  395. if resp.Response.Data.Events == nil {
  396. resp.Response.Data.Events = make(map[string]interface{})
  397. }
  398. resp.Response.Data.Events["myInfo"] = myInfoPayload
  399. }
  400. for _, event := range events {
  401. switch types.EventType(event) {
  402. case types.EventTypeBuddyList:
  403. buddyGroups := []WebAPIBuddyGroup{}
  404. if authToken != "" && h.BuddyListManager != nil {
  405. var err error
  406. buddyGroups, err = h.BuddyListManager.GetBuddyListForUser(ctx, session)
  407. if err != nil {
  408. h.Logger.ErrorContext(ctx, "failed to get buddy list", "err", err.Error())
  409. buddyGroups = []WebAPIBuddyGroup{}
  410. }
  411. }
  412. if buddyGroups == nil {
  413. buddyGroups = []WebAPIBuddyGroup{}
  414. }
  415. blPayload := map[string]interface{}{"groups": buddyGroups}
  416. if resp.Response.Data.Events == nil {
  417. resp.Response.Data.Events = make(map[string]interface{})
  418. }
  419. resp.Response.Data.Events["buddylist"] = blPayload
  420. if authToken != "" {
  421. session.EventQueue.Push(types.EventTypeBuddyList, blPayload)
  422. }
  423. case types.EventTypePreference:
  424. prefPayload := map[string]interface{}{
  425. "showGroups": true,
  426. "showOfflineGrp": true,
  427. "sortBuddyList": false,
  428. "globalOTR": false,
  429. }
  430. if resp.Response.Data.Events == nil {
  431. resp.Response.Data.Events = make(map[string]interface{})
  432. }
  433. resp.Response.Data.Events["preference"] = prefPayload
  434. if authToken != "" {
  435. session.EventQueue.Push(types.EventTypePreference, prefPayload)
  436. }
  437. }
  438. }
  439. // Check response format
  440. format := r.URL.Query().Get("f")
  441. if format == "" {
  442. format = "json" // default to JSON
  443. }
  444. // Send response in requested format
  445. if format == "xml" {
  446. // Build XML response
  447. xmlResp := StartSessionXMLResponse{}
  448. xmlResp.StatusCode = 200
  449. xmlResp.StatusText = "OK"
  450. xmlResp.Data.AimSID = session.AimSID
  451. xmlResp.Data.FetchTimeout = timeout
  452. xmlResp.Data.TimeToNextFetch = 500
  453. // Gromit expects fetchBaseURL directly in data
  454. xmlResp.Data.FetchBaseURL = fmt.Sprintf("%s/aim/fetchEvents?aimsid=%s&seqNum=0", baseURL, session.AimSID)
  455. // Add wellKnownUrls for other clients
  456. xmlBase := baseURL + "/"
  457. xmlResp.Data.WellKnownUrls = &struct {
  458. WebApiBase string `xml:"webApiBase"`
  459. FetchBaseURL string `xml:"fetchBaseURL"`
  460. LifestreamApiBase string `xml:"lifestreamApiBase"`
  461. }{
  462. WebApiBase: xmlBase,
  463. FetchBaseURL: baseURL + "/aim/fetchEvents",
  464. LifestreamApiBase: xmlBase,
  465. }
  466. // Add myInfo with user data
  467. xmlResp.Data.MyInfo = &struct {
  468. AimID string `xml:"aimId"`
  469. DisplayID string `xml:"displayId"`
  470. Buddylist struct {
  471. Groups *[]BuddyGroup `xml:"group,omitempty"`
  472. } `xml:"buddylist,omitempty"`
  473. }{
  474. AimID: session.ScreenName.String(),
  475. DisplayID: session.ScreenName.String(),
  476. }
  477. // Add buddy list if requested in myInfo or events
  478. for _, event := range events {
  479. if event == "buddylist" || event == "myInfo" {
  480. var buddyGroups []BuddyGroup
  481. if authToken != "" && h.BuddyListManager != nil {
  482. // Fetch actual buddy list from service
  483. webAPIGroups, err := h.BuddyListManager.GetBuddyListForUser(ctx, session)
  484. if err != nil {
  485. h.Logger.ErrorContext(ctx, "failed to get buddy list for XML response", "err", err.Error())
  486. buddyGroups = []BuddyGroup{}
  487. } else {
  488. // Convert WebAPIBuddyGroup to handler.BuddyGroup
  489. for _, webGroup := range webAPIGroups {
  490. group := BuddyGroup{
  491. Name: webGroup.Name,
  492. Buddies: []Buddy{},
  493. }
  494. for _, webBuddy := range webGroup.Buddies {
  495. buddy := Buddy{
  496. AimID: webBuddy.AimID,
  497. State: webBuddy.State,
  498. StatusMsg: webBuddy.StatusMsg,
  499. AwayMsg: webBuddy.AwayMsg,
  500. UserType: webBuddy.UserType,
  501. }
  502. group.Buddies = append(group.Buddies, buddy)
  503. }
  504. buddyGroups = append(buddyGroups, group)
  505. }
  506. }
  507. } else {
  508. buddyGroups = []BuddyGroup{}
  509. }
  510. // Add to myInfo buddylist
  511. xmlResp.Data.MyInfo.Buddylist.Groups = &buddyGroups
  512. // Also add to events if specifically requested
  513. if event == "buddylist" {
  514. if xmlResp.Data.Events == nil {
  515. xmlResp.Data.Events = &struct {
  516. BuddyList struct {
  517. Groups *[]BuddyGroup `xml:"group,omitempty"`
  518. } `xml:"buddylist"`
  519. }{}
  520. }
  521. xmlResp.Data.Events.BuddyList.Groups = &buddyGroups
  522. }
  523. break
  524. }
  525. }
  526. // Send XML response
  527. w.Header().Set("Content-Type", "text/xml; charset=utf-8")
  528. // Build complete XML string first
  529. xmlData, err := xml.Marshal(xmlResp)
  530. if err != nil {
  531. h.Logger.Error("failed to marshal XML response", "error", err)
  532. h.sendError(w, r, http.StatusInternalServerError, "internal server error")
  533. return
  534. }
  535. // Write XML declaration and data as one response
  536. xmlOutput := fmt.Sprintf(`<?xml version="1.0" encoding="UTF-8"?>%s`, xmlData)
  537. w.Header().Set("Content-Length", strconv.Itoa(len(xmlOutput)))
  538. _, _ = fmt.Fprint(w, xmlOutput)
  539. } else {
  540. // Send response in requested format (JSON, JSONP, or AMF)
  541. SendResponse(w, r, resp, h.Logger)
  542. }
  543. h.Logger.DebugContext(ctx, "session started",
  544. "aimsid", session.AimSID,
  545. "screen_name", screenName,
  546. "dev_id", apiKey.DevID,
  547. "events", events,
  548. "format", format,
  549. )
  550. }
  551. // EndSession handles GET /aim/endSession requests.
  552. func (h *SessionHandler) EndSession(w http.ResponseWriter, r *http.Request) {
  553. ctx := r.Context()
  554. // Get session ID from parameters
  555. aimsid := r.URL.Query().Get("aimsid")
  556. if aimsid == "" {
  557. h.sendError(w, r, http.StatusBadRequest, "missing aimsid parameter")
  558. return
  559. }
  560. // Get session
  561. session, err := h.SessionManager.GetSession(r.Context(), aimsid)
  562. if err != nil {
  563. switch err {
  564. case state.ErrNoWebAPISession:
  565. h.sendError(w, r, http.StatusNotFound, "session not found")
  566. case state.ErrWebAPISessionExpired:
  567. h.sendError(w, r, http.StatusGone, "session expired")
  568. default:
  569. h.sendError(w, r, http.StatusInternalServerError, "internal server error")
  570. }
  571. return
  572. }
  573. // Clean up OSCAR session if present
  574. if session.OSCARSession != nil && h.OSCARSessionManager != nil {
  575. // Broadcast departure to OSCAR clients
  576. if h.BuddyBroadcaster != nil {
  577. if err := h.BuddyBroadcaster.BroadcastBuddyDeparted(ctx, session.OSCARSession.IdentScreenName()); err != nil {
  578. h.Logger.ErrorContext(ctx, "failed to broadcast buddy departure", "err", err.Error())
  579. }
  580. }
  581. // Unregister buddy list
  582. if h.BuddyListRegistry != nil {
  583. if err := h.BuddyListRegistry.UnregisterBuddyList(ctx, session.ScreenName.IdentScreenName()); err != nil {
  584. h.Logger.ErrorContext(ctx, "failed to unregister buddy list", "err", err.Error())
  585. }
  586. }
  587. // Remove OSCAR session
  588. h.OSCARSessionManager.RemoveSession(session.OSCARSession.Session())
  589. session.OSCARSession = nil
  590. }
  591. // Remove session
  592. if err := h.SessionManager.RemoveSession(r.Context(), aimsid); err != nil {
  593. h.Logger.ErrorContext(ctx, "failed to remove session", "err", err.Error())
  594. h.sendError(w, r, http.StatusInternalServerError, "failed to end session")
  595. return
  596. }
  597. // Send response
  598. resp := EndSessionResponse{}
  599. resp.Response.StatusCode = 200
  600. resp.Response.StatusText = "OK"
  601. // Send response in requested format (JSON, JSONP, or AMF)
  602. SendResponse(w, r, resp, h.Logger)
  603. h.Logger.DebugContext(ctx, "session ended",
  604. "aimsid", aimsid,
  605. "screen_name", session.ScreenName,
  606. )
  607. }
  608. // sendError sends a Web AIM API error envelope, honoring JSONP when requested.
  609. func (h *SessionHandler) sendError(w http.ResponseWriter, r *http.Request, statusCode int, message string) {
  610. resp := BaseResponse{}
  611. resp.Response.StatusCode = statusCode
  612. resp.Response.StatusText = message
  613. SendResponse(w, r, resp, h.Logger)
  614. }
  615. func requestScheme(r *http.Request) string {
  616. if r.TLS != nil {
  617. return "https"
  618. }
  619. if proto := r.Header.Get("X-Forwarded-Proto"); proto != "" {
  620. return proto
  621. }
  622. return "http"
  623. }
  624. func shuttingDown(ctx context.Context) bool {
  625. select {
  626. case <-ctx.Done():
  627. // server is shutting down, don't send buddy notifications
  628. return true
  629. default:
  630. }
  631. return false
  632. }