jazz-tools 0.13.31 → 0.14.0

package/src/implementation/zodSchema/schemaTypes/RichTextSchema.ts

@@ -0,0 +1,25 @@
+import z from "zod/v4";
+import { Account, CoRichText, Group } from "../../../internal.js";
+import { AnonymousJazzAgent } from "../../anonymousJazzAgent.js";
+
+export type RichTextSchema = z.core.$ZodCustom<CoRichText, unknown> & {
+  collaborative: true;
+  builtin: "CoRichText";
+  create(
+    text: string,
+    options?: { owner: Account | Group } | Account | Group,
+  ): CoRichText;
+  load(
+    id: string,
+    options: { loadAs: Account | AnonymousJazzAgent },
+  ): Promise<CoRichText>;
+  subscribe(
+    id: string,
+    options: { loadAs: Account | AnonymousJazzAgent },
+    listener: (value: CoRichText, unsubscribe: () => void) => void,
+  ): () => void;
+  subscribe(
+    id: string,
+    listener: (value: CoRichText, unsubscribe: () => void) => void,
+  ): () => void;
+};

package/src/implementation/zodSchema/typeConverters/InstanceOfSchema.ts

@@ -0,0 +1,61 @@
+import z from "zod/v4";
+import {
+  Account,
+  CoFeed,
+  CoList,
+  CoMap,
+  CoPlainText,
+  CoRichText,
+  CoValueClass,
+  FileStream,
+} from "../../../internal.js";
+import { AnyCoFeedSchema } from "../schemaTypes/CoFeedSchema.js";
+import { AnyCoListSchema } from "../schemaTypes/CoListSchema.js";
+import { AnyCoMapSchema } from "../schemaTypes/CoMapSchema.js";
+import { FileStreamSchema } from "../schemaTypes/FileStreamSchema.js";
+import { PlainTextSchema } from "../schemaTypes/PlainTextSchema.js";
+import { RichTextSchema } from "../schemaTypes/RichTextSchema.js";
+import { InstanceOrPrimitiveOfSchema } from "./InstanceOrPrimitiveOfSchema.js";
+
+export type InstanceOfSchema<S extends CoValueClass | z.core.$ZodType> =
+  S extends z.core.$ZodType
+    ? S extends z.core.$ZodObject<infer Shape> & {
+        collaborative: true;
+        builtin: "Account";
+      }
+      ? {
+          [key in keyof Shape]: InstanceOrPrimitiveOfSchema<Shape[key]>;
+        } & Account
+      : S extends z.core.$ZodRecord<infer K, infer V> & {
+            collaborative: true;
+          }
+        ? {
+            [key in z.output<K> & string]: InstanceOrPrimitiveOfSchema<V>;
+          } & CoMap
+        : S extends AnyCoMapSchema<infer Shape, infer Config>
+          ? {
+              [key in keyof Shape]: InstanceOrPrimitiveOfSchema<Shape[key]>;
+            } & (unknown extends Config["out"][string]
+              ? {}
+              : {
+                  [key: string]: Config["out"][string];
+                }) &
+              CoMap
+          : S extends AnyCoListSchema<infer T>
+            ? CoList<InstanceOrPrimitiveOfSchema<T>>
+            : S extends AnyCoFeedSchema<infer T>
+              ? CoFeed<InstanceOrPrimitiveOfSchema<T>>
+              : S extends PlainTextSchema
+                ? CoPlainText
+                : S extends RichTextSchema
+                  ? CoRichText
+                  : S extends FileStreamSchema
+                    ? FileStream
+                    : S extends z.core.$ZodOptional<infer Inner>
+                      ? InstanceOrPrimitiveOfSchema<Inner>
+                      : S extends z.core.$ZodUnion<infer Members>
+                        ? InstanceOrPrimitiveOfSchema<Members[number]>
+                        : never
+    : S extends CoValueClass
+      ? InstanceType<S>
+      : never;

package/src/implementation/zodSchema/typeConverters/InstanceOfSchemaCoValuesNullable.ts

@@ -0,0 +1,77 @@
+import z from "zod/v4";
+import {
+  Account,
+  AnyCoFeedSchema,
+  AnyCoListSchema,
+  AnyCoMapSchema,
+  CoFeed,
+  CoList,
+  CoMap,
+  CoPlainText,
+  CoRichText,
+  CoValueClass,
+  FileStream,
+} from "../../../internal.js";
+import { FileStreamSchema } from "../schemaTypes/FileStreamSchema.js";
+import { PlainTextSchema } from "../schemaTypes/PlainTextSchema.js";
+import { RichTextSchema } from "../schemaTypes/RichTextSchema.js";
+import { InstanceOrPrimitiveOfSchemaCoValuesNullable } from "./InstanceOrPrimitiveOfSchemaCoValuesNullable.js";
+
+export type InstanceOfSchemaCoValuesNullable<
+  S extends CoValueClass | z.core.$ZodType,
+> = S extends z.core.$ZodType
+  ? S extends z.core.$ZodObject<infer Shape> & {
+      collaborative: true;
+      builtin: "Account";
+    }
+    ?
+        | ({
+            [key in keyof Shape]: InstanceOrPrimitiveOfSchemaCoValuesNullable<
+              Shape[key]
+            >;
+          } & Account)
+        | null
+    : S extends z.core.$ZodRecord<infer K, infer V> & {
+          collaborative: true;
+        }
+      ?
+          | ({
+              [key in z.output<K> &
+                string]: InstanceOrPrimitiveOfSchemaCoValuesNullable<V>;
+            } & CoMap)
+          | null
+      : S extends AnyCoMapSchema<infer Shape, infer Config>
+        ?
+            | ({
+                [key in keyof Shape]: InstanceOrPrimitiveOfSchemaCoValuesNullable<
+                  Shape[key]
+                >;
+              } & (unknown extends Config["out"][string]
+                ? {}
+                : {
+                    [key: string]: Config["out"][string];
+                  }) &
+                CoMap)
+            | null
+        : S extends AnyCoListSchema<infer T>
+          ? CoList<InstanceOrPrimitiveOfSchemaCoValuesNullable<T>> | null
+          : S extends AnyCoFeedSchema<infer T>
+            ? CoFeed<InstanceOrPrimitiveOfSchemaCoValuesNullable<T>> | null
+            : S extends PlainTextSchema
+              ? CoPlainText | null
+              : S extends RichTextSchema
+                ? CoRichText | null
+                : S extends FileStreamSchema
+                  ? FileStream | null
+                  : S extends z.core.$ZodOptional<infer Inner>
+                    ?
+                        | InstanceOrPrimitiveOfSchemaCoValuesNullable<Inner>
+                        | undefined
+                    : S extends z.core.$ZodUnion<infer Members>
+                      ? InstanceOrPrimitiveOfSchemaCoValuesNullable<
+                          Members[number]
+                        >
+                      : never
+  : S extends CoValueClass
+    ? InstanceType<S> | null
+    : never;

package/src/implementation/zodSchema/typeConverters/InstanceOrPrimitiveOfSchema.ts

@@ -0,0 +1,90 @@
+import z from "zod/v4";
+import {
+  Account,
+  CoFeed,
+  CoList,
+  CoMap,
+  CoPlainText,
+  CoRichText,
+  CoValueClass,
+  FileStream,
+  Profile,
+} from "../../../internal.js";
+import { AnyCoFeedSchema } from "../schemaTypes/CoFeedSchema.js";
+import { AnyCoListSchema } from "../schemaTypes/CoListSchema.js";
+import { AnyCoMapSchema } from "../schemaTypes/CoMapSchema.js";
+import { FileStreamSchema } from "../schemaTypes/FileStreamSchema.js";
+import { PlainTextSchema } from "../schemaTypes/PlainTextSchema.js";
+import { RichTextSchema } from "../schemaTypes/RichTextSchema.js";
+
+export type InstanceOrPrimitiveOfSchema<
+  S extends CoValueClass | z.core.$ZodType,
+> = S extends z.core.$ZodType
+  ? S extends z.core.$ZodObject<infer Shape> & {
+      collaborative: true;
+      builtin: "Account";
+    }
+    ? {
+        -readonly [key in keyof Shape]: InstanceOrPrimitiveOfSchema<Shape[key]>;
+      } & { profile: Profile } & Account
+    : S extends z.core.$ZodRecord<infer K, infer V> & {
+          collaborative: true;
+        }
+      ? {
+          -readonly [key in z.output<K> &
+            string]: InstanceOrPrimitiveOfSchema<V>;
+        } & CoMap
+      : S extends AnyCoMapSchema<infer Shape, infer Config>
+        ? {
+            -readonly [key in keyof Shape]: InstanceOrPrimitiveOfSchema<
+              Shape[key]
+            >;
+          } & (unknown extends Config["out"][string]
+            ? {}
+            : {
+                [key: string]: Config["out"][string];
+              }) &
+            CoMap
+        : S extends AnyCoListSchema<infer T>
+          ? CoList<InstanceOrPrimitiveOfSchema<T>>
+          : S extends AnyCoFeedSchema<infer T>
+            ? CoFeed<InstanceOrPrimitiveOfSchema<T>>
+            : S extends PlainTextSchema
+              ? CoPlainText
+              : S extends RichTextSchema
+                ? CoRichText
+                : S extends FileStreamSchema
+                  ? FileStream
+                  : S extends z.core.$ZodOptional<infer Inner>
+                    ? InstanceOrPrimitiveOfSchema<Inner> | undefined
+                    : S extends z.core.$ZodUnion<infer Members>
+                      ? InstanceOrPrimitiveOfSchema<Members[number]>
+                      : // primitives below here - we manually traverse to ensure we only allow what we can handle
+                        S extends z.core.$ZodObject<infer Shape>
+                        ? {
+                            -readonly [key in keyof Shape]: InstanceOrPrimitiveOfSchema<
+                              Shape[key]
+                            >;
+                          }
+                        : S extends z.core.$ZodArray<infer Item>
+                          ? InstanceOrPrimitiveOfSchema<Item>[]
+                          : S extends z.core.$ZodTuple<infer Items>
+                            ? {
+                                [key in keyof Items]: InstanceOrPrimitiveOfSchema<
+                                  Items[key]
+                                >;
+                              }
+                            : S extends z.core.$ZodString
+                              ? string
+                              : S extends z.core.$ZodNumber
+                                ? number
+                                : S extends z.core.$ZodBoolean
+                                  ? boolean
+                                  : S extends z.core.$ZodLiteral<infer Literal>
+                                    ? Literal
+                                    : S extends z.core.$ZodDate
+                                      ? Date
+                                      : never
+  : S extends CoValueClass
+    ? InstanceType<S>
+    : never;

package/src/implementation/zodSchema/typeConverters/InstanceOrPrimitiveOfSchemaCoValuesNullable.ts

@@ -0,0 +1,103 @@
+import z from "zod/v4";
+import {
+  Account,
+  CoFeed,
+  CoList,
+  CoMap,
+  CoPlainText,
+  CoRichText,
+  CoValueClass,
+  FileStream,
+  InstanceOrPrimitiveOfSchema,
+  Profile,
+} from "../../../internal.js";
+import { AnyCoFeedSchema } from "../schemaTypes/CoFeedSchema.js";
+import { AnyCoListSchema } from "../schemaTypes/CoListSchema.js";
+import { AnyCoMapSchema } from "../schemaTypes/CoMapSchema.js";
+import { FileStreamSchema } from "../schemaTypes/FileStreamSchema.js";
+import { PlainTextSchema } from "../schemaTypes/PlainTextSchema.js";
+import { RichTextSchema } from "../schemaTypes/RichTextSchema.js";
+
+export type InstanceOrPrimitiveOfSchemaCoValuesNullable<
+  S extends CoValueClass | z.core.$ZodType,
+> = S extends z.core.$ZodType
+  ? S extends z.core.$ZodObject<infer Shape> & {
+      collaborative: true;
+      builtin: "Account";
+    }
+    ?
+        | ({
+            -readonly [key in keyof Shape]: InstanceOrPrimitiveOfSchemaCoValuesNullable<
+              Shape[key]
+            >;
+          } & { profile: Profile | null } & Account)
+        | null
+    : S extends z.core.$ZodRecord<infer K, infer V> & {
+          collaborative: true;
+        }
+      ?
+          | ({
+              -readonly [key in z.output<K> &
+                string]: InstanceOrPrimitiveOfSchemaCoValuesNullable<V>;
+            } & CoMap)
+          | null
+      : S extends AnyCoMapSchema<infer Shape, infer Config>
+        ?
+            | ({
+                -readonly [key in keyof Shape]: InstanceOrPrimitiveOfSchemaCoValuesNullable<
+                  Shape[key]
+                >;
+              } & (unknown extends Config["out"][string]
+                ? {}
+                : {
+                    [key: string]: Config["out"][string];
+                  }) &
+                CoMap)
+            | null
+        : S extends AnyCoListSchema<infer T>
+          ? CoList<InstanceOrPrimitiveOfSchemaCoValuesNullable<T>> | null
+          : S extends AnyCoFeedSchema<infer T>
+            ? CoFeed<InstanceOrPrimitiveOfSchemaCoValuesNullable<T>> | null
+            : S extends PlainTextSchema
+              ? CoPlainText | null
+              : S extends RichTextSchema
+                ? CoRichText | null
+                : S extends FileStreamSchema
+                  ? FileStream | null
+                  : S extends z.core.$ZodOptional<infer Inner>
+                    ?
+                        | InstanceOrPrimitiveOfSchemaCoValuesNullable<Inner>
+                        | undefined
+                    : S extends z.core.$ZodUnion<infer Members>
+                      ? InstanceOrPrimitiveOfSchemaCoValuesNullable<
+                          Members[number]
+                        >
+                      : // primitives below here - we manually traverse to ensure we only allow what we can handle
+                        S extends z.core.$ZodObject<infer Shape>
+                        ? {
+                            -readonly [key in keyof Shape]: InstanceOrPrimitiveOfSchema<
+                              Shape[key]
+                            >;
+                          }
+                        : S extends z.core.$ZodArray<infer Item>
+                          ? InstanceOrPrimitiveOfSchema<Item>[]
+                          : S extends z.core.$ZodTuple<infer Items>
+                            ? {
+                                [key in keyof Items]: InstanceOrPrimitiveOfSchema<
+                                  Items[key]
+                                >;
+                              }
+                            : S extends z.core.$ZodString
+                              ? string
+                              : S extends z.core.$ZodNumber
+                                ? number
+                                : S extends z.core.$ZodBoolean
+                                  ? boolean
+                                  : S extends z.core.$ZodLiteral<infer Literal>
+                                    ? Literal
+                                    : S extends z.core.$ZodDate
+                                      ? Date
+                                      : never
+  : S extends CoValueClass
+    ? InstanceType<S> | null
+    : never;

package/src/implementation/zodSchema/unionUtils.ts

@@ -0,0 +1,139 @@
+import { RawAccount, RawCoList, RawCoMap } from "cojson";
+import z from "zod/v4";
+import { zodSchemaToCoSchema } from "./runtimeConverters/zodSchemaToCoSchema.js";
+
+export function schemaUnionDiscriminatorFor(
+  schema: z.core.$ZodDiscriminatedUnion,
+) {
+  if (isUnionOfCoMapsDeeply(schema)) {
+    if (!schema._zod.disc || schema._zod.disc.size == 0) {
+      throw new Error(
+        "z.union() of collaborative types is not supported, use z.discriminatedUnion() instead",
+      );
+    }
+    if (![...schema._zod.disc.keys()].every((key) => typeof key === "string")) {
+      throw new Error(
+        "z.discriminatedUnion() of collaborative types with non-string discriminator key is not supported",
+      );
+    }
+    if (
+      ![...schema._zod.disc.values()].every((v) =>
+        [...v.values].every((value) => typeof value === "string"),
+      )
+    ) {
+      throw new Error(
+        "z.discriminatedUnion() of collaborative types with non-string discriminator value is not supported",
+      );
+    }
+
+    const flattendOptions = [
+      ...schema._zod.def.options.flatMap((option) => {
+        if (option._zod.def.type === "object") {
+          return [option];
+        } else if (option._zod.def.type === "union") {
+          return [...(option as z.core.$ZodUnion)._zod.def.options];
+        } else {
+          throw new Error(
+            "Unsupported zod type in z.discriminatedUnion() of collaborative types",
+          );
+        }
+      }),
+    ];
+
+    const determineSchema = (_raw: RawCoMap | RawAccount | RawCoList) => {
+      if (_raw instanceof RawCoList) {
+        throw new Error(
+          "z.discriminatedUnion() of collaborative types is not supported for CoLists",
+        );
+      }
+      let remainingOptions = [...flattendOptions];
+      for (const discriminator of schema._zod.disc.keys()) {
+        const discriminatorValue = (_raw as RawCoMap).get(
+          discriminator as string,
+        );
+        if (typeof discriminatorValue !== "string") {
+          throw new Error("Discriminator must be a string");
+        }
+        remainingOptions = remainingOptions.filter((option) => {
+          if (option._zod.def.type !== "object") {
+            return false;
+          }
+          const discriminatorDef = (option as z.core.$ZodObject)._zod.def.shape[
+            discriminator as string
+          ];
+          if (!discriminatorDef) {
+            return false;
+          }
+
+          console.log(discriminatorDef._zod.def);
+
+          if (discriminatorDef._zod.def.type !== "literal") {
+            console.warn(
+              "Non-literal discriminator found in z.discriminatedUnion() of collaborative types",
+            );
+            return false;
+          }
+          if (
+            (discriminatorDef._zod.def as z.core.$ZodLiteralDef).values
+              .length !== 1
+          ) {
+            console.warn(
+              "Literal discriminator with more than one value found in z.discriminatedUnion() of collaborative types",
+            );
+            return false;
+          }
+          return (
+            (discriminatorDef._zod.def as z.core.$ZodLiteralDef).values[0] ===
+            discriminatorValue
+          );
+        });
+        if (remainingOptions.length === 1) {
+          return zodSchemaToCoSchema(remainingOptions[0]!);
+        }
+      }
+      throw new Error(
+        "z.discriminatedUnion() of collaborative types with no matching discriminator value found",
+      );
+    };
+
+    return determineSchema;
+  } else {
+    throw new Error(
+      "z.discriminatedUnion() of non-collaborative types is not supported",
+    );
+  }
+}
+
+export function isUnionOfCoMapsDeeply(
+  schema: z.core.$ZodType,
+): schema is z.core.$ZodDiscriminatedUnion {
+  if (schema instanceof z.core.$ZodUnion) {
+    return schema._zod.def.options.every(isCoMapOrUnionOfCoMapsDeeply);
+  } else {
+    return false;
+  }
+}
+
+function isCoMapOrUnionOfCoMapsDeeply(
+  schema: z.core.$ZodType,
+): schema is z.core.$ZodDiscriminatedUnion {
+  if (
+    schema instanceof z.core.$ZodObject &&
+    "collaborative" in schema &&
+    schema.collaborative
+  ) {
+    return true;
+  } else if (schema instanceof z.core.$ZodUnion) {
+    return schema._zod.def.options.every(isCoMapOrUnionOfCoMapsDeeply);
+  } else {
+    return false;
+  }
+}
+
+export function isUnionOfPrimitivesDeeply(schema: z.core.$ZodType) {
+  if (schema instanceof z.core.$ZodUnion) {
+    return schema._zod.def.options.every(isUnionOfPrimitivesDeeply);
+  } else {
+    return !("collaborative" in schema);
+  }
+}