ygggo/internal/imapserver/user.go

/*
 *  Copyright (c) 2021 Neil Alexander
 *
 *  This Source Code Form is subject to the terms of the Mozilla Public
 *  License, v. 2.0. If a copy of the MPL was not distributed with this
 *  file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */

package imapserver

import (
	"errors"
	"fmt"
	"log"

	"github.com/emersion/go-imap"
	"github.com/emersion/go-imap/backend"
	"github.com/neilalexander/yggmail/internal/account"
	"github.com/neilalexander/yggmail/internal/storage"
)

type User struct {
	//backend  *Backend
	storage  storage.Storage
	username string
	session  *account.Session
	conn     *imap.ConnInfo
	log      *log.Logger
}

func (u *User) Username() string {
	return u.session.Username()
}

func (u *User) ListMailboxes(subscribed bool) (mailboxes []backend.Mailbox, err error) {
	names, err := u.storage.MailboxList(u.username, subscribed)
	if err != nil {
		return nil, err
	}

	for _, mailbox := range names {
		mailboxes = append(mailboxes, &Mailbox{
			storage: u.storage,
			user:    u,
			name:    mailbox,
		})
	}

	return
}

func (u *User) GetMailbox(name string) (mailbox backend.Mailbox, err error) {
	if name == "" {
		return &Mailbox{
			storage: u.storage,
			user:    u,
			name:    "",
		}, nil
	}
	ok, _ := u.storage.MailboxSelect(u.username, name)
	if !ok {
		return nil, fmt.Errorf("mailbox %q not found", name)
	}
	return &Mailbox{
		storage: u.storage,
		user:    u,
		name:    name,
	}, nil
}

func (u *User) CreateMailbox(name string) error {
	u.log.Printf("Creating mailbox '%s'...\n", name)

	if e := u.storage.MailboxCreate(u.username, name); e != nil {
		u.log.Printf("Error creating mailbox '%s': %v\n", name, e)
		return e
	}

	u.log.Printf("Created mailbox '%s'\n", name)
	return nil
}

func (u *User) DeleteMailbox(name string) error {
	switch name {
	case "INBOX", "Outbox", "Sent":
		return errors.New("Cannot delete " + name)
	default:
		if e := u.storage.MailboxDelete(u.username, name); e != nil {
			u.log.Printf("Error deleting mailbox '%s': %v\n", name, e)
			return e
		} else {
			u.log.Printf("Deleted mailbox '%s'\n", name)
			return e
		}
	}
}

func (u *User) RenameMailbox(existingName, newName string) error {
	switch existingName {
	case "INBOX", "Outbox", "Sent":
		return errors.New("Cannot rename " + existingName)
	default:
		return u.storage.MailboxRename(u.username, existingName, newName)
	}
}

func (u *User) Logout() error {
	return nil
}