Update ui_extension to support Unified config

packages/app/src/cli/models/extensions/specifications/ui_extension.ts

@@ -1,5 +1,11 @@
 import {ExtensionFeature, createExtensionSpecification} from '../specification.js'
-import {NewExtensionPointSchemaType, NewExtensionPointsSchema, BaseSchema} from '../schemas.js'
+import {
+  NewExtensionPointSchemaType,
+  NewExtensionPointsSchema,
+  BaseSchema,
+  CapabilitiesSchema,
+  MetafieldSchema,
+} from '../schemas.js'
 import {loadLocalesConfig} from '../../../utilities/extensions/locales-configuration.js'
 import {configurationFileNames} from '../../../constants.js'
 import {getExtensionPointTargetSurface} from '../../../services/dev/extension/utilities.js'
@@ -8,10 +14,11 @@ import {err, ok, Result} from '@shopify/cli-kit/node/result'
 import {fileExists} from '@shopify/cli-kit/node/fs'
 import {joinPath} from '@shopify/cli-kit/node/path'
 import {outputContent, outputToken} from '@shopify/cli-kit/node/output'
+import {getArrayRejectingUndefined} from '@shopify/cli-kit/common/array'
 
 const dependency = '@shopify/checkout-ui-extensions'
 
-const UIExtensionSchema = BaseSchema.extend({
+const UIExtensionLegacySchema = BaseSchema.extend({
   settings: zod
     .object({
       fields: zod.any().optional(),
@@ -20,13 +27,73 @@ const UIExtensionSchema = BaseSchema.extend({
   extension_points: NewExtensionPointsSchema,
 })
 
+type UIExtensionLegacySchemaType = zod.infer<typeof UIExtensionLegacySchema>
+
+const UnifiedSettingsSchema = zod
+  .object({
+    fields: zod
+      .array(
+        zod.object({
+          key: zod.string().optional(),
+          name: zod.string().optional(),
+          description: zod.string().optional(),
+          required: zod.boolean().optional(),
+          type: zod.string(),
+        }),
+      )
+      .optional(),
+  })
+  .optional()
+
+const UIExtensionSchema = BaseSchema.extend({
+  type: zod.literal('ui_extension'),
+  name: zod.string().optional(),
+  description: zod.string().optional(),
+  api_version: zod.string().optional(),
+  capabilities: CapabilitiesSchema.optional(),
+  settings: UnifiedSettingsSchema,
+  targeting: zod.array(
+    zod.object({
+      target: zod.string(),
+      module: zod.string(),
+      metafields: zod.array(MetafieldSchema).optional().default([]),
+    }),
+  ),
+})
+
+const UIExtensionUnifiedSchema = BaseSchema.extend({
+  settings: UnifiedSettingsSchema,
+  extensions: zod.array(UIExtensionSchema).min(1).max(1),
+}).transform((config) => {
+  const extensionPoints = config.extensions[0]?.targeting.map((targeting) => {
+    return {
+      target: targeting.target,
+      module: targeting.module,
+      metafields: targeting.metafields ?? config.extensions[0]?.metafields ?? [],
+    }
+  })
+  const newConfig: UIExtensionLegacySchemaType = {
+    name: config.extensions[0]?.name ?? config.name,
+    type: config.extensions[0]?.type ?? config.type,
+    description: config.extensions[0]?.description ?? config.description,
+    api_version: config.extensions[0]?.api_version ?? config.api_version,
+    extension_points: extensionPoints ?? [],
+    capabilities: config.extensions[0]?.capabilities,
+    metafields: config.extensions[0]?.metafields ?? [],
+    settings: config.extensions[0]?.settings,
+  }
+  return newConfig
+})
+
+const UnionSchema = zod.union([UIExtensionUnifiedSchema, UIExtensionLegacySchema])
+
 const spec = createExtensionSpecification({
   identifier: 'ui_extension',
   surface: 'all',
   dependency,
   partnersWebIdentifier: 'ui_extension',
   singleEntryPath: false,
-  schema: UIExtensionSchema,
+  schema: UnionSchema,
   appModuleFeatures: (config) => {
     const basic: ExtensionFeature[] = ['ui_preview', 'bundling', 'esbuild']
     const needsCart =