mautrix-discord/database/thread.go

package database

import (
	"database/sql"
	"errors"

	log "maunium.net/go/maulogger/v2"

	"maunium.net/go/mautrix/id"
	"maunium.net/go/mautrix/util/dbutil"
)

type ThreadQuery struct {
	db  *Database
	log log.Logger
}

const (
	threadSelect = "SELECT dcid, parent_chan_id, root_msg_dcid, root_msg_mxid FROM thread"
)

func (tq *ThreadQuery) New() *Thread {
	return &Thread{
		db:  tq.db,
		log: tq.log,
	}
}

func (tq *ThreadQuery) GetByDiscordID(discordID string) *Thread {
	query := threadSelect + " WHERE dcid=$1"

	row := tq.db.QueryRow(query, discordID)
	if row == nil {
		return nil
	}

	return tq.New().Scan(row)
}

//func (tq *ThreadQuery) GetByDiscordRootMsg(channelID, messageID string) *Thread {
//	query := messageSelect + " WHERE parent_chan_id=$1 AND root_msg_dcid=$2"
//
//	row := tq.db.QueryRow(query, channelID, messageID)
//	if row == nil {
//		return nil
//	}
//
//	return tq.New().Scan(row)
//}

func (tq *ThreadQuery) GetByMatrixRootMsg(mxid id.EventID) *Thread {
	query := threadSelect + " WHERE root_msg_mxid=$1"

	row := tq.db.QueryRow(query, mxid)
	if row == nil {
		return nil
	}

	return tq.New().Scan(row)
}

type Thread struct {
	db  *Database
	log log.Logger

	ID       string
	ParentID string

	RootDiscordID string
	RootMXID      id.EventID
}

func (t *Thread) Scan(row dbutil.Scannable) *Thread {
	err := row.Scan(&t.ID, &t.ParentID, &t.RootDiscordID, &t.RootMXID)
	if err != nil {
		if !errors.Is(err, sql.ErrNoRows) {
			t.log.Errorln("Database scan failed:", err)
			panic(err)
		}
		return nil
	}
	return t
}

func (t *Thread) Insert() {
	query := "INSERT INTO thread (dcid, parent_chan_id, root_msg_dcid, root_msg_mxid) VALUES ($1, $2, $3, $4)"

	_, err := t.db.Exec(query, t.ID, t.ParentID, t.RootDiscordID, t.RootMXID)

	if err != nil {
		t.log.Warnfln("Failed to insert %s@%s: %v", t.ID, t.ParentID, err)
		panic(err)
	}
}

func (t *Thread) Delete() {
	query := "DELETE FROM thread WHERE dcid=$1 AND parent_chan_id=$2"

	_, err := t.db.Exec(query, t.ID, t.ParentID)

	if err != nil {
		t.log.Warnfln("Failed to delete %s@%s: %v", t.ID, t.ParentID, err)
		panic(err)
	}
}