domain

type ConcurrentSpamClassifier interface {
	CheckAll(mails [][]byte, concurrency int) []*SpamResult
	LearnAll(learnType LearnType, mails [][]byte, concurrency int) []error
}

type ImapConnector interface {
	Select(folder string) (uint32, error)
	ListUids() ([]uint32, error)
	FetchMails(uids []uint32) ([]*RawImapMail, error)
	FetchIdHeaders(uids []uint32) ([]*ImapIdInfo, error)
	Put(body []byte, folder string) error
	DeleteReady() (error, error)
	Delete(uids []uint32) error
	MoveReady() (error, error)
	Move(uids []uint32, folder string) error

	Close() error
}

type ImapFolder struct {
	Name        string
	UidValidity uint32
}

type ImapIdInfo struct {
	Uid        uint32
	Subject    string
	MailIdHash string
}

type LearnType string

type MailClass int

type Persistence interface {
	Close() error
	AllFolders() ([]*ImapFolder, error)
	SaveFolder(name string, uidValidity uint32) error
	GetMailsInFolder(class MailClass, folder string) ([]*SavedImapMail, error)
	FindMailByHash(class MailClass, folder string, mailIdHash string) (*SavedImapMail, error)
	UpdateUid(id int64, uid uint32) error
	SaveMails(mails []SaveMail) error
}

type RawImapMail struct {
	Uid        uint32
	Subject    string
	MailIdHash string
	RawMail    []byte
}

type SaveMail struct {
	Class      MailClass
	Uid        uint32
	MailIdHash string
	FolderName string
	Subject    string
	IsSpam     *bool
	Score      *float64
}

type SavedImapMail struct {
	Id         int64
	Class      MailClass
	Uid        uint32
	MailIdHash string
	FolderName string
	Subject    string
	IsSpam     bool
	Score      float64
}

type SpamClassifier interface {
	Check(rawMail []byte) *SpamResult
	Learn(learnType LearnType, rawMail []byte) error
}

type SpamResult struct {
	IsSpam bool
	Score  float64
	Body   []byte
	Error  error
}