//
// Implements the mautrix.Storer interface on StateStore
//

package store

import (
	log "github.com/sirupsen/logrus"
	"maunium.net/go/mautrix"
	mid "maunium.net/go/mautrix/id"
)

func (store *StateStore) SaveFilterID(userID mid.UserID, filterID string) {
	log.Debug("Upserting row into user_filter_ids")
	tx, err := store.DB.Begin()
	if err != nil {
		tx.Rollback()
		return
	}

	update := "UPDATE user_filter_ids SET filter_id = ? WHERE user_id = ?"
	if _, err := tx.Exec(update, filterID, userID); err != nil {
		tx.Rollback()
		return
	}

	insert := "INSERT OR IGNORE INTO user_filter_ids VALUES (?, ?)"
	if _, err := tx.Exec(insert, userID, filterID); err != nil {
		tx.Rollback()
		return
	}

	tx.Commit()
}

func (store *StateStore) LoadFilterID(userID mid.UserID) string {
	row := store.DB.QueryRow("SELECT filter_id FROM user_filter_ids WHERE user_id = ?", userID)
	var filterID string
	if err := row.Scan(&filterID); err != nil {
		return ""
	}
	return filterID
}

func (store *StateStore) SaveNextBatch(userID mid.UserID, nextBatchToken string) {
	log.Debug("Upserting row into user_batch_tokens")
	tx, err := store.DB.Begin()
	if err != nil {
		tx.Rollback()
		return
	}

	update := "UPDATE user_batch_tokens SET next_batch_token = ? WHERE user_id = ?"
	if _, err := tx.Exec(update, nextBatchToken, userID); err != nil {
		tx.Rollback()
		return
	}

	insert := "INSERT OR IGNORE INTO user_batch_tokens VALUES (?, ?)"
	if _, err := tx.Exec(insert, userID, nextBatchToken); err != nil {
		tx.Rollback()
		return
	}

	tx.Commit()
}

func (store *StateStore) LoadNextBatch(userID mid.UserID) string {
	row := store.DB.QueryRow("SELECT next_batch_token FROM user_batch_tokens WHERE user_id = ?", userID)
	var batchToken string
	if err := row.Scan(&batchToken); err != nil {
		return ""
	}
	return batchToken
}

func (store *StateStore) GetRoomMembers(roomId mid.RoomID) []mid.UserID {
	rows, err := store.DB.Query("SELECT user_id FROM room_members WHERE room_id = ?", roomId)
	users := make([]mid.UserID, 0)
	if err != nil {
		return users
	}
	defer rows.Close()

	var userId mid.UserID
	for rows.Next() {
		if err := rows.Scan(&userId); err == nil {
			users = append(users, userId)
		}
	}
	return users
}

func (store *StateStore) SaveRoom(room *mautrix.Room) {
	// This isn't really used at all.
}

func (store *StateStore) LoadRoom(roomId mid.RoomID) *mautrix.Room {
	// This isn't really used at all.
	return mautrix.NewRoom(roomId)
}