mirror of
https://github.com/stashapp/stash.git
synced 2026-05-09 05:05:29 +02:00
2b29207f1e
* Bump go version in go.mod * Update compiler image. Changed github download url since existing one didn't have version 12 of the SDK. * Update macOS requirements in README for v0.32.0 * Update lint action * Bump golangci-lint version * Migrate golangci-lint config * Fix QF1012 errors (Use fmt.Fprintf(...) instead of WriteString(fmt.Sprintf(...))) * Fix QF1003 errors (could use tagged switch) * Fix ST1005 errors (error string capitalisation) * Fix ST1011 errors (seconds suffix) * Fix QF1006 errors (lift into loop condition) * Fix QF1002 errors (switch condition) * Fix gocritic error (deprecated paragraph) * Fix incorrect nolint directive * Ignore specific checks noctx should be addressed in a later PR --------- Co-authored-by: DogmaDragon <103123951+DogmaDragon@users.noreply.github.com> Co-authored-by: feederbox826 <me@feederbox.cc>
106 lines
3.1 KiB
Go
106 lines
3.1 KiB
Go
package identify
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
|
|
"github.com/stashapp/stash/pkg/models"
|
|
"github.com/stashapp/stash/pkg/scraper"
|
|
)
|
|
|
|
type Source struct {
|
|
Source *scraper.Source `json:"source"`
|
|
// Options defined for a source override the defaults
|
|
Options *MetadataOptions `json:"options"`
|
|
}
|
|
|
|
type Options struct {
|
|
// An ordered list of sources to identify items with. Only the first source that finds a match is used.
|
|
Sources []*Source `json:"sources"`
|
|
// Options defined here override the configured defaults
|
|
Options *MetadataOptions `json:"options"`
|
|
// scene ids to identify
|
|
SceneIDs []string `json:"sceneIDs"`
|
|
// paths of scenes to identify - ignored if scene ids are set
|
|
Paths []string `json:"paths"`
|
|
}
|
|
|
|
type MetadataOptions struct {
|
|
// any fields missing from here are defaulted to MERGE and createMissing false
|
|
FieldOptions []*FieldOptions `json:"fieldOptions"`
|
|
// defaults to true if not provided
|
|
SetCoverImage *bool `json:"setCoverImage"`
|
|
SetOrganized *bool `json:"setOrganized"`
|
|
// defaults to true if not provided
|
|
|
|
// Deprecated: use PerformerGenders instead
|
|
IncludeMalePerformers *bool `json:"includeMalePerformers"`
|
|
|
|
// Filter to only include performers with these genders. If not provided, all genders are included.
|
|
PerformerGenders []models.GenderEnum `json:"performerGenders"`
|
|
// defaults to true if not provided
|
|
SkipMultipleMatches *bool `json:"skipMultipleMatches"`
|
|
// ID of tag to tag skipped multiple matches with
|
|
SkipMultipleMatchTag *string `json:"skipMultipleMatchTag"`
|
|
// defaults to true if not provided
|
|
SkipSingleNamePerformers *bool `json:"skipSingleNamePerformers"`
|
|
// ID of tag to tag skipped single name performers with
|
|
SkipSingleNamePerformerTag *string `json:"skipSingleNamePerformerTag"`
|
|
}
|
|
|
|
type FieldOptions struct {
|
|
Field string `json:"field"`
|
|
Strategy FieldStrategy `json:"strategy"`
|
|
// creates missing objects if needed - only applicable for performers, tags and studios
|
|
CreateMissing *bool `json:"createMissing"`
|
|
}
|
|
|
|
type FieldStrategy string
|
|
|
|
const (
|
|
// Never sets the field value
|
|
FieldStrategyIgnore FieldStrategy = "IGNORE"
|
|
// For multi-value fields, merge with existing.
|
|
// For single-value fields, ignore if already set
|
|
FieldStrategyMerge FieldStrategy = "MERGE"
|
|
// Always replaces the value if a value is found.
|
|
// For multi-value fields, any existing values are removed and replaced with the
|
|
// scraped values.
|
|
FieldStrategyOverwrite FieldStrategy = "OVERWRITE"
|
|
)
|
|
|
|
var AllFieldStrategy = []FieldStrategy{
|
|
FieldStrategyIgnore,
|
|
FieldStrategyMerge,
|
|
FieldStrategyOverwrite,
|
|
}
|
|
|
|
func (e FieldStrategy) IsValid() bool {
|
|
switch e {
|
|
case FieldStrategyIgnore, FieldStrategyMerge, FieldStrategyOverwrite:
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (e FieldStrategy) String() string {
|
|
return string(e)
|
|
}
|
|
|
|
func (e *FieldStrategy) UnmarshalGQL(v interface{}) error {
|
|
str, ok := v.(string)
|
|
if !ok {
|
|
return fmt.Errorf("enums must be strings")
|
|
}
|
|
|
|
*e = FieldStrategy(str)
|
|
if !e.IsValid() {
|
|
return fmt.Errorf("%s is not a valid IdentifyFieldStrategy", str)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (e FieldStrategy) MarshalGQL(w io.Writer) {
|
|
fmt.Fprint(w, strconv.Quote(e.String()))
|
|
}
|