76 lines
1.7 KiB
Go
76 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/micromata/swd/app"
|
|
log "github.com/sirupsen/logrus"
|
|
"golang.org/x/net/webdav"
|
|
"net/http"
|
|
"errors"
|
|
)
|
|
|
|
func main() {
|
|
log.SetFormatter(&log.TextFormatter{})
|
|
|
|
config := app.ParseConfig()
|
|
|
|
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) {
|
|
var err error
|
|
defer func() {
|
|
r := recover()
|
|
switch t := r.(type) {
|
|
case string:
|
|
err = errors.New(t)
|
|
case error:
|
|
err = t
|
|
default:
|
|
err = errors.New("Unknown error")
|
|
}
|
|
|
|
log.WithError(err).Error("An error occurred handling a webdav request")
|
|
http.Error(w, "Internal server error", http.StatusInternalServerError)
|
|
}()
|
|
|
|
handler.ServeHTTP(w, r)
|
|
})
|
|
}
|