1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
package koushincarddav
import (
"fmt"
"net/http"
"net/url"
"git.sr.ht/~emersion/koushin"
koushinbase "git.sr.ht/~emersion/koushin/plugins/base"
"github.com/emersion/go-vcard"
"github.com/emersion/go-webdav/carddav"
)
type authRoundTripper struct {
upstream http.RoundTripper
session *koushin.Session
}
func (rt *authRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
rt.session.SetHTTPBasicAuth(req)
return rt.upstream.RoundTrip(req)
}
func newPlugin(srv *koushin.Server) (koushin.Plugin, error) {
u, err := srv.Upstream("https", "http+insecure")
if _, ok := err.(*koushin.NoUpstreamError); ok {
srv.Logger().Print("carddav: no upstream server provided")
return nil, nil
} else if err != nil {
return nil, fmt.Errorf("carddav: failed to parse upstream CardDAV server: %v", err)
}
if u.Scheme == "http+insecure" {
u.Scheme = "http"
}
if u.Scheme == "" {
s, err := carddav.Discover(u.Host)
if err != nil {
srv.Logger().Printf("carddav: failed to discover CardDAV server: %v", err)
return nil, nil
}
u, err = url.Parse(s)
if err != nil {
return nil, fmt.Errorf("carddav: Discover returned an invalid URL: %v", err)
}
}
srv.Logger().Printf("Configured upstream CardDAV server: %v", u)
p := koushin.GoPlugin{Name: "carddav"}
p.Inject("compose.html", func(ctx *koushin.Context, _data koushin.RenderData) error {
data := _data.(*koushinbase.ComposeRenderData)
rt := authRoundTripper{
upstream: http.DefaultTransport,
session: ctx.Session,
}
c, err := carddav.NewClient(&http.Client{Transport: &rt}, u.String())
if err != nil {
return fmt.Errorf("failed to create CardDAV client: %v", err)
}
principal, err := c.FindCurrentUserPrincipal()
if err != nil {
return fmt.Errorf("failed to query CardDAV principal: %v", err)
}
addressBookHomeSet, err := c.FindAddressBookHomeSet(principal)
if err != nil {
return fmt.Errorf("failed to query CardDAV address book home set: %v", err)
}
addressBooks, err := c.FindAddressBooks(addressBookHomeSet)
if err != nil {
return fmt.Errorf("failed to query CardDAV address books: %v", err)
}
if len(addressBooks) == 0 {
return nil
}
addressBook := addressBooks[0]
query := carddav.AddressBookQuery{
DataRequest: carddav.AddressDataRequest{
Props: []string{vcard.FieldFormattedName, vcard.FieldEmail},
},
}
addrs, err := c.QueryAddressBook(addressBook.Path, &query)
if err != nil {
return fmt.Errorf("failed to query CardDAV addresses: %v", err)
}
// TODO: cache the results
emails := make([]string, 0, len(addrs))
for _, addr := range addrs {
cardEmails := addr.Card.Values(vcard.FieldEmail)
emails = append(emails, cardEmails...)
}
data.Extra["EmailSuggestions"] = emails
return nil
})
return p.Plugin(), nil
}
func init() {
koushin.RegisterPluginLoader(func(s *koushin.Server) ([]koushin.Plugin, error) {
p, err := newPlugin(s)
if err != nil {
return nil, err
}
if p == nil {
return nil, nil
}
return []koushin.Plugin{p}, err
})
}
|