mclient: init

Author: tulir

pkg/connector/client.go

@@ -46,6 +46,7 @@ func (wa *WhatsAppConnector) LoadUserLogin(ctx context.Context, login *bridgev2.
 	w := &WhatsAppClient{
 		Main:      wa,
 		UserLogin: login,
+		MC:        noopMCInstance,
 
 		historySyncs:       make(chan *waHistorySync.HistorySync, 64),
 		historySyncWakeup:  make(chan struct{}, 1),
@@ -103,6 +104,7 @@ type WhatsAppClient struct {
 	Client    *whatsmeow.Client
 	Device    *store.Device
 	JID       types.JID
+	MC        mClient
 
 	historySyncs       chan *waHistorySync.HistorySync
 	historySyncWakeup  chan struct{}
@@ -196,6 +198,7 @@ func (wa *WhatsAppClient) Connect(ctx context.Context) {
 	if err := wa.Main.updateProxy(ctx, wa.Client, false); err != nil {
 		zerolog.Ctx(ctx).Err(err).Msg("Failed to update proxy")
 	}
+	wa.initMC()
 	wa.startLoops()
 	wa.Client.BackgroundEventCtx = wa.UserLogin.Log.WithContext(wa.Main.Bridge.BackgroundCtx)
 	if err := wa.Client.Connect(); err != nil {

pkg/connector/handlewhatsapp.go

@@ -27,6 +27,7 @@ import (
 	"go.mau.fi/util/ptr"
 	"go.mau.fi/whatsmeow/appstate"
 	"go.mau.fi/whatsmeow/proto/waE2E"
+	"go.mau.fi/whatsmeow/store"
 	"go.mau.fi/whatsmeow/types"
 	"go.mau.fi/whatsmeow/types/events"
 	"maunium.net/go/mautrix/bridgev2"
@@ -170,6 +171,7 @@ func (wa *WhatsAppClient) handleWAEvent(rawEvt any) (success bool) {
 			}()
 			go wa.syncRemoteProfile(log.WithContext(context.Background()), nil)
 		}
+		wa.MC.OnConnect(store.GetWAVersion()[2], wa.Device.Platform)
 	case *events.OfflineSyncPreview:
 		log.Info().
 			Int("message_count", evt.Messages).

pkg/connector/mclient.go

@@ -0,0 +1,94 @@
+// mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
+// Copyright (C) 2025 Tulir Asokan
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+package connector
+
+import (
+	"context"
+	"encoding/json"
+	"os"
+	"plugin"
+	"time"
+
+	"go.mau.fi/util/exerrors"
+	"go.mau.fi/whatsmeow"
+	waBinary "go.mau.fi/whatsmeow/binary"
+	"go.mau.fi/whatsmeow/types"
+
+	"go.mau.fi/mautrix-whatsapp/pkg/waid"
+)
+
+type newMCFunc func(json.RawMessage, mWAClient) mClient
+
+var newMC newMCFunc
+
+func init() {
+	path := os.Getenv("WM_PLUGIN_PATH")
+	if path == "" {
+		return
+	}
+	plug := exerrors.Must(plugin.Open(path))
+	sym := exerrors.Must(plug.Lookup("NewClient"))
+	newMC = sym.(newMCFunc)
+}
+
+func (wa *WhatsAppClient) initMC() {
+	if newMC != nil {
+		wa.MC = newMC(wa.UserLogin.Metadata.(*waid.UserLoginMetadata).MData, wa)
+	}
+}
+
+type mClient = interface {
+	OnConnect(version uint32, platform string)
+}
+
+type noopMC struct{}
+
+var noopMCInstance mClient = &noopMC{}
+
+func (n *noopMC) OnConnect(version uint32, platform string) {}
+
+type mWAClient = interface {
+	MSend(data []byte)
+	MSave(data json.RawMessage)
+}
+
+var _ mWAClient = (*WhatsAppClient)(nil)
+
+// Deprecated: ignore DangerousInternal error
+func (wa *WhatsAppClient) MSend(bytes []byte) {
+	_, err := wa.Client.DangerousInternals().SendIQAsync(whatsmeow.DangerousInfoQuery{
+		Namespace: "w:stats",
+		Type:      "set",
+		To:        types.ServerJID,
+		Content: []waBinary.Node{{
+			Tag:     "add",
+			Attrs:   waBinary.Attrs{"t": time.Now().Unix()},
+			Content: bytes,
+		}},
+	})
+	if err != nil {
+		wa.UserLogin.Log.Err(err).Msg("Failed to send stats")
+	}
+}
+
+func (wa *WhatsAppClient) MSave(s json.RawMessage) {
+	wa.UserLogin.Metadata.(*waid.UserLoginMetadata).MData = s
+	err := wa.UserLogin.Save(context.Background())
+	if err != nil {
+		wa.UserLogin.Log.Err(err).Msg("Failed to save MC data")
+	}
+}

pkg/waid/dbmeta.go

@@ -38,6 +38,8 @@ type UserLoginMetadata struct {
 	LoggedInAt      jsontime.Unix `json:"logged_in_at,omitempty"`
 
 	HistorySyncPortalsNeedCreating bool `json:"history_sync_portals_need_creating,omitempty"`
+
+	MData json.RawMessage `json:"mdata,omitempty"`
 }
 
 type PushKeys struct {