mirror of https://github.com/stashapp/stash.git
52 lines
1.6 KiB
Go
52 lines
1.6 KiB
Go
package autotag
|
|
|
|
import (
|
|
"github.com/stashapp/stash/pkg/match"
|
|
"github.com/stashapp/stash/pkg/models"
|
|
"github.com/stashapp/stash/pkg/scene"
|
|
)
|
|
|
|
func getSceneFileTagger(s *models.Scene, cache *match.Cache) tagger {
|
|
return tagger{
|
|
ID: s.ID,
|
|
Type: "scene",
|
|
Name: s.GetTitle(),
|
|
Path: s.Path,
|
|
cache: cache,
|
|
}
|
|
}
|
|
|
|
// ScenePerformers tags the provided scene with performers whose name matches the scene's path.
|
|
func ScenePerformers(s *models.Scene, rw models.SceneReaderWriter, performerReader models.PerformerReader, cache *match.Cache) error {
|
|
t := getSceneFileTagger(s, cache)
|
|
|
|
return t.tagPerformers(performerReader, func(subjectID, otherID int) (bool, error) {
|
|
return scene.AddPerformer(rw, subjectID, otherID)
|
|
})
|
|
}
|
|
|
|
// SceneStudios tags the provided scene with the first studio whose name matches the scene's path.
|
|
//
|
|
// Scenes will not be tagged if studio is already set.
|
|
func SceneStudios(s *models.Scene, rw models.SceneReaderWriter, studioReader models.StudioReader, cache *match.Cache) error {
|
|
if s.StudioID.Valid {
|
|
// don't modify
|
|
return nil
|
|
}
|
|
|
|
t := getSceneFileTagger(s, cache)
|
|
|
|
return t.tagStudios(studioReader, func(subjectID, otherID int) (bool, error) {
|
|
return addSceneStudio(rw, subjectID, otherID)
|
|
})
|
|
}
|
|
|
|
// SceneTags tags the provided scene with tags whose name matches the scene's path.
|
|
func SceneTags(s *models.Scene, rw models.SceneReaderWriter, tagReader models.TagReader, cache *match.Cache) error {
|
|
t := getSceneFileTagger(s, cache)
|
|
|
|
return t.tagTags(tagReader, func(subjectID, otherID int) (bool, error) {
|
|
return scene.AddTag(rw, subjectID, otherID)
|
|
})
|
|
}
|