use custom staged upload mutation-

Author: odacrem

packages/store/src/apis/admin/index.ts

@@ -1,25 +1,24 @@
-import {Shop} from './types.js'
-import {StagedUploadsCreate} from '../../cli/api/graphql/admin/generated/staged_uploads_create.js'
+import {Shop, SQLiteStagedUploadsCreate} from './types.js'
 import {ShopDetails} from '../../cli/api/graphql/admin/generated/shop_details.js'
 import {adminRequestDoc} from '@shopify/cli-kit/node/api/admin'
 import {ensureAuthenticatedAdmin} from '@shopify/cli-kit/node/session'
+import type {StagedUploadsCreateMutation} from '../../cli/api/graphql/admin/generated/staged_uploads_create.js'
 import type {
-  StagedUploadsCreateMutation,
-  StagedUploadsCreateMutationVariables,
-} from '../../cli/api/graphql/admin/generated/staged_uploads_create.js'
-import type {StagedUploadInput} from '../../cli/api/graphql/admin/generated/types.js'
-
+  SQLiteStagedUploadInput,
+  SQLiteStagedUploadsCreateMutationVariables,
+  SQLiteStagedUploadsCreateMutation,
+} from './types.js'
 import type {ShopDetailsQuery, ShopDetailsQueryVariables} from '../../cli/api/graphql/admin/generated/shop_details.js'
 
 export async function createStagedUploadAdmin(
   storeFqdn: string,
-  input: StagedUploadInput[],
+  input: SQLiteStagedUploadInput[],
   version?: string,
 ): Promise<StagedUploadsCreateMutation> {
   const adminSession = await ensureAuthenticatedAdmin(storeFqdn)
 
-  return adminRequestDoc<StagedUploadsCreateMutation, StagedUploadsCreateMutationVariables>({
-    query: StagedUploadsCreate,
+  return adminRequestDoc<SQLiteStagedUploadsCreateMutation, SQLiteStagedUploadsCreateMutationVariables>({
+    query: SQLiteStagedUploadsCreate,
     session: adminSession,
     variables: {
       input,

packages/store/src/apis/admin/types.ts

@@ -1,4 +1,42 @@
+/* eslint-disable @typescript-eslint/consistent-type-definitions */
+
+import {StagedUploadsCreate} from '../../cli/api/graphql/admin/generated/staged_uploads_create.js'
+import {TypedDocumentNode as DocumentNode} from '@graphql-typed-document-node/core'
+
 export interface Shop {
   id: string
   name: string
 }
+export type SQLiteStagedUploadInput = {
+  filename: string
+  mimeType: 'application/x-sqlite3'
+  httpMethod: 'POST'
+  fileSize?: string
+  resource: 'SQLITE_DATABASE'
+}
+export type SQLiteStagedUploadsCreateMutationVariables = {
+  input: SQLiteStagedUploadInput[]
+}
+
+export type SQLiteStagedUploadsCreateMutation = {
+  stagedUploadsCreate?: {
+    stagedTargets?:
+      | {
+          url?: string | null
+          resourceUrl?: string | null
+          parameters: {name: string; value: string}[]
+        }[]
+      | null
+    userErrors: {field?: string[] | null; message: string}[]
+  } | null
+}
+
+export const SQLiteStagedUploadsCreate = {
+  ...StagedUploadsCreate,
+  definitions: [
+    {
+      ...StagedUploadsCreate.definitions[0],
+      name: {kind: 'Name', value: 'SQLiteStagedUploadsCreate'},
+    },
+  ],
+} as unknown as DocumentNode<SQLiteStagedUploadsCreateMutation, SQLiteStagedUploadsCreateMutationVariables>

packages/store/src/cli/api/graphql/admin/generated/types.d.ts

@@ -232,10 +232,3 @@ export type StagedUploadTargetGenerateUploadResource =
    * [fileCreate mutation](https://shopify.dev/api/admin-graphql/latest/mutations/fileCreate).
    */
   | 'VIDEO'
-  /**
-   * A SQLite database file.
-   *
-   * For example, after uploading the database file, you can use the
-   * bulkDataImportStart mutation to import the data into a Shopify store.
-   */
-  | 'SQLITE_DATABASE'

packages/store/src/services/store/utils/file-uploader.ts

@@ -1,4 +1,5 @@
-import {StagedUploadInput, createStagedUploadAdmin} from '../../../apis/admin/index.js'
+import {createStagedUploadAdmin} from '../../../apis/admin/index.js'
+import {SQLiteStagedUploadInput} from '../../../apis/admin/types.js'
 import {ValidationError, OperationError, ErrorCodes} from '../errors/errors.js'
 import {fetch} from '@shopify/cli-kit/node/http'
 import {fileExistsSync, fileSize, isDirectory, readFileSync} from '@shopify/cli-kit/node/fs'
@@ -12,7 +13,7 @@ export class FileUploader {
 
     const fileBuffer = readFileSync(filePath)
     const sizeOfFile = await fileSize(filePath)
-    const uploadInput: StagedUploadInput = {
+    const uploadInput: SQLiteStagedUploadInput = {
       resource: 'SQLITE_DATABASE',
       filename: 'database.sqlite',
       mimeType: 'application/x-sqlite3',