81 lines
1.8 KiB
Go
81 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/micromata/dave/app"
|
|
log "github.com/sirupsen/logrus"
|
|
"golang.org/x/net/webdav"
|
|
syslog "log"
|
|
"net/http"
|
|
)
|
|
|
|
func main() {
|
|
config := app.ParseConfig()
|
|
|
|
// Set formatter for logrus
|
|
formatter := &log.TextFormatter{}
|
|
log.SetFormatter(formatter)
|
|
|
|
// Set formatter for default log outputs
|
|
logger := log.New()
|
|
logger.Formatter = formatter
|
|
writer := logger.Writer()
|
|
defer writer.Close()
|
|
syslog.SetOutput(writer)
|
|
|
|
wdHandler := &webdav.Handler{
|
|
Prefix: config.Prefix,
|
|
FileSystem: &app.Dir{
|
|
Config: config,
|
|
},
|
|
LockSystem: webdav.NewMemLS(),
|
|
Logger: func(request *http.Request, err error) {
|
|
if config.Log.Error && err != nil {
|
|
log.Error(err)
|
|
}
|
|
},
|
|
}
|
|
|
|
a := &app.App{
|
|
Config: config,
|
|
Handler: wdHandler,
|
|
}
|
|
|
|
http.Handle("/", wrapRecovery(app.NewBasicAuthWebdavHandler(a)))
|
|
connAddr := fmt.Sprintf("%s:%s", config.Address, config.Port)
|
|
|
|
if config.TLS != nil {
|
|
log.WithFields(log.Fields{
|
|
"address": config.Address,
|
|
"port": config.Port,
|
|
"security": "TLS",
|
|
}).Info("Server is starting and listening")
|
|
log.Fatal(http.ListenAndServeTLS(connAddr, config.TLS.CertFile, config.TLS.KeyFile, nil))
|
|
|
|
} else {
|
|
log.WithFields(log.Fields{
|
|
"address": config.Address,
|
|
"port": config.Port,
|
|
"security": "none",
|
|
}).Info("Server is starting and listening")
|
|
log.Fatal(http.ListenAndServe(connAddr, nil))
|
|
}
|
|
}
|
|
|
|
func wrapRecovery(handler http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
defer func() {
|
|
if err := recover(); err != nil {
|
|
switch t := err.(type) {
|
|
case string:
|
|
log.WithError(errors.New(t)).Error("An error occurred handling a webdav request")
|
|
case error:
|
|
log.WithError(t).Error("An error occurred handling a webdav request")
|
|
}
|
|
}
|
|
}()
|
|
|
|
handler.ServeHTTP(w, r)
|
|
})
|
|
}
|