chat_nav.go 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201
  1. package server
  2. import (
  3. "errors"
  4. "fmt"
  5. "github.com/google/uuid"
  6. "github.com/mkaminski/goaim/oscar"
  7. "io"
  8. "time"
  9. )
  10. const (
  11. ChatNavErr uint16 = 0x0001
  12. ChatNavRequestChatRights = 0x0002
  13. ChatNavRequestExchangeInfo = 0x0003
  14. ChatNavRequestRoomInfo = 0x0004
  15. ChatNavRequestMoreRoomInfo = 0x0005
  16. ChatNavRequestOccupantList = 0x0006
  17. ChatNavSearchForRoom = 0x0007
  18. ChatNavCreateRoom = 0x0008
  19. ChatNavNavInfo = 0x0009
  20. )
  21. func routeChatNav(sess *Session, cr *ChatRegistry, snac oscar.SnacFrame, r io.Reader, w io.Writer, sequence *uint32) error {
  22. switch snac.SubGroup {
  23. case ChatNavRequestChatRights:
  24. return SendAndReceiveNextChatRights(snac, w, sequence)
  25. case ChatNavRequestRoomInfo:
  26. return SendAndReceiveRequestRoomInfo(cr, snac, r, w, sequence)
  27. case ChatNavCreateRoom:
  28. return SendAndReceiveCreateRoom(sess, cr, NewChatRoom, snac, r, w, sequence)
  29. default:
  30. return ErrUnsupportedSubGroup
  31. }
  32. }
  33. type ChatCookie struct {
  34. Cookie []byte `len_prefix:"uint16"`
  35. SessID string `len_prefix:"uint16"`
  36. }
  37. func SendAndReceiveNextChatRights(snac oscar.SnacFrame, w io.Writer, sequence *uint32) error {
  38. fmt.Printf("sendAndReceiveNextChatRights read SNAC frame: %+v\n", snac)
  39. snacFrameOut := oscar.SnacFrame{
  40. FoodGroup: CHAT_NAV,
  41. SubGroup: ChatNavNavInfo,
  42. }
  43. snacPayloadOut := oscar.SNAC_0x0D_0x09_ChatNavNavInfo{
  44. TLVRestBlock: oscar.TLVRestBlock{
  45. TLVList: oscar.TLVList{
  46. {
  47. TType: 0x02,
  48. Val: uint8(10),
  49. },
  50. {
  51. TType: 0x03,
  52. Val: oscar.SNAC_0x0D_0x09_TLVExchangeInfo{
  53. Identifier: 4,
  54. TLVBlock: oscar.TLVBlock{
  55. TLVList: oscar.TLVList{
  56. {
  57. TType: 0x0002,
  58. Val: uint16(0x0010),
  59. },
  60. {
  61. TType: 0x00c9,
  62. Val: uint16(15),
  63. },
  64. {
  65. TType: 0x00d3,
  66. Val: "default Exchange",
  67. },
  68. {
  69. TType: 0x00d5,
  70. Val: uint8(2),
  71. },
  72. {
  73. TType: 0xd6,
  74. Val: "us-ascii",
  75. },
  76. {
  77. TType: 0xd7,
  78. Val: "en",
  79. },
  80. {
  81. TType: 0xd8,
  82. Val: "us-ascii",
  83. },
  84. {
  85. TType: 0xd9,
  86. Val: "en",
  87. },
  88. },
  89. },
  90. },
  91. },
  92. },
  93. },
  94. }
  95. return writeOutSNAC(snac, snacFrameOut, snacPayloadOut, sequence, w)
  96. }
  97. func NewChatRoom() ChatRoom {
  98. return ChatRoom{
  99. Cookie: uuid.New().String(),
  100. CreateTime: time.Now(),
  101. SessionManager: NewSessionManager(),
  102. }
  103. }
  104. type ChatRoomFactory func() ChatRoom
  105. func SendAndReceiveCreateRoom(sess *Session, cr *ChatRegistry, newChatRoom ChatRoomFactory, snac oscar.SnacFrame, r io.Reader, w io.Writer, sequence *uint32) error {
  106. snacPayloadIn := oscar.SNAC_0x0E_0x02_ChatRoomInfoUpdate{}
  107. if err := oscar.Unmarshal(&snacPayloadIn, r); err != nil {
  108. return err
  109. }
  110. name, hasName := snacPayloadIn.GetString(oscar.ChatTLVRoomName)
  111. if !hasName {
  112. return errors.New("unable to find chat name")
  113. }
  114. room := newChatRoom()
  115. room.DetailLevel = snacPayloadIn.DetailLevel
  116. room.Exchange = snacPayloadIn.Exchange
  117. room.InstanceNumber = snacPayloadIn.InstanceNumber
  118. room.Name = name
  119. cr.Register(room)
  120. // add user to chat room
  121. room.NewSessionWithSN(sess.ID, sess.ScreenName)
  122. snacFrameOut := oscar.SnacFrame{
  123. FoodGroup: CHAT_NAV,
  124. SubGroup: ChatNavNavInfo,
  125. }
  126. snacPayloadOut := oscar.SNAC_0x0D_0x09_ChatNavNavInfo{
  127. TLVRestBlock: oscar.TLVRestBlock{
  128. TLVList: oscar.TLVList{
  129. {
  130. TType: oscar.ChatNavTLVRoomInfo,
  131. Val: oscar.SNAC_0x0E_0x02_ChatRoomInfoUpdate{
  132. Exchange: snacPayloadIn.Exchange,
  133. Cookie: room.Cookie,
  134. InstanceNumber: snacPayloadIn.InstanceNumber,
  135. DetailLevel: snacPayloadIn.DetailLevel,
  136. TLVBlock: oscar.TLVBlock{
  137. TLVList: room.TLVList(),
  138. },
  139. },
  140. },
  141. },
  142. },
  143. }
  144. return writeOutSNAC(snac, snacFrameOut, snacPayloadOut, sequence, w)
  145. }
  146. func SendAndReceiveRequestRoomInfo(cr *ChatRegistry, snac oscar.SnacFrame, r io.Reader, w io.Writer, sequence *uint32) error {
  147. fmt.Printf("SendAndReceiveRequestRoomInfo read SNAC frame: %+v\n", snac)
  148. snacPayloadIn := oscar.SNAC_0x0D_0x04_ChatNavRequestRoomInfo{}
  149. if err := oscar.Unmarshal(&snacPayloadIn, r); err != nil {
  150. return err
  151. }
  152. room, err := cr.Retrieve(string(snacPayloadIn.Cookie))
  153. if err != nil {
  154. return err
  155. }
  156. snacFrameOut := oscar.SnacFrame{
  157. FoodGroup: CHAT_NAV,
  158. SubGroup: ChatNavNavInfo,
  159. }
  160. snacPayloadOut := oscar.SNAC_0x0D_0x09_ChatNavNavInfo{
  161. TLVRestBlock: oscar.TLVRestBlock{
  162. TLVList: oscar.TLVList{
  163. {
  164. TType: 0x04,
  165. Val: oscar.SNAC_0x0E_0x02_ChatRoomInfoUpdate{
  166. Exchange: 4,
  167. Cookie: room.Cookie,
  168. InstanceNumber: 100,
  169. DetailLevel: 2,
  170. TLVBlock: oscar.TLVBlock{
  171. TLVList: room.TLVList(),
  172. },
  173. },
  174. },
  175. },
  176. },
  177. }
  178. return writeOutSNAC(snac, snacFrameOut, snacPayloadOut, sequence, w)
  179. }