mirror of
https://github.com/tickstep/aliyunpan.git
synced 2025-02-03 05:47:16 +08:00
92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package webdav
|
||
|
||
import (
|
||
"github.com/tickstep/aliyunpan/internal/config"
|
||
"github.com/tickstep/library-go/logger"
|
||
"golang.org/x/net/webdav"
|
||
"log"
|
||
"net"
|
||
"net/http"
|
||
"strconv"
|
||
"strings"
|
||
)
|
||
|
||
type WebdavUser struct {
|
||
Username string `json:"username"`
|
||
Password string `json:"password"`
|
||
Scope string `json:"scope"`
|
||
}
|
||
|
||
type WebdavConfig struct {
|
||
// 指定Webdav使用哪个账号的云盘资源
|
||
PanUserId string `json:"panUserId"`
|
||
PanDriveId string `json:"panDriveId"`
|
||
PanUser *config.PanUser `json:"-"`
|
||
UploadChunkSize int `json:"uploadChunkSize"` // 上传文件分片数据块大小，该数值不能太小，建议大于等于512KB
|
||
TransferUrlType int `json:"transferUrlType"` // 上传/下载URL类别，1-默认，2-阿里云ECS
|
||
|
||
Address string `json:"address"`
|
||
Port int `json:"port"`
|
||
Prefix string `json:"prefix"`
|
||
Users []WebdavUser `json:"users"`
|
||
}
|
||
|
||
func (w *WebdavConfig) StartServer() {
|
||
users := map[string]*User{}
|
||
for _,u := range w.Users {
|
||
fileItem,e := w.PanUser.PanClient().FileInfoByPath(w.PanDriveId, u.Scope)
|
||
if e != nil {
|
||
logger.Verboseln("scope not existed, shutting server")
|
||
return
|
||
}
|
||
wdfi := NewWebDavFileInfo(fileItem)
|
||
if wdfi.fullPath != "/" && strings.Index(wdfi.fullPath, "/") != 0 {
|
||
wdfi.fullPath = "/" + wdfi.fullPath
|
||
}
|
||
panClientProxy := &PanClientProxy{
|
||
PanUser: w.PanUser,
|
||
PanDriveId: w.PanDriveId,
|
||
PanTransferUrlType: w.TransferUrlType,
|
||
}
|
||
users[u.Username] = &User{
|
||
Username: u.Username,
|
||
Password: u.Password,
|
||
Scope: u.Scope,
|
||
Modify: true,
|
||
Rules: nil,
|
||
Handler: &webdav.Handler{
|
||
Prefix: w.Prefix,
|
||
FileSystem: WebDavDir{
|
||
Dir: webdav.Dir(u.Scope),
|
||
NoSniff: false,
|
||
panClientProxy: panClientProxy,
|
||
fileInfo: wdfi,
|
||
uploadChunkSize: w.UploadChunkSize,
|
||
},
|
||
LockSystem: webdav.NewMemLS(),
|
||
},
|
||
}
|
||
// load & cache root folder info
|
||
_, _ = panClientProxy.FileListGetAll(u.Scope)
|
||
}
|
||
cfg := &Config{
|
||
Auth: true,
|
||
NoSniff: false,
|
||
Cors: CorsCfg{
|
||
Enabled: false,
|
||
Credentials: false,
|
||
},
|
||
Users: users,
|
||
LogFormat: "",
|
||
}
|
||
|
||
listener, err := net.Listen("tcp", w.Address + ":" + strconv.Itoa(w.Port))
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
if err := http.Serve(listener, cfg); err != nil {
|
||
logger.Verboseln("shutting server", err)
|
||
}
|
||
}
|
||
|