package socket import ( "fmt" "reflect" "github.com/beevik/etree" "github.com/ectrc/snow/aid" "github.com/ectrc/snow/person" "github.com/gofiber/contrib/websocket" ) type JabberData struct { JabberID string LastPresence string } var jabberHandlers = map[string]func(*Socket[JabberData], *etree.Document) error { "open": jabberOpenHandler, "iq": jabberIqRootHandler, "presence": jabberPresenceHandler, "message": jabberMessageHandler, } func HandleNewJabberSocket(identifier string) { socket, ok := JabberSockets.Get(identifier) if !ok { return } defer JabberSockets.Delete(socket.ID) for { _, message, failed := socket.Connection.ReadMessage() if failed != nil { break } aid.Print(string(message)) parsed := etree.NewDocument() if err := parsed.ReadFromBytes(message); err != nil { return } if handler, ok := jabberHandlers[parsed.Root().Tag]; ok { if err := handler(socket, parsed); err != nil { socket.Connection.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, err.Error())) return } } } } func jabberOpenHandler(socket *Socket[JabberData], parsed *etree.Document) error { socket.Write([]byte(``)) socket.Write([]byte(``)) return nil } func jabberIqRootHandler(socket *Socket[JabberData], parsed *etree.Document) error { redirect := map[string]func(*Socket[JabberData], *etree.Document) error { "set": jabberIqSetHandler, "get": jabberIqGetHandler, } if handler, ok := redirect[parsed.Root().SelectAttr("type").Value]; ok { if err := handler(socket, parsed); err != nil { return err } } return nil } func jabberIqSetHandler(socket *Socket[JabberData], parsed *etree.Document) error { snowId, err := aid.GetSnowFromToken(parsed.FindElement("/iq/query/password").Text()) if err != nil { return err } person := person.Find(snowId) if person == nil { return fmt.Errorf("person not found") } JabberSockets.ChangeKey(socket.ID, person.ID) socket.ID = person.ID socket.Person = person socket.Data.JabberID = snowId + "@prod.ol.epicgames.com/" + parsed.FindElement("/iq/query/resource").Text() socket.Write([]byte(``)) return nil } func jabberIqGetHandler(socket *Socket[JabberData], parsed *etree.Document) error { socket.Write([]byte(``)) socket.JabberNotifyFriends() return nil } func jabberPresenceHandler(socket *Socket[JabberData], parsed *etree.Document) error { socket.Data.LastPresence = parsed.FindElement("/presence/status").Text() socket.JabberNotifyFriends() return nil } func jabberMessageHandler(socket *Socket[JabberData], parsed *etree.Document) error { return nil } func (s *Socket[T]) JabberSendMessageToPerson(data aid.JSON) { if reflect.TypeOf(s.Data) != reflect.TypeOf(&JabberData{}) { return } jabberSocket, ok := JabberSockets.Get(s.ID) if !ok { aid.Print("jabber socket not found even though it should be") return } s.Write([]byte(` `+ string(data.ToBytes()) +` `)) } func (s *Socket[T]) JabberNotifyFriends() { if reflect.TypeOf(s.Data) != reflect.TypeOf(&JabberData{}) { return } jabberSocket, ok := JabberSockets.Get(s.ID) if !ok { aid.Print("jabber socket not found even though it should be") return } s.Person.Relationships.Range(func(key string, value *person.Relationship) bool { friendSocket, found := JabberSockets.Get(value.From.ID) if value.Direction == person.RelationshipOutboundDirection { friendSocket, found = JabberSockets.Get(value.Towards.ID) } if !found { return true } friendSocket.Write([]byte(` `+ jabberSocket.Data.LastPresence +` `)) jabberSocket.Write([]byte(` `+ friendSocket.Data.LastPresence +` `)) return true }) }