Submit fields director and (studio) code to scene drafts (stashap…

…p#544)
Maista6969 · Jun 24, 2023 · 678e1f1 · 678e1f1
1 parent 0fd15d8
commit 678e1f1
Show file tree

Hide file tree

Showing 10 changed files with 153 additions and 4 deletions.
diff --git a/frontend/src/graphql/mutations/AddScene.gql b/frontend/src/graphql/mutations/AddScene.gql
@@ -3,7 +3,9 @@ mutation AddScene($sceneData: SceneCreateInput!) {
     id
     release_date
     title
+    code
     details
+    director
     urls {
       url
       site {

diff --git a/frontend/src/graphql/queries/Draft.gql b/frontend/src/graphql/queries/Draft.gql
@@ -34,7 +34,9 @@ query Draft($id: ID!) {
       ... on SceneDraft {
         id
         title
+        code
         details
+        director
         date
         url {
           ...URLFragment

diff --git a/frontend/src/graphql/types.ts b/frontend/src/graphql/types.ts
@@ -1291,8 +1291,10 @@ export type SceneDestroyInput = {
 
 export type SceneDraft = {
   __typename: "SceneDraft";
+  code?: Maybe<Scalars["String"]>;
   date?: Maybe<Scalars["String"]>;
   details?: Maybe<Scalars["String"]>;
+  director?: Maybe<Scalars["String"]>;
   fingerprints: Array<DraftFingerprint>;
   id?: Maybe<Scalars["ID"]>;
   image?: Maybe<Image>;
@@ -1304,8 +1306,10 @@ export type SceneDraft = {
 };
 
 export type SceneDraftInput = {
+  code?: InputMaybe<Scalars["String"]>;
   date?: InputMaybe<Scalars["String"]>;
   details?: InputMaybe<Scalars["String"]>;
+  director?: InputMaybe<Scalars["String"]>;
   fingerprints: Array<FingerprintInput>;
   id?: InputMaybe<Scalars["ID"]>;
   image?: InputMaybe<Scalars["Upload"]>;
@@ -3044,7 +3048,9 @@ export type AddSceneMutation = {
     id: string;
     release_date?: string | null;
     title?: string | null;
+    code?: string | null;
     details?: string | null;
+    director?: string | null;
     urls: Array<{
       __typename: "URL";
       url: string;
@@ -13859,7 +13865,9 @@ export type DraftQuery = {
           __typename: "SceneDraft";
           id?: string | null;
           title?: string | null;
+          code?: string | null;
           details?: string | null;
+          director?: string | null;
           date?: string | null;
           url?: {
             __typename: "URL";

diff --git a/frontend/src/pages/drafts/parse.ts b/frontend/src/pages/drafts/parse.ts
@@ -69,8 +69,8 @@ export const parseSceneDraft = (
     details: draft.details,
     urls: joinURLs(draft.url, existingScene?.urls),
     studio: draft.studio?.__typename === "Studio" ? draft.studio : null,
-    director: null,
-    code: null,
+    director: draft.director,
+    code: draft.code,
     duration: draft.fingerprints?.[0]?.duration ?? null,
     images: draft.image ? [draft.image] : existingScene?.images,
     tags: joinTags(

diff --git a/graphql/schema/types/scene.graphql b/graphql/schema/types/scene.graphql
@@ -223,7 +223,9 @@ union SceneDraftTag = Tag | DraftEntity
 type SceneDraft {
   id: ID
   title: String
+  code: String
   details: String
+  director: String
   url: URL
   date: String
   studio: SceneDraftStudio
@@ -236,7 +238,9 @@ type SceneDraft {
 input SceneDraftInput {
   id: ID
   title: String
+  code: String
   details: String
+  director: String
   url: String
   date: String
   studio: DraftEntityInput

diff --git a/main.go b/main.go
@@ -13,7 +13,8 @@ import (
 	"github.com/stashapp/stash-box/pkg/user"
 )
 
-//nolint
+// nolint
+//
 //go:embed frontend/build
 var ui embed.FS
 

diff --git a/pkg/api/resolver_mutation_draft.go b/pkg/api/resolver_mutation_draft.go
@@ -23,7 +23,9 @@ func (r *mutationResolver) SubmitSceneDraft(ctx context.Context, input models.Sc
 	data := models.SceneDraft{
 		ID:           input.ID,
 		Title:        input.Title,
+		Code:         input.Code,
 		Details:      input.Details,
+		Director:     input.Director,
 		URL:          input.URL,
 		Date:         input.Date,
 		Studio:       translateDraftEntity(input.Studio),

diff --git a/pkg/models/generated_exec.go b/pkg/models/generated_exec.go
diff --git a/pkg/models/generated_models.go b/pkg/models/generated_models.go
diff --git a/pkg/models/model_draft.go b/pkg/models/model_draft.go
@@ -29,7 +29,9 @@ func (DraftEntity) IsSceneDraftStudio()    {}
 type SceneDraft struct {
 	ID           *uuid.UUID         `json:"id,omitempty"`
 	Title        *string            `json:"title,omitempty"`
+	Code         *string            `json:"code,omitempty"`
 	Details      *string            `json:"details,omitempty"`
+	Director     *string            `json:"director,omitempty"`
 	URL          *string            `json:"url,omitempty"`
 	Date         *string            `json:"date,omitempty"`
 	Studio       *DraftEntity       `json:"studio,omitempty"`