Add network field to auth data

mautrix · Feb 23, 2024 · 1df5dfe · 1df5dfe
1 parent e3f965e
commit 1df5dfe
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/libgm/client.go b/libgm/client.go
@@ -38,6 +38,13 @@ type AuthData struct {
 	Cookies   map[string]string `json:"cookies,omitempty"`
 }
 
+func (ad *AuthData) AuthNetwork() string {
+	if ad.Cookies != nil {
+		return util.GoogleNetwork
+	}
+	return ""
+}
+
 const RefreshTachyonBuffer = 1 * time.Hour
 
 type Proxy func(*http.Request) (*url.URL, error)
@@ -292,6 +299,7 @@ func (c *Client) refreshAuthToken() error {
 		MessageAuth: &gmproto.AuthMessage{
 			RequestID:        requestID,
 			TachyonAuthToken: c.AuthData.TachyonAuthToken,
+			Network:          c.AuthData.AuthNetwork(),
 			ConfigVersion:    util.ConfigMessage,
 		},
 		CurrBrowserDevice: c.AuthData.Browser,

diff --git a/libgm/longpoll.go b/libgm/longpoll.go
@@ -122,6 +122,7 @@ func (c *Client) doLongPoll(loggedIn bool) {
 			Auth: &gmproto.AuthMessage{
 				RequestID:        listenReqID,
 				TachyonAuthToken: c.AuthData.TachyonAuthToken,
+				Network:          c.AuthData.AuthNetwork(),
 				ConfigVersion:    util.ConfigMessage,
 			},
 			Unknown: &gmproto.ReceiveMessagesRequest_UnknownEmptyObject2{

diff --git a/libgm/session_handler.go b/libgm/session_handler.go
@@ -280,6 +280,7 @@ func (s *SessionHandler) sendAckRequest() {
 		AuthData: &gmproto.AuthMessage{
 			RequestID:        uuid.NewString(),
 			TachyonAuthToken: s.client.AuthData.TachyonAuthToken,
+			Network:          s.client.AuthData.AuthNetwork(),
 			ConfigVersion:    util.ConfigMessage,
 		},
 		EmptyArr: &gmproto.EmptyArr{},