mirror of
https://github.com/stashapp/stash.git
synced 2025-12-06 08:26:00 +01:00
db7d45792e
* Move stashbox package under pkg * Remove StashBox from method names * Add fingerprint conversion methods to Fingerprint Refactor Fingerprints methods * Make FindSceneByFingerprints accept fingerprints not scene ids * Refactor SubmitSceneDraft to not require readers * Have SubmitFingerprints accept scenes Remove SceneReader dependency * Move ScrapedScene to models package * Move ScrapedImage into models package * Move ScrapedGallery into models package * Move Scene relationship matching out of stashbox package This is now expected to be done in the client code * Remove TagFinder dependency from stashbox.Client * Make stashbox scene find full hierarchy of studios * Move studio resolution into separate method * Move studio matching out of stashbox package This is now client code responsibility * Move performer matching out of FindPerformerByID and FindPerformerByName * Refactor performer querying logic and remove unused stashbox models Renames FindStashBoxPerformersByPerformerNames to QueryPerformers and accepts names instead of performer ids * Refactor SubmitPerformerDraft to not load relationships This will be the responsibility of the calling code * Remove repository references
78 lines
1.6 KiB
Go
78 lines
1.6 KiB
Go
package stashbox
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stashapp/stash/pkg/models"
|
|
"github.com/stashapp/stash/pkg/stashbox/graphql"
|
|
)
|
|
|
|
func (c Client) resolveStudio(ctx context.Context, s *graphql.StudioFragment) (*models.ScrapedStudio, error) {
|
|
scraped := studioFragmentToScrapedStudio(*s)
|
|
|
|
if s.Parent != nil {
|
|
parentStudio, err := c.client.FindStudio(ctx, &s.Parent.ID, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if parentStudio.FindStudio == nil {
|
|
return scraped, nil
|
|
}
|
|
|
|
scraped.Parent, err = c.resolveStudio(ctx, parentStudio.FindStudio)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return scraped, nil
|
|
}
|
|
|
|
func (c Client) FindStudio(ctx context.Context, query string) (*models.ScrapedStudio, error) {
|
|
var studio *graphql.FindStudio
|
|
|
|
_, err := uuid.Parse(query)
|
|
if err == nil {
|
|
// Confirmed the user passed in a Stash ID
|
|
studio, err = c.client.FindStudio(ctx, &query, nil)
|
|
} else {
|
|
// Otherwise assume they're searching on a name
|
|
studio, err = c.client.FindStudio(ctx, nil, &query)
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var ret *models.ScrapedStudio
|
|
if studio.FindStudio != nil {
|
|
ret, err = c.resolveStudio(ctx, studio.FindStudio)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
func studioFragmentToScrapedStudio(s graphql.StudioFragment) *models.ScrapedStudio {
|
|
images := []string{}
|
|
for _, image := range s.Images {
|
|
images = append(images, image.URL)
|
|
}
|
|
|
|
st := &models.ScrapedStudio{
|
|
Name: s.Name,
|
|
URL: findURL(s.Urls, "HOME"),
|
|
Images: images,
|
|
RemoteSiteID: &s.ID,
|
|
}
|
|
|
|
if len(st.Images) > 0 {
|
|
st.Image = &st.Images[0]
|
|
}
|
|
|
|
return st
|
|
}
|