zones as nodes

packages/core/src/schema/index.ts

@@ -9,10 +9,8 @@ export { LevelNode } from './nodes/level'
 // Nodes
 export { SiteNode } from './nodes/site'
 export { WallNode } from './nodes/wall'
+// Zones
+export type { ZoneNode, ZonePolygon } from './nodes/zone'
 export type { AnyNodeId, AnyNodeType } from './types'
 // Union types
 export { AnyNode } from './types'
-
-// Zones
-export type { Zone, ZonePolygon } from './zone'
-export { ZoneSchema } from './zone'

packages/core/src/schema/nodes/level.ts

@@ -2,11 +2,13 @@ import dedent from 'dedent'
 import { z } from 'zod'
 import { BaseNode, nodeType, objectId } from '../base'
 import { WallNode } from './wall'
+import { ZoneNode } from './zone'
+
 
 export const LevelNode = BaseNode.extend({
   id: objectId('level'),
   type: nodeType('level'),
-  children: z.array(WallNode.shape.id).default([]),
+  children: z.array(z.union([WallNode.shape.id, ZoneNode.shape.id])).default([]),
   // Specific props
   level: z.number().default(0),
 }).describe(

packages/core/src/schema/nodes/zone.ts

@@ -0,0 +1,31 @@
+import dedent from 'dedent'
+import { z } from 'zod'
+import { BaseNode, nodeType, objectId } from '../base'
+
+// Polygon boundary for zone area - array of [x, z] coordinates
+export const ZonePolygon = z.array(z.tuple([z.number(), z.number()]))
+
+export const ZoneNode = BaseNode.extend({
+  id: objectId('zone'),
+  type: nodeType('zone'),
+  name: z.string(),
+  // Polygon boundary - array of [x, z] coordinates defining the zone
+  polygon: ZonePolygon,
+  // Visual styling
+  color: z.string().default('#3b82f6'), // Default blue
+  metadata: z.json().optional().default({}),
+}).describe(
+  dedent`
+  Zone schema - a polygon zone attached to a level
+  - object: "zone"
+  - id: zone id
+  - levelId: level this zone is attached to
+  - name: zone name
+  - polygon: array of [x, z] points defining the zone boundary
+  - color: hex color for visual styling
+  - metadata: zone metadata (optional)
+  `,
+)
+
+export type ZoneNode = z.infer<typeof ZoneNode>
+export type ZonePolygon = z.infer<typeof ZonePolygon>

packages/core/src/schema/types.ts

@@ -4,6 +4,7 @@ import { ItemNode } from './nodes/item'
 import { LevelNode } from './nodes/level'
 import { SiteNode } from './nodes/site'
 import { WallNode } from './nodes/wall'
+import { ZoneNode } from './nodes/zone'
 
 export const AnyNode = z.discriminatedUnion('type', [
   SiteNode,
@@ -11,6 +12,7 @@ export const AnyNode = z.discriminatedUnion('type', [
   LevelNode,
   WallNode,
   ItemNode,
+  ZoneNode,
 ])
 
 export type AnyNode = z.infer<typeof AnyNode>

packages/core/src/schema/zone.ts

@@ -1,35 +0,0 @@
-import dedent from 'dedent'
-import { z } from 'zod'
-import { objectId } from './base'
-import { LevelNode } from './nodes/level'
-
-// Polygon boundary for zone area - array of [x, z] coordinates
-export const ZonePolygon = z.array(z.tuple([z.number(), z.number()]))
-
-export const ZoneSchema = z
-  .object({
-    id: objectId('zone'),
-    object: z.literal('zone').default('zone'),
-    levelId: LevelNode.shape.id, // Required - must be attached to a level
-    name: z.string(),
-    // Polygon boundary - array of [x, z] coordinates defining the zone
-    polygon: ZonePolygon,
-    // Visual styling
-    color: z.string().default('#3b82f6'), // Default blue
-    metadata: z.json().optional().default({}),
-  })
-  .describe(
-    dedent`
-  Zone schema - a polygon zone attached to a level
-  - object: "zone"
-  - id: zone id
-  - levelId: level this zone is attached to
-  - name: zone name
-  - polygon: array of [x, z] points defining the zone boundary
-  - color: hex color for visual styling
-  - metadata: zone metadata (optional)
-  `,
-  )
-
-export type Zone = z.infer<typeof ZoneSchema>
-export type ZonePolygon = z.infer<typeof ZonePolygon>