mirror of
https://github.com/stashapp/stash.git
synced 2025-12-17 12:24:38 +03:00
add functionality to clean button
This commit is contained in:
42
pkg/manager/task_clean.go
Normal file
42
pkg/manager/task_clean.go
Normal 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()
|
||||
}
|
||||
Reference in New Issue
Block a user