formstand 0.23.0 → 0.24.0

package/dist/index.js

@@ -221,12 +221,32 @@ var fieldSchemaForIntrospection = (schema, path) => {
   );
 };
 var containerKindAt = (schema) => (prefix) => {
-  const sub = fieldSchemaForIntrospection(schema, prefix);
-  if (sub === null) return null;
   const unwrapAll = (s) => {
     const inner2 = unwrapForWalk(s);
     return inner2 === null ? s : unwrapAll(inner2);
   };
+  const step = (s, segment) => {
+    const inner2 = unwrapAll(s);
+    if (inner2 instanceof z.ZodObject && typeof segment === "string") {
+      const shape = inner2.shape;
+      return Object.hasOwn(shape, segment) ? shape[segment] ?? null : null;
+    }
+    if (inner2 instanceof z.ZodArray && typeof segment === "number") {
+      return inner2.element;
+    }
+    if (inner2 instanceof z.ZodTuple && typeof segment === "number") {
+      return inner2.def.items[segment] ?? null;
+    }
+    if (inner2 instanceof z.ZodRecord) {
+      return inner2.valueType;
+    }
+    return null;
+  };
+  const sub = parsePath(prefix).reduce(
+    (current, segment) => current === null ? null : step(current, segment),
+    schema
+  );
+  if (sub === null) return null;
   const inner = unwrapAll(sub);
   if (inner instanceof z.ZodArray || inner instanceof z.ZodTuple) {
     return "array";