100 lines
2.5 KiB
Go
100 lines
2.5 KiB
Go
|
package imap
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net/url"
|
||
|
|
"strconv"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"git.sr.ht/~rjarry/aerc/worker/types"
|
||
|
|
"golang.org/x/oauth2"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (w *IMAPWorker) handleConfigure(msg *types.Configure) error {
|
||
|
|
u, err := url.Parse(msg.Config.Source)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
w.config.scheme = u.Scheme
|
||
|
|
if strings.HasSuffix(w.config.scheme, "+insecure") {
|
||
|
|
w.config.scheme = strings.TrimSuffix(w.config.scheme, "+insecure")
|
||
|
|
w.config.insecure = true
|
||
|
|
}
|
||
|
|
|
||
|
|
if strings.HasSuffix(w.config.scheme, "+oauthbearer") {
|
||
|
|
w.config.scheme = strings.TrimSuffix(w.config.scheme, "+oauthbearer")
|
||
|
|
w.config.oauthBearer.Enabled = true
|
||
|
|
q := u.Query()
|
||
|
|
|
||
|
|
oauth2 := &oauth2.Config{}
|
||
|
|
if q.Get("token_endpoint") != "" {
|
||
|
|
oauth2.ClientID = q.Get("client_id")
|
||
|
|
oauth2.ClientSecret = q.Get("client_secret")
|
||
|
|
oauth2.Scopes = []string{q.Get("scope")}
|
||
|
|
oauth2.Endpoint.TokenURL = q.Get("token_endpoint")
|
||
|
|
}
|
||
|
|
w.config.oauthBearer.OAuth2 = oauth2
|
||
|
|
}
|
||
|
|
|
||
|
|
w.config.addr = u.Host
|
||
|
|
if !strings.ContainsRune(w.config.addr, ':') {
|
||
|
|
w.config.addr += ":" + w.config.scheme
|
||
|
|
}
|
||
|
|
|
||
|
|
w.config.user = u.User
|
||
|
|
w.config.folders = msg.Config.Folders
|
||
|
|
w.config.idle_timeout = 10 * time.Second
|
||
|
|
w.config.connection_timeout = 30 * time.Second
|
||
|
|
w.config.keepalive_period = 0 * time.Second
|
||
|
|
w.config.keepalive_probes = 3
|
||
|
|
w.config.keepalive_interval = 3
|
||
|
|
for key, value := range msg.Config.Params {
|
||
|
|
switch key {
|
||
|
|
case "idle-timeout":
|
||
|
|
val, err := time.ParseDuration(value)
|
||
|
|
if err != nil || val < 0 {
|
||
|
|
return fmt.Errorf(
|
||
|
|
"invalid idle-timeout value %v: %v",
|
||
|
|
value, err)
|
||
|
|
}
|
||
|
|
w.config.idle_timeout = val
|
||
|
|
case "connection-timeout":
|
||
|
|
val, err := time.ParseDuration(value)
|
||
|
|
if err != nil || val < 0 {
|
||
|
|
return fmt.Errorf(
|
||
|
|
"invalid connection-timeout value %v: %v",
|
||
|
|
value, err)
|
||
|
|
}
|
||
|
|
w.config.connection_timeout = val
|
||
|
|
case "keepalive-period":
|
||
|
|
val, err := time.ParseDuration(value)
|
||
|
|
if err != nil || val < 0 {
|
||
|
|
return fmt.Errorf(
|
||
|
|
"invalid keepalive-period value %v: %v",
|
||
|
|
value, err)
|
||
|
|
}
|
||
|
|
w.config.keepalive_period = val
|
||
|
|
case "keepalive-probes":
|
||
|
|
val, err := strconv.Atoi(value)
|
||
|
|
if err != nil || val < 0 {
|
||
|
|
return fmt.Errorf(
|
||
|
|
"invalid keepalive-probes value %v: %v",
|
||
|
|
value, err)
|
||
|
|
}
|
||
|
|
w.config.keepalive_probes = val
|
||
|
|
case "keepalive-interval":
|
||
|
|
val, err := time.ParseDuration(value)
|
||
|
|
if err != nil || val < 0 {
|
||
|
|
return fmt.Errorf(
|
||
|
|
"invalid keepalive-interval value %v: %v",
|
||
|
|
value, err)
|
||
|
|
}
|
||
|
|
w.config.keepalive_interval = int(val.Seconds())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|