🧹 CLI: Add missing schema [3/N] (#707)

LayerZero-Labs · Jul 17, 2024 · 8ed825a · 8ed825a
1 parent 059fbc8
commit 8ed825a
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/.changeset/selfish-tables-shout.md b/.changeset/selfish-tables-shout.md
@@ -0,0 +1,5 @@
+---
+"@layerzerolabs/ua-devtools": patch
+---
+
+Add OAppOmniGraphSchema to ua-devtools
diff --git a/packages/ua-devtools/src/oapp/schema.ts b/packages/ua-devtools/src/oapp/schema.ts
@@ -1,5 +1,12 @@
 import { z } from 'zod'
-import { AddressSchema, UIntBigIntSchema, UIntNumberSchema } from '@layerzerolabs/devtools'
+import {
+    AddressSchema,
+    createOmniEdgeSchema,
+    createOmniGraphSchema,
+    createOmniNodeSchema,
+    UIntBigIntSchema,
+    UIntNumberSchema,
+} from '@layerzerolabs/devtools'
 import { Uln302ExecutorConfigSchema, Uln302UlnUserConfigSchema, TimeoutSchema } from '@layerzerolabs/protocol-devtools'
 import {
     ExecutorComposeOption,
@@ -88,3 +95,8 @@ export const OAppEdgeConfigSchema = z
     // We'll pass all unknown properties through without validating them
     .passthrough()
     .partial() satisfies z.ZodSchema<OAppEdgeConfig, z.ZodTypeDef, unknown>
+
+export const OAppOmniGraphSchema = createOmniGraphSchema(
+    createOmniNodeSchema(OAppNodeConfigSchema),
+    createOmniEdgeSchema(OAppEdgeConfigSchema)
+)