2017-09-10 09:39:55 +01:00
|
|
|
// Package carddav provides a CardDAV server implementation, as defined in
|
|
|
|
// RFC 6352.
|
2017-09-03 19:11:36 +01:00
|
|
|
package carddav
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"encoding/xml"
|
|
|
|
"net/http"
|
2017-09-11 18:10:12 +01:00
|
|
|
"net/url"
|
2017-09-03 19:11:36 +01:00
|
|
|
"os"
|
|
|
|
|
|
|
|
"github.com/emersion/go-vcard"
|
|
|
|
"github.com/emersion/go-webdav"
|
|
|
|
"golang.org/x/net/context"
|
|
|
|
|
|
|
|
"log"
|
|
|
|
)
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
var addressDataName = xml.Name{Space: "urn:ietf:params:xml:ns:carddav", Local: "address-data"}
|
2017-09-03 19:11:36 +01:00
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
type responseWriter struct {
|
|
|
|
http.ResponseWriter
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
func (w responseWriter) Write(b []byte) (int, error) {
|
|
|
|
os.Stdout.Write(b)
|
|
|
|
return w.ResponseWriter.Write(b)
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
type Handler struct {
|
2017-09-11 18:10:53 +01:00
|
|
|
ab AddressBook
|
2017-09-11 18:10:12 +01:00
|
|
|
webdav *webdav.Handler
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
func NewHandler(ab AddressBook) *Handler {
|
|
|
|
return &Handler{
|
|
|
|
ab: ab,
|
|
|
|
webdav: &webdav.Handler{
|
|
|
|
FileSystem: &fileSystem{ab},
|
|
|
|
Logger: func(req *http.Request, err error) {
|
|
|
|
if err != nil {
|
|
|
|
log.Println("ERROR", req, err)
|
|
|
|
}
|
|
|
|
},
|
|
|
|
},
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
|
|
log.Printf("%+v\n", r)
|
|
|
|
if r.Method == http.MethodOptions {
|
|
|
|
w.Header().Add("DAV", "addressbook")
|
|
|
|
}
|
2017-09-03 19:11:36 +01:00
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
w = responseWriter{w}
|
|
|
|
switch r.Method {
|
|
|
|
case "REPORT":
|
|
|
|
status, _ := h.handleReport(w, r)
|
|
|
|
if status != 0 {
|
|
|
|
w.WriteHeader(status)
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|
2017-09-11 18:10:12 +01:00
|
|
|
case "OPTIONS":
|
|
|
|
w.Header().Add("Allow", "REPORT")
|
|
|
|
fallthrough
|
|
|
|
default:
|
|
|
|
h.webdav.ServeHTTP(w, r)
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
func (h *Handler) handleReport(w http.ResponseWriter, r *http.Request) (int, error) {
|
|
|
|
var mg addressbookMultiget
|
|
|
|
if err := xml.NewDecoder(r.Body).Decode(&mg); err != nil {
|
|
|
|
return http.StatusBadRequest, err
|
2017-09-04 11:06:06 +01:00
|
|
|
}
|
2017-09-11 18:10:12 +01:00
|
|
|
log.Printf("%#v\n", mg)
|
2017-09-04 11:06:06 +01:00
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
mw := webdav.NewMultistatusWriter(w)
|
|
|
|
defer mw.Close()
|
2017-09-03 19:11:36 +01:00
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
if len(mg.Href) == 0 {
|
|
|
|
mg.Href = []string{r.URL.Path}
|
|
|
|
}
|
|
|
|
for _, href := range mg.Href {
|
|
|
|
pstats, status, _ := multiget(r.Context(), h.webdav.FileSystem, h.webdav.LockSystem, href, []xml.Name(mg.Prop), mg.Allprop != nil)
|
|
|
|
// TODO: error handling
|
|
|
|
|
|
|
|
resp := &webdav.Response{
|
|
|
|
Href: []string{(&url.URL{Path: href}).EscapedPath()},
|
|
|
|
Status: status,
|
|
|
|
Propstat: pstats,
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
if err := mw.Write(resp); err != nil {
|
|
|
|
return http.StatusInternalServerError, err
|
|
|
|
}
|
|
|
|
}
|
2017-09-04 11:06:06 +01:00
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
return 0, nil
|
|
|
|
}
|
2017-09-04 11:06:06 +01:00
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
func multiget(ctx context.Context, fs webdav.FileSystem, ls webdav.LockSystem, name string, pnames []xml.Name, allprop bool) ([]webdav.Propstat, int, error) {
|
|
|
|
wantAddressData := false
|
|
|
|
for i, pname := range pnames {
|
|
|
|
if pname == addressDataName {
|
|
|
|
pnames = append(pnames[:i], pnames[i+1:]...)
|
|
|
|
wantAddressData = true
|
|
|
|
break
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
var pstats []webdav.Propstat
|
|
|
|
var err error
|
|
|
|
if allprop {
|
|
|
|
wantAddressData = true
|
|
|
|
pstats, err = webdav.Allprop(ctx, fs, ls, name, pnames)
|
|
|
|
} else {
|
|
|
|
pstats, err = webdav.Props(ctx, fs, ls, name, pnames)
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|
2017-09-11 18:10:12 +01:00
|
|
|
if err != nil {
|
|
|
|
return pstats, http.StatusInternalServerError, err
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
// TODO: locking stuff
|
2017-09-03 19:11:36 +01:00
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
f, err := fs.OpenFile(ctx, name, os.O_RDONLY, 0)
|
2017-09-09 15:45:31 +01:00
|
|
|
if err != nil {
|
2017-09-11 18:10:12 +01:00
|
|
|
return nil, http.StatusNotFound, err
|
|
|
|
}
|
|
|
|
defer f.Close()
|
|
|
|
fi, err := f.Stat()
|
|
|
|
if err != nil {
|
|
|
|
return nil, http.StatusNotFound, err
|
2017-09-09 15:45:31 +01:00
|
|
|
}
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
if wantAddressData {
|
|
|
|
if fi.IsDir() {
|
|
|
|
// TODO
|
|
|
|
return nil, http.StatusNotFound, err
|
|
|
|
}
|
2017-09-03 19:11:36 +01:00
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
prop, status, _ := addressdata(f.(*file).ao)
|
|
|
|
if status == 0 {
|
|
|
|
status = http.StatusOK
|
|
|
|
}
|
2017-09-03 19:11:36 +01:00
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
inserted := false
|
|
|
|
for i, pstat := range pstats {
|
|
|
|
if pstat.Status == status {
|
|
|
|
pstats[i].Props = append(pstat.Props, prop)
|
|
|
|
inserted = true
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
2017-09-03 19:11:36 +01:00
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
if !inserted {
|
|
|
|
pstats = append(pstats, webdav.Propstat{
|
2017-09-11 18:10:53 +01:00
|
|
|
Props: []webdav.Property{prop},
|
2017-09-11 18:10:12 +01:00
|
|
|
Status: status,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
2017-09-03 19:11:36 +01:00
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
return pstats, 0, nil
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
func addressdata(ao AddressObject) (webdav.Property, int, error) {
|
|
|
|
prop := webdav.Property{XMLName: addressDataName}
|
2017-09-03 19:11:36 +01:00
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
card, err := ao.Card()
|
2017-09-03 19:11:36 +01:00
|
|
|
if err != nil {
|
2017-09-11 18:10:12 +01:00
|
|
|
return prop, http.StatusInternalServerError, err
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
// TODO: restrict to requested props
|
2017-09-03 19:11:36 +01:00
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
var b bytes.Buffer
|
|
|
|
if err := vcard.NewEncoder(&b).Encode(card); err != nil {
|
|
|
|
return prop, http.StatusInternalServerError, err
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
var escaped bytes.Buffer
|
|
|
|
if err := xml.EscapeText(&escaped, b.Bytes()); err != nil {
|
|
|
|
return prop, http.StatusInternalServerError, err
|
2017-09-09 15:37:49 +01:00
|
|
|
}
|
|
|
|
|
2017-09-11 18:10:12 +01:00
|
|
|
prop.InnerXML = escaped.Bytes()
|
|
|
|
return prop, 0, nil
|
2017-09-03 19:11:36 +01:00
|
|
|
}
|