Handle users leaving from rooms

1 files changed, 18 insertions, 0 deletions

diff --git a/appservice/db.go b/appservice/db.go
index cbd1d11..e5b2d02 100644
--- a/appservice/db.go
+++ b/appservice/db.go
@@ -182,3 +182,21 @@ func dbGetMxUser(protocol string, userId connector.UserID) (string, error) {
 
 	return user.MxUserID, nil
 }
+
+func dbIsPmRoom(mxRoomId string) *DbPmRoomMap {
+	var pm_room DbPmRoomMap
+	if db.First(&pm_room, DbPmRoomMap{MxRoomID: mxRoomId}).RecordNotFound() {
+		return nil
+	} else {
+		return &pm_room
+	}
+}
+
+func dbIsPublicRoom(mxRoomId string) *DbRoomMap {
+	var room DbRoomMap
+	if db.First(&room, DbRoomMap{MxRoomID: mxRoomId}).RecordNotFound() {
+		return nil
+	} else {
+		return &room
+	}
+}