add functionality to clean button

This commit is contained in:
daposy
2019-10-16 16:45:17 -07:00
parent c8e0ac7d20
commit 6aa54c2775
5 changed files with 102 additions and 3 deletions

42
pkg/manager/task_clean.go Normal file
View File

@@ -0,0 +1,42 @@
package manager
import (
"context"
"github.com/stashapp/stash/pkg/database"
"github.com/stashapp/stash/pkg/logger"
"github.com/stashapp/stash/pkg/models"
"os"
"strconv"
"sync"
)
type CleanTask struct {
Scene models.Scene
}
func (t *CleanTask) Start(wg *sync.WaitGroup) {
defer wg.Done()
if t.fileExists(t.Scene.Path) {
logger.Debugf("Found: %s", t.Scene.Path)
} else {
logger.Debugf("Deleting missing file: %s", t.Scene.Path)
t.deleteScene(strconv.Itoa(t.Scene.ID))
}
}
func (t *CleanTask) deleteScene(id string) {
ctx := context.TODO()
qb := models.NewSceneQueryBuilder()
tx := database.DB.MustBeginTx(ctx, nil)
defer tx.Commit()
qb.Destroy(strconv.Itoa(t.Scene.ID), tx)
}
func (t *CleanTask) fileExists(filename string) bool {
info, err := os.Stat(filename)
if os.IsNotExist(err) {
return false
}
return !info.IsDir()
}