/* Package entity implementing sharing with WebDAV servers. Additional information can be found in our Developer Guide: https://github.com/photoprism/photoprism/wiki */ package webdav import ( "fmt" "io/ioutil" "os" "path" "time" "github.com/photoprism/photoprism/internal/event" "github.com/photoprism/photoprism/pkg/fs" "github.com/studio-b12/gowebdav" ) var log = event.Log type Client struct { client *gowebdav.Client } // New creates a new WebDAV client. func New(url, user, pass string) Client { clt := gowebdav.NewClient(url, user, pass) clt.SetTimeout(10 * time.Minute) // TODO: Change timeout if needed result := Client{client: clt} return result } func (c Client) readDir(path string) ([]os.FileInfo, error) { if path == "" { path = "/" } return c.client.ReadDir(path) } // Files returns all files in path as string slice. func (c Client) Files(dir string) (result fs.FileInfos, err error) { files, err := c.readDir(dir) if err != nil { return result, err } for _, file := range files { if !file.Mode().IsRegular() { continue } info := fs.NewFileInfo(file, dir) result = append(result, info) } return result, nil } // Directories returns all sub directories in path as string slice. func (c Client) Directories(root string, recursive bool) (result fs.FileInfos, err error) { files, err := c.readDir(root) if err != nil { return result, err } if root == "/" { root = "" } for _, file := range files { if !file.Mode().IsDir() { continue } info := fs.NewFileInfo(file, root) result = append(result, info) if recursive { subDirs, err := c.Directories(info.Abs, true) if err != nil { return result, err } result = append(result, subDirs...) } } return result, nil } // Download downloads a single file to the given location. func (c Client) Download(from, to string, force bool) error { if _, err := os.Stat(to); err == nil && !force { return fmt.Errorf("webdav: download skipped, %s already exists", to) } dir := path.Dir(to) dirInfo, err := os.Stat(dir) if err != nil { // Create directory if err := os.MkdirAll(dir, os.ModePerm); err != nil { return fmt.Errorf("webdav: can't create %s (%s)", dir, err) } } else if !dirInfo.IsDir() { return fmt.Errorf("webdav: %s is not a folder", dir) } bytes, err := c.client.Read(from) if err != nil { return err } return ioutil.WriteFile(to, bytes, 0644) } // DownloadDir downloads all files from a remote to a local directory. func (c Client) DownloadDir(from, to string, recursive, force bool) (errs []error) { files, err := c.Files(from) if err != nil { return append(errs, err) } for _, file := range files { dest := to + string(os.PathSeparator) + file.Abs if _, err := os.Stat(dest); err == nil { // File exists msg := fmt.Errorf("webdav: %s exists", dest) errs = append(errs, msg) log.Error(msg) continue } if err := c.Download(file.Abs, dest, force); err != nil { msg := fmt.Errorf("webdav: %s", err) errs = append(errs, msg) log.Error(msg) continue } } if !recursive { return errs } dirs, err := c.Directories(from, false) for _, dir := range dirs { errs = append(errs, c.DownloadDir(dir.Abs, to, true, force)...) } return errs } // CreateDir recursively creates directories if they don't exist. func (c Client) CreateDir(dir string) error { if dir == "" || dir == "/" || dir == "." { return nil } return c.client.MkdirAll(dir, os.ModePerm) } // Upload uploads a single file to the remote server. func (c Client) Upload(from, to string) error { file, err := os.Open(from) if err != nil || file == nil { return err } defer file.Close() return c.client.WriteStream(to, file, 0644) } // Delete deletes a single file or directory on a remote server. func (c Client) Delete(path string) error { return c.client.Remove(path) }