46 lines
1.2 KiB
Go
46 lines
1.2 KiB
Go
package webdav
|
|
|
|
import (
|
|
"github.com/Xhofe/alist/drivers/base"
|
|
"github.com/Xhofe/alist/drivers/webdav/odrvcookie"
|
|
"github.com/Xhofe/alist/model"
|
|
"github.com/Xhofe/alist/utils"
|
|
"github.com/Xhofe/alist/utils/cookie"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/studio-b12/gowebdav"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
func (driver WebDav) NewClient(account *model.Account) *gowebdav.Client {
|
|
c := gowebdav.NewClient(account.SiteUrl, account.Username, account.Password)
|
|
if account.InternalType == "sharepoint" {
|
|
|
|
ca := odrvcookie.New(account.Username, account.Password, account.SiteUrl)
|
|
tokenConf, err := ca.Cookies()
|
|
log.Debugln(err, tokenConf)
|
|
if err != nil {
|
|
c.SetInterceptor(func(method string, rq *http.Request) {
|
|
rq.Header.Del("Authorization")
|
|
//rq.AddCookie(&tokenConf.FedAuth)
|
|
//rq.AddCookie(&tokenConf.RtFa)
|
|
rq.Header.Set("Cookie", cookie.ToString([]*http.Cookie{&tokenConf.RtFa, &tokenConf.FedAuth}))
|
|
if method == "PROPFIND" {
|
|
rq.Header.Set("Depth", "0")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
return c
|
|
}
|
|
|
|
func (driver WebDav) WebDavPath(path string) string {
|
|
path = utils.ParsePath(path)
|
|
path = strings.TrimPrefix(path, "/")
|
|
return path
|
|
}
|
|
|
|
func init() {
|
|
base.RegisterDriver(&WebDav{})
|
|
}
|