Initial commit

2026-04-18 16:16:29 +03:00 · 2021-07-07 18:15:07 +01:00
commit ceffe7612d
26 changed files with 2130 additions and 0 deletions
--- a/internal/storage/sqlite3/table_mailboxes.go
+++ b/internal/storage/sqlite3/table_mailboxes.go
@@ -0,0 +1,152 @@
+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
+}