fix: handle intentory level

gatsbyjs · Sep 24, 2024 · c778f00 · c778f00
1 parent e54e070
commit c778f00
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 7 deletions.
diff --git a/packages/gatsby-source-shopify/src/create-schema-customization.ts b/packages/gatsby-source-shopify/src/create-schema-customization.ts
@@ -49,6 +49,29 @@ export function createSchemaCustomization(
       }
     `)
   }
+  actions.createFieldExtension({
+    name: `selectQuantityByName`,
+    args: {
+      name: `String!`,
+    },
+    extend({ name }, fieldConfig) {
+      return {
+        async resolve(source, args, context, info): Promise<number> {
+          const resolver = fieldConfig.resolve || context.defaultFieldResolver
+          const quantities = await resolver(source, args, context, info)
 
+          if (quantities && Array.isArray(quantities)) {
+            for (const quantity of quantities) {
+              if (quantity?.name === name) {
+                return quantity.quantity
+              }
+            }
+          }
+
+          return 0
+        },
+      }
+    },
+  })
   actions.createTypes(typeDefs)
 }
diff --git a/packages/gatsby-source-shopify/src/query-builders/product-variants-query.ts b/packages/gatsby-source-shopify/src/query-builders/product-variants-query.ts
@@ -47,7 +47,7 @@ export class ProductVariantsQuery extends BulkQuery {
                 inventoryLevels {
                   edges {
                     node {
-                      quantities(names: ["available"]) {
+                      quantities(names: ["incoming", "available", "committed", "reserved", "damaged", "safety_stock", "quality_control"]) {
                         name
                         quantity
                       }

diff --git a/packages/gatsby-source-shopify/src/type-builders/location-type.ts b/packages/gatsby-source-shopify/src/type-builders/location-type.ts
@@ -24,9 +24,9 @@ export function locationTypeBuilder(prefix: string): string {
         duplicateSkuCount: Int!
         harmonizedSystemCode: String
         inventoryHistoryUrl: String
-        inventoryLevels: [${prefix}InventoryLevel!]! @link(from: "inventoryLevels___NODE", by: "id")
+        inventoryLevels: [${prefix}InventoryLevel!]! @link(by: "id") @proxy(from: "inventoryLevels___NODE", fromNode: true)
         legacyResourceId: String!
-        locationsCount: Count!
+        locationsCount: ${prefix}Count!
         provinceCodeOfOrigin: String
         requiresShipping: Boolean!
         shopifyId: String!
@@ -38,12 +38,21 @@ export function locationTypeBuilder(prefix: string): string {
         variant: ${prefix}ProductVariantConnection!
       }
 
+      type ${prefix}InventoryQuantity {
+        name: String!
+        quantity: Int!
+      }
+
       type ${prefix}InventoryLevel implements Node @dontInfer {
         _location: String! # Temporary field so we don't break existing users
-        quantities(names: ["available"]) {
-          name
-          quantity
-        }
+        quantities: [${prefix}InventoryQuantity!]!
+        available: Int! @proxy(from: "quantities") @selectQuantityByName(name: "available")
+        incoming: Int! @proxy(from: "quantities") @selectQuantityByName(name: "incoming")
+        committed: Int! @proxy(from: "quantities") @selectQuantityByName(name: "committed")
+        reserved: Int! @proxy(from: "quantities") @selectQuantityByName(name: "reserved")
+        damaged: Int! @proxy(from: "quantities") @selectQuantityByName(name: "damaged")
+        safety_stock: Int! @proxy(from: "quantities") @selectQuantityByName(name: "safety_stock")
+        quality_control: Int! @proxy(from: "quantities") @selectQuantityByName(name: "quality_control")
         id: ID!
         location: ${prefix}Location! @link(from: "_location", by: "id")
         shopifyId: String!