aboutsummaryrefslogtreecommitdiff
path: root/commands/msg
diff options
context:
space:
mode:
Diffstat (limited to 'commands/msg')
-rw-r--r--commands/msg/unsubscribe.go103
-rw-r--r--commands/msg/unsubscribe_test.go41
2 files changed, 144 insertions, 0 deletions
diff --git a/commands/msg/unsubscribe.go b/commands/msg/unsubscribe.go
new file mode 100644
index 0000000..d4a7e9a
--- /dev/null
+++ b/commands/msg/unsubscribe.go
@@ -0,0 +1,103 @@
+package msg
+
+import (
+ "bufio"
+ "errors"
+ "net/url"
+ "strings"
+
+ "git.sr.ht/~sircmpwn/aerc/lib"
+ "git.sr.ht/~sircmpwn/aerc/widgets"
+)
+
+// Unsubscribe helps people unsubscribe from mailing lists by way of the
+// List-Unsubscribe header.
+type Unsubscribe struct{}
+
+func init() {
+ register(Unsubscribe{})
+}
+
+// Aliases returns a list of aliases for the :unsubscribe command
+func (Unsubscribe) Aliases() []string {
+ return []string{"unsubscribe"}
+}
+
+// Complete returns a list of completions
+func (Unsubscribe) Complete(aerc *widgets.Aerc, args []string) []string {
+ return nil
+}
+
+// Execute runs the Unsubscribe command
+func (Unsubscribe) Execute(aerc *widgets.Aerc, args []string) error {
+ if len(args) != 1 {
+ return errors.New("Usage: unsubscribe")
+ }
+ widget := aerc.SelectedTab().(widgets.ProvidesMessage)
+ headers := widget.SelectedMessage().RFC822Headers
+ if !headers.Has("list-unsubscribe") {
+ return errors.New("No List-Unsubscribe header found")
+ }
+ methods := parseUnsubscribeMethods(headers.Get("list-unsubscribe"))
+ aerc.Logger().Printf("found %d unsubscribe methods", len(methods))
+ for _, method := range methods {
+ aerc.Logger().Printf("trying to unsubscribe using %v", method)
+ switch method.Scheme {
+ case "mailto":
+ return unsubscribeMailto(aerc, method)
+ case "http", "https":
+ return unsubscribeHTTP(method)
+ default:
+ aerc.Logger().Printf("skipping unrecognized scheme: %s", method.Scheme)
+ }
+ }
+ return errors.New("no supported unsubscribe methods found")
+}
+
+// parseUnsubscribeMethods reads the list-unsubscribe header and parses it as a
+// list of angle-bracket <> deliminated URLs. See RFC 2369.
+func parseUnsubscribeMethods(header string) (methods []*url.URL) {
+ r := bufio.NewReader(strings.NewReader(header))
+ for {
+ // discard until <
+ _, err := r.ReadSlice('<')
+ if err != nil {
+ return
+ }
+ // read until <
+ m, err := r.ReadSlice('>')
+ if err != nil {
+ return
+ }
+ m = m[:len(m)-1]
+ if u, err := url.Parse(string(m)); err == nil {
+ methods = append(methods, u)
+ }
+ }
+}
+
+func unsubscribeMailto(aerc *widgets.Aerc, u *url.URL) error {
+ widget := aerc.SelectedTab().(widgets.ProvidesMessage)
+ acct := widget.SelectedAccount()
+ composer := widgets.NewComposer(aerc.Config(), acct.AccountConfig(),
+ acct.Worker())
+ composer.Defaults(map[string]string{
+ "To": u.Opaque,
+ "Subject": u.Query().Get("subject"),
+ })
+ composer.SetContents(strings.NewReader(u.Query().Get("body")))
+ tab := aerc.NewTab(composer, "unsubscribe")
+ composer.OnSubjectChange(func(subject string) {
+ if subject == "" {
+ tab.Name = "unsubscribe"
+ } else {
+ tab.Name = subject
+ }
+ tab.Content.Invalidate()
+ })
+ return nil
+}
+
+func unsubscribeHTTP(u *url.URL) error {
+ return lib.OpenFile(u.String())
+}
diff --git a/commands/msg/unsubscribe_test.go b/commands/msg/unsubscribe_test.go
new file mode 100644
index 0000000..e4e6f25
--- /dev/null
+++ b/commands/msg/unsubscribe_test.go
@@ -0,0 +1,41 @@
+package msg
+
+import (
+ "testing"
+)
+
+func TestParseUnsubscribe(t *testing.T) {
+ type tc struct {
+ hdr string
+ expected []string
+ }
+ cases := []*tc{
+ &tc{"", []string{}},
+ &tc{"invalid", []string{}},
+ &tc{"<https://example.com>, <http://example.com>", []string{
+ "https://example.com", "http://example.com",
+ }},
+ &tc{"<https://example.com> is a URL", []string{
+ "https://example.com",
+ }},
+ &tc{"<mailto:user@host?subject=unsubscribe>, <https://example.com>",
+ []string{
+ "mailto:user@host?subject=unsubscribe", "https://example.com",
+ }},
+ &tc{"<>, <https://example> ", []string{
+ "", "https://example",
+ }},
+ }
+ for _, c := range cases {
+ result := parseUnsubscribeMethods(c.hdr)
+ if len(result) != len(c.expected) {
+ t.Errorf("expected %d methods but got %d", len(c.expected), len(result))
+ continue
+ }
+ for idx := 0; idx < len(result); idx++ {
+ if result[idx].String() != c.expected[idx] {
+ t.Errorf("expected %v but got %v", c.expected[idx], result[idx])
+ }
+ }
+ }
+}