Add support for submitting performer/scene drafts to stash-box (#2234)

* Add support for submitting performer/scene drafts to stash-box

Co-authored-by: Kermie <kermie@isinthe.house>

pkg/scraper/stashbox/graphql/generated_client.go

@@ -31,6 +31,8 @@ type Query struct {
 	FindScenesByFingerprints     []*Scene                     "json:\"findScenesByFingerprints\" graphql:\"findScenesByFingerprints\""
 	FindScenesByFullFingerprints []*Scene                     "json:\"findScenesByFullFingerprints\" graphql:\"findScenesByFullFingerprints\""
 	QueryScenes                  QueryScenesResultType        "json:\"queryScenes\" graphql:\"queryScenes\""
+	FindSite                     *Site                        "json:\"findSite\" graphql:\"findSite\""
+	QuerySites                   QuerySitesResultType         "json:\"querySites\" graphql:\"querySites\""
 	FindEdit                     *Edit                        "json:\"findEdit\" graphql:\"findEdit\""
 	QueryEdits                   QueryEditsResultType         "json:\"queryEdits\" graphql:\"queryEdits\""
 	FindUser                     *User                        "json:\"findUser\" graphql:\"findUser\""
@@ -38,48 +40,57 @@ type Query struct {
 	Me                           *User                        "json:\"me\" graphql:\"me\""
 	SearchPerformer              []*Performer                 "json:\"searchPerformer\" graphql:\"searchPerformer\""
 	SearchScene                  []*Scene                     "json:\"searchScene\" graphql:\"searchScene\""
+	FindDraft                    *Draft                       "json:\"findDraft\" graphql:\"findDraft\""
+	FindDrafts                   []*Draft                     "json:\"findDrafts\" graphql:\"findDrafts\""
 	Version                      Version                      "json:\"version\" graphql:\"version\""
+	GetConfig                    StashBoxConfig               "json:\"getConfig\" graphql:\"getConfig\""
 }
 
 type Mutation struct {
-	SceneCreate        *Scene       "json:\"sceneCreate\" graphql:\"sceneCreate\""
-	SceneUpdate        *Scene       "json:\"sceneUpdate\" graphql:\"sceneUpdate\""
-	SceneDestroy       bool         "json:\"sceneDestroy\" graphql:\"sceneDestroy\""
-	PerformerCreate    *Performer   "json:\"performerCreate\" graphql:\"performerCreate\""
-	PerformerUpdate    *Performer   "json:\"performerUpdate\" graphql:\"performerUpdate\""
-	PerformerDestroy   bool         "json:\"performerDestroy\" graphql:\"performerDestroy\""
-	StudioCreate       *Studio      "json:\"studioCreate\" graphql:\"studioCreate\""
-	StudioUpdate       *Studio      "json:\"studioUpdate\" graphql:\"studioUpdate\""
-	StudioDestroy      bool         "json:\"studioDestroy\" graphql:\"studioDestroy\""
-	TagCreate          *Tag         "json:\"tagCreate\" graphql:\"tagCreate\""
-	TagUpdate          *Tag         "json:\"tagUpdate\" graphql:\"tagUpdate\""
-	TagDestroy         bool         "json:\"tagDestroy\" graphql:\"tagDestroy\""
-	UserCreate         *User        "json:\"userCreate\" graphql:\"userCreate\""
-	UserUpdate         *User        "json:\"userUpdate\" graphql:\"userUpdate\""
-	UserDestroy        bool         "json:\"userDestroy\" graphql:\"userDestroy\""
-	ImageCreate        *Image       "json:\"imageCreate\" graphql:\"imageCreate\""
-	ImageDestroy       bool         "json:\"imageDestroy\" graphql:\"imageDestroy\""
-	NewUser            *string      "json:\"newUser\" graphql:\"newUser\""
-	ActivateNewUser    *User        "json:\"activateNewUser\" graphql:\"activateNewUser\""
-	GenerateInviteCode string       "json:\"generateInviteCode\" graphql:\"generateInviteCode\""
-	RescindInviteCode  bool         "json:\"rescindInviteCode\" graphql:\"rescindInviteCode\""
-	GrantInvite        int          "json:\"grantInvite\" graphql:\"grantInvite\""
-	RevokeInvite       int          "json:\"revokeInvite\" graphql:\"revokeInvite\""
-	TagCategoryCreate  *TagCategory "json:\"tagCategoryCreate\" graphql:\"tagCategoryCreate\""
-	TagCategoryUpdate  *TagCategory "json:\"tagCategoryUpdate\" graphql:\"tagCategoryUpdate\""
-	TagCategoryDestroy bool         "json:\"tagCategoryDestroy\" graphql:\"tagCategoryDestroy\""
-	RegenerateAPIKey   string       "json:\"regenerateAPIKey\" graphql:\"regenerateAPIKey\""
-	ResetPassword      bool         "json:\"resetPassword\" graphql:\"resetPassword\""
-	ChangePassword     bool         "json:\"changePassword\" graphql:\"changePassword\""
-	SceneEdit          Edit         "json:\"sceneEdit\" graphql:\"sceneEdit\""
-	PerformerEdit      Edit         "json:\"performerEdit\" graphql:\"performerEdit\""
-	StudioEdit         Edit         "json:\"studioEdit\" graphql:\"studioEdit\""
-	TagEdit            Edit         "json:\"tagEdit\" graphql:\"tagEdit\""
-	EditVote           Edit         "json:\"editVote\" graphql:\"editVote\""
-	EditComment        Edit         "json:\"editComment\" graphql:\"editComment\""
-	ApplyEdit          Edit         "json:\"applyEdit\" graphql:\"applyEdit\""
-	CancelEdit         Edit         "json:\"cancelEdit\" graphql:\"cancelEdit\""
-	SubmitFingerprint  bool         "json:\"submitFingerprint\" graphql:\"submitFingerprint\""
+	SceneCreate          *Scene                "json:\"sceneCreate\" graphql:\"sceneCreate\""
+	SceneUpdate          *Scene                "json:\"sceneUpdate\" graphql:\"sceneUpdate\""
+	SceneDestroy         bool                  "json:\"sceneDestroy\" graphql:\"sceneDestroy\""
+	PerformerCreate      *Performer            "json:\"performerCreate\" graphql:\"performerCreate\""
+	PerformerUpdate      *Performer            "json:\"performerUpdate\" graphql:\"performerUpdate\""
+	PerformerDestroy     bool                  "json:\"performerDestroy\" graphql:\"performerDestroy\""
+	StudioCreate         *Studio               "json:\"studioCreate\" graphql:\"studioCreate\""
+	StudioUpdate         *Studio               "json:\"studioUpdate\" graphql:\"studioUpdate\""
+	StudioDestroy        bool                  "json:\"studioDestroy\" graphql:\"studioDestroy\""
+	TagCreate            *Tag                  "json:\"tagCreate\" graphql:\"tagCreate\""
+	TagUpdate            *Tag                  "json:\"tagUpdate\" graphql:\"tagUpdate\""
+	TagDestroy           bool                  "json:\"tagDestroy\" graphql:\"tagDestroy\""
+	UserCreate           *User                 "json:\"userCreate\" graphql:\"userCreate\""
+	UserUpdate           *User                 "json:\"userUpdate\" graphql:\"userUpdate\""
+	UserDestroy          bool                  "json:\"userDestroy\" graphql:\"userDestroy\""
+	ImageCreate          *Image                "json:\"imageCreate\" graphql:\"imageCreate\""
+	ImageDestroy         bool                  "json:\"imageDestroy\" graphql:\"imageDestroy\""
+	NewUser              *string               "json:\"newUser\" graphql:\"newUser\""
+	ActivateNewUser      *User                 "json:\"activateNewUser\" graphql:\"activateNewUser\""
+	GenerateInviteCode   *string               "json:\"generateInviteCode\" graphql:\"generateInviteCode\""
+	RescindInviteCode    bool                  "json:\"rescindInviteCode\" graphql:\"rescindInviteCode\""
+	GrantInvite          int                   "json:\"grantInvite\" graphql:\"grantInvite\""
+	RevokeInvite         int                   "json:\"revokeInvite\" graphql:\"revokeInvite\""
+	TagCategoryCreate    *TagCategory          "json:\"tagCategoryCreate\" graphql:\"tagCategoryCreate\""
+	TagCategoryUpdate    *TagCategory          "json:\"tagCategoryUpdate\" graphql:\"tagCategoryUpdate\""
+	TagCategoryDestroy   bool                  "json:\"tagCategoryDestroy\" graphql:\"tagCategoryDestroy\""
+	SiteCreate           *Site                 "json:\"siteCreate\" graphql:\"siteCreate\""
+	SiteUpdate           *Site                 "json:\"siteUpdate\" graphql:\"siteUpdate\""
+	SiteDestroy          bool                  "json:\"siteDestroy\" graphql:\"siteDestroy\""
+	RegenerateAPIKey     string                "json:\"regenerateAPIKey\" graphql:\"regenerateAPIKey\""
+	ResetPassword        bool                  "json:\"resetPassword\" graphql:\"resetPassword\""
+	ChangePassword       bool                  "json:\"changePassword\" graphql:\"changePassword\""
+	SceneEdit            Edit                  "json:\"sceneEdit\" graphql:\"sceneEdit\""
+	PerformerEdit        Edit                  "json:\"performerEdit\" graphql:\"performerEdit\""
+	StudioEdit           Edit                  "json:\"studioEdit\" graphql:\"studioEdit\""
+	TagEdit              Edit                  "json:\"tagEdit\" graphql:\"tagEdit\""
+	EditVote             Edit                  "json:\"editVote\" graphql:\"editVote\""
+	EditComment          Edit                  "json:\"editComment\" graphql:\"editComment\""
+	ApplyEdit            Edit                  "json:\"applyEdit\" graphql:\"applyEdit\""
+	CancelEdit           Edit                  "json:\"cancelEdit\" graphql:\"cancelEdit\""
+	SubmitFingerprint    bool                  "json:\"submitFingerprint\" graphql:\"submitFingerprint\""
+	SubmitSceneDraft     DraftSubmissionStatus "json:\"submitSceneDraft\" graphql:\"submitSceneDraft\""
+	SubmitPerformerDraft DraftSubmissionStatus "json:\"submitPerformerDraft\" graphql:\"submitPerformerDraft\""
+	DestroyDraft         bool                  "json:\"destroyDraft\" graphql:\"destroyDraft\""
 }
 type URLFragment struct {
 	URL  string "json:\"url\" graphql:\"url\""
@@ -185,12 +196,26 @@ type Me struct {
 		Name string "json:\"name\" graphql:\"name\""
 	} "json:\"me\" graphql:\"me\""
 }
+type SubmitSceneDraftPayload struct {
+	SubmitSceneDraft struct {
+		ID *string "json:\"id\" graphql:\"id\""
+	} "json:\"submitSceneDraft\" graphql:\"submitSceneDraft\""
+}
+type SubmitPerformerDraftPayload struct {
+	SubmitPerformerDraft struct {
+		ID *string "json:\"id\" graphql:\"id\""
+	} "json:\"submitPerformerDraft\" graphql:\"submitPerformerDraft\""
+}
 
 const FindSceneByFingerprintQuery = `query FindSceneByFingerprint ($fingerprint: FingerprintQueryInput!) {
 	findSceneByFingerprint(fingerprint: $fingerprint) {
 		... SceneFragment
 	}
 }
+fragment BodyModificationFragment on BodyModification {
+	location
+	description
+}
 fragment FingerprintFragment on Fingerprint {
 	algorithm
 	hash
@@ -200,27 +225,9 @@ fragment URLFragment on URL {
 	url
 	type
 }
-fragment ImageFragment on Image {
-	id
-	url
-	width
-	height
-}
-fragment StudioFragment on Studio {
+fragment TagFragment on Tag {
 	name
 	id
-	urls {
-		... URLFragment
-	}
-	images {
-		... ImageFragment
-	}
-}
-fragment PerformerAppearanceFragment on PerformerAppearance {
-	as
-	performer {
-		... PerformerFragment
-	}
 }
 fragment PerformerFragment on Performer {
 	id
@@ -256,9 +263,15 @@ fragment PerformerFragment on Performer {
 		... BodyModificationFragment
 	}
 }
-fragment BodyModificationFragment on BodyModification {
-	location
-	description
+fragment FuzzyDateFragment on FuzzyDate {
+	date
+	accuracy
+}
+fragment MeasurementsFragment on Measurements {
+	band_size
+	cup_size
+	waist
+	hip
 }
 fragment SceneFragment on Scene {
 	id
@@ -285,19 +298,27 @@ fragment SceneFragment on Scene {
 		... FingerprintFragment
 	}
 }
-fragment TagFragment on Tag {
+fragment ImageFragment on Image {
+	id
+	url
+	width
+	height
+}
+fragment StudioFragment on Studio {
 	name
 	id
+	urls {
+		... URLFragment
+	}
+	images {
+		... ImageFragment
+	}
 }
-fragment FuzzyDateFragment on FuzzyDate {
-	date
-	accuracy
-}
-fragment MeasurementsFragment on Measurements {
-	band_size
-	cup_size
-	waist
-	hip
+fragment PerformerAppearanceFragment on PerformerAppearance {
+	as
+	performer {
+		... PerformerFragment
+	}
 }
 `
 
@@ -367,10 +388,36 @@ fragment FuzzyDateFragment on FuzzyDate {
 	date
 	accuracy
 }
+fragment MeasurementsFragment on Measurements {
+	band_size
+	cup_size
+	waist
+	hip
+}
 fragment BodyModificationFragment on BodyModification {
 	location
 	description
 }
+fragment ImageFragment on Image {
+	id
+	url
+	width
+	height
+}
+fragment URLFragment on URL {
+	url
+	type
+}
+fragment TagFragment on Tag {
+	name
+	id
+}
+fragment PerformerAppearanceFragment on PerformerAppearance {
+	as
+	performer {
+		... PerformerFragment
+	}
+}
 fragment FingerprintFragment on Fingerprint {
 	algorithm
 	hash
@@ -401,32 +448,6 @@ fragment SceneFragment on Scene {
 		... FingerprintFragment
 	}
 }
-fragment URLFragment on URL {
-	url
-	type
-}
-fragment ImageFragment on Image {
-	id
-	url
-	width
-	height
-}
-fragment TagFragment on Tag {
-	name
-	id
-}
-fragment PerformerAppearanceFragment on PerformerAppearance {
-	as
-	performer {
-		... PerformerFragment
-	}
-}
-fragment MeasurementsFragment on Measurements {
-	band_size
-	cup_size
-	waist
-	hip
-}
 `
 
 func (c *Client) FindScenesByFullFingerprints(ctx context.Context, fingerprints []*FingerprintQueryInput, httpRequestOptions ...client.HTTPRequestOption) (*FindScenesByFullFingerprints, error) {
@@ -447,6 +468,11 @@ const SearchSceneQuery = `query SearchScene ($term: String!) {
 		... SceneFragment
 	}
 }
+fragment FingerprintFragment on Fingerprint {
+	algorithm
+	hash
+	duration
+}
 fragment URLFragment on URL {
 	url
 	type
@@ -457,9 +483,15 @@ fragment ImageFragment on Image {
 	width
 	height
 }
-fragment FuzzyDateFragment on FuzzyDate {
-	date
-	accuracy
+fragment TagFragment on Tag {
+	name
+	id
+}
+fragment PerformerAppearanceFragment on PerformerAppearance {
+	as
+	performer {
+		... PerformerFragment
+	}
 }
 fragment MeasurementsFragment on Measurements {
 	band_size
@@ -506,16 +538,6 @@ fragment StudioFragment on Studio {
 		... ImageFragment
 	}
 }
-fragment TagFragment on Tag {
-	name
-	id
-}
-fragment PerformerAppearanceFragment on PerformerAppearance {
-	as
-	performer {
-		... PerformerFragment
-	}
-}
 fragment PerformerFragment on Performer {
 	id
 	name
@@ -550,10 +572,9 @@ fragment PerformerFragment on Performer {
 		... BodyModificationFragment
 	}
 }
-fragment FingerprintFragment on Fingerprint {
-	algorithm
-	hash
-	duration
+fragment FuzzyDateFragment on FuzzyDate {
+	date
+	accuracy
 }
 `
 
@@ -653,6 +674,30 @@ const FindPerformerByIDQuery = `query FindPerformerByID ($id: ID!) {
 		... PerformerFragment
 	}
 }
+fragment URLFragment on URL {
+	url
+	type
+}
+fragment ImageFragment on Image {
+	id
+	url
+	width
+	height
+}
+fragment FuzzyDateFragment on FuzzyDate {
+	date
+	accuracy
+}
+fragment MeasurementsFragment on Measurements {
+	band_size
+	cup_size
+	waist
+	hip
+}
+fragment BodyModificationFragment on BodyModification {
+	location
+	description
+}
 fragment PerformerFragment on Performer {
 	id
 	name
@@ -687,30 +732,6 @@ fragment PerformerFragment on Performer {
 		... BodyModificationFragment
 	}
 }
-fragment URLFragment on URL {
-	url
-	type
-}
-fragment ImageFragment on Image {
-	id
-	url
-	width
-	height
-}
-fragment FuzzyDateFragment on FuzzyDate {
-	date
-	accuracy
-}
-fragment MeasurementsFragment on Measurements {
-	band_size
-	cup_size
-	waist
-	hip
-}
-fragment BodyModificationFragment on BodyModification {
-	location
-	description
-}
 `
 
 func (c *Client) FindPerformerByID(ctx context.Context, id string, httpRequestOptions ...client.HTTPRequestOption) (*FindPerformerByID, error) {
@@ -731,22 +752,10 @@ const FindSceneByIDQuery = `query FindSceneByID ($id: ID!) {
 		... SceneFragment
 	}
 }
-fragment ImageFragment on Image {
-	id
-	url
-	width
-	height
-}
 fragment TagFragment on Tag {
 	name
 	id
 }
-fragment PerformerAppearanceFragment on PerformerAppearance {
-	as
-	performer {
-		... PerformerFragment
-	}
-}
 fragment FuzzyDateFragment on FuzzyDate {
 	date
 	accuracy
@@ -762,58 +771,6 @@ fragment FingerprintFragment on Fingerprint {
 	hash
 	duration
 }
-fragment URLFragment on URL {
-	url
-	type
-}
-fragment StudioFragment on Studio {
-	name
-	id
-	urls {
-		... URLFragment
-	}
-	images {
-		... ImageFragment
-	}
-}
-fragment PerformerFragment on Performer {
-	id
-	name
-	disambiguation
-	aliases
-	gender
-	merged_ids
-	urls {
-		... URLFragment
-	}
-	images {
-		... ImageFragment
-	}
-	birthdate {
-		... FuzzyDateFragment
-	}
-	ethnicity
-	country
-	eye_color
-	hair_color
-	height
-	measurements {
-		... MeasurementsFragment
-	}
-	breast_type
-	career_start_year
-	career_end_year
-	tattoos {
-		... BodyModificationFragment
-	}
-	piercings {
-		... BodyModificationFragment
-	}
-}
-fragment BodyModificationFragment on BodyModification {
-	location
-	description
-}
 fragment SceneFragment on Scene {
 	id
 	title
@@ -839,6 +796,70 @@ fragment SceneFragment on Scene {
 		... FingerprintFragment
 	}
 }
+fragment URLFragment on URL {
+	url
+	type
+}
+fragment ImageFragment on Image {
+	id
+	url
+	width
+	height
+}
+fragment StudioFragment on Studio {
+	name
+	id
+	urls {
+		... URLFragment
+	}
+	images {
+		... ImageFragment
+	}
+}
+fragment PerformerAppearanceFragment on PerformerAppearance {
+	as
+	performer {
+		... PerformerFragment
+	}
+}
+fragment PerformerFragment on Performer {
+	id
+	name
+	disambiguation
+	aliases
+	gender
+	merged_ids
+	urls {
+		... URLFragment
+	}
+	images {
+		... ImageFragment
+	}
+	birthdate {
+		... FuzzyDateFragment
+	}
+	ethnicity
+	country
+	eye_color
+	hair_color
+	height
+	measurements {
+		... MeasurementsFragment
+	}
+	breast_type
+	career_start_year
+	career_end_year
+	tattoos {
+		... BodyModificationFragment
+	}
+	piercings {
+		... BodyModificationFragment
+	}
+}
+fragment BodyModificationFragment on BodyModification {
+	location
+	description
+}
 `
 
 func (c *Client) FindSceneByID(ctx context.Context, id string, httpRequestOptions ...client.HTTPRequestOption) (*FindSceneByID, error) {
@@ -889,3 +910,43 @@ func (c *Client) Me(ctx context.Context, httpRequestOptions ...client.HTTPReques
 
 	return &res, nil
 }
+
+const SubmitSceneDraftQuery = `mutation SubmitSceneDraft ($input: SceneDraftInput!) {
+	submitSceneDraft(input: $input) {
+		id
+	}
+}
+`
+
+func (c *Client) SubmitSceneDraft(ctx context.Context, input SceneDraftInput, httpRequestOptions ...client.HTTPRequestOption) (*SubmitSceneDraftPayload, error) {
+	vars := map[string]interface{}{
+		"input": input,
+	}
+
+	var res SubmitSceneDraftPayload
+	if err := c.Client.Post(ctx, SubmitSceneDraftQuery, &res, vars, httpRequestOptions...); err != nil {
+		return nil, err
+	}
+
+	return &res, nil
+}
+
+const SubmitPerformerDraftQuery = `mutation SubmitPerformerDraft ($input: PerformerDraftInput!) {
+	submitPerformerDraft(input: $input) {
+		id
+	}
+}
+`
+
+func (c *Client) SubmitPerformerDraft(ctx context.Context, input PerformerDraftInput, httpRequestOptions ...client.HTTPRequestOption) (*SubmitPerformerDraftPayload, error) {
+	vars := map[string]interface{}{
+		"input": input,
+	}
+
+	var res SubmitPerformerDraftPayload
+	if err := c.Client.Post(ctx, SubmitPerformerDraftQuery, &res, vars, httpRequestOptions...); err != nil {
+		return nil, err
+	}
+
+	return &res, nil
+}