2022-07-11 20:11:21 +02:00
|
|
|
package account
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"errors"
|
|
|
|
"fmt"
|
|
|
|
"io"
|
|
|
|
"os"
|
|
|
|
"path/filepath"
|
|
|
|
"sync/atomic"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"git.sr.ht/~rjarry/aerc/commands"
|
2022-07-19 22:31:51 +02:00
|
|
|
"git.sr.ht/~rjarry/aerc/logging"
|
2022-07-11 20:11:21 +02:00
|
|
|
"git.sr.ht/~rjarry/aerc/models"
|
|
|
|
"git.sr.ht/~rjarry/aerc/widgets"
|
|
|
|
mboxer "git.sr.ht/~rjarry/aerc/worker/mbox"
|
|
|
|
"git.sr.ht/~rjarry/aerc/worker/types"
|
|
|
|
)
|
|
|
|
|
|
|
|
type ImportMbox struct{}
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
register(ImportMbox{})
|
|
|
|
}
|
|
|
|
|
|
|
|
func (ImportMbox) Aliases() []string {
|
|
|
|
return []string{"import-mbox"}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (ImportMbox) Complete(aerc *widgets.Aerc, args []string) []string {
|
|
|
|
return commands.CompletePath(filepath.Join(args...))
|
|
|
|
}
|
|
|
|
|
|
|
|
func (ImportMbox) Execute(aerc *widgets.Aerc, args []string) error {
|
|
|
|
if len(args) != 2 {
|
|
|
|
return importFolderUsage(args[0])
|
|
|
|
}
|
|
|
|
filename := args[1]
|
|
|
|
|
|
|
|
acct := aerc.SelectedAccount()
|
|
|
|
if acct == nil {
|
|
|
|
return errors.New("No account selected")
|
|
|
|
}
|
|
|
|
store := acct.Store()
|
|
|
|
if store == nil {
|
|
|
|
return errors.New("No message store selected")
|
|
|
|
}
|
|
|
|
|
|
|
|
folder := acct.SelectedDirectory()
|
|
|
|
if folder == "" {
|
|
|
|
return errors.New("No directory selected")
|
|
|
|
}
|
|
|
|
|
|
|
|
importFolder := func() {
|
|
|
|
statusInfo := fmt.Sprintln("Importing", filename, "to folder", folder)
|
|
|
|
aerc.PushStatus(statusInfo, 10*time.Second)
|
2022-07-19 22:31:51 +02:00
|
|
|
logging.Infof(statusInfo)
|
2022-07-11 20:11:21 +02:00
|
|
|
f, err := os.Open(filename)
|
|
|
|
if err != nil {
|
|
|
|
aerc.PushError(err.Error())
|
|
|
|
return
|
|
|
|
}
|
|
|
|
defer f.Close()
|
|
|
|
|
|
|
|
messages, err := mboxer.Read(f)
|
|
|
|
if err != nil {
|
|
|
|
aerc.PushError(err.Error())
|
|
|
|
return
|
|
|
|
}
|
|
|
|
worker := acct.Worker()
|
|
|
|
|
|
|
|
var appended uint32
|
|
|
|
for i, m := range messages {
|
|
|
|
done := make(chan bool)
|
|
|
|
var retries int = 4
|
|
|
|
for retries > 0 {
|
|
|
|
var buf bytes.Buffer
|
|
|
|
r, err := m.NewReader()
|
|
|
|
if err != nil {
|
2022-07-19 22:31:51 +02:00
|
|
|
logging.Errorf("could not get reader for uid %d", m.UID())
|
2022-07-11 20:11:21 +02:00
|
|
|
break
|
|
|
|
}
|
|
|
|
nbytes, _ := io.Copy(&buf, r)
|
|
|
|
worker.PostAction(&types.AppendMessage{
|
|
|
|
Destination: folder,
|
|
|
|
Flags: []models.Flag{models.SeenFlag},
|
|
|
|
Date: time.Now(),
|
|
|
|
Reader: &buf,
|
|
|
|
Length: int(nbytes),
|
|
|
|
}, func(msg types.WorkerMessage) {
|
|
|
|
switch msg := msg.(type) {
|
|
|
|
case *types.Unsupported:
|
|
|
|
errMsg := fmt.Sprintf("%s: AppendMessage is unsupported", args[0])
|
2022-07-19 22:31:51 +02:00
|
|
|
logging.Errorf(errMsg)
|
2022-07-11 20:11:21 +02:00
|
|
|
aerc.PushError(errMsg)
|
|
|
|
return
|
|
|
|
case *types.Error:
|
2022-07-19 22:31:51 +02:00
|
|
|
logging.Errorf("AppendMessage failed: %v", msg.Error)
|
2022-07-11 20:11:21 +02:00
|
|
|
done <- false
|
|
|
|
case *types.Done:
|
|
|
|
atomic.AddUint32(&appended, 1)
|
|
|
|
done <- true
|
|
|
|
}
|
|
|
|
})
|
|
|
|
|
|
|
|
select {
|
|
|
|
case ok := <-done:
|
|
|
|
if ok {
|
|
|
|
retries = 0
|
|
|
|
} else {
|
|
|
|
// error encountered; try to append again after a quick nap
|
|
|
|
retries -= 1
|
|
|
|
sleeping := time.Duration((5 - retries) * 1e9)
|
2022-07-19 22:31:51 +02:00
|
|
|
|
|
|
|
logging.Debugf("sleeping for %s before append message %d again", sleeping, i)
|
2022-07-11 20:11:21 +02:00
|
|
|
time.Sleep(sleeping)
|
|
|
|
}
|
|
|
|
case <-time.After(30 * time.Second):
|
2022-07-19 22:31:51 +02:00
|
|
|
logging.Warnf("timed-out; appended %d of %d", appended, len(messages))
|
2022-07-11 20:11:21 +02:00
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
infoStr := fmt.Sprintf("%s: imported %d of %d sucessfully.", args[0], appended, len(messages))
|
2022-07-19 22:31:51 +02:00
|
|
|
logging.Infof(infoStr)
|
2022-07-11 20:11:21 +02:00
|
|
|
aerc.SetStatus(infoStr)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(store.Uids()) > 0 {
|
|
|
|
confirm := widgets.NewSelectorDialog(
|
|
|
|
"Selected directory is not empty",
|
|
|
|
fmt.Sprintf("Import mbox file to %s anyways?", folder),
|
|
|
|
[]string{"No", "Yes"}, 0, aerc.SelectedAccountUiConfig(),
|
|
|
|
func(option string, err error) {
|
|
|
|
aerc.CloseDialog()
|
|
|
|
aerc.Invalidate()
|
2022-07-31 14:32:48 +02:00
|
|
|
if option == "Yes" {
|
2022-07-11 20:11:21 +02:00
|
|
|
go importFolder()
|
|
|
|
}
|
|
|
|
},
|
|
|
|
)
|
|
|
|
aerc.AddDialog(confirm)
|
|
|
|
} else {
|
|
|
|
go importFolder()
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func importFolderUsage(cmd string) error {
|
|
|
|
return fmt.Errorf("Usage: %s <filename>", cmd)
|
|
|
|
}
|