mirror of
https://github.com/stashapp/stash.git
synced 2025-12-08 09:23:38 +01:00
add functionality to clean button
This commit is contained in:
parent
c8e0ac7d20
commit
6aa54c2775
5 changed files with 102 additions and 3 deletions
|
|
@ -28,5 +28,6 @@ func (r *queryResolver) MetadataGenerate(ctx context.Context, input models.Gener
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *queryResolver) MetadataClean(ctx context.Context) (string, error) {
|
func (r *queryResolver) MetadataClean(ctx context.Context) (string, error) {
|
||||||
panic("not implemented")
|
manager.GetInstance().Clean()
|
||||||
|
return "todo", nil
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -133,6 +133,41 @@ func (s *singleton) Generate(sprites bool, previews bool, markers bool, transcod
|
||||||
}()
|
}()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (s *singleton) Clean() {
|
||||||
|
if s.Status != Idle {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
s.Status = Clean
|
||||||
|
|
||||||
|
qb := models.NewSceneQueryBuilder()
|
||||||
|
go func() {
|
||||||
|
defer s.returnToIdleState()
|
||||||
|
|
||||||
|
logger.Infof("Starting cleaning of tracked files")
|
||||||
|
scenes, err := qb.All()
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("failed to fetch list of scenes for cleaning")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
for _, scene := range scenes {
|
||||||
|
if scene == nil {
|
||||||
|
logger.Errorf("nil scene, skipping generate")
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
wg.Add(1)
|
||||||
|
|
||||||
|
task := CleanTask{Scene: *scene}
|
||||||
|
go task.Start(&wg)
|
||||||
|
wg.Wait()
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.Info("Finished Cleaning")
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
func (s *singleton) returnToIdleState() {
|
func (s *singleton) returnToIdleState() {
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
logger.Info("recovered from ", r)
|
logger.Info("recovered from ", r)
|
||||||
|
|
|
||||||
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()
|
||||||
|
}
|
||||||
|
|
@ -7,6 +7,7 @@ import (
|
||||||
|
|
||||||
"github.com/jmoiron/sqlx"
|
"github.com/jmoiron/sqlx"
|
||||||
"github.com/stashapp/stash/pkg/database"
|
"github.com/stashapp/stash/pkg/database"
|
||||||
|
"github.com/stashapp/stash/pkg/logger"
|
||||||
)
|
)
|
||||||
|
|
||||||
const scenesForPerformerQuery = `
|
const scenesForPerformerQuery = `
|
||||||
|
|
@ -292,3 +293,19 @@ func (qb *SceneQueryBuilder) queryScenes(query string, args []interface{}, tx *s
|
||||||
|
|
||||||
return scenes, nil
|
return scenes, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (qb *SceneQueryBuilder) Destroy(id string, tx *sqlx.Tx) error {
|
||||||
|
|
||||||
|
_, err := tx.Exec("DELETE FROM performers_scenes WHERE scene_id = ?", id)
|
||||||
|
if err != nil {
|
||||||
|
logger.Debugf("error deleting performers_scenes for scene_id: %s", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
_, err = tx.Exec("DELETE FROM scenes_tags WHERE scene_id = ?", id)
|
||||||
|
if err != nil {
|
||||||
|
logger.Debugf("error deleting scenes_tags for scene_id: %s", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return executeDeleteQuery("scenes", id, tx)
|
||||||
|
}
|
||||||
|
|
|
||||||
|
|
@ -75,11 +75,15 @@ export const SettingsTasksPanel: FunctionComponent<IProps> = (props: IProps) =>
|
||||||
<H4>Generated Content</H4>
|
<H4>Generated Content</H4>
|
||||||
<GenerateButton />
|
<GenerateButton />
|
||||||
<FormGroup
|
<FormGroup
|
||||||
helperText="TODO"
|
helperText="Check for missing files and remove them from the database. This is a destructive action"
|
||||||
labelFor="clean"
|
labelFor="clean"
|
||||||
inline={true}
|
inline={true}
|
||||||
>
|
>
|
||||||
<Button id="clean" text="Clean" onClick={() => StashService.queryMetadataClean()} />
|
<Button
|
||||||
|
id="clean"
|
||||||
|
text="Clean"
|
||||||
|
onClick={() => StashService.queryMetadataClean()}
|
||||||
|
/>
|
||||||
</FormGroup>
|
</FormGroup>
|
||||||
<Divider />
|
<Divider />
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue