124 lines
3.4 KiB
Go
124 lines
3.4 KiB
Go
// Copyright (c) 2021 Proton Technologies AG
|
|
//
|
|
// This file is part of ProtonMail Bridge.
|
|
//
|
|
// ProtonMail Bridge is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// ProtonMail Bridge is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with ProtonMail Bridge. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package transfer
|
|
|
|
import (
|
|
"net"
|
|
"strings"
|
|
|
|
"github.com/emersion/go-imap"
|
|
"github.com/emersion/go-sasl"
|
|
)
|
|
|
|
type IMAPClientProvider interface {
|
|
Capability() (map[string]bool, error)
|
|
Support(capability string) (bool, error)
|
|
State() imap.ConnState
|
|
SupportAuth(mech string) (bool, error)
|
|
Authenticate(auth sasl.Client) error
|
|
Login(username, password string) error
|
|
List(ref, name string, ch chan *imap.MailboxInfo) error
|
|
Select(name string, readOnly bool) (*imap.MailboxStatus, error)
|
|
Fetch(seqset *imap.SeqSet, items []imap.FetchItem, ch chan *imap.Message) error
|
|
UidFetch(seqset *imap.SeqSet, items []imap.FetchItem, ch chan *imap.Message) error
|
|
}
|
|
|
|
// IMAPProvider implements export from IMAP server.
|
|
type IMAPProvider struct {
|
|
username string
|
|
password string
|
|
addr string
|
|
|
|
clientDialer func(addr string) (IMAPClientProvider, error)
|
|
client IMAPClientProvider
|
|
|
|
timeIt *timeIt
|
|
}
|
|
|
|
// NewIMAPProvider returns new IMAPProvider.
|
|
func NewIMAPProvider(username, password, host, port string) (*IMAPProvider, error) {
|
|
return newIMAPProvider(imapClientDial, username, password, host, port)
|
|
}
|
|
|
|
func newIMAPProvider(clientDialer func(string) (IMAPClientProvider, error), username, password, host, port string) (*IMAPProvider, error) {
|
|
p := &IMAPProvider{
|
|
username: username,
|
|
password: password,
|
|
addr: net.JoinHostPort(host, port),
|
|
|
|
timeIt: newTimeIt("imap"),
|
|
|
|
clientDialer: clientDialer,
|
|
}
|
|
|
|
if err := p.auth(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return p, nil
|
|
}
|
|
|
|
// ID is used for generating transfer ID by combining source and target ID.
|
|
// We want to keep the same rules for import from any IMAP server, therefore
|
|
// it returns constant.
|
|
func (p *IMAPProvider) ID() string {
|
|
return "imap"
|
|
}
|
|
|
|
// Mailboxes returns all available folder names from root of EML files.
|
|
// In case the same folder name is used more than once (for example root/a/foo
|
|
// and root/b/foo), it's treated as the same folder.
|
|
func (p *IMAPProvider) Mailboxes(includeEmpty, includeAllMail bool) ([]Mailbox, error) {
|
|
mailboxesInfo, err := p.list()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
mailboxes := []Mailbox{}
|
|
for _, mailbox := range mailboxesInfo {
|
|
hasNoSelect := false
|
|
for _, attrib := range mailbox.Attributes {
|
|
if strings.ToLower(attrib) == "\\noselect" {
|
|
hasNoSelect = true
|
|
break
|
|
}
|
|
}
|
|
if hasNoSelect {
|
|
continue
|
|
}
|
|
|
|
if !includeEmpty || true {
|
|
mailboxStatus, err := p.selectIn(mailbox.Name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if mailboxStatus.Messages == 0 {
|
|
continue
|
|
}
|
|
}
|
|
|
|
mailboxes = append(mailboxes, Mailbox{
|
|
ID: "",
|
|
Name: mailbox.Name,
|
|
Color: "",
|
|
IsExclusive: false,
|
|
})
|
|
}
|
|
return mailboxes, nil
|
|
}
|