aboutsummaryrefslogtreecommitdiff
path: root/worker/imap
diff options
context:
space:
mode:
authorDrew DeVault <sir@cmpwn.com>2018-02-01 18:42:03 -0500
committerDrew DeVault <sir@cmpwn.com>2018-02-01 18:42:03 -0500
commitd24e4712a45e03d79fa1ccb71d00c5b830d5a305 (patch)
tree61e2ba149d7b1197bdc5180200a6e087844131fe /worker/imap
parentee73c419507ef74a78ddd6a3466b605cba140b68 (diff)
Reduce boilerplate in worker/UI
Diffstat (limited to 'worker/imap')
-rw-r--r--worker/imap/worker.go49
1 files changed, 15 insertions, 34 deletions
diff --git a/worker/imap/worker.go b/worker/imap/worker.go
index ceff34d..9fbaf0c 100644
--- a/worker/imap/worker.go
+++ b/worker/imap/worker.go
@@ -4,7 +4,6 @@ import (
"crypto/tls"
"crypto/x509"
"fmt"
- "log"
"net/url"
"strings"
@@ -23,9 +22,6 @@ type imapClient struct {
}
type IMAPWorker struct {
- messages chan types.WorkerMessage
- actions chan types.WorkerMessage
-
config struct {
scheme string
insecure bool
@@ -33,33 +29,18 @@ type IMAPWorker struct {
user *url.Userinfo
}
+ worker *types.Worker
client *imapClient
updates chan client.Update
- logger *log.Logger
}
-func NewIMAPWorker(logger *log.Logger) *IMAPWorker {
+func NewIMAPWorker(worker *types.Worker) *IMAPWorker {
return &IMAPWorker{
- messages: make(chan types.WorkerMessage, 50),
- actions: make(chan types.WorkerMessage, 50),
- updates: make(chan client.Update, 50),
- logger: logger,
+ worker: worker,
+ updates: make(chan client.Update, 50),
}
}
-func (w *IMAPWorker) GetMessages() chan types.WorkerMessage {
- return w.messages
-}
-
-func (w *IMAPWorker) PostAction(msg types.WorkerMessage) {
- w.actions <- msg
-}
-
-func (w *IMAPWorker) postMessage(msg types.WorkerMessage) {
- w.logger.Printf("=> %T\n", msg)
- w.messages <- msg
-}
-
func (w *IMAPWorker) verifyPeerCert(msg types.WorkerMessage) func(
rawCerts [][]byte, _ [][]*x509.Certificate) error {
@@ -77,9 +58,9 @@ func (w *IMAPWorker) verifyPeerCert(msg types.WorkerMessage) func(
Message: types.RespondTo(msg),
CertPool: pool,
}
- w.postMessage(request)
+ w.worker.PostMessage(request, nil)
- response := <-w.actions
+ response := <-w.worker.Actions
if response.InResponseTo() != request {
return fmt.Errorf("Expected UI to answer cert request")
}
@@ -176,24 +157,24 @@ func (w *IMAPWorker) handleMessage(msg types.WorkerMessage) error {
func (w *IMAPWorker) Run() {
for {
select {
- case msg := <-w.actions:
- w.logger.Printf("<= %T\n", msg)
+ case msg := <-w.worker.Actions:
+ msg = w.worker.ProcessAction(msg)
if err := w.handleMessage(msg); err == errUnsupported {
- w.postMessage(types.Unsupported{
+ w.worker.PostMessage(types.Unsupported{
Message: types.RespondTo(msg),
- })
+ }, nil)
} else if err != nil {
- w.postMessage(types.Error{
+ w.worker.PostMessage(types.Error{
Message: types.RespondTo(msg),
Error: err,
- })
+ }, nil)
} else {
- w.postMessage(types.Ack{
+ w.worker.PostMessage(types.Ack{
Message: types.RespondTo(msg),
- })
+ }, nil)
}
case update := <-w.updates:
- w.logger.Printf("[= %T", update)
+ w.worker.Logger.Printf("(= %T", update)
}
}
}