83 lines
1.8 KiB
Go
83 lines
1.8 KiB
Go
|
package workers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/photoprism/photoprism/internal/entity"
|
||
|
|
"github.com/photoprism/photoprism/internal/mutex"
|
||
|
|
"github.com/photoprism/photoprism/internal/remote"
|
||
|
|
"github.com/photoprism/photoprism/internal/remote/webdav"
|
||
|
|
"github.com/photoprism/photoprism/pkg/fs"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Updates the local list of remote files so that they can be downloaded in batches
|
||
|
|
func (s *Sync) refresh(a entity.Account) (complete bool, err error) {
|
||
|
|
if a.AccType != remote.ServiceWebDAV {
|
||
|
|
return false, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
db := s.conf.Db()
|
||
|
|
client := webdav.New(a.AccURL, a.AccUser, a.AccPass)
|
||
|
|
|
||
|
|
subDirs, err := client.Directories(a.SyncPath, true)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
log.Error(err)
|
||
|
|
return false, err
|
||
|
|
}
|
||
|
|
|
||
|
|
dirs := append(subDirs.Abs(), a.SyncPath)
|
||
|
|
|
||
|
|
for _, dir := range dirs {
|
||
|
|
if mutex.Sync.Canceled() {
|
||
|
|
return false, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
files, err := client.Files(dir)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
log.Error(err)
|
||
|
|
return false, err
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, file := range files {
|
||
|
|
if mutex.Sync.Canceled() {
|
||
|
|
return false, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
f := entity.NewFileSync(a.ID, file.Abs)
|
||
|
|
|
||
|
|
f.Status = entity.FileSyncIgnore
|
||
|
|
f.RemoteDate = file.Date
|
||
|
|
f.RemoteSize = file.Size
|
||
|
|
|
||
|
|
// Select supported types for download
|
||
|
|
mediaType := fs.GetMediaType(file.Name)
|
||
|
|
switch mediaType {
|
||
|
|
case fs.MediaImage:
|
||
|
|
f.Status = entity.FileSyncNew
|
||
|
|
case fs.MediaSidecar:
|
||
|
|
f.Status = entity.FileSyncNew
|
||
|
|
case fs.MediaRaw:
|
||
|
|
if a.SyncRaw {
|
||
|
|
f.Status = entity.FileSyncNew
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
f.FirstOrCreate(db)
|
||
|
|
|
||
|
|
if f.Status == entity.FileSyncIgnore && mediaType == fs.MediaRaw && a.SyncRaw {
|
||
|
|
f.Status = entity.FileSyncNew
|
||
|
|
db.Save(&f)
|
||
|
|
}
|
||
|
|
|
||
|
|
if f.Status == entity.FileSyncDownloaded && !f.RemoteDate.Equal(file.Date) {
|
||
|
|
f.Status = entity.FileSyncNew
|
||
|
|
f.RemoteDate = file.Date
|
||
|
|
f.RemoteSize = file.Size
|
||
|
|
db.Save(&f)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return true, nil
|
||
|
|
}
|