Add DM rooms to space

database/userportal.go

@@ -0,0 +1,102 @@
+package database
+
+import (
+	"database/sql"
+	"errors"
+	"time"
+
+	log "maunium.net/go/maulogger/v2"
+	"maunium.net/go/mautrix/util/dbutil"
+)
+
+const (
+	UserPortalTypeDM    = "dm"
+	UserPortalTypeGuild = "guild"
+)
+
+type UserPortal struct {
+	DiscordID string
+	Type      string
+	Timestamp time.Time
+	InSpace   bool
+}
+
+func (up UserPortal) Scan(l log.Logger, row dbutil.Scannable) *UserPortal {
+	var ts int64
+	err := row.Scan(&up.DiscordID, &up.Type, &ts, &up.InSpace)
+	if err != nil {
+		l.Errorln("Error scanning user portal:", err)
+		panic(err)
+		return nil
+	}
+	up.Timestamp = time.UnixMilli(ts)
+	return &up
+}
+
+func (u *User) scanUserPortals(rows *sql.Rows) []UserPortal {
+	var ups []UserPortal
+	for rows.Next() {
+		up := UserPortal{}.Scan(u.log, rows)
+		if up != nil {
+			ups = append(ups, *up)
+		}
+	}
+	return ups
+}
+
+func (u *User) GetPortals() []UserPortal {
+	rows, err := u.db.Query("SELECT discord_id, type, timestamp, in_space FROM user_portal WHERE user_mxid=$1", u.MXID)
+	if err != nil {
+		u.log.Errorln("Failed to get portals:", err)
+		panic(err)
+		return nil
+	}
+	return u.scanUserPortals(rows)
+}
+
+func (u *User) IsInSpace(discordID string) (isIn bool) {
+	query := `SELECT in_space FROM user_portal WHERE user_mxid=$1 AND discord_id=$2`
+	err := u.db.QueryRow(query, u.MXID, discordID).Scan(&isIn)
+	if err != nil && !errors.Is(err, sql.ErrNoRows) {
+		u.log.Warnfln("Failed to scan in_space for %s/%s: %v", u.MXID, discordID, err)
+		panic(err)
+	}
+	return
+}
+
+func (u *User) MarkInPortal(portal UserPortal) {
+	query := `
+		INSERT INTO user_portal (discord_id, type, user_mxid, timestamp, in_space)
+		VALUES ($1, $2, $3, $4, $5)
+		ON CONFLICT (discord_id, user_mxid) DO UPDATE
+		    SET timestamp=excluded.timestamp, in_space=excluded.in_space
+	`
+	_, err := u.db.Exec(query, portal.DiscordID, portal.Type, u.MXID, portal.Timestamp.UnixMilli(), portal.InSpace)
+	if err != nil {
+		u.log.Errorfln("Failed to insert user portal %s/%s: %v", u.MXID, portal.DiscordID, err)
+		panic(err)
+	}
+}
+
+func (u *User) MarkNotInPortal(discordID string) {
+	query := `DELETE FROM user_portal WHERE user_mxid=$1 AND discord_id=$2`
+	_, err := u.db.Exec(query, u.MXID, discordID)
+	if err != nil {
+		u.log.Errorfln("Failed to remove user portal %s/%s: %v", u.MXID, discordID, err)
+		panic(err)
+	}
+}
+
+func (u *User) PrunePortalList(beforeTS time.Time) []UserPortal {
+	query := `
+		DELETE FROM user_portal
+			WHERE user_mxid=$1 AND timestamp<$2
+			RETURNING discord_id, type, timestamp, in_space
+	`
+	rows, err := u.db.Query(query, u.MXID, beforeTS.UnixMilli())
+	if err != nil {
+		u.log.Errorln("Failed to prune user guild list:", err)
+		panic(err)
+	}
+	return u.scanUserPortals(rows)
+}