mirror of https://github.com/Xhofe/alist
53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
package webdav
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"net/http"
|
|
"net/http/cookiejar"
|
|
|
|
"github.com/alist-org/alist/v3/drivers/webdav/odrvcookie"
|
|
"github.com/alist-org/alist/v3/internal/model"
|
|
"github.com/alist-org/alist/v3/pkg/gowebdav"
|
|
)
|
|
|
|
// do others that not defined in Driver interface
|
|
|
|
func (d *WebDav) isSharepoint() bool {
|
|
return d.Vendor == "sharepoint"
|
|
}
|
|
|
|
func (d *WebDav) setClient() error {
|
|
c := gowebdav.NewClient(d.Address, d.Username, d.Password)
|
|
c.SetTransport(&http.Transport{
|
|
Proxy: http.ProxyFromEnvironment,
|
|
TLSClientConfig: &tls.Config{InsecureSkipVerify: d.TlsInsecureSkipVerify},
|
|
})
|
|
if d.isSharepoint() {
|
|
cookie, err := odrvcookie.GetCookie(d.Username, d.Password, d.Address)
|
|
if err == nil {
|
|
c.SetInterceptor(func(method string, rq *http.Request) {
|
|
rq.Header.Del("Authorization")
|
|
rq.Header.Set("Cookie", cookie)
|
|
})
|
|
} else {
|
|
return err
|
|
}
|
|
} else {
|
|
cookieJar, err := cookiejar.New(nil)
|
|
if err == nil {
|
|
c.SetJar(cookieJar)
|
|
} else {
|
|
return err
|
|
}
|
|
}
|
|
d.client = c
|
|
return nil
|
|
}
|
|
|
|
func getPath(obj model.Obj) string {
|
|
if obj.IsDir() {
|
|
return obj.GetPath() + "/"
|
|
}
|
|
return obj.GetPath()
|
|
}
|