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>
107 lines
2.3 KiB
Go
107 lines
2.3 KiB
Go
package manager
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
|
|
"github.com/stashapp/stash/pkg/logger"
|
|
)
|
|
|
|
type ImportDuplicateEnum string
|
|
|
|
const (
|
|
ImportDuplicateEnumIgnore ImportDuplicateEnum = "IGNORE"
|
|
ImportDuplicateEnumOverwrite ImportDuplicateEnum = "OVERWRITE"
|
|
ImportDuplicateEnumFail ImportDuplicateEnum = "FAIL"
|
|
)
|
|
|
|
var AllImportDuplicateEnum = []ImportDuplicateEnum{
|
|
ImportDuplicateEnumIgnore,
|
|
ImportDuplicateEnumOverwrite,
|
|
ImportDuplicateEnumFail,
|
|
}
|
|
|
|
func (e ImportDuplicateEnum) IsValid() bool {
|
|
switch e {
|
|
case ImportDuplicateEnumIgnore, ImportDuplicateEnumOverwrite, ImportDuplicateEnumFail:
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (e ImportDuplicateEnum) String() string {
|
|
return string(e)
|
|
}
|
|
|
|
func (e *ImportDuplicateEnum) UnmarshalGQL(v interface{}) error {
|
|
str, ok := v.(string)
|
|
if !ok {
|
|
return fmt.Errorf("enums must be strings")
|
|
}
|
|
|
|
*e = ImportDuplicateEnum(str)
|
|
if !e.IsValid() {
|
|
return fmt.Errorf("%s is not a valid ImportDuplicateEnum", str)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (e ImportDuplicateEnum) MarshalGQL(w io.Writer) {
|
|
fmt.Fprint(w, strconv.Quote(e.String()))
|
|
}
|
|
|
|
type importer interface {
|
|
PreImport(ctx context.Context) error
|
|
PostImport(ctx context.Context, id int) error
|
|
Name() string
|
|
FindExistingID(ctx context.Context) (*int, error)
|
|
Create(ctx context.Context) (*int, error)
|
|
Update(ctx context.Context, id int) error
|
|
}
|
|
|
|
func performImport(ctx context.Context, i importer, duplicateBehaviour ImportDuplicateEnum) error {
|
|
if err := i.PreImport(ctx); err != nil {
|
|
return err
|
|
}
|
|
|
|
// try to find an existing object with the same name
|
|
name := i.Name()
|
|
existing, err := i.FindExistingID(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("error finding existing objects: %v", err)
|
|
}
|
|
|
|
var id int
|
|
|
|
if existing != nil {
|
|
switch duplicateBehaviour {
|
|
case ImportDuplicateEnumFail:
|
|
return fmt.Errorf("existing object with name '%s'", name)
|
|
case ImportDuplicateEnumIgnore:
|
|
logger.Infof("Skipping existing object %q", name)
|
|
return nil
|
|
}
|
|
|
|
// must be overwriting
|
|
id = *existing
|
|
if err := i.Update(ctx, id); err != nil {
|
|
return fmt.Errorf("error updating existing object: %v", err)
|
|
}
|
|
} else {
|
|
// creating
|
|
createdID, err := i.Create(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("error creating object: %v", err)
|
|
}
|
|
|
|
id = *createdID
|
|
}
|
|
|
|
if err := i.PostImport(ctx, id); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|