dm support

libmxclient/mxclient/client.go

@@ -8,6 +8,7 @@ import (
 	"errors"
 	"fmt"
 	"mxclientlib/determinant/mxpassfile"
+	"slices"
 
 	_ "github.com/mattn/go-sqlite3"
 	"github.com/rs/zerolog"
@@ -26,10 +27,74 @@ type MXClient struct {
 	*mautrix.Client
 	OnEvent    func(string)
 	OnMessage  func(string)
+	_directMap map[id.RoomID][]id.UserID
 }
 
-func (mxc *MXClient) _onEvent(ctx context.Context, evt *event.Event) {
+func (mxc *MXClient) _onAccountDataDM(ctx context.Context, evt *event.Event) { // event.DirectChatsEventContent
+	// TODO
+	fmt.Printf("\nTODO: Got event account data dm: %#v\n", evt)
+}
+
+func (mxc *MXClient) AddDirectRoom(uid id.UserID, roomid id.RoomID) {
+	room, ok := mxc._directMap[roomid]
+	if ok {
+		if slices.Contains(room, uid) {
+			log.Error().Str("room", roomid.String()).Str("uid", uid.String()).Msg("direct mapping already exists.")
+		} else {
+			log.Warn().Str("room", roomid.String()).Str("uid", uid.String()).Msg("room already a dm.")
+			mxc._directMap[roomid] = append(room, uid)
+		}
+	} else {
+		mxc._directMap[roomid] = []id.UserID{uid}
+	}
+}
+
+func (mxc *MXClient) IsDirectRoom(roomid id.RoomID) bool {
+	_, ok := mxc._directMap[roomid]
+	return ok
+}
+
+func (mxc *MXClient) _loadDirectMap() error {
+	var directChats event.DirectChatsEventContent
+	err := mxc.GetAccountData(context.Background(), event.AccountDataDirectChats.Type, &directChats)
+	if err != nil {
+		return err
+	}
+
+	new_directMap := make(map[id.RoomID][]id.UserID)
+	for uid, rooms := range directChats {
+		for _, room := range rooms {
+			new_directMap[room] = append(new_directMap[room], uid)
+		}
+	}
+	mxc._directMap = new_directMap
+	return nil
+}
+
+func (mxc *MXClient) _storeDirectMap() error {
+	directChats := make(event.DirectChatsEventContent)
+
+	for room, uids := range mxc._directMap {
+		for _, uid := range uids {
+			directChats[uid] = append(directChats[uid], room)
+		}
+	}
+	err := mxc.SetAccountData(context.Background(), event.AccountDataDirectChats.Type, &directChats)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (mxc *MXClient) _onEventMember(ctx context.Context, evt *event.Event) {
 	if evt.GetStateKey() == mxc.UserID.String() && evt.Content.AsMember().Membership == event.MembershipInvite {
+		if evt.Content.AsMember().IsDirect {
+			mxc.AddDirectRoom(evt.Sender, evt.RoomID)
+			err := mxc._storeDirectMap()
+			if err != nil {
+				log.Error().Err(err).Msg("failed to store direct chats account data")
+			}
+		}
 		_, err := mxc.JoinRoomByID(ctx, evt.RoomID)
 		if err == nil {
 			log.Info().
@@ -43,15 +108,16 @@ func (mxc *MXClient) _onEvent(ctx context.Context, evt *event.Event) {
 				Msg("Failed to join room after invite")
 		}
 	} else {
-		fmt.Printf("Got event: %#v\n", evt)
+		fmt.Printf("\nGot member event: %#v\n", evt)
 	}
 }
 
 func (mxc *MXClient) _onMessage(ctx context.Context, evt *event.Event) {
-	out, err := json.Marshal(map[string]interface{}{"sender": evt.Sender.String(),
+	out, err := json.Marshal(map[string]any{"sender": evt.Sender.String(),
 		"type":      evt.Type.String(),
 		"id":        evt.ID.String(),
 		"roomid":    evt.RoomID.String(),
+		"is_direct": mxc.IsDirectRoom(evt.RoomID),
 		"content":   evt.Content.Raw})
 
 	if err != nil {
@@ -75,6 +141,22 @@ func (mxc *MXClient) _onMessage(ctx context.Context, evt *event.Event) {
 	*/
 }
 
+func (mxc *MXClient) LeaveRoomAndForget(ctx context.Context, room id.RoomID) error {
+	_, err := mxc.LeaveRoom(ctx, room)
+	if err != nil {
+		return err
+	}
+	if mxc.IsDirectRoom(room) {
+		delete(mxc._directMap, room)
+		mxc._storeDirectMap()
+	}
+	_, err = mxc.ForgetRoom(ctx, room)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
 type sendmessage_data struct {
 	RoomId  id.RoomID      `json:"roomid"`
 	Type    event.Type     `json:"type"`
@@ -158,13 +240,16 @@ func NewMXClient(homeserverURL string, userID id.UserID, accessToken string) (*M
 
 	client.Store = cryptoStore
 
-	mxclient := &MXClient{client, nil, nil}
+	mxclient := &MXClient{client, nil, nil, make(map[id.RoomID][]id.UserID)}
 
 	syncer.ParseEventContent = true
 	syncer.OnEvent(client.StateStoreSyncHandler)
 
 	syncer.OnEventType(event.EventMessage, mxclient._onMessage)
-	syncer.OnEventType(event.StateMember, mxclient._onEvent)
+	syncer.OnEventType(event.StateMember, mxclient._onEventMember)
+	syncer.OnEventType(event.AccountDataDirectChats, mxclient._onAccountDataDM)
+
+	mxclient._loadDirectMap()
 
 	return mxclient, nil
 }

libmxclient/mxclientlib.go

@@ -293,11 +293,7 @@ func apiv0_leaveroom(cid C.int, roomid *C.char) *C.char {
 	if err != nil {
 		return C.CString(fmt.Sprintf("ERR: %v", err))
 	}
-	_, err = cli.LeaveRoom(context.Background(), id.RoomID(C.GoString(roomid)))
+	err = cli.LeaveRoomAndForget(context.Background(), id.RoomID(C.GoString(roomid)))
-	if err != nil {
-		return C.CString(fmt.Sprintf("ERR: %v", err))
-	}
-	_, err = cli.ForgetRoom(context.Background(), id.RoomID(C.GoString(roomid)))
 	if err != nil {
 		return C.CString(fmt.Sprintf("ERR: %v", err))
 	}

smal/src/demobot/demobot.py

@@ -51,16 +51,16 @@ class DemoBot(SMALBot):
             return
 
         if msg["content"]["body"].startswith("!echo"):
-            logger.error(f"reply to this: {msg}")
 
             txt = msg["content"]["body"][5:].strip()
 
-            if txt.strip() == "":
+            if txt == "":
-                txt = "Are you kidding me?"
+                txt = "Empty text? Are you kidding me?"
 
-            # self.sendmessage(msg["roomid"], txt)
+            if msg["is_direct"]:
+                self.sendmessage(msg["roomid"], txt)
+            else:
                 self.sendmessagereply(msg["roomid"], msg["id"], msg["sender"], txt)
-
             return
 
         logger.info(f"ignored a message: {msg}")