mgmt_api.go 67 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990
  1. package http
  2. import (
  3. "bytes"
  4. "context"
  5. "encoding/base64"
  6. "encoding/hex"
  7. "encoding/json"
  8. "errors"
  9. "fmt"
  10. "io"
  11. "log/slog"
  12. "math"
  13. "math/rand"
  14. "net/http"
  15. "strconv"
  16. "strings"
  17. "time"
  18. "github.com/google/uuid"
  19. "github.com/mk6i/open-oscar-server/config"
  20. "github.com/mk6i/open-oscar-server/state"
  21. "github.com/mk6i/open-oscar-server/wire"
  22. )
  23. func NewManagementAPI(bld config.Build, listener string, userManager UserManager, sessionRetriever SessionRetriever, buddyBroadcaster BuddyBroadcaster, chatRoomRetriever ChatRoomRetriever, chatRoomCreator ChatRoomCreator, chatRoomDeleter ChatRoomDeleter, chatSessionRetriever ChatSessionRetriever, directoryManager DirectoryManager, messageRelayer MessageRelayer, bartAssetManager BARTAssetManager, feedbagRetriever FeedBagRetriever, feedbagManager FeedbagManager, accountManager AccountManager, profileRetriever ProfileRetriever, webAPIKeyManager WebAPIKeyManager, icqProfileManager ICQProfileManager, createAccount state.CreateAccountFunc, logger *slog.Logger) *Server {
  24. mux := http.NewServeMux()
  25. // Handlers for '/user' route
  26. mux.HandleFunc("DELETE /user", func(w http.ResponseWriter, r *http.Request) {
  27. deleteUserHandler(w, r, userManager, logger)
  28. })
  29. mux.HandleFunc("GET /user", func(w http.ResponseWriter, r *http.Request) {
  30. getUserHandler(w, r, userManager, logger)
  31. })
  32. mux.HandleFunc("POST /user", func(w http.ResponseWriter, r *http.Request) {
  33. postUserHandler(w, r, createAccount, logger)
  34. })
  35. // Handlers for '/user/password' route
  36. mux.HandleFunc("PUT /user/password", func(w http.ResponseWriter, r *http.Request) {
  37. putUserPasswordHandler(w, r, userManager, logger)
  38. })
  39. // Handlers for '/user/login' route
  40. mux.HandleFunc("GET /user/login", func(w http.ResponseWriter, r *http.Request) {
  41. getUserLoginHandler(w, r, userManager, logger)
  42. })
  43. // Handlers for '/user/{screenname}/account' route
  44. mux.HandleFunc("GET /user/{screenname}/account", func(w http.ResponseWriter, r *http.Request) {
  45. getUserAccountHandler(w, r, userManager, accountManager, profileRetriever, logger)
  46. })
  47. mux.HandleFunc("PATCH /user/{screenname}/account", func(w http.ResponseWriter, r *http.Request) {
  48. patchUserAccountHandler(w, r, userManager, accountManager, logger)
  49. })
  50. // Handlers for '/user/{screenname}/icon' route
  51. mux.HandleFunc("GET /user/{screenname}/icon", func(w http.ResponseWriter, r *http.Request) {
  52. getUserBuddyIconHandler(w, r, userManager, feedbagRetriever, bartAssetManager, logger)
  53. })
  54. // Handlers for '/user/{screenname}/icq' route
  55. mux.HandleFunc("GET /user/{screenname}/icq", func(w http.ResponseWriter, r *http.Request) {
  56. getICQProfileHandler(w, r, icqProfileManager, logger)
  57. })
  58. mux.HandleFunc("PUT /user/{screenname}/icq", func(w http.ResponseWriter, r *http.Request) {
  59. putICQProfileHandler(w, r, icqProfileManager, logger)
  60. })
  61. // Handlers for '/session' route
  62. mux.HandleFunc("GET /session", func(w http.ResponseWriter, r *http.Request) {
  63. getSessionHandler(w, r, sessionRetriever, time.Now)
  64. })
  65. // Handlers for '/session/{screenname}' route
  66. mux.HandleFunc("GET /session/{screenname}", func(w http.ResponseWriter, r *http.Request) {
  67. getSessionHandler(w, r, sessionRetriever, time.Now)
  68. })
  69. mux.HandleFunc("DELETE /session/{screenname}", func(w http.ResponseWriter, r *http.Request) {
  70. deleteSessionHandler(w, r, sessionRetriever)
  71. })
  72. // Handlers for '/chat/room/public' route
  73. mux.HandleFunc("GET /chat/room/public", func(w http.ResponseWriter, r *http.Request) {
  74. getPublicChatHandler(w, r, chatRoomRetriever, chatSessionRetriever, logger)
  75. })
  76. mux.HandleFunc("POST /chat/room/public", func(w http.ResponseWriter, r *http.Request) {
  77. postPublicChatHandler(w, r, chatRoomCreator, logger)
  78. })
  79. mux.HandleFunc("DELETE /chat/room/public", func(w http.ResponseWriter, r *http.Request) {
  80. deletePublicChatHandler(w, r, chatRoomDeleter, logger)
  81. })
  82. // Handlers for '/chat/room/private' route
  83. mux.HandleFunc("GET /chat/room/private", func(w http.ResponseWriter, r *http.Request) {
  84. getPrivateChatHandler(w, r, chatRoomRetriever, chatSessionRetriever, logger)
  85. })
  86. // Handlers for '/instant-message' route
  87. mux.HandleFunc("POST /instant-message", func(w http.ResponseWriter, r *http.Request) {
  88. postInstantMessageHandler(w, r, messageRelayer, logger)
  89. })
  90. // Handlers for '/version' route
  91. mux.HandleFunc("GET /version", func(w http.ResponseWriter, r *http.Request) {
  92. getVersionHandler(w, bld)
  93. })
  94. // Handlers for '/admin/webapi/keys' route - Web API key management
  95. mux.HandleFunc("POST /admin/webapi/keys", func(w http.ResponseWriter, r *http.Request) {
  96. postWebAPIKeyHandler(w, r, webAPIKeyManager, uuid.New, logger)
  97. })
  98. mux.HandleFunc("GET /admin/webapi/keys", func(w http.ResponseWriter, r *http.Request) {
  99. getWebAPIKeysHandler(w, r, webAPIKeyManager, logger)
  100. })
  101. mux.HandleFunc("GET /admin/webapi/keys/{id}", func(w http.ResponseWriter, r *http.Request) {
  102. getWebAPIKeyHandler(w, r, webAPIKeyManager, logger)
  103. })
  104. mux.HandleFunc("PUT /admin/webapi/keys/{id}", func(w http.ResponseWriter, r *http.Request) {
  105. putWebAPIKeyHandler(w, r, webAPIKeyManager, logger)
  106. })
  107. mux.HandleFunc("DELETE /admin/webapi/keys/{id}", func(w http.ResponseWriter, r *http.Request) {
  108. deleteWebAPIKeyHandler(w, r, webAPIKeyManager, logger)
  109. })
  110. // Handlers for '/directory/category' route
  111. mux.HandleFunc("GET /directory/category", func(w http.ResponseWriter, r *http.Request) {
  112. getDirectoryCategoryHandler(w, r, directoryManager, logger)
  113. })
  114. mux.HandleFunc("POST /directory/category", func(w http.ResponseWriter, r *http.Request) {
  115. postDirectoryCategoryHandler(w, r, directoryManager, logger)
  116. })
  117. // Handlers for '/directory/category/{id}' route
  118. mux.HandleFunc("DELETE /directory/category/{id}", func(w http.ResponseWriter, r *http.Request) {
  119. deleteDirectoryCategoryHandler(w, r, directoryManager, logger)
  120. })
  121. // Handlers for '/directory/category/{id}/keyword' route
  122. mux.HandleFunc("GET /directory/category/{id}/keyword", func(w http.ResponseWriter, r *http.Request) {
  123. getDirectoryCategoryKeywordHandler(w, r, directoryManager, logger)
  124. })
  125. // Handlers for '/directory/keyword' route
  126. mux.HandleFunc("POST /directory/keyword", func(w http.ResponseWriter, r *http.Request) {
  127. postDirectoryKeywordHandler(w, r, directoryManager, logger)
  128. })
  129. // Handlers for '/directory/keyword/{id}' route
  130. mux.HandleFunc("DELETE /directory/keyword/{id}", func(w http.ResponseWriter, r *http.Request) {
  131. deleteDirectoryKeywordHandler(w, r, directoryManager, logger)
  132. })
  133. // Handlers for '/bart' route
  134. mux.HandleFunc("GET /bart", func(w http.ResponseWriter, r *http.Request) {
  135. getBARTByTypeHandler(w, r, bartAssetManager, logger)
  136. })
  137. // Handlers for '/bart/{hash}' route
  138. mux.HandleFunc("GET /bart/{hash}", func(w http.ResponseWriter, r *http.Request) {
  139. getBARTHandler(w, r, bartAssetManager, logger)
  140. })
  141. mux.HandleFunc("POST /bart/{hash}", func(w http.ResponseWriter, r *http.Request) {
  142. postBARTHandler(w, r, bartAssetManager, logger)
  143. })
  144. mux.HandleFunc("DELETE /bart/{hash}", func(w http.ResponseWriter, r *http.Request) {
  145. deleteBARTHandler(w, r, bartAssetManager, logger)
  146. })
  147. // Handlers for '/feedbag/{screen_name}/group' route
  148. mux.HandleFunc("GET /feedbag/{screen_name}/group", func(w http.ResponseWriter, r *http.Request) {
  149. getFeedbagBuddyHandler(w, r, feedbagManager, logger)
  150. })
  151. // Handlers for '/feedbag/{screen_name}/group/{group_name}' route
  152. mux.HandleFunc("PUT /feedbag/{screen_name}/group/{group_name}", func(w http.ResponseWriter, r *http.Request) {
  153. putFeedbagGroupHandler(w, r, feedbagManager, messageRelayer, logger, rand.Intn)
  154. })
  155. // Handlers for '/feedbag/{screen_name}/group/{group_id}/buddy/{buddy_screen_name}' route
  156. mux.HandleFunc("PUT /feedbag/{screen_name}/group/{group_id}/buddy/{buddy_screen_name}", func(w http.ResponseWriter, r *http.Request) {
  157. putFeedbagBuddyHandler(w, r, buddyBroadcaster, feedbagManager, sessionRetriever, messageRelayer, logger, rand.Intn)
  158. })
  159. mux.HandleFunc("DELETE /feedbag/{screen_name}/group/{group_id}/buddy/{buddy_screen_name}", func(w http.ResponseWriter, r *http.Request) {
  160. deleteFeedbagBuddyHandler(w, r, buddyBroadcaster, feedbagManager, sessionRetriever, messageRelayer, logger)
  161. })
  162. return &Server{
  163. server: http.Server{
  164. Addr: listener,
  165. Handler: mux,
  166. },
  167. logger: logger,
  168. }
  169. }
  170. type Server struct {
  171. server http.Server
  172. logger *slog.Logger
  173. }
  174. func (s *Server) ListenAndServe() error {
  175. s.logger.Info("starting server", "addr", s.server.Addr)
  176. if err := s.server.ListenAndServe(); !errors.Is(err, http.ErrServerClosed) {
  177. return fmt.Errorf("unable to start management API server: %w", err)
  178. }
  179. return nil
  180. }
  181. func (s *Server) Shutdown(ctx context.Context) error {
  182. defer s.logger.Info("shutdown complete")
  183. return s.server.Shutdown(ctx)
  184. }
  185. // deleteUserHandler handles the DELETE /user endpoint.
  186. func deleteUserHandler(w http.ResponseWriter, r *http.Request, manager UserManager, logger *slog.Logger) {
  187. user, err := userFromBody(r)
  188. if err != nil {
  189. http.Error(w, err.Error(), http.StatusBadRequest)
  190. return
  191. }
  192. err = manager.DeleteUser(r.Context(), state.NewIdentScreenName(user.ScreenName))
  193. switch {
  194. case errors.Is(err, state.ErrNoUser):
  195. http.Error(w, "user does not exist", http.StatusNotFound)
  196. return
  197. case err != nil:
  198. logger.Error("error deleting user DELETE /user", "err", err.Error())
  199. http.Error(w, "internal server error", http.StatusInternalServerError)
  200. return
  201. }
  202. w.WriteHeader(http.StatusNoContent)
  203. _, _ = fmt.Fprintln(w, "User account successfully deleted.")
  204. }
  205. // putUserPasswordHandler handles the PUT /user/password endpoint.
  206. func putUserPasswordHandler(w http.ResponseWriter, r *http.Request, userManager UserManager, logger *slog.Logger) {
  207. input, err := userFromBody(r)
  208. if err != nil {
  209. http.Error(w, err.Error(), http.StatusBadRequest)
  210. return
  211. }
  212. sn := state.NewIdentScreenName(input.ScreenName)
  213. if err := userManager.SetUserPassword(r.Context(), sn, input.Password); err != nil {
  214. switch {
  215. case errors.Is(err, state.ErrNoUser):
  216. http.Error(w, "user does not exist", http.StatusNotFound)
  217. return
  218. case errors.Is(err, state.ErrPasswordInvalid):
  219. http.Error(w, err.Error(), http.StatusBadRequest)
  220. return
  221. default:
  222. logger.Error("error updating user password PUT /user/password", "err", err.Error())
  223. http.Error(w, "internal server error", http.StatusInternalServerError)
  224. return
  225. }
  226. }
  227. w.WriteHeader(http.StatusNoContent)
  228. _, _ = fmt.Fprintln(w, "Password successfully reset.")
  229. }
  230. // getSessionHandler handles GET /session
  231. func getSessionHandler(w http.ResponseWriter, r *http.Request, sessionRetriever SessionRetriever, nowFn func() time.Time) {
  232. w.Header().Set("Content-Type", "application/json")
  233. var allSessions []*state.Session
  234. if screenName := r.PathValue("screenname"); screenName != "" {
  235. session := sessionRetriever.RetrieveSession(state.NewIdentScreenName(screenName))
  236. if session == nil {
  237. http.Error(w, "session not found", http.StatusNotFound)
  238. return
  239. }
  240. allSessions = append(allSessions, session)
  241. } else {
  242. // AllSessions returns all sessions
  243. allSessions = sessionRetriever.AllSessions()
  244. }
  245. ou := onlineUsers{
  246. Count: len(allSessions),
  247. Sessions: make([]sessionHandle, len(allSessions)),
  248. }
  249. for i, s := range allSessions {
  250. instances := s.Instances()
  251. instanceHandles := make([]instanceHandle, len(instances))
  252. for j, inst := range instances {
  253. instanceIdleSeconds := 0
  254. if inst.Idle() {
  255. instanceIdleSeconds = int(nowFn().Sub(inst.IdleTime()).Seconds())
  256. }
  257. awayMsg, _ := inst.AwayMessage()
  258. instanceHandles[j] = instanceHandle{
  259. Num: int(inst.Num()),
  260. IdleSeconds: instanceIdleSeconds,
  261. IsAway: inst.Away(),
  262. AwayMessage: awayMsg,
  263. IsInvisible: inst.Invisible(),
  264. }
  265. ra := inst.RemoteAddr()
  266. if ra != nil {
  267. instanceHandles[j].RemoteAddr = ra.Addr().String()
  268. instanceHandles[j].RemotePort = int(ra.Port())
  269. }
  270. }
  271. sessionIdleSeconds := 0
  272. if s.Idle() {
  273. sessionIdleSeconds = int(nowFn().Sub(s.IdleTime()).Seconds())
  274. }
  275. allAway := s.Away()
  276. awayMessage := ""
  277. if allAway {
  278. awayMessage = s.AwayMessage()
  279. }
  280. ou.Sessions[i] = sessionHandle{
  281. ID: s.IdentScreenName().String(),
  282. ScreenName: s.DisplayScreenName().String(),
  283. OnlineSeconds: int(nowFn().Sub(s.SignonTime()).Seconds()),
  284. IsAway: allAway,
  285. AwayMessage: awayMessage,
  286. IdleSeconds: sessionIdleSeconds,
  287. IsInvisible: s.Invisible(),
  288. IsICQ: s.UIN() > 0,
  289. InstanceCount: s.InstanceCount(),
  290. Instances: instanceHandles,
  291. }
  292. }
  293. if err := json.NewEncoder(w).Encode(ou); err != nil {
  294. http.Error(w, err.Error(), http.StatusInternalServerError)
  295. return
  296. }
  297. }
  298. // deleteSessionHandler handles DELETE /session/{screenname}
  299. func deleteSessionHandler(w http.ResponseWriter, r *http.Request, sessionRetriever SessionRetriever) {
  300. w.Header().Set("Content-Type", "application/json")
  301. if screenName := r.PathValue("screenname"); screenName != "" {
  302. session := sessionRetriever.RetrieveSession(state.NewIdentScreenName(screenName))
  303. if session == nil {
  304. errorMsg(w, "session not found", http.StatusNotFound)
  305. return
  306. }
  307. session.CloseSession()
  308. }
  309. w.WriteHeader(http.StatusNoContent)
  310. }
  311. // getUserHandler handles the GET /user endpoint.
  312. func getUserHandler(w http.ResponseWriter, r *http.Request, userManager UserManager, logger *slog.Logger) {
  313. w.Header().Set("Content-Type", "application/json")
  314. users, err := userManager.AllUsers(r.Context())
  315. if err != nil {
  316. logger.Error("error in GET /user", "err", err.Error())
  317. http.Error(w, "internal server error", http.StatusInternalServerError)
  318. return
  319. }
  320. out := make([]userHandle, len(users))
  321. for i, u := range users {
  322. suspendedStatus, err := getSuspendedStatusErrCodeToText(u.SuspendedStatus)
  323. if err != nil {
  324. logger.Error("error getting suspended status in GET /user", "err", err.Error())
  325. http.Error(w, "internal server error", http.StatusInternalServerError)
  326. return
  327. }
  328. out[i] = userHandle{
  329. ID: u.IdentScreenName.String(),
  330. ScreenName: u.DisplayScreenName.String(),
  331. IsICQ: u.IsICQ,
  332. SuspendedStatus: suspendedStatus,
  333. IsBot: u.IsBot,
  334. }
  335. }
  336. if err := json.NewEncoder(w).Encode(out); err != nil {
  337. http.Error(w, err.Error(), http.StatusInternalServerError)
  338. return
  339. }
  340. }
  341. // postUserHandler handles the POST /user endpoint.
  342. func postUserHandler(w http.ResponseWriter, r *http.Request, createAccount state.CreateAccountFunc, logger *slog.Logger) {
  343. input, err := userFromBody(r)
  344. if err != nil {
  345. http.Error(w, err.Error(), http.StatusBadRequest)
  346. return
  347. }
  348. sn := state.DisplayScreenName(input.ScreenName)
  349. err = createAccount(r.Context(), sn, input.Password)
  350. switch {
  351. case errors.Is(err, state.ErrDupUser):
  352. http.Error(w, "user already exists", http.StatusConflict)
  353. return
  354. case errors.Is(err, state.ErrAIMHandleInvalidFormat), errors.Is(err, state.ErrAIMHandleLength):
  355. http.Error(w, fmt.Sprintf("invalid screen name: %s", err), http.StatusBadRequest)
  356. return
  357. case errors.Is(err, state.ErrICQUINInvalidFormat):
  358. http.Error(w, fmt.Sprintf("invalid uin: %s", err), http.StatusBadRequest)
  359. return
  360. case errors.Is(err, state.ErrPasswordInvalid):
  361. http.Error(w, fmt.Sprintf("invalid password: %s", err), http.StatusBadRequest)
  362. return
  363. case err != nil:
  364. logger.Error("error inserting user POST /user", "err", err.Error())
  365. http.Error(w, "internal server error", http.StatusInternalServerError)
  366. return
  367. }
  368. w.WriteHeader(http.StatusCreated)
  369. _, _ = fmt.Fprintln(w, "User account created successfully.")
  370. }
  371. func userFromBody(r *http.Request) (userWithPassword, error) {
  372. user := userWithPassword{}
  373. if err := json.NewDecoder(r.Body).Decode(&user); err != nil {
  374. return userWithPassword{}, errors.New("malformed input")
  375. }
  376. return user, nil
  377. }
  378. // getUserLoginHandler is a temporary endpoint for validating user credentials.
  379. // do not rely on this endpoint, as it will be eventually removed.
  380. func getUserLoginHandler(w http.ResponseWriter, r *http.Request, userManager UserManager, logger *slog.Logger) {
  381. authHeader := r.Header.Get("Authorization")
  382. if authHeader == "" {
  383. // No authentication header found
  384. w.WriteHeader(http.StatusUnauthorized)
  385. w.Header().Set("WWW-Authenticate", `Basic realm="User Login"`)
  386. _, _ = w.Write([]byte("401 Unauthorized\n"))
  387. return
  388. }
  389. auth := strings.SplitN(authHeader, " ", 2)
  390. if len(auth) != 2 || auth[0] != "Basic" {
  391. w.WriteHeader(http.StatusUnauthorized)
  392. _, _ = w.Write([]byte("401 Unauthorized: Missing Basic prefix\n"))
  393. return
  394. }
  395. payload, err := base64.StdEncoding.DecodeString(auth[1])
  396. if err != nil {
  397. w.WriteHeader(http.StatusUnauthorized)
  398. _, _ = w.Write([]byte("401 Unauthorized: Invalid Base64 Encoding\n"))
  399. return
  400. }
  401. pair := strings.SplitN(string(payload), ":", 2)
  402. if len(pair) != 2 {
  403. w.WriteHeader(http.StatusUnauthorized)
  404. _, _ = w.Write([]byte("401 Unauthorized: Invalid Authentication Token\n"))
  405. return
  406. }
  407. username, password := state.NewIdentScreenName(pair[0]), pair[1]
  408. user, err := userManager.User(r.Context(), username)
  409. if err != nil {
  410. w.WriteHeader(http.StatusInternalServerError)
  411. _, _ = w.Write([]byte("500 InternalServerError\n"))
  412. logger.Error("error getting user", "err", err.Error())
  413. return
  414. }
  415. if user == nil || !user.ValidateHash(wire.StrongMD5PasswordHash(password, user.AuthKey)) {
  416. w.WriteHeader(http.StatusUnauthorized)
  417. _, _ = w.Write([]byte("401 Unauthorized: Invalid Credentials\n"))
  418. return
  419. }
  420. // Successfully authenticated
  421. w.WriteHeader(http.StatusOK)
  422. _, _ = w.Write([]byte("200 OK: Successfully Authenticated\n"))
  423. }
  424. // getPublicChatHandler handles the GET /chat/room/public endpoint.
  425. func getPublicChatHandler(w http.ResponseWriter, r *http.Request, chatRoomRetriever ChatRoomRetriever, chatSessionRetriever ChatSessionRetriever, logger *slog.Logger) {
  426. w.Header().Set("Content-Type", "application/json")
  427. rooms, err := chatRoomRetriever.AllChatRooms(r.Context(), state.PublicExchange)
  428. if err != nil {
  429. logger.Error("error in GET /chat/rooms/public", "err", err.Error())
  430. http.Error(w, "internal server error", http.StatusInternalServerError)
  431. return
  432. }
  433. out := make([]chatRoom, len(rooms))
  434. for i, room := range rooms {
  435. sessions := chatSessionRetriever.AllSessions(room.Cookie())
  436. cr := chatRoom{
  437. CreateTime: room.CreateTime(),
  438. Name: room.Name(),
  439. Participants: make([]aimChatUserHandle, len(sessions)),
  440. URL: room.URL().String(),
  441. }
  442. for j, sess := range sessions {
  443. cr.Participants[j] = aimChatUserHandle{
  444. ID: sess.IdentScreenName().String(),
  445. ScreenName: sess.DisplayScreenName().String(),
  446. }
  447. }
  448. out[i] = cr
  449. }
  450. writeUnescapeChatURL(w, out)
  451. }
  452. // postPublicChatHandler handles the POST /chat/room/public endpoint.
  453. func postPublicChatHandler(w http.ResponseWriter, r *http.Request, chatRoomCreator ChatRoomCreator, logger *slog.Logger) {
  454. input := chatRoomCreate{}
  455. if err := json.NewDecoder(r.Body).Decode(&input); err != nil {
  456. http.Error(w, "invalid input", http.StatusBadRequest)
  457. return
  458. }
  459. input.Name = strings.TrimSpace(input.Name)
  460. if input.Name == "" || len(input.Name) > 50 {
  461. http.Error(w, "chat room name must be between 1 and 50 characters", http.StatusBadRequest)
  462. return
  463. }
  464. cr := state.NewChatRoom(input.Name, state.NewIdentScreenName("system"), state.PublicExchange)
  465. err := chatRoomCreator.CreateChatRoom(r.Context(), &cr)
  466. switch {
  467. case errors.Is(err, state.ErrDupChatRoom):
  468. http.Error(w, "Chat room already exists.", http.StatusConflict)
  469. return
  470. case err != nil:
  471. logger.Error("error inserting chat room POST /chat/room/public", "err", err.Error())
  472. http.Error(w, "internal server error", http.StatusInternalServerError)
  473. return
  474. }
  475. w.WriteHeader(http.StatusCreated)
  476. _, _ = fmt.Fprintln(w, "Chat room created successfully.")
  477. }
  478. // getPrivateChatHandler handles the GET /chat/room/private endpoint.
  479. func getPrivateChatHandler(w http.ResponseWriter, r *http.Request, chatRoomRetriever ChatRoomRetriever, chatSessionRetriever ChatSessionRetriever, logger *slog.Logger) {
  480. w.Header().Set("Content-Type", "application/json")
  481. rooms, err := chatRoomRetriever.AllChatRooms(r.Context(), state.PrivateExchange)
  482. if err != nil {
  483. logger.Error("error in GET /chat/rooms/private", "err", err.Error())
  484. http.Error(w, "internal server error", http.StatusInternalServerError)
  485. return
  486. }
  487. out := make([]chatRoom, len(rooms))
  488. for i, room := range rooms {
  489. sessions := chatSessionRetriever.AllSessions(room.Cookie())
  490. cr := chatRoom{
  491. CreateTime: room.CreateTime(),
  492. CreatorID: room.Creator().String(),
  493. Name: room.Name(),
  494. Participants: make([]aimChatUserHandle, len(sessions)),
  495. URL: room.URL().String(),
  496. }
  497. for j, sess := range sessions {
  498. cr.Participants[j] = aimChatUserHandle{
  499. ID: sess.IdentScreenName().String(),
  500. ScreenName: sess.DisplayScreenName().String(),
  501. }
  502. }
  503. out[i] = cr
  504. }
  505. writeUnescapeChatURL(w, out)
  506. }
  507. // deletePublicChatHandler handles the DELETE /chat/room/public endpoint.
  508. func deletePublicChatHandler(w http.ResponseWriter, r *http.Request, chatRoomDeleter ChatRoomDeleter, logger *slog.Logger) {
  509. input := chatRoomDelete{}
  510. if err := json.NewDecoder(r.Body).Decode(&input); err != nil {
  511. http.Error(w, "malformed input", http.StatusBadRequest)
  512. return
  513. }
  514. if len(input.Names) == 0 {
  515. http.Error(w, "no chat room names provided", http.StatusBadRequest)
  516. return
  517. }
  518. err := chatRoomDeleter.DeleteChatRooms(r.Context(), state.PublicExchange, input.Names)
  519. if err != nil {
  520. logger.Error("error deleting public chat rooms DELETE /chat/room/public", "err", err.Error())
  521. http.Error(w, "internal server error", http.StatusInternalServerError)
  522. return
  523. }
  524. w.WriteHeader(http.StatusNoContent)
  525. _, _ = fmt.Fprintln(w, "Chat rooms deleted successfully.")
  526. }
  527. // writeUnescapeChatURL writes a JSON-encoded list of chat rooms with unescaped
  528. // ampersands preceding the exchange query param.
  529. //
  530. // before: aim:gochat?roomname=Office+Hijinks\u0026exchange=5
  531. // after: aim:gochat?roomname=Office+Hijinks&exchange=5
  532. //
  533. // This makes it easier to copy the gochat URL into AIM, which does not
  534. // recognize the ampersand unicode character \u0026.
  535. func writeUnescapeChatURL(w http.ResponseWriter, out []chatRoom) {
  536. buf := &bytes.Buffer{}
  537. if err := json.NewEncoder(buf).Encode(out); err != nil {
  538. http.Error(w, err.Error(), http.StatusInternalServerError)
  539. return
  540. }
  541. b := bytes.ReplaceAll(buf.Bytes(), []byte(`\u0026exchange`), []byte(`&exchange`))
  542. if _, err := w.Write(b); err != nil {
  543. http.Error(w, err.Error(), http.StatusInternalServerError)
  544. return
  545. }
  546. }
  547. // postIMHandler handles the POST /instant-message endpoint.
  548. func postInstantMessageHandler(w http.ResponseWriter, r *http.Request, messageRelayer MessageRelayer, logger *slog.Logger) {
  549. input := instantMessage{}
  550. if err := json.NewDecoder(r.Body).Decode(&input); err != nil {
  551. http.Error(w, "malformed input", http.StatusBadRequest)
  552. return
  553. }
  554. tlv, err := wire.ICBMFragmentList(input.Text)
  555. if err != nil {
  556. logger.Error("error sending message POST /instant-message", "err", err.Error())
  557. http.Error(w, "internal server error", http.StatusInternalServerError)
  558. return
  559. }
  560. msg := wire.SNACMessage{
  561. Frame: wire.SNACFrame{
  562. FoodGroup: wire.ICBM,
  563. SubGroup: wire.ICBMChannelMsgToClient,
  564. },
  565. Body: wire.SNAC_0x04_0x07_ICBMChannelMsgToClient{
  566. ChannelID: 1,
  567. TLVUserInfo: wire.TLVUserInfo{
  568. ScreenName: input.From,
  569. },
  570. TLVRestBlock: wire.TLVRestBlock{
  571. TLVList: wire.TLVList{
  572. wire.NewTLVBE(wire.ICBMTLVAOLIMData, tlv),
  573. },
  574. },
  575. },
  576. }
  577. messageRelayer.RelayToScreenName(context.Background(), state.NewIdentScreenName(input.To), msg)
  578. w.WriteHeader(http.StatusOK)
  579. _, _ = fmt.Fprintln(w, "Message sent successfully.")
  580. }
  581. // getUserBuddyIconHandler handles the GET /user/{screenname}/icon endpoint.
  582. func getUserBuddyIconHandler(w http.ResponseWriter, r *http.Request, u UserManager, f FeedBagRetriever, b BARTAssetManager, logger *slog.Logger) {
  583. screenName := state.NewIdentScreenName(r.PathValue("screenname"))
  584. user, err := u.User(r.Context(), screenName)
  585. if err != nil {
  586. logger.Error("error retrieving user", "err", err.Error())
  587. http.Error(w, "internal server error", http.StatusInternalServerError)
  588. return
  589. }
  590. if user == nil {
  591. http.Error(w, "user not found", http.StatusNotFound)
  592. return
  593. }
  594. iconRef, err := f.BuddyIconMetadata(r.Context(), screenName)
  595. if err != nil {
  596. logger.Error("error retrieving buddy icon ref", "err", err.Error())
  597. http.Error(w, "internal server error", http.StatusInternalServerError)
  598. return
  599. }
  600. if iconRef == nil || iconRef.HasClearIconHash() {
  601. http.Error(w, "icon not found", http.StatusNotFound)
  602. return
  603. }
  604. icon, err := b.BARTItem(r.Context(), iconRef.Hash)
  605. if err != nil {
  606. logger.Error("error retrieving buddy icon bart item", "err", err.Error())
  607. http.Error(w, "internal server error", http.StatusInternalServerError)
  608. return
  609. }
  610. w.Header().Set("Content-Type", http.DetectContentType(icon))
  611. _, _ = w.Write(icon)
  612. }
  613. // getUserAccountHandler handles the GET /user/{screenname}/account endpoint.
  614. func getUserAccountHandler(w http.ResponseWriter, r *http.Request, userManager UserManager, a AccountManager, p ProfileRetriever, logger *slog.Logger) {
  615. w.Header().Set("Content-Type", "application/json")
  616. screenName := r.PathValue("screenname")
  617. user, err := userManager.User(r.Context(), state.NewIdentScreenName(screenName))
  618. if err != nil {
  619. logger.Error("error in GET /user/{screenname}/account", "err", err.Error())
  620. http.Error(w, "internal server error", http.StatusInternalServerError)
  621. return
  622. }
  623. if user == nil {
  624. http.Error(w, "user not found", http.StatusNotFound)
  625. return
  626. }
  627. emailAddress := ""
  628. email, err := a.EmailAddress(r.Context(), user.IdentScreenName)
  629. if err != nil {
  630. emailAddress = ""
  631. } else {
  632. emailAddress = email.String()
  633. }
  634. regStatus, err := a.RegStatus(r.Context(), user.IdentScreenName)
  635. if err != nil {
  636. logger.Error("error in GET /user/*/account RegStatus", "err", err.Error())
  637. http.Error(w, "internal server error", http.StatusInternalServerError)
  638. return
  639. }
  640. confirmStatus, err := a.ConfirmStatus(r.Context(), user.IdentScreenName)
  641. if err != nil {
  642. logger.Error("error in GET /user/*/account ConfirmStatus", "err", err.Error())
  643. http.Error(w, "internal server error", http.StatusInternalServerError)
  644. return
  645. }
  646. profile, err := p.Profile(r.Context(), user.IdentScreenName)
  647. if err != nil {
  648. logger.Error("error in GET /user/*/account Profile", "err", err.Error())
  649. http.Error(w, "internal server error", http.StatusInternalServerError)
  650. return
  651. }
  652. suspendedStatusText, err := getSuspendedStatusErrCodeToText(user.SuspendedStatus)
  653. if err != nil {
  654. logger.Error("error in GET /user/{screenname}/account", "err", err.Error())
  655. http.Error(w, "internal server error", http.StatusInternalServerError)
  656. }
  657. out := userAccountHandle{
  658. ID: user.IdentScreenName.String(),
  659. ScreenName: user.DisplayScreenName.String(),
  660. EmailAddress: emailAddress,
  661. RegStatus: regStatus,
  662. Confirmed: confirmStatus,
  663. Profile: profile.ProfileText,
  664. IsICQ: user.IsICQ,
  665. SuspendedStatus: suspendedStatusText,
  666. IsBot: user.IsBot,
  667. }
  668. if err := json.NewEncoder(w).Encode(out); err != nil {
  669. http.Error(w, err.Error(), http.StatusInternalServerError)
  670. return
  671. }
  672. }
  673. // patchUserAccountHandler handles the PATCH /user/{screenname}/account endpoint.
  674. func patchUserAccountHandler(w http.ResponseWriter, r *http.Request, userManager UserManager, a AccountManager, logger *slog.Logger) {
  675. w.Header().Set("Content-Type", "application/json")
  676. screenName := r.PathValue("screenname")
  677. user, err := userManager.User(r.Context(), state.NewIdentScreenName(screenName))
  678. if err != nil {
  679. logger.Error("error in PATCH /user/{screenname}/account", "err", err.Error())
  680. http.Error(w, "internal server error", http.StatusInternalServerError)
  681. return
  682. }
  683. if user == nil {
  684. http.Error(w, "user not found", http.StatusNotFound)
  685. return
  686. }
  687. input := userAccountPatch{}
  688. d := json.NewDecoder(r.Body)
  689. d.DisallowUnknownFields()
  690. if err := d.Decode(&input); err != nil {
  691. errorMsg(w, err.Error(), http.StatusBadRequest)
  692. return
  693. }
  694. modifiedUser := false
  695. if input.SuspendedStatusText != nil {
  696. switch *input.SuspendedStatusText {
  697. case
  698. "", "deleted", "expired",
  699. "suspended", "suspended_age":
  700. suspendedStatus, err := getSuspendedStatusTextToErrCode(*input.SuspendedStatusText)
  701. if err != nil {
  702. logger.Error("error in PATCH /user/{screenname}/account", "err", err.Error())
  703. http.Error(w, "internal server error", http.StatusInternalServerError)
  704. return
  705. }
  706. if suspendedStatus != user.SuspendedStatus {
  707. if err := a.UpdateSuspendedStatus(r.Context(), suspendedStatus, user.IdentScreenName); err != nil {
  708. logger.Error("error in PATCH /user/{screenname}/account", "err", err.Error())
  709. http.Error(w, "internal server error", http.StatusInternalServerError)
  710. return
  711. }
  712. modifiedUser = true
  713. }
  714. default:
  715. errorMsg(w, "suspended_status must be empty str or one of deleted,expired,suspended,suspended_age", http.StatusBadRequest)
  716. return
  717. }
  718. }
  719. if input.IsBot != nil && user.IsBot != *input.IsBot {
  720. if err := a.SetBotStatus(r.Context(), *input.IsBot, user.IdentScreenName); err != nil {
  721. logger.Error("error in PATCH /user/{screenname}/account", "err", err.Error())
  722. http.Error(w, "internal server error", http.StatusInternalServerError)
  723. return
  724. }
  725. modifiedUser = true
  726. }
  727. if !modifiedUser {
  728. w.WriteHeader(http.StatusNotModified)
  729. return
  730. }
  731. w.WriteHeader(http.StatusNoContent)
  732. }
  733. // getSuspendedStatusTextToErrCode maps the given suspendedStatusText to
  734. // the appropriate error code, or 0x0 for none.
  735. func getSuspendedStatusTextToErrCode(suspendedStatusText string) (uint16, error) {
  736. suspendedStatusTextMap := map[string]uint16{
  737. "": 0x0,
  738. "deleted": wire.LoginErrDeletedAccount,
  739. "expired": wire.LoginErrExpiredAccount,
  740. "suspended": wire.LoginErrSuspendedAccount,
  741. "suspended_age": wire.LoginErrSuspendedAccountAge,
  742. }
  743. suspendedStatus, ok := suspendedStatusTextMap[suspendedStatusText]
  744. if !ok {
  745. return 0x0, errors.New("unable to map suspendedText to error code")
  746. }
  747. return suspendedStatus, nil
  748. }
  749. // getSuspendedStatusErrCodeToText maps the given suspendedStatus to
  750. // the appropriate text, or "" for none.
  751. func getSuspendedStatusErrCodeToText(suspendedStatus uint16) (string, error) {
  752. suspendedStatusTextMap := map[uint16]string{
  753. 0x0: "",
  754. wire.LoginErrDeletedAccount: "deleted",
  755. wire.LoginErrExpiredAccount: "expired",
  756. wire.LoginErrSuspendedAccount: "suspended",
  757. wire.LoginErrSuspendedAccountAge: "suspended_age",
  758. }
  759. st, ok := suspendedStatusTextMap[suspendedStatus]
  760. if !ok {
  761. return "", errors.New("unable to map error code to suspendedText")
  762. }
  763. return st, nil
  764. }
  765. // getVersionHandler handles the GET /version endpoint.
  766. func getVersionHandler(w http.ResponseWriter, bld config.Build) {
  767. w.Header().Set("Content-Type", "application/json")
  768. if err := json.NewEncoder(w).Encode(bld); err != nil {
  769. http.Error(w, err.Error(), http.StatusInternalServerError)
  770. return
  771. }
  772. }
  773. // getDirectoryCategoryHandler handles the GET /directory/category endpoint.
  774. func getDirectoryCategoryHandler(w http.ResponseWriter, r *http.Request, manager DirectoryManager, logger *slog.Logger) {
  775. w.Header().Set("Content-Type", "application/json")
  776. categories, err := manager.Categories(r.Context())
  777. if err != nil {
  778. logger.Error("error in GET /directory/category", "err", err.Error())
  779. errorMsg(w, "internal server error", http.StatusInternalServerError)
  780. return
  781. }
  782. out := make([]directoryCategory, len(categories))
  783. for i, category := range categories {
  784. out[i] = directoryCategory{
  785. ID: category.ID,
  786. Name: category.Name,
  787. }
  788. }
  789. if err := json.NewEncoder(w).Encode(out); err != nil {
  790. errorMsg(w, err.Error(), http.StatusInternalServerError)
  791. }
  792. }
  793. // postDirectoryCategoryHandler handles the POST /directory/category endpoint.
  794. func postDirectoryCategoryHandler(w http.ResponseWriter, r *http.Request, manager DirectoryManager, logger *slog.Logger) {
  795. input := directoryCategoryCreate{}
  796. if err := json.NewDecoder(r.Body).Decode(&input); err != nil {
  797. errorMsg(w, "malformed input", http.StatusBadRequest)
  798. return
  799. }
  800. category, err := manager.CreateCategory(r.Context(), input.Name)
  801. if err != nil {
  802. if errors.Is(err, state.ErrKeywordCategoryExists) {
  803. errorMsg(w, "category already exists", http.StatusConflict)
  804. } else {
  805. logger.Error("error in POST /directory/category", "err", err.Error())
  806. errorMsg(w, "internal server error", http.StatusInternalServerError)
  807. }
  808. return
  809. }
  810. w.WriteHeader(http.StatusCreated)
  811. dc := directoryCategory{
  812. ID: category.ID,
  813. Name: category.Name,
  814. }
  815. if err := json.NewEncoder(w).Encode(dc); err != nil {
  816. errorMsg(w, err.Error(), http.StatusBadRequest)
  817. }
  818. }
  819. // deleteDirectoryCategoryHandler handles the DELETE /directory/category/{id} endpoint.
  820. func deleteDirectoryCategoryHandler(w http.ResponseWriter, r *http.Request, manager DirectoryManager, logger *slog.Logger) {
  821. categoryID, err := strconv.ParseUint(r.PathValue("id"), 10, 8)
  822. if err != nil {
  823. http.Error(w, "invalid category ID", http.StatusBadRequest)
  824. return
  825. }
  826. if err := manager.DeleteCategory(r.Context(), uint8(categoryID)); err != nil {
  827. switch {
  828. case errors.Is(err, state.ErrKeywordCategoryNotFound):
  829. errorMsg(w, "category not found", http.StatusNotFound)
  830. return
  831. case errors.Is(err, state.ErrKeywordInUse):
  832. errorMsg(w, "can't delete because category in use by a user", http.StatusConflict)
  833. return
  834. default:
  835. logger.Error("error in DELETE /directory/category/{id}", "err", err.Error())
  836. errorMsg(w, "internal server error", http.StatusInternalServerError)
  837. return
  838. }
  839. }
  840. w.WriteHeader(http.StatusNoContent)
  841. }
  842. // getDirectoryCategoryKeywordHandler handles the GET /directory/category/{id}/keyword endpoint.
  843. func getDirectoryCategoryKeywordHandler(w http.ResponseWriter, r *http.Request, manager DirectoryManager, logger *slog.Logger) {
  844. w.Header().Set("Content-Type", "application/json")
  845. categoryID, err := strconv.ParseUint(r.PathValue("id"), 10, 8)
  846. if err != nil {
  847. errorMsg(w, "invalid category ID", http.StatusBadRequest)
  848. return
  849. }
  850. categories, err := manager.KeywordsByCategory(r.Context(), uint8(categoryID))
  851. if err != nil {
  852. if errors.Is(err, state.ErrKeywordCategoryNotFound) {
  853. errorMsg(w, "category not found", http.StatusNotFound)
  854. } else {
  855. logger.Error("error in GET /directory/category/{id}/keyword", "err", err.Error())
  856. errorMsg(w, "internal server error", http.StatusInternalServerError)
  857. }
  858. return
  859. }
  860. out := make([]directoryCategory, len(categories))
  861. for i, category := range categories {
  862. out[i] = directoryCategory{
  863. ID: category.ID,
  864. Name: category.Name,
  865. }
  866. }
  867. if err := json.NewEncoder(w).Encode(out); err != nil {
  868. errorMsg(w, err.Error(), http.StatusInternalServerError)
  869. }
  870. }
  871. // postDirectoryKeywordHandler handles the POST /directory/keyword endpoint.
  872. func postDirectoryKeywordHandler(w http.ResponseWriter, r *http.Request, manager DirectoryManager, logger *slog.Logger) {
  873. w.Header().Set("Content-Type", "application/json")
  874. input := directoryKeywordCreate{}
  875. if err := json.NewDecoder(r.Body).Decode(&input); err != nil {
  876. errorMsg(w, "malformed input", http.StatusBadRequest)
  877. return
  878. }
  879. kw, err := manager.CreateKeyword(r.Context(), input.Name, input.CategoryID)
  880. if err != nil {
  881. switch {
  882. case errors.Is(err, state.ErrKeywordCategoryNotFound):
  883. errorMsg(w, "category not found", http.StatusNotFound)
  884. return
  885. case errors.Is(err, state.ErrKeywordExists):
  886. errorMsg(w, "keyword already exists", http.StatusConflict)
  887. return
  888. default:
  889. logger.Error("error in POST /directory/keyword", "err", err.Error())
  890. errorMsg(w, "internal server error", http.StatusInternalServerError)
  891. return
  892. }
  893. }
  894. w.WriteHeader(http.StatusCreated)
  895. dc := directoryKeyword{
  896. ID: kw.ID,
  897. Name: kw.Name,
  898. }
  899. if err := json.NewEncoder(w).Encode(dc); err != nil {
  900. errorMsg(w, err.Error(), http.StatusBadRequest)
  901. }
  902. }
  903. // deleteDirectoryKeywordHandler handles the DELETE /directory/keyword/{id} endpoint.
  904. func deleteDirectoryKeywordHandler(w http.ResponseWriter, r *http.Request, manager DirectoryManager, logger *slog.Logger) {
  905. keywordID, err := strconv.ParseUint(r.PathValue("id"), 10, 8)
  906. if err != nil {
  907. errorMsg(w, "invalid keyword ID", http.StatusBadRequest)
  908. return
  909. }
  910. if err := manager.DeleteKeyword(r.Context(), uint8(keywordID)); err != nil {
  911. switch {
  912. case errors.Is(err, state.ErrKeywordInUse):
  913. errorMsg(w, "can't delete because category in use by a user", http.StatusConflict)
  914. return
  915. case errors.Is(err, state.ErrKeywordNotFound):
  916. errorMsg(w, "keyword not found", http.StatusNotFound)
  917. return
  918. default:
  919. logger.Error("error in DELETE /directory/keyword/{id}", "err", err.Error())
  920. errorMsg(w, "internal server error", http.StatusInternalServerError)
  921. return
  922. }
  923. }
  924. w.WriteHeader(http.StatusNoContent)
  925. }
  926. // getICQProfileHandler handles the GET /user/{screenname}/icq endpoint.
  927. func getICQProfileHandler(w http.ResponseWriter, r *http.Request, mgr ICQProfileManager, logger *slog.Logger) {
  928. w.Header().Set("Content-Type", "application/json")
  929. screenName := r.PathValue("screenname")
  930. user, err := mgr.User(r.Context(), state.NewIdentScreenName(screenName))
  931. if err != nil {
  932. logger.Error("error in GET /user/{screenname}/icq", "err", err.Error())
  933. http.Error(w, "internal server error", http.StatusInternalServerError)
  934. return
  935. }
  936. if user == nil {
  937. http.Error(w, "user not found", http.StatusNotFound)
  938. return
  939. }
  940. if !user.IsICQ {
  941. http.Error(w, "user is not an ICQ account", http.StatusBadRequest)
  942. return
  943. }
  944. out := icqProfileHandle{
  945. UIN: user.IdentScreenName.UIN(),
  946. BasicInfo: icqBasicInfoHandle{
  947. Nickname: user.ICQInfo.Basic.Nickname,
  948. FirstName: user.ICQInfo.Basic.FirstName,
  949. LastName: user.ICQInfo.Basic.LastName,
  950. EmailAddress: user.ICQInfo.Basic.EmailAddress,
  951. City: user.ICQInfo.Basic.City,
  952. State: user.ICQInfo.Basic.State,
  953. Phone: user.ICQInfo.Basic.Phone,
  954. Fax: user.ICQInfo.Basic.Fax,
  955. Address: user.ICQInfo.Basic.Address,
  956. CellPhone: user.ICQInfo.Basic.CellPhone,
  957. ZIPCode: user.ICQInfo.Basic.ZIPCode,
  958. CountryCode: user.ICQInfo.Basic.CountryCode,
  959. GMTOffset: user.ICQInfo.Basic.GMTOffset,
  960. PublishEmail: user.ICQInfo.Basic.PublishEmail,
  961. OriginCity: user.ICQInfo.Basic.OriginallyFromCity,
  962. OriginState: user.ICQInfo.Basic.OriginallyFromState,
  963. OriginCountryCode: user.ICQInfo.Basic.OriginallyFromCountryCode,
  964. },
  965. MoreInfo: icqMoreInfoHandle{
  966. Gender: user.ICQInfo.More.Gender,
  967. HomePageAddr: user.ICQInfo.More.HomePageAddr,
  968. BirthYear: user.ICQInfo.More.BirthYear,
  969. BirthMonth: user.ICQInfo.More.BirthMonth,
  970. BirthDay: user.ICQInfo.More.BirthDay,
  971. Lang1: user.ICQInfo.More.Lang1,
  972. Lang2: user.ICQInfo.More.Lang2,
  973. Lang3: user.ICQInfo.More.Lang3,
  974. },
  975. WorkInfo: icqWorkInfoHandle{
  976. Company: user.ICQInfo.Work.Company,
  977. Department: user.ICQInfo.Work.Department,
  978. Position: user.ICQInfo.Work.Position,
  979. OccupationCode: user.ICQInfo.Work.OccupationCode,
  980. Address: user.ICQInfo.Work.Address,
  981. City: user.ICQInfo.Work.City,
  982. State: user.ICQInfo.Work.State,
  983. ZIPCode: user.ICQInfo.Work.ZIPCode,
  984. CountryCode: user.ICQInfo.Work.CountryCode,
  985. Phone: user.ICQInfo.Work.Phone,
  986. Fax: user.ICQInfo.Work.Fax,
  987. WebPage: user.ICQInfo.Work.WebPage,
  988. },
  989. Notes: user.ICQInfo.Notes.Notes,
  990. Interests: icqInterestsHandle{
  991. Code1: user.ICQInfo.Interests.Code1,
  992. Keyword1: user.ICQInfo.Interests.Keyword1,
  993. Code2: user.ICQInfo.Interests.Code2,
  994. Keyword2: user.ICQInfo.Interests.Keyword2,
  995. Code3: user.ICQInfo.Interests.Code3,
  996. Keyword3: user.ICQInfo.Interests.Keyword3,
  997. Code4: user.ICQInfo.Interests.Code4,
  998. Keyword4: user.ICQInfo.Interests.Keyword4,
  999. },
  1000. Affiliations: icqAffiliationsHandle{
  1001. PastCode1: user.ICQInfo.Affiliations.PastCode1,
  1002. PastKeyword1: user.ICQInfo.Affiliations.PastKeyword1,
  1003. PastCode2: user.ICQInfo.Affiliations.PastCode2,
  1004. PastKeyword2: user.ICQInfo.Affiliations.PastKeyword2,
  1005. PastCode3: user.ICQInfo.Affiliations.PastCode3,
  1006. PastKeyword3: user.ICQInfo.Affiliations.PastKeyword3,
  1007. CurrentCode1: user.ICQInfo.Affiliations.CurrentCode1,
  1008. CurrentKeyword1: user.ICQInfo.Affiliations.CurrentKeyword1,
  1009. CurrentCode2: user.ICQInfo.Affiliations.CurrentCode2,
  1010. CurrentKeyword2: user.ICQInfo.Affiliations.CurrentKeyword2,
  1011. CurrentCode3: user.ICQInfo.Affiliations.CurrentCode3,
  1012. CurrentKeyword3: user.ICQInfo.Affiliations.CurrentKeyword3,
  1013. },
  1014. Permissions: icqPermissionsHandle{
  1015. AuthRequired: user.ICQInfo.Permissions.AuthRequired,
  1016. WebAware: user.ICQInfo.Permissions.WebAware,
  1017. AllowSpam: user.ICQInfo.Permissions.AllowSpam,
  1018. },
  1019. }
  1020. if err := json.NewEncoder(w).Encode(out); err != nil {
  1021. http.Error(w, err.Error(), http.StatusInternalServerError)
  1022. }
  1023. }
  1024. // putICQProfileHandler handles the PUT /user/{screenname}/icq endpoint.
  1025. func putICQProfileHandler(w http.ResponseWriter, r *http.Request, mgr ICQProfileManager, logger *slog.Logger) {
  1026. w.Header().Set("Content-Type", "application/json")
  1027. screenName := r.PathValue("screenname")
  1028. user, err := mgr.User(r.Context(), state.NewIdentScreenName(screenName))
  1029. if err != nil {
  1030. logger.Error("error in PUT /user/{screenname}/icq", "err", err.Error())
  1031. http.Error(w, "internal server error", http.StatusInternalServerError)
  1032. return
  1033. }
  1034. if user == nil {
  1035. http.Error(w, "user not found", http.StatusNotFound)
  1036. return
  1037. }
  1038. if !user.IsICQ {
  1039. http.Error(w, "user is not an ICQ account", http.StatusBadRequest)
  1040. return
  1041. }
  1042. var input icqProfileHandle
  1043. d := json.NewDecoder(r.Body)
  1044. if err := d.Decode(&input); err != nil {
  1045. errorMsg(w, "malformed input: "+err.Error(), http.StatusBadRequest)
  1046. return
  1047. }
  1048. // Validate field lengths. ICQ clients typically handle:
  1049. // - String fields: max 127 chars (some fields shorter)
  1050. // - Nickname: max 20 chars
  1051. // - Phone/fax: max 30 chars
  1052. // - Email: max 64 chars
  1053. // - Address/city/state: max 64 chars
  1054. // - Country codes (basic_info.country_code, basic_info.origin_country_code,
  1055. // work_info.country_code): uint16 in JSON, not string length checks
  1056. // - ZIP: max 12 chars
  1057. // - Homepage: max 127 chars
  1058. // - Notes: max 450 chars (v5 limit)
  1059. // - Interest/affiliation keywords: max 64 chars
  1060. // - Company/department/position: max 64 chars
  1061. // - Web page: max 127 chars
  1062. type fieldCheck struct {
  1063. name string
  1064. val string
  1065. max int
  1066. }
  1067. checks := []fieldCheck{
  1068. {"basic_info.nickname", input.BasicInfo.Nickname, 20},
  1069. {"basic_info.first_name", input.BasicInfo.FirstName, 64},
  1070. {"basic_info.last_name", input.BasicInfo.LastName, 64},
  1071. {"basic_info.email", input.BasicInfo.EmailAddress, 64},
  1072. {"basic_info.city", input.BasicInfo.City, 64},
  1073. {"basic_info.state", input.BasicInfo.State, 64},
  1074. {"basic_info.phone", input.BasicInfo.Phone, 30},
  1075. {"basic_info.fax", input.BasicInfo.Fax, 30},
  1076. {"basic_info.address", input.BasicInfo.Address, 64},
  1077. {"basic_info.cell_phone", input.BasicInfo.CellPhone, 30},
  1078. {"basic_info.zip", input.BasicInfo.ZIPCode, 12},
  1079. {"basic_info.origin_city", input.BasicInfo.OriginCity, 64},
  1080. {"basic_info.origin_state", input.BasicInfo.OriginState, 64},
  1081. // basic_info.origin_country_code is uint16 (like basic_info.country_code); validated by JSON binding only.
  1082. {"more_info.homepage", input.MoreInfo.HomePageAddr, 127},
  1083. {"work_info.company", input.WorkInfo.Company, 64},
  1084. {"work_info.department", input.WorkInfo.Department, 64},
  1085. {"work_info.position", input.WorkInfo.Position, 64},
  1086. {"work_info.address", input.WorkInfo.Address, 64},
  1087. {"work_info.city", input.WorkInfo.City, 64},
  1088. {"work_info.state", input.WorkInfo.State, 64},
  1089. {"work_info.zip", input.WorkInfo.ZIPCode, 12},
  1090. {"work_info.phone", input.WorkInfo.Phone, 30},
  1091. {"work_info.fax", input.WorkInfo.Fax, 30},
  1092. {"work_info.web_page", input.WorkInfo.WebPage, 127},
  1093. {"notes", input.Notes, 450},
  1094. {"interests.keyword1", input.Interests.Keyword1, 64},
  1095. {"interests.keyword2", input.Interests.Keyword2, 64},
  1096. {"interests.keyword3", input.Interests.Keyword3, 64},
  1097. {"interests.keyword4", input.Interests.Keyword4, 64},
  1098. {"affiliations.past_keyword1", input.Affiliations.PastKeyword1, 64},
  1099. {"affiliations.past_keyword2", input.Affiliations.PastKeyword2, 64},
  1100. {"affiliations.past_keyword3", input.Affiliations.PastKeyword3, 64},
  1101. {"affiliations.current_keyword1", input.Affiliations.CurrentKeyword1, 64},
  1102. {"affiliations.current_keyword2", input.Affiliations.CurrentKeyword2, 64},
  1103. {"affiliations.current_keyword3", input.Affiliations.CurrentKeyword3, 64},
  1104. }
  1105. var validationErrors []string
  1106. for _, c := range checks {
  1107. if len(c.val) > c.max {
  1108. msg := fmt.Sprintf("field %s exceeds max length of %d (got %d)", c.name, c.max, len(c.val))
  1109. validationErrors = append(validationErrors, msg)
  1110. logger.Warn("ICQ profile field exceeds max length",
  1111. "screenname", screenName,
  1112. "field", c.name,
  1113. "max", c.max,
  1114. "got", len(c.val),
  1115. )
  1116. }
  1117. }
  1118. // Validate gender (0=not specified, 1=female, 2=male)
  1119. if input.MoreInfo.Gender > 2 {
  1120. validationErrors = append(validationErrors, "more_info.gender must be 0 (unspecified), 1 (female), or 2 (male)")
  1121. logger.Warn("ICQ profile invalid gender value", "screenname", screenName, "gender", input.MoreInfo.Gender)
  1122. }
  1123. // Validate birth date
  1124. if input.MoreInfo.BirthMonth > 12 {
  1125. validationErrors = append(validationErrors, "more_info.birth_month must be 0-12")
  1126. logger.Warn("ICQ profile invalid birth month", "screenname", screenName, "birth_month", input.MoreInfo.BirthMonth)
  1127. }
  1128. if input.MoreInfo.BirthDay > 31 {
  1129. validationErrors = append(validationErrors, "more_info.birth_day must be 0-31")
  1130. logger.Warn("ICQ profile invalid birth day", "screenname", screenName, "birth_day", input.MoreInfo.BirthDay)
  1131. }
  1132. if len(validationErrors) > 0 {
  1133. errorMsg(w, strings.Join(validationErrors, "; "), http.StatusBadRequest)
  1134. return
  1135. }
  1136. sn := user.IdentScreenName
  1137. if err := mgr.SetBasicInfo(r.Context(), sn, state.ICQBasicInfo{
  1138. Nickname: input.BasicInfo.Nickname,
  1139. FirstName: input.BasicInfo.FirstName,
  1140. LastName: input.BasicInfo.LastName,
  1141. EmailAddress: input.BasicInfo.EmailAddress,
  1142. City: input.BasicInfo.City,
  1143. State: input.BasicInfo.State,
  1144. Phone: input.BasicInfo.Phone,
  1145. Fax: input.BasicInfo.Fax,
  1146. Address: input.BasicInfo.Address,
  1147. CellPhone: input.BasicInfo.CellPhone,
  1148. ZIPCode: input.BasicInfo.ZIPCode,
  1149. CountryCode: input.BasicInfo.CountryCode,
  1150. GMTOffset: input.BasicInfo.GMTOffset,
  1151. PublishEmail: input.BasicInfo.PublishEmail,
  1152. }); err != nil {
  1153. logger.Error("error setting basic info", "err", err.Error())
  1154. http.Error(w, "internal server error", http.StatusInternalServerError)
  1155. return
  1156. }
  1157. if err := mgr.SetMoreInfo(r.Context(), sn, state.ICQMoreInfo{
  1158. Gender: input.MoreInfo.Gender,
  1159. HomePageAddr: input.MoreInfo.HomePageAddr,
  1160. BirthYear: input.MoreInfo.BirthYear,
  1161. BirthMonth: input.MoreInfo.BirthMonth,
  1162. BirthDay: input.MoreInfo.BirthDay,
  1163. Lang1: input.MoreInfo.Lang1,
  1164. Lang2: input.MoreInfo.Lang2,
  1165. Lang3: input.MoreInfo.Lang3,
  1166. }); err != nil {
  1167. logger.Error("error setting more info", "err", err.Error())
  1168. http.Error(w, "internal server error", http.StatusInternalServerError)
  1169. return
  1170. }
  1171. if err := mgr.SetWorkInfo(r.Context(), sn, state.ICQWorkInfo{
  1172. Company: input.WorkInfo.Company,
  1173. Department: input.WorkInfo.Department,
  1174. Position: input.WorkInfo.Position,
  1175. OccupationCode: input.WorkInfo.OccupationCode,
  1176. Address: input.WorkInfo.Address,
  1177. City: input.WorkInfo.City,
  1178. State: input.WorkInfo.State,
  1179. ZIPCode: input.WorkInfo.ZIPCode,
  1180. CountryCode: input.WorkInfo.CountryCode,
  1181. Phone: input.WorkInfo.Phone,
  1182. Fax: input.WorkInfo.Fax,
  1183. WebPage: input.WorkInfo.WebPage,
  1184. }); err != nil {
  1185. logger.Error("error setting work info", "err", err.Error())
  1186. http.Error(w, "internal server error", http.StatusInternalServerError)
  1187. return
  1188. }
  1189. if err := mgr.SetUserNotes(r.Context(), sn, state.ICQUserNotes{
  1190. Notes: input.Notes,
  1191. }); err != nil {
  1192. logger.Error("error setting notes", "err", err.Error())
  1193. http.Error(w, "internal server error", http.StatusInternalServerError)
  1194. return
  1195. }
  1196. if err := mgr.SetInterests(r.Context(), sn, state.ICQInterests{
  1197. Code1: input.Interests.Code1,
  1198. Keyword1: input.Interests.Keyword1,
  1199. Code2: input.Interests.Code2,
  1200. Keyword2: input.Interests.Keyword2,
  1201. Code3: input.Interests.Code3,
  1202. Keyword3: input.Interests.Keyword3,
  1203. Code4: input.Interests.Code4,
  1204. Keyword4: input.Interests.Keyword4,
  1205. }); err != nil {
  1206. logger.Error("error setting interests", "err", err.Error())
  1207. http.Error(w, "internal server error", http.StatusInternalServerError)
  1208. return
  1209. }
  1210. if err := mgr.SetAffiliations(r.Context(), sn, state.ICQAffiliations{
  1211. PastCode1: input.Affiliations.PastCode1,
  1212. PastKeyword1: input.Affiliations.PastKeyword1,
  1213. PastCode2: input.Affiliations.PastCode2,
  1214. PastKeyword2: input.Affiliations.PastKeyword2,
  1215. PastCode3: input.Affiliations.PastCode3,
  1216. PastKeyword3: input.Affiliations.PastKeyword3,
  1217. CurrentCode1: input.Affiliations.CurrentCode1,
  1218. CurrentKeyword1: input.Affiliations.CurrentKeyword1,
  1219. CurrentCode2: input.Affiliations.CurrentCode2,
  1220. CurrentKeyword2: input.Affiliations.CurrentKeyword2,
  1221. CurrentCode3: input.Affiliations.CurrentCode3,
  1222. CurrentKeyword3: input.Affiliations.CurrentKeyword3,
  1223. }); err != nil {
  1224. logger.Error("error setting affiliations", "err", err.Error())
  1225. http.Error(w, "internal server error", http.StatusInternalServerError)
  1226. return
  1227. }
  1228. if err := mgr.SetPermissions(r.Context(), sn, state.ICQPermissions{
  1229. AuthRequired: input.Permissions.AuthRequired,
  1230. WebAware: input.Permissions.WebAware,
  1231. AllowSpam: input.Permissions.AllowSpam,
  1232. }); err != nil {
  1233. logger.Error("error setting permissions", "err", err.Error())
  1234. http.Error(w, "internal server error", http.StatusInternalServerError)
  1235. return
  1236. }
  1237. merged, err := mgr.User(r.Context(), sn)
  1238. if err != nil {
  1239. logger.Error("error reloading user after ICQ profile update", "err", err.Error())
  1240. http.Error(w, "internal server error", http.StatusInternalServerError)
  1241. return
  1242. }
  1243. if merged == nil {
  1244. http.Error(w, "user not found", http.StatusNotFound)
  1245. return
  1246. }
  1247. merged.ICQInfo.Basic.OriginallyFromCity = input.BasicInfo.OriginCity
  1248. merged.ICQInfo.Basic.OriginallyFromState = input.BasicInfo.OriginState
  1249. merged.ICQInfo.Basic.OriginallyFromCountryCode = input.BasicInfo.OriginCountryCode
  1250. if err := mgr.SetICQInfo(r.Context(), sn, merged.ICQInfo); err != nil {
  1251. logger.Error("error setting ICQ info (origin fields)", "err", err.Error())
  1252. http.Error(w, "internal server error", http.StatusInternalServerError)
  1253. return
  1254. }
  1255. w.WriteHeader(http.StatusNoContent)
  1256. }
  1257. // errorMsg sends an error response message and code.
  1258. func errorMsg(w http.ResponseWriter, error string, code int) {
  1259. msg := messageBody{Message: error}
  1260. w.WriteHeader(code)
  1261. if err := json.NewEncoder(w).Encode(msg); err != nil {
  1262. http.Error(w, err.Error(), http.StatusInternalServerError)
  1263. }
  1264. }
  1265. // BARTAsset represents a BART asset entry.
  1266. type BARTAsset struct {
  1267. Hash string `json:"hash"`
  1268. Type uint16 `json:"type"`
  1269. }
  1270. // getBARTByTypeHandler handles the GET /bart endpoint.
  1271. func getBARTByTypeHandler(w http.ResponseWriter, r *http.Request, bartAssetManager BARTAssetManager, logger *slog.Logger) {
  1272. w.Header().Set("Content-Type", "application/json")
  1273. // Get type from query parameter (required)
  1274. typeStr := r.URL.Query().Get("type")
  1275. if typeStr == "" {
  1276. errorMsg(w, "type query parameter is required", http.StatusBadRequest)
  1277. return
  1278. }
  1279. typeVal, err := strconv.ParseUint(typeStr, 10, 16)
  1280. if err != nil {
  1281. errorMsg(w, "invalid type ID", http.StatusBadRequest)
  1282. return
  1283. }
  1284. itemType := uint16(typeVal)
  1285. // Get BART items, filtered by type
  1286. items, err := bartAssetManager.ListBARTItems(r.Context(), itemType)
  1287. if err != nil {
  1288. logger.Error("error listing BART items", "err", err.Error())
  1289. errorMsg(w, "internal server error", http.StatusInternalServerError)
  1290. return
  1291. }
  1292. // Convert to BARTAsset format
  1293. assets := make([]BARTAsset, 0, len(items))
  1294. for _, item := range items {
  1295. assets = append(assets, BARTAsset{
  1296. Hash: item.Hash,
  1297. Type: item.Type,
  1298. })
  1299. }
  1300. if err := json.NewEncoder(w).Encode(assets); err != nil {
  1301. logger.Error("error encoding response", "err", err.Error())
  1302. }
  1303. }
  1304. // getBARTHandler handles the GET /bart/{hash} endpoint.
  1305. func getBARTHandler(w http.ResponseWriter, r *http.Request, bartAssetManager BARTAssetManager, logger *slog.Logger) {
  1306. hashStr := r.PathValue("hash")
  1307. if hashStr == "" {
  1308. errorMsg(w, "hash is required", http.StatusBadRequest)
  1309. return
  1310. }
  1311. hashBytes, err := hex.DecodeString(hashStr)
  1312. if err != nil {
  1313. errorMsg(w, "invalid hash format", http.StatusBadRequest)
  1314. return
  1315. }
  1316. body, err := bartAssetManager.BARTItem(r.Context(), hashBytes)
  1317. if err != nil {
  1318. logger.Error("error retrieving BART asset", "err", err.Error())
  1319. errorMsg(w, "internal server error", http.StatusInternalServerError)
  1320. return
  1321. }
  1322. if len(body) == 0 {
  1323. errorMsg(w, "BART asset not found", http.StatusNotFound)
  1324. return
  1325. }
  1326. w.Header().Set("Content-Type", "application/octet-stream")
  1327. _, _ = w.Write(body)
  1328. }
  1329. // postBARTHandler handles the POST /bart endpoint.
  1330. func postBARTHandler(w http.ResponseWriter, r *http.Request, bartAssetManager BARTAssetManager, logger *slog.Logger) {
  1331. w.Header().Set("Content-Type", "application/json")
  1332. // Extract hash from URL path
  1333. hashStr := r.PathValue("hash")
  1334. if hashStr == "" {
  1335. errorMsg(w, "hash path parameter is required", http.StatusBadRequest)
  1336. return
  1337. }
  1338. hashBytes, err := hex.DecodeString(hashStr)
  1339. if err != nil {
  1340. errorMsg(w, "invalid hash format", http.StatusBadRequest)
  1341. return
  1342. }
  1343. typeStr := r.URL.Query().Get("type")
  1344. if typeStr == "" {
  1345. errorMsg(w, "type query parameter is required", http.StatusBadRequest)
  1346. return
  1347. }
  1348. typeVal, err := strconv.ParseUint(typeStr, 10, 16)
  1349. if err != nil {
  1350. errorMsg(w, "invalid type ID", http.StatusBadRequest)
  1351. return
  1352. }
  1353. bartType := uint16(typeVal)
  1354. data, err := io.ReadAll(r.Body)
  1355. if err != nil {
  1356. errorMsg(w, "failed to read request body", http.StatusBadRequest)
  1357. return
  1358. }
  1359. if err := bartAssetManager.InsertBARTItem(r.Context(), hashBytes, data, bartType); err != nil {
  1360. if errors.Is(err, state.ErrBARTItemExists) {
  1361. errorMsg(w, "BART asset already exists", http.StatusConflict)
  1362. return
  1363. }
  1364. logger.Error("error in POST /bart", "err", err.Error())
  1365. errorMsg(w, "internal server error", http.StatusInternalServerError)
  1366. return
  1367. }
  1368. w.WriteHeader(http.StatusCreated)
  1369. response := BARTAsset{
  1370. Hash: hex.EncodeToString(hashBytes),
  1371. Type: bartType,
  1372. }
  1373. _ = json.NewEncoder(w).Encode(response)
  1374. }
  1375. // deleteBARTHandler handles the DELETE /bart/{hash} endpoint.
  1376. func deleteBARTHandler(w http.ResponseWriter, r *http.Request, bartAssetManager BARTAssetManager, logger *slog.Logger) {
  1377. w.Header().Set("Content-Type", "application/json")
  1378. // Extract hash from URL path
  1379. hashStr := r.PathValue("hash")
  1380. if hashStr == "" {
  1381. errorMsg(w, "hash path parameter is required", http.StatusBadRequest)
  1382. return
  1383. }
  1384. hashBytes, err := hex.DecodeString(hashStr)
  1385. if err != nil {
  1386. errorMsg(w, "invalid hash format", http.StatusBadRequest)
  1387. return
  1388. }
  1389. if err := bartAssetManager.DeleteBARTItem(r.Context(), hashBytes); err != nil {
  1390. if errors.Is(err, state.ErrBARTItemNotFound) {
  1391. errorMsg(w, "BART asset not found", http.StatusNotFound)
  1392. return
  1393. }
  1394. logger.Error("error in DELETE /bart", "err", err.Error())
  1395. errorMsg(w, "internal server error", http.StatusInternalServerError)
  1396. return
  1397. }
  1398. msg := messageBody{Message: "BART asset deleted successfully."}
  1399. _ = json.NewEncoder(w).Encode(msg)
  1400. }
  1401. // getFeedbagBuddyHandler handles the GET /feedbag/{screen_name}/group endpoint.
  1402. func getFeedbagBuddyHandler(w http.ResponseWriter, r *http.Request, feedbagManager FeedbagManager, logger *slog.Logger) {
  1403. w.Header().Set("Content-Type", "application/json")
  1404. screenName := r.PathValue("screen_name")
  1405. if screenName == "" {
  1406. errorMsg(w, "screen_name is required", http.StatusBadRequest)
  1407. return
  1408. }
  1409. items, err := feedbagManager.Feedbag(r.Context(), state.NewIdentScreenName(screenName))
  1410. if err != nil {
  1411. logger.Error("error retrieving feedbag", "err", err.Error())
  1412. errorMsg(w, "internal server error", http.StatusInternalServerError)
  1413. return
  1414. }
  1415. if len(items) == 0 {
  1416. errorMsg(w, "feedbag not found", http.StatusNotFound)
  1417. return
  1418. }
  1419. buddyMap := make(map[uint16][]*wire.FeedbagItem)
  1420. for _, item := range items {
  1421. switch item.ClassID {
  1422. case wire.FeedbagClassIdBuddy:
  1423. buddyMap[item.GroupID] = append(buddyMap[item.GroupID], &item)
  1424. }
  1425. }
  1426. type buddyItem struct {
  1427. Name string `json:"name"`
  1428. ItemID uint16 `json:"item_id"`
  1429. }
  1430. type groupItem struct {
  1431. feedbagGroupHandle
  1432. Buddies []buddyItem `json:"buddies"`
  1433. }
  1434. response := make([]groupItem, 0)
  1435. for _, item := range items {
  1436. switch item.ClassID {
  1437. case wire.FeedbagClassIdGroup:
  1438. if item.GroupID == 0 {
  1439. // can't add buddies to the root group
  1440. continue
  1441. }
  1442. group := groupItem{
  1443. feedbagGroupHandle: feedbagGroupFromItem(item),
  1444. Buddies: make([]buddyItem, 0, len(buddyMap[item.GroupID])),
  1445. }
  1446. for _, buddy := range buddyMap[item.GroupID] {
  1447. group.Buddies = append(group.Buddies, buddyItem{
  1448. Name: buddy.Name,
  1449. ItemID: buddy.ItemID,
  1450. })
  1451. }
  1452. response = append(response, group)
  1453. }
  1454. }
  1455. if err := json.NewEncoder(w).Encode(response); err != nil {
  1456. logger.Error("error encoding response", "err", err.Error())
  1457. }
  1458. }
  1459. // putFeedbagGroupHandler handles the PUT /feedbag/{screen_name}/group/{group_name} endpoint.
  1460. func putFeedbagGroupHandler(w http.ResponseWriter, r *http.Request, feedbagManager FeedbagManager, messageRelayer MessageRelayer, logger *slog.Logger, intn func(n int) int) {
  1461. w.Header().Set("Content-Type", "application/json")
  1462. screenName := r.PathValue("screen_name")
  1463. if screenName == "" {
  1464. errorMsg(w, "screen_name is required", http.StatusBadRequest)
  1465. return
  1466. }
  1467. groupName := r.PathValue("group_name")
  1468. if groupName == "" {
  1469. errorMsg(w, "group_name is required", http.StatusBadRequest)
  1470. return
  1471. }
  1472. user := state.NewIdentScreenName(screenName)
  1473. items, err := feedbagManager.Feedbag(r.Context(), user)
  1474. if err != nil {
  1475. logger.Error("error retrieving feedbag", "err", err.Error())
  1476. errorMsg(w, "internal server error", http.StatusInternalServerError)
  1477. return
  1478. }
  1479. var groupItem wire.FeedbagItem
  1480. var alreadyExist bool
  1481. for _, item := range items {
  1482. if item.ClassID == wire.FeedbagClassIdGroup && item.Name == groupName {
  1483. alreadyExist = true
  1484. groupItem = item
  1485. break
  1486. }
  1487. }
  1488. if !alreadyExist {
  1489. fb := state.NewFeedbagList(items, intn)
  1490. groupItem = fb.AddGroup(groupName)
  1491. pending := fb.PendingUpdates()
  1492. if len(pending) > 0 {
  1493. if err := feedbagManager.FeedbagUpsert(r.Context(), user, pending); err != nil {
  1494. logger.Error("error inserting feedbag item", "err", err.Error())
  1495. errorMsg(w, "internal server error", http.StatusInternalServerError)
  1496. return
  1497. }
  1498. messageRelayer.RelayToScreenName(r.Context(), user, wire.SNACMessage{
  1499. Frame: wire.SNACFrame{
  1500. FoodGroup: wire.Feedbag,
  1501. SubGroup: wire.FeedbagInsertItem,
  1502. RequestID: wire.ReqIDFromServer,
  1503. },
  1504. Body: wire.SNAC_0x13_0x09_FeedbagUpdateItem{
  1505. Items: pending,
  1506. },
  1507. })
  1508. }
  1509. }
  1510. if alreadyExist {
  1511. w.WriteHeader(http.StatusOK)
  1512. } else {
  1513. w.WriteHeader(http.StatusCreated)
  1514. }
  1515. if err := json.NewEncoder(w).Encode(feedbagGroupFromItem(groupItem)); err != nil {
  1516. logger.Error("error encoding response", "err", err.Error())
  1517. }
  1518. }
  1519. // putFeedbagBuddyHandler handles the PUT /feedbag/{screen_name}/group/{group_id}/buddy/{buddy_screen_name} endpoint.
  1520. func putFeedbagBuddyHandler(w http.ResponseWriter, r *http.Request, buddyBroadcaster BuddyBroadcaster, feedbagManager FeedbagManager, sessionRetriever SessionRetriever, messageRelayer MessageRelayer, logger *slog.Logger, randInt func(n int) int) {
  1521. w.Header().Set("Content-Type", "application/json")
  1522. gid, err := strconv.ParseUint(r.PathValue("group_id"), 10, 16)
  1523. if err != nil {
  1524. errorMsg(w, "invalid group_id", http.StatusBadRequest)
  1525. return
  1526. }
  1527. groupID := uint16(gid)
  1528. if groupID == 0 {
  1529. errorMsg(w, "can't add buddies to root group", http.StatusBadRequest)
  1530. return
  1531. }
  1532. screenName := r.PathValue("screen_name")
  1533. if screenName == "" {
  1534. errorMsg(w, "screen_name is required", http.StatusBadRequest)
  1535. return
  1536. }
  1537. me := state.NewIdentScreenName(screenName)
  1538. buddyScreenName := r.PathValue("buddy_screen_name")
  1539. if buddyScreenName == "" {
  1540. errorMsg(w, "buddy_screen_name is required", http.StatusBadRequest)
  1541. return
  1542. }
  1543. newBuddy := state.DisplayScreenName(buddyScreenName)
  1544. if newBuddy.IsUIN() {
  1545. if err := newBuddy.ValidateUIN(); err != nil {
  1546. errorMsg(w, fmt.Sprintf("invalid uin: %s", err), http.StatusBadRequest)
  1547. return
  1548. }
  1549. } else {
  1550. if err := newBuddy.ValidateAIMHandle(); err != nil {
  1551. errorMsg(w, fmt.Sprintf("invalid screen name: %s", err), http.StatusBadRequest)
  1552. return
  1553. }
  1554. }
  1555. items, err := feedbagManager.Feedbag(r.Context(), me)
  1556. if err != nil {
  1557. logger.Error("error retrieving feedbag", "err", err.Error())
  1558. errorMsg(w, "internal server error", http.StatusInternalServerError)
  1559. return
  1560. }
  1561. var group *wire.FeedbagItem
  1562. count := 0
  1563. for _, item := range items {
  1564. switch {
  1565. case item.ClassID == wire.FeedbagClassIdGroup && item.GroupID == groupID:
  1566. group = &item
  1567. case item.ClassID == wire.FeedbagClassIdBuddy && item.GroupID == groupID:
  1568. count++
  1569. if item.Name == newBuddy.IdentScreenName().String() {
  1570. response := struct {
  1571. Name string `json:"name"`
  1572. GroupID uint16 `json:"group_id"`
  1573. ItemID uint16 `json:"item_id"`
  1574. }{
  1575. Name: buddyScreenName,
  1576. GroupID: groupID,
  1577. ItemID: item.ItemID,
  1578. }
  1579. w.WriteHeader(http.StatusOK)
  1580. if err := json.NewEncoder(w).Encode(response); err != nil {
  1581. logger.Error("error encoding response", "err", err.Error())
  1582. }
  1583. return
  1584. }
  1585. }
  1586. }
  1587. if count >= 30 {
  1588. errorMsg(w, "too many buddies in group. max: 30", http.StatusBadRequest)
  1589. return
  1590. }
  1591. if group == nil {
  1592. errorMsg(w, "group not found", http.StatusNotFound)
  1593. return
  1594. }
  1595. buddyItem := wire.FeedbagItem{
  1596. Name: buddyScreenName,
  1597. GroupID: groupID,
  1598. ItemID: randItemID(randInt, items),
  1599. ClassID: wire.FeedbagClassIdBuddy,
  1600. }
  1601. if buddyItem.ItemID == 0 {
  1602. errorMsg(w, "maximum items reached", http.StatusConflict)
  1603. return
  1604. }
  1605. group.AppendOrderMembers(buddyItem.ItemID)
  1606. updates := []wire.FeedbagItem{
  1607. buddyItem,
  1608. *group,
  1609. }
  1610. if err := feedbagManager.FeedbagUpsert(r.Context(), me, updates); err != nil {
  1611. logger.Error("error inserting feedbag item", "err", err.Error())
  1612. errorMsg(w, "internal server error", http.StatusInternalServerError)
  1613. return
  1614. }
  1615. session := sessionRetriever.RetrieveSession(me)
  1616. if session != nil {
  1617. messageRelayer.RelayToScreenName(r.Context(), me, wire.SNACMessage{
  1618. Frame: wire.SNACFrame{
  1619. FoodGroup: wire.Feedbag,
  1620. SubGroup: wire.FeedbagInsertItem,
  1621. RequestID: wire.ReqIDFromServer,
  1622. },
  1623. Body: wire.SNAC_0x13_0x09_FeedbagUpdateItem{
  1624. Items: []wire.FeedbagItem{buddyItem},
  1625. },
  1626. })
  1627. messageRelayer.RelayToScreenName(r.Context(), me, wire.SNACMessage{
  1628. Frame: wire.SNACFrame{
  1629. FoodGroup: wire.Feedbag,
  1630. SubGroup: wire.FeedbagUpdateItem,
  1631. RequestID: wire.ReqIDFromServer,
  1632. },
  1633. Body: wire.SNAC_0x13_0x09_FeedbagUpdateItem{
  1634. Items: []wire.FeedbagItem{*group},
  1635. },
  1636. })
  1637. instances := session.Instances()
  1638. if len(instances) > 0 {
  1639. if err := buddyBroadcaster.BroadcastVisibility(r.Context(), instances[0], []state.IdentScreenName{newBuddy.IdentScreenName()}, false); err != nil {
  1640. logger.Error("error broadcasting visibility", "err", err.Error())
  1641. }
  1642. }
  1643. }
  1644. response := struct {
  1645. Name string `json:"name"`
  1646. GroupID uint16 `json:"group_id"`
  1647. ItemID uint16 `json:"item_id"`
  1648. }{
  1649. Name: buddyItem.Name,
  1650. GroupID: buddyItem.GroupID,
  1651. ItemID: buddyItem.ItemID,
  1652. }
  1653. w.WriteHeader(http.StatusOK)
  1654. if err := json.NewEncoder(w).Encode(response); err != nil {
  1655. logger.Error("error encoding response", "err", err.Error())
  1656. }
  1657. }
  1658. func randItemID(randInt func(n int) int, items []wire.FeedbagItem) uint16 {
  1659. num := uint16(randInt(math.MaxUint16))
  1660. for itemID := num; itemID != num-1; itemID++ {
  1661. if itemID == 0 {
  1662. continue
  1663. }
  1664. exists := false
  1665. for _, item := range items {
  1666. if item.GroupID == itemID || item.ItemID == itemID {
  1667. exists = true
  1668. break
  1669. }
  1670. }
  1671. if !exists {
  1672. return itemID
  1673. }
  1674. }
  1675. return 0
  1676. }
  1677. // deleteFeedbagBuddyHandler handles the DELETE /feedbag/{screen_name}/group/{group_id}/buddy/{buddy_screen_name} endpoint.
  1678. func deleteFeedbagBuddyHandler(w http.ResponseWriter, r *http.Request, buddyBroadcaster BuddyBroadcaster, feedbagManager FeedbagManager, sessionRetriever SessionRetriever, messageRelayer MessageRelayer, logger *slog.Logger) {
  1679. gid, err := strconv.ParseUint(r.PathValue("group_id"), 10, 16)
  1680. if err != nil {
  1681. errorMsg(w, "invalid group_id", http.StatusBadRequest)
  1682. return
  1683. }
  1684. groupID := uint16(gid)
  1685. if groupID == 0 {
  1686. errorMsg(w, "can't add buddies to root group", http.StatusBadRequest)
  1687. return
  1688. }
  1689. screenName := r.PathValue("screen_name")
  1690. if screenName == "" {
  1691. errorMsg(w, "screen_name is required", http.StatusBadRequest)
  1692. return
  1693. }
  1694. me := state.NewIdentScreenName(screenName)
  1695. buddyScreenName := r.PathValue("buddy_screen_name")
  1696. if buddyScreenName == "" {
  1697. errorMsg(w, "buddy_screen_name is required", http.StatusBadRequest)
  1698. return
  1699. }
  1700. deleteBuddy := state.NewIdentScreenName(buddyScreenName)
  1701. items, err := feedbagManager.Feedbag(r.Context(), me)
  1702. if err != nil {
  1703. logger.Error("error retrieving feedbag", "err", err.Error())
  1704. errorMsg(w, "internal server error", http.StatusInternalServerError)
  1705. return
  1706. }
  1707. var itemToDelete *wire.FeedbagItem
  1708. var parentGroup *wire.FeedbagItem
  1709. for _, item := range items {
  1710. switch {
  1711. case item.ClassID == wire.FeedbagClassIdGroup && item.GroupID == groupID:
  1712. parentGroup = &item
  1713. case item.ClassID == wire.FeedbagClassIdBuddy && item.Name == buddyScreenName && item.GroupID == groupID:
  1714. itemToDelete = &item
  1715. }
  1716. }
  1717. switch {
  1718. case parentGroup == nil:
  1719. errorMsg(w, "group not found", http.StatusNotFound)
  1720. return
  1721. case itemToDelete == nil:
  1722. errorMsg(w, "buddy not found", http.StatusNotFound)
  1723. return
  1724. }
  1725. // Remove the buddy from the parent group's order TLV (same as TOC feedbag_list.DeleteBuddy).
  1726. parentGroup.RemoveOrderMembers(itemToDelete.ItemID)
  1727. if err := feedbagManager.FeedbagUpsert(r.Context(), me, []wire.FeedbagItem{*parentGroup}); err != nil {
  1728. logger.Error("error updating feedbag group order", "err", err.Error())
  1729. errorMsg(w, "internal server error", http.StatusInternalServerError)
  1730. return
  1731. }
  1732. if err := feedbagManager.FeedbagDelete(r.Context(), me, []wire.FeedbagItem{*itemToDelete}); err != nil {
  1733. logger.Error("error deleting feedbag item", "err", err.Error())
  1734. errorMsg(w, "internal server error", http.StatusInternalServerError)
  1735. return
  1736. }
  1737. session := sessionRetriever.RetrieveSession(me)
  1738. if session != nil {
  1739. messageRelayer.RelayToScreenName(r.Context(), me, wire.SNACMessage{
  1740. Frame: wire.SNACFrame{
  1741. FoodGroup: wire.Feedbag,
  1742. SubGroup: wire.FeedbagDeleteItem,
  1743. RequestID: wire.ReqIDFromServer,
  1744. },
  1745. Body: wire.SNAC_0x13_0x0A_FeedbagDeleteItem{
  1746. Items: []wire.FeedbagItem{*itemToDelete},
  1747. },
  1748. })
  1749. messageRelayer.RelayToScreenName(r.Context(), me, wire.SNACMessage{
  1750. Frame: wire.SNACFrame{
  1751. FoodGroup: wire.Feedbag,
  1752. SubGroup: wire.FeedbagUpdateItem,
  1753. RequestID: wire.ReqIDFromServer,
  1754. },
  1755. Body: wire.SNAC_0x13_0x09_FeedbagUpdateItem{
  1756. Items: []wire.FeedbagItem{*parentGroup},
  1757. },
  1758. })
  1759. instances := session.Instances()
  1760. if len(instances) > 0 {
  1761. if err := buddyBroadcaster.BroadcastVisibility(r.Context(), instances[0], []state.IdentScreenName{deleteBuddy}, true); err != nil {
  1762. logger.Error("error broadcasting visibility", "err", err.Error())
  1763. }
  1764. }
  1765. }
  1766. w.WriteHeader(http.StatusNoContent)
  1767. }