mirror of
https://github.com/OpenListTeam/OpenList.git
synced 2025-09-19 20:26:26 +08:00
fdcc2f136e
* Enable blank issue * chore(README.md): update docs (temporally) * Update FUNDING.yml * chore: purge README.md * chore: change module name to OpenListTeam/OpenList * fix: fix link errors * chore: remove v3 in module name * fix: resolve some conficts * fix: resolve conficts * docs: update with latest file --------- Co-authored-by: ShenLin <773933146@qq.com> Co-authored-by: Hantong Chen <cxwdyx620@gmail.com> Co-authored-by: joshua <i@joshua.su> Co-authored-by: Hantong Chen <70561268+cxw620@users.noreply.github.com>
110 lines
2.8 KiB
Go
110 lines
2.8 KiB
Go
package webdav
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"time"
|
|
|
|
"github.com/OpenListTeam/OpenList/internal/driver"
|
|
"github.com/OpenListTeam/OpenList/internal/model"
|
|
"github.com/OpenListTeam/OpenList/pkg/cron"
|
|
"github.com/OpenListTeam/OpenList/pkg/gowebdav"
|
|
"github.com/OpenListTeam/OpenList/pkg/utils"
|
|
)
|
|
|
|
type WebDav struct {
|
|
model.Storage
|
|
Addition
|
|
client *gowebdav.Client
|
|
cron *cron.Cron
|
|
}
|
|
|
|
func (d *WebDav) Config() driver.Config {
|
|
return config
|
|
}
|
|
|
|
func (d *WebDav) GetAddition() driver.Additional {
|
|
return &d.Addition
|
|
}
|
|
|
|
func (d *WebDav) Init(ctx context.Context) error {
|
|
err := d.setClient()
|
|
if err == nil {
|
|
d.cron = cron.NewCron(time.Hour * 12)
|
|
d.cron.Do(func() {
|
|
_ = d.setClient()
|
|
})
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (d *WebDav) Drop(ctx context.Context) error {
|
|
if d.cron != nil {
|
|
d.cron.Stop()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *WebDav) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
|
|
files, err := d.client.ReadDir(dir.GetPath())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return utils.SliceConvert(files, func(src os.FileInfo) (model.Obj, error) {
|
|
return &model.Object{
|
|
Name: src.Name(),
|
|
Size: src.Size(),
|
|
Modified: src.ModTime(),
|
|
IsFolder: src.IsDir(),
|
|
}, nil
|
|
})
|
|
}
|
|
|
|
func (d *WebDav) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
|
url, header, err := d.client.Link(file.GetPath())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &model.Link{
|
|
URL: url,
|
|
Header: header,
|
|
}, nil
|
|
}
|
|
|
|
func (d *WebDav) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
|
return d.client.MkdirAll(path.Join(parentDir.GetPath(), dirName), 0644)
|
|
}
|
|
|
|
func (d *WebDav) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
|
return d.client.Rename(getPath(srcObj), path.Join(dstDir.GetPath(), srcObj.GetName()), true)
|
|
}
|
|
|
|
func (d *WebDav) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
|
return d.client.Rename(getPath(srcObj), path.Join(path.Dir(srcObj.GetPath()), newName), true)
|
|
}
|
|
|
|
func (d *WebDav) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
|
return d.client.Copy(getPath(srcObj), path.Join(dstDir.GetPath(), srcObj.GetName()), true)
|
|
}
|
|
|
|
func (d *WebDav) Remove(ctx context.Context, obj model.Obj) error {
|
|
return d.client.RemoveAll(getPath(obj))
|
|
}
|
|
|
|
func (d *WebDav) Put(ctx context.Context, dstDir model.Obj, s model.FileStreamer, up driver.UpdateProgress) error {
|
|
callback := func(r *http.Request) {
|
|
r.Header.Set("Content-Type", s.GetMimetype())
|
|
r.ContentLength = s.GetSize()
|
|
}
|
|
reader := driver.NewLimitedUploadStream(ctx, &driver.ReaderUpdatingProgress{
|
|
Reader: s,
|
|
UpdateProgress: up,
|
|
})
|
|
err := d.client.WriteStream(path.Join(dstDir.GetPath(), s.GetName()), reader, 0644, callback)
|
|
return err
|
|
}
|
|
|
|
var _ driver.Driver = (*WebDav)(nil)
|