go-webdav/carddav/client.go

185 lines
3.9 KiB
Go
Raw Normal View History

2020-01-14 21:19:54 +00:00
package carddav
import (
2020-01-14 22:44:21 +00:00
"bytes"
2020-01-14 21:19:54 +00:00
"encoding/xml"
"net/http"
2020-01-14 22:44:21 +00:00
"github.com/emersion/go-vcard"
2020-01-14 21:19:54 +00:00
"github.com/emersion/go-webdav"
"github.com/emersion/go-webdav/internal"
)
type Client struct {
*webdav.Client
ic *internal.Client
}
func NewClient(c *http.Client, endpoint string) (*Client, error) {
wc, err := webdav.NewClient(c, endpoint)
if err != nil {
return nil, err
}
ic, err := internal.NewClient(c, endpoint)
if err != nil {
return nil, err
}
return &Client{wc, ic}, nil
}
2020-01-15 22:45:37 +00:00
func (c *Client) SetBasicAuth(username, password string) {
c.Client.SetBasicAuth(username, password)
c.ic.SetBasicAuth(username, password)
}
2020-01-14 22:13:23 +00:00
func (c *Client) FindAddressBookHomeSet(principal string) (string, error) {
2020-01-14 21:19:54 +00:00
name := xml.Name{namespace, "addressbook-home-set"}
propfind := internal.NewPropNamePropfind(name)
2020-01-14 21:19:54 +00:00
resp, err := c.ic.PropfindFlat(principal, propfind)
if err != nil {
return "", err
}
var prop addressbookHomeSet
if err := resp.DecodeProp(&prop); err != nil {
2020-01-14 21:19:54 +00:00
return "", err
}
return prop.Href, nil
}
2020-01-14 22:13:23 +00:00
func (c *Client) FindAddressBooks(addressBookHomeSet string) ([]AddressBook, error) {
resTypeName := xml.Name{"DAV:", "resourcetype"}
descName := xml.Name{namespace, "addressbook-description"}
propfind := internal.NewPropNamePropfind(resTypeName, descName)
2020-01-14 22:13:23 +00:00
2020-01-15 11:30:42 +00:00
ms, err := c.ic.Propfind(addressBookHomeSet, internal.DepthOne, propfind)
2020-01-14 22:13:23 +00:00
if err != nil {
return nil, err
}
l := make([]AddressBook, 0, len(ms.Responses))
2020-01-14 22:44:21 +00:00
for _, resp := range ms.Responses {
2020-01-14 22:13:23 +00:00
href, err := resp.Href()
if err != nil {
return nil, err
}
var resTypeProp internal.ResourceType
if err := resp.DecodeProp(&resTypeProp); err != nil {
2020-01-14 22:13:23 +00:00
return nil, err
}
if !resTypeProp.Is(addressBookName) {
continue
}
var descProp addressbookDescription
if err := resp.DecodeProp(&descProp); err != nil {
2020-01-14 22:13:23 +00:00
return nil, err
}
l = append(l, AddressBook{
Href: href,
Description: descProp.Data,
})
}
return l, nil
}
2020-01-14 22:44:21 +00:00
func decodeAddressList(ms *internal.Multistatus) ([]Address, error) {
addrs := make([]Address, 0, len(ms.Responses))
for _, resp := range ms.Responses {
href, err := resp.Href()
if err != nil {
return nil, err
}
var addrData addressDataResp
if err := resp.DecodeProp(&addrData); err != nil {
return nil, err
}
r := bytes.NewReader(addrData.Data)
card, err := vcard.NewDecoder(r).Decode()
if err != nil {
return nil, err
}
addrs = append(addrs, Address{
Href: href,
Card: card,
})
}
return addrs, nil
}
2020-01-14 22:44:21 +00:00
func (c *Client) QueryAddressBook(addressBook string, query *AddressBookQuery) ([]Address, error) {
var addrDataReq addressDataReq
if query != nil {
for _, name := range query.Props {
addrDataReq.Props = append(addrDataReq.Props, prop{Name: name})
}
2020-01-14 22:44:21 +00:00
}
propReq, err := internal.EncodeProp(&addrDataReq)
if err != nil {
return nil, err
2020-01-14 22:44:21 +00:00
}
addressbookQuery := addressbookQuery{Prop: propReq}
2020-01-14 22:44:21 +00:00
req, err := c.ic.NewXMLRequest("REPORT", addressBook, &addressbookQuery)
if err != nil {
return nil, err
}
req.Header.Add("Depth", "1")
ms, err := c.ic.DoMultiStatus(req)
if err != nil {
return nil, err
}
return decodeAddressList(ms)
}
2020-01-14 22:44:21 +00:00
func (c *Client) MultiGetAddressBook(href string, multiGet *AddressBookMultiGet) ([]Address, error) {
var addrDataReq addressDataReq
if multiGet != nil {
for _, name := range multiGet.Props {
addrDataReq.Props = append(addrDataReq.Props, prop{Name: name})
2020-01-14 22:44:21 +00:00
}
}
2020-01-14 22:44:21 +00:00
propReq, err := internal.EncodeProp(&addrDataReq)
if err != nil {
return nil, err
}
2020-01-14 22:44:21 +00:00
addressbookMultiget := addressbookMultiget{Prop: propReq}
if multiGet == nil || len(multiGet.Hrefs) == 0 {
addressbookMultiget.Hrefs = []string{href}
} else {
addressbookMultiget.Hrefs = multiGet.Hrefs
2020-01-14 22:44:21 +00:00
}
req, err := c.ic.NewXMLRequest("REPORT", href, &addressbookMultiget)
if err != nil {
return nil, err
}
req.Header.Add("Depth", "1")
ms, err := c.ic.DoMultiStatus(req)
if err != nil {
return nil, err
}
return decodeAddressList(ms)
2020-01-14 22:44:21 +00:00
}