yggmail/internal/storage/sqlite3/table_mailboxes.go

package sqlite3

import (
	"database/sql"
	"fmt"
)

type TableMailboxes struct {
	db                      *sql.DB
	selectMailboxes         *sql.Stmt
	listMailboxes           *sql.Stmt
	listMailboxesSubscribed *sql.Stmt
	createMailbox           *sql.Stmt
	renameMailbox           *sql.Stmt
	deleteMailbox           *sql.Stmt
	subscribeMailbox        *sql.Stmt
}

const mailboxesSchema = `
	CREATE TABLE IF NOT EXISTS mailboxes (
		username 	TEXT NOT NULL REFERENCES users(username) ON DELETE CASCADE ON UPDATE CASCADE,
		mailbox 	TEXT NOT NULL DEFAULT('INBOX'),
		subscribed  BOOLEAN NOT NULL DEFAULT 1,
		PRIMARY 	KEY(username, mailbox)
	);
`

const mailboxesList = `
	SELECT mailbox FROM mailboxes WHERE username = $1
`

const mailboxesListSubscribed = `
	SELECT mailbox FROM mailboxes WHERE username = $1 AND subscribed = 1
`

const mailboxesSelect = `
	SELECT mailbox FROM mailboxes WHERE username = $1 AND mailbox = $2
`

const mailboxesCreate = `
	INSERT INTO mailboxes (username, mailbox) VALUES($1, $2)
`

const mailboxesRename = `
	UPDATE mailboxes SET mailbox = $1 WHERE username = $2 AND mailbox = $3
`

const mailboxesDelete = `
	DELETE FROM mailboxes WHERE username = $1 AND mailbox = $2
`

const mailboxesSubscribe = `
	UPDATE mailboxes SET subscribed = $1 WHERE username = $2 AND mailbox = $3
`

func NewTableMailboxes(db *sql.DB) (*TableMailboxes, error) {
	t := &TableMailboxes{
		db: db,
	}
	_, err := db.Exec(mailboxesSchema)
	if err != nil {
		return nil, fmt.Errorf("db.Exec: %w", err)
	}
	t.listMailboxes, err = db.Prepare(mailboxesList)
	if err != nil {
		return nil, fmt.Errorf("db.Prepare(mailboxesCreate): %w", err)
	}
	t.listMailboxesSubscribed, err = db.Prepare(mailboxesListSubscribed)
	if err != nil {
		return nil, fmt.Errorf("db.Prepare(mailboxesCreate): %w", err)
	}
	t.selectMailboxes, err = db.Prepare(mailboxesSelect)
	if err != nil {
		return nil, fmt.Errorf("db.Prepare(mailboxesSelect): %w", err)
	}
	t.createMailbox, err = db.Prepare(mailboxesCreate)
	if err != nil {
		return nil, fmt.Errorf("db.Prepare(mailboxesCreate): %w", err)
	}
	t.deleteMailbox, err = db.Prepare(mailboxesDelete)
	if err != nil {
		return nil, fmt.Errorf("db.Prepare(mailboxesDelete): %w", err)
	}
	t.renameMailbox, err = db.Prepare(mailboxesRename)
	if err != nil {
		return nil, fmt.Errorf("db.Prepare(mailboxesRename): %w", err)
	}
	t.subscribeMailbox, err = db.Prepare(mailboxesSubscribe)
	if err != nil {
		return nil, fmt.Errorf("db.Prepare(mailboxesSubscribe): %w", err)
	}
	return t, nil
}

func (t *TableMailboxes) MailboxList(user string, onlySubscribed bool) ([]string, error) {
	stmt := t.listMailboxes
	if onlySubscribed {
		stmt = t.listMailboxesSubscribed
	}
	rows, err := stmt.Query(user)
	if err != nil {
		return nil, fmt.Errorf("t.listMailboxes.Query: %w", err)
	}
	defer rows.Close()
	var mailboxes []string
	for rows.Next() {
		var mailbox string
		if err := rows.Scan(&mailbox); err != nil {
			return nil, fmt.Errorf("rows.Scan: %w", err)
		}
		mailboxes = append(mailboxes, mailbox)
	}
	return mailboxes, nil
}

func (t *TableMailboxes) MailboxSelect(user, mailbox string) (bool, error) {
	row := t.selectMailboxes.QueryRow(user, mailbox)
	if err := row.Err(); err != nil && err != sql.ErrNoRows {
		return false, fmt.Errorf("row.Err: %w", err)
	} else if err == sql.ErrNoRows {
		return false, nil
	}
	var got string
	if err := row.Scan(&got); err != nil {
		return false, fmt.Errorf("row.Scan: %w", err)
	}
	return mailbox == got, nil
}

func (t *TableMailboxes) MailboxCreate(user, name string) error {
	_, err := t.createMailbox.Exec(user, name)
	return err
}

func (t *TableMailboxes) MailboxRename(user, old, new string) error {
	_, err := t.renameMailbox.Exec(new, user, old)
	return err
}

func (t *TableMailboxes) MailboxDelete(user, name string) error {
	_, err := t.deleteMailbox.Exec(user, name)
	return err
}

func (t *TableMailboxes) MailboxSubscribe(user, name string, subscribed bool) error {
	sn := 1
	if !subscribed {
		sn = 0
	}
	_, err := t.subscribeMailbox.Exec(sn, user, name)
	return err
}