Calls AI

client/client.go

@@ -43,6 +43,8 @@ const (
 	WSCallLoweredHandEvent    EventType = "WSCallLoweredHand"
 	WSCallScreenOnEvent       EventType = "WSCallScreenOn"
 	WSCallScreenOffEvent      EventType = "WSCallScreenOff"
+	WSSummonAIEvent           EventType = "WSSummonAI"
+	WSAIPostUpdateEvent       EventType = "WSAIPostUpdate"
 )
 
 func (e EventType) IsValid() bool {
@@ -58,7 +60,9 @@ func (e EventType) IsValid() bool {
 		WSCallRaisedHandEvent, WSCallLoweredHandEvent,
 		WSCallScreenOnEvent, WSCallScreenOffEvent,
 		WSCallJobStateEvent,
-		WSJobStopEvent:
+		WSJobStopEvent,
+		WSSummonAIEvent,
+		WSAIPostUpdateEvent:
 		return true
 	default:
 		return false

client/websocket.go

@@ -59,6 +59,8 @@ const (
 	wsEventUserScreenOn    = wsEvPrefix + "user_screen_on"
 	wsEventUserScreenOff   = wsEvPrefix + "user_screen_off"
 	wsEventUserReacted     = wsEvPrefix + "user_reacted"
+	wsEventSummonAI        = wsEvPrefix + "summon_ai"
+	wsEventAIPostUpdate    = "custom_mattermost-ai_postupdate"
 )
 
 var (
@@ -295,6 +297,15 @@ func (c *Client) handleWSMsg(msg ws.Message) error {
 				evType = WSCallScreenOffEvent
 			}
 			c.emit(evType, sessionID)
+		case wsEventSummonAI:
+			channelID, _ := ev.GetData()["channel_id"].(string)
+			if channelID != c.cfg.ChannelID {
+				return nil
+			}
+			authToken, _ := ev.GetData()["auth_token"].(string)
+			c.emit(WSSummonAIEvent, authToken)
+		case wsEventAIPostUpdate:
+			c.emit(WSAIPostUpdateEvent, ev.GetData())
 		default:
 		}
 	case ws.BinaryMessage: