package api import ( "net/http" "os" "path" "path/filepath" "strings" "time" "github.com/dustin/go-humanize/english" "github.com/gin-gonic/gin" "github.com/photoprism/photoprism/internal/acl" "github.com/photoprism/photoprism/internal/entity" "github.com/photoprism/photoprism/internal/event" "github.com/photoprism/photoprism/internal/form" "github.com/photoprism/photoprism/internal/i18n" "github.com/photoprism/photoprism/internal/photoprism" "github.com/photoprism/photoprism/internal/query" "github.com/photoprism/photoprism/internal/service" "github.com/photoprism/photoprism/pkg/clean" "github.com/photoprism/photoprism/pkg/fs" ) const ( UploadPath = "/upload" ) // StartImport imports media files from a directory and converts/indexes them as needed. // // POST /api/v1/import* func StartImport(router *gin.RouterGroup) { router.POST("/import/*path", func(c *gin.Context) { s := AuthAny(c, acl.ResourceFiles, acl.Permissions{acl.ActionManage, acl.ActionUpload}) if s.Abort(c) { return } conf := service.Config() if conf.ReadOnly() || !conf.Settings().Features.Import { AbortFeatureDisabled(c) return } start := time.Now() var f form.ImportOptions if err := c.BindJSON(&f); err != nil { AbortBadRequest(c) return } srcFolder := "" importPath := conf.ImportPath() // Import from sub-folder? if srcFolder = clean.Path(c.Param("path")); srcFolder != "" && srcFolder != "/" { srcFolder = strings.Replace(srcFolder, ".", "", -1) } else if f.Path != "" { srcFolder = strings.Replace(f.Path, ".", "", -1) } // To avoid conflicts, uploads are imported from "import_path/upload/session_ref/timestamp". if token := path.Base(srcFolder); token != "" && path.Dir(srcFolder) == UploadPath { srcFolder = path.Join(UploadPath, s.RefID+token) event.AuditInfo([]string{ClientIP(c), "session %s", "import uploads from %s as %s", "granted"}, s.RefID, clean.Log(srcFolder), s.User().AclRole().String()) } else if acl.Resources.Deny(acl.ResourceFiles, s.User().AclRole(), acl.ActionManage) { event.AuditErr([]string{ClientIP(c), "session %s", "import files from %s as %s", "denied"}, s.RefID, clean.Log(srcFolder), s.User().AclRole().String()) AbortForbidden(c) return } importPath = path.Join(importPath, srcFolder) imp := service.Import() RemoveFromFolderCache(entity.RootImport) var destFolder string if destFolder = s.User().UploadPath; destFolder == "" { destFolder = conf.ImportDest() } var opt photoprism.ImportOptions // Copy or move files to the destination folder? if f.Move { event.InfoMsg(i18n.MsgMovingFilesFrom, clean.Log(filepath.Base(importPath))) opt = photoprism.ImportOptionsMove(importPath, destFolder) } else { event.InfoMsg(i18n.MsgCopyingFilesFrom, clean.Log(filepath.Base(importPath))) opt = photoprism.ImportOptionsCopy(importPath, destFolder) } // Add imported files to albums if allowed. if len(f.Albums) > 0 && acl.Resources.AllowAny(acl.ResourceAlbums, s.User().AclRole(), acl.Permissions{acl.ActionCreate, acl.ActionUpload}) { log.Debugf("import: adding files to album %s", clean.Log(strings.Join(f.Albums, " and "))) opt.Albums = f.Albums } // Set user UID if known. if s.UserUID != "" { opt.UserUID = s.UserUID } // Start import. imported := imp.Start(opt) // Delete empty import directory. if srcFolder != "" && importPath != conf.ImportPath() && fs.DirIsEmpty(importPath) { if err := os.Remove(importPath); err != nil { log.Errorf("import: failed deleting empty folder %s: %s", clean.Log(importPath), err) } else { log.Infof("import: deleted empty folder %s", clean.Log(importPath)) } } // Update moments if files have been imported. if n := len(imported); n == 0 { log.Infof("import: no new files found to import", clean.Log(importPath)) } else { log.Infof("import: imported %s", english.Plural(n, "file", "files")) if moments := service.Moments(); moments == nil { log.Warnf("import: moments service not set - possible bug") } else if err := moments.Start(); err != nil { log.Warnf("moments: %s", err) } } elapsed := int(time.Since(start).Seconds()) // Show success message. msg := i18n.Msg(i18n.MsgImportCompletedIn, elapsed) event.Success(msg) event.Publish("import.completed", event.Data{"path": importPath, "seconds": elapsed}) event.Publish("index.completed", event.Data{"path": importPath, "seconds": elapsed}) for _, uid := range f.Albums { PublishAlbumEvent(EntityUpdated, uid, c) } // Update the user interface. UpdateClientConfig() // Update album, label, and subject cover thumbs. if err := query.UpdateCovers(); err != nil { log.Warnf("index: %s (update covers)", err) } c.JSON(http.StatusOK, i18n.Response{Code: http.StatusOK, Msg: msg}) }) } // CancelImport stops the current import operation. // // DELETE /api/v1/import func CancelImport(router *gin.RouterGroup) { router.DELETE("/import", func(c *gin.Context) { s := Auth(c, acl.ResourceFiles, acl.ActionManage) if s.Abort(c) { return } conf := service.Config() if conf.ReadOnly() || !conf.Settings().Features.Import { AbortFeatureDisabled(c) return } imp := service.Import() imp.Cancel() c.JSON(http.StatusOK, i18n.NewResponse(http.StatusOK, i18n.MsgImportCanceled)) }) }