PLAT-25704: Add megabridge login flows

.gitignore

@@ -6,8 +6,11 @@
 
 *.json
 *.db
+*.db-shm
+*.db-wal
 *.log
 *.bak
 
 /mautrix-meta
+/mautrix-meta-v2
 /start

build-v2.sh

@@ -6,4 +6,4 @@ if [[ -z "$LIBRARY_PATH" && -d /opt/homebrew ]]; then
 fi
 export MAUTRIX_VERSION=$(cat go.mod | grep 'maunium.net/go/mautrix ' | awk '{ print $2 }')
 export GO_LDFLAGS="-s -w -X main.Tag=$(git describe --exact-match --tags 2>/dev/null) -X main.Commit=$(git rev-parse HEAD) -X 'main.BuildTime=`date '+%b %_d %Y, %H:%M:%S'`' -X 'maunium.net/go/mautrix.GoModVersion=$MAUTRIX_VERSION'"
-go build -ldflags "$GO_LDFLAGS" -o mautrix-meta ./cmd/mautrix-meta "$@"
+go build -ldflags "$GO_LDFLAGS" -o mautrix-meta-v2 ./cmd/mautrix-meta-v2 "$@"

cmd/lscli/go.mod

@@ -9,7 +9,7 @@ require (
 	github.com/tidwall/gjson v1.17.1
 	github.com/zyedidia/clipboard v1.0.4
 	go.mau.fi/mautrix-meta v0.2.0
-	go.mau.fi/util v0.5.1-0.20240625181823-38eefa626984
+	go.mau.fi/util v0.5.1-0.20240629220711-4fa40bf64652
 )
 
 require (

cmd/lscli/go.sum

@@ -36,8 +36,8 @@ github.com/zyedidia/clipboard v1.0.4 h1:r6GUQOyPtIaApRLeD56/U+2uJbXis6ANGbKWCljU
 github.com/zyedidia/clipboard v1.0.4/go.mod h1:zykFnZUXX0ErxqvYLUFEq7QDJKId8rmh2FgD0/Y8cjA=
 go.mau.fi/libsignal v0.1.0 h1:vAKI/nJ5tMhdzke4cTK1fb0idJzz1JuEIpmjprueC+c=
 go.mau.fi/libsignal v0.1.0/go.mod h1:R8ovrTezxtUNzCQE5PH30StOQWWeBskBsWE55vMfY9I=
-go.mau.fi/util v0.5.1-0.20240625181823-38eefa626984 h1:63X00R1qL5G+m2LWxdKzAS8cyihFu5IaaOzJ7QHvkzs=
-go.mau.fi/util v0.5.1-0.20240625181823-38eefa626984/go.mod h1:DsJzUrJAG53lCZnnYvq9/mOyLuPScWwYhvETiTrpdP4=
+go.mau.fi/util v0.5.1-0.20240629220711-4fa40bf64652 h1:/wY7vpwOE6he5Qlf6lICHuOUs+nAQdsC7qwRsDbsh14=
+go.mau.fi/util v0.5.1-0.20240629220711-4fa40bf64652/go.mod h1:DsJzUrJAG53lCZnnYvq9/mOyLuPScWwYhvETiTrpdP4=
 go.mau.fi/whatsmeow v0.0.0-20240625083845-6acab596dd8c h1:yiULssyKHJcFA1fae2NJkwU7QW4EHQs7QEWoIqfqilA=
 go.mau.fi/whatsmeow v0.0.0-20240625083845-6acab596dd8c/go.mod h1:0+65CYaE6r4dWzr0dN8i+UZKy0gIfJ79VuSqIl0nKRM=
 golang.org/x/crypto v0.24.0 h1:mnl8DM0o513X8fdIkmyFE/5hTYxbwYOjDS/+rK6qpRI=

go.mod

@@ -12,7 +12,7 @@ require (
 	github.com/rs/zerolog v1.33.0
 	github.com/tidwall/gjson v1.17.1
 	go.mau.fi/libsignal v0.1.0
-	go.mau.fi/util v0.5.1-0.20240625181823-38eefa626984
+	go.mau.fi/util v0.5.1-0.20240629220711-4fa40bf64652
 	go.mau.fi/whatsmeow v0.0.0-20240625083845-6acab596dd8c
 	golang.org/x/crypto v0.24.0
 	golang.org/x/exp v0.0.0-20240613232115-7f521ea00fb8

go.sum

@@ -51,8 +51,8 @@ github.com/yuin/goldmark v1.7.2 h1:NjGd7lO7zrUn/A7eKwn5PEOt4ONYGqpxSEeZuduvgxc=
 github.com/yuin/goldmark v1.7.2/go.mod h1:uzxRWxtg69N339t3louHJ7+O03ezfj6PlliRlaOzY1E=
 go.mau.fi/libsignal v0.1.0 h1:vAKI/nJ5tMhdzke4cTK1fb0idJzz1JuEIpmjprueC+c=
 go.mau.fi/libsignal v0.1.0/go.mod h1:R8ovrTezxtUNzCQE5PH30StOQWWeBskBsWE55vMfY9I=
-go.mau.fi/util v0.5.1-0.20240625181823-38eefa626984 h1:63X00R1qL5G+m2LWxdKzAS8cyihFu5IaaOzJ7QHvkzs=
-go.mau.fi/util v0.5.1-0.20240625181823-38eefa626984/go.mod h1:DsJzUrJAG53lCZnnYvq9/mOyLuPScWwYhvETiTrpdP4=
+go.mau.fi/util v0.5.1-0.20240629220711-4fa40bf64652 h1:/wY7vpwOE6he5Qlf6lICHuOUs+nAQdsC7qwRsDbsh14=
+go.mau.fi/util v0.5.1-0.20240629220711-4fa40bf64652/go.mod h1:DsJzUrJAG53lCZnnYvq9/mOyLuPScWwYhvETiTrpdP4=
 go.mau.fi/whatsmeow v0.0.0-20240625083845-6acab596dd8c h1:yiULssyKHJcFA1fae2NJkwU7QW4EHQs7QEWoIqfqilA=
 go.mau.fi/whatsmeow v0.0.0-20240625083845-6acab596dd8c/go.mod h1:0+65CYaE6r4dWzr0dN8i+UZKy0gIfJ79VuSqIl0nKRM=
 go.mau.fi/zeroconfig v0.1.2 h1:DKOydWnhPMn65GbXZOafgkPm11BvFashZWLct0dGFto=

messagix/cookies/cookies.go

@@ -54,6 +54,15 @@ type Cookies struct {
 	IGWWWClaim string
 }
 
+func (c *Cookies) UpdateValues(newValues map[string]string) {
+	c.lock.Lock()
+	defer c.lock.Unlock()
+	c.values = make(map[MetaCookieName]string)
+	for k, v := range newValues {
+		c.values[MetaCookieName(k)] = v
+	}
+}
+
 func (c *Cookies) MarshalJSON() ([]byte, error) {
 	return json.Marshal(c.values)
 }

pkg/connector/connector.go

@@ -32,14 +32,37 @@ func (m *MetaConnector) GetCapabilities() *bridgev2.NetworkGeneralCapabilities {
 }
 
 func (s *MetaConnector) GetName() bridgev2.BridgeName {
-	// Would be nice if we could use MetaConfig.Mode here, but it's not available yet?
-	return bridgev2.BridgeName{
-		DisplayName:      "Meta",
-		NetworkURL:       "https://meta.com",
-		NetworkIcon:      "mxc://maunium.net/JxjlbZUlCPULEeHZSwleUXQv", // Instagram icon
-		NetworkID:        "meta",
-		BeeperBridgeType: "meta",
-		DefaultPort:      29319,
+	if s.Config == nil || s.Config.Mode == "" {
+		return bridgev2.BridgeName{
+			DisplayName:      "Meta",
+			NetworkURL:       "https://meta.com",
+			NetworkIcon:      "mxc://maunium.net/DxpVrwwzPUwaUSazpsjXgcKB",
+			NetworkID:        "meta",
+			BeeperBridgeType: "meta",
+			DefaultPort:      29319,
+		}
+	} else {
+		if s.Config.Mode == "instagram" {
+			return bridgev2.BridgeName{
+				DisplayName:      "Instagram",
+				NetworkURL:       "https://instagram.com",
+				NetworkIcon:      "mxc://maunium.net/JxjlbZUlCPULEeHZSwleUXQv",
+				NetworkID:        "instagram",
+				BeeperBridgeType: "meta",
+				DefaultPort:      29319,
+			}
+		} else if s.Config.Mode == "facebook" {
+			return bridgev2.BridgeName{
+				DisplayName:      "Facebook",
+				NetworkURL:       "https://www.facebook.com/messenger",
+				NetworkIcon:      "mxc://maunium.net/ygtkteZsXnGJLJHRchUwYWak",
+				NetworkID:        "facebook",
+				BeeperBridgeType: "meta",
+				DefaultPort:      29319,
+			}
+		} else {
+			panic("unknown mode")
+		}
 	}
 }
 

pkg/connector/example-config.yaml

@@ -1,3 +1,2 @@
 # Which service is this bridge for? Either "facebook" or "instagram".
 # Remember to change the appservice id, bot profile info and bridge username_template too.
-mode: instagram

pkg/connector/login.go

@@ -5,43 +5,141 @@ import (
 	"fmt"
 
 	"maunium.net/go/mautrix/bridgev2"
+
+	"go.mau.fi/util/exslices"
+
+	"go.mau.fi/mautrix-meta/messagix"
+	"go.mau.fi/mautrix-meta/messagix/cookies"
+	"go.mau.fi/mautrix-meta/messagix/table"
+	"go.mau.fi/mautrix-meta/messagix/types"
 )
 
+const FlowIDFacebookCookies = "cookies-facebook"
+const FlowIDInstagramCookies = "cookies-instagram"
+
 func (m *MetaConnector) CreateLogin(ctx context.Context, user *bridgev2.User, flowID string) (bridgev2.LoginProcess, error) {
-	if flowID != "cookies" {
+	if flowID != FlowIDFacebookCookies && flowID != FlowIDInstagramCookies {
 		return nil, fmt.Errorf("unknown flow ID %s", flowID)
 	}
 
-	return &CookieLogin{}, nil
+	return &MetaCookieLogin{
+		Flow: flowID,
+		User: user,
+		Main: m,
+	}, nil
 }
 
 func (m *MetaConnector) GetLoginFlows() []bridgev2.LoginFlow {
-	return []bridgev2.LoginFlow{{
-		Name:        "Cookies",
-		Description: "Login using cookies from a web browser",
-		ID:          "cookies",
-	}}
+	facebook := bridgev2.LoginFlow{
+		Name:        "Facebook Cookies",
+		Description: "Login using cookies from Facebook Messenger",
+		ID:          FlowIDFacebookCookies,
+	}
+	instagram := bridgev2.LoginFlow{
+		Name:        "Instagram Cookies",
+		Description: "Login using cookies from Instagram",
+		ID:          FlowIDInstagramCookies,
+	}
+	if m.Config == nil || m.Config.Mode == "" {
+		return []bridgev2.LoginFlow{facebook, instagram}
+	} else if m.Config.Mode == "facebook" {
+		return []bridgev2.LoginFlow{facebook}
+	} else if m.Config.Mode == "instagram" {
+		return []bridgev2.LoginFlow{instagram}
+	} else {
+		panic("unknown mode in config")
+	}
 }
 
-type CookieLogin struct{}
+type MetaCookieLogin struct {
+	Flow string
+	User *bridgev2.User
+	Main *MetaConnector
+}
 
-var _ bridgev2.LoginProcessCookies = (*CookieLogin)(nil)
+var _ bridgev2.LoginProcessCookies = (*MetaCookieLogin)(nil)
 
-func (p *CookieLogin) Start(ctx context.Context) (*bridgev2.LoginStep, error) {
-	return &bridgev2.LoginStep{
-		Type:          bridgev2.LoginStepTypeCookies,
-		StepID:        "fi.mau.meta.cookies",
-		Instructions:  "Please enter cookies from your browser",
-		CookiesParams: &bridgev2.LoginCookiesParams{},
-	}, nil
+func (m *MetaCookieLogin) Start(ctx context.Context) (*bridgev2.LoginStep, error) {
+	if m.Flow == FlowIDFacebookCookies {
+		return &bridgev2.LoginStep{
+			Type:         bridgev2.LoginStepTypeCookies,
+			StepID:       "fi.mau.meta.cookies",
+			Instructions: "Please enter cookies from your browser",
+			CookiesParams: &bridgev2.LoginCookiesParams{
+				URL:          "https://www.facebook.com/",
+				UserAgent:    messagix.UserAgent,
+				CookieDomain: "www.facebook.com",
+				CookieKeys:   exslices.CastToString[string](cookies.FBRequiredCookies),
+			},
+		}, nil
+	} else if m.Flow == FlowIDInstagramCookies {
+		return &bridgev2.LoginStep{
+			Type:         bridgev2.LoginStepTypeCookies,
+			StepID:       "fi.mau.meta.cookies",
+			Instructions: "Please enter cookies from your browser",
+			CookiesParams: &bridgev2.LoginCookiesParams{
+				URL:          "https://www.instagram.com/",
+				UserAgent:    messagix.UserAgent,
+				CookieDomain: "www.instagram.com",
+				CookieKeys:   exslices.CastToString[string](cookies.IGRequiredCookies),
+			},
+		}, nil
+	} else {
+		return nil, fmt.Errorf("unknown flow ID %s", m.Flow)
+	}
 }
 
-func (c *CookieLogin) Cancel() {}
+func (m *MetaCookieLogin) Cancel() {}
+
+func (m *MetaCookieLogin) getFBID(currentUser types.UserInfo, tbl *table.LSTable) int64 {
+	var newFBID int64
+	// TODO figure out why the contact IDs for self is different than the fbid in the ready event
+	for _, row := range tbl.LSVerifyContactRowExists {
+		if row.IsSelf && row.ContactId != newFBID {
+			if newFBID != 0 {
+				// Hopefully this won't happen
+				m.User.Log.Warn().Int64("prev_fbid", newFBID).Int64("new_fbid", row.ContactId).Msg("Got multiple fbids for self")
+			} else {
+				m.User.Log.Debug().Int64("fbid", row.ContactId).Msg("Found own fbid")
+			}
+			newFBID = row.ContactId
+		}
+	}
+	if newFBID == 0 {
+		newFBID = currentUser.GetFBID()
+		m.User.Log.Warn().Int64("fbid", newFBID).Msg("Own contact entry not found, falling back to fbid in current user object")
+	}
+	return newFBID
+}
+
+func (m *MetaCookieLogin) SubmitCookies(ctx context.Context, strCookies map[string]string) (*bridgev2.LoginStep, error) {
+	c := &cookies.Cookies{
+		Platform: types.Instagram,
+	}
+	if m.Flow == FlowIDFacebookCookies {
+		c.Platform = types.Facebook
+	}
+
+	c.UpdateValues(strCookies)
+
+	// Check if the cookies are valid
+	if !c.IsLoggedIn() {
+		return nil, fmt.Errorf("invalid cookies")
+	}
+
+	log := m.User.Log.With().Str("component", "messagix").Logger()
+	client := messagix.NewClient(c, log)
+
+	currentUser, initialTable, err := client.LoadMessagesPage()
+	if err != nil {
+		return nil, err
+	}
+
+	FBID := m.getFBID(currentUser, initialTable)
 
-func (p *CookieLogin) SubmitCookies(ctx context.Context, cookies map[string]string) (*bridgev2.LoginStep, error) {
 	return &bridgev2.LoginStep{
 		Type:         bridgev2.LoginStepTypeComplete,
 		StepID:       "fi.mau.meta.complete",
-		Instructions: "Login successful",
+		Instructions: fmt.Sprintf("Logged in as %s (%d)", currentUser.GetName(), FBID),
 	}, nil
 }