import {
  Account,
  AnyZodOrCoValueSchema,
  CoDiscriminatedUnionSchema,
  CoFeed,
  CoList,
  CoMap,
  CoPlainText,
  CoRichText,
  CoVector,
  CoValueClass,
  CoreAccountSchema,
  CoreCoRecordSchema,
  CoreCoVectorSchema,
  FileStream,
  Group,
  CoreSnapshotRefSchema,
  SnapshotRef,
} from "../../../internal.js";
import { CoreGroupSchema } from "../schemaTypes/GroupSchema.js";
import { CoreCoFeedSchema } from "../schemaTypes/CoFeedSchema.js";
import { CoreCoListSchema } from "../schemaTypes/CoListSchema.js";
import { CoreCoMapSchema } from "../schemaTypes/CoMapSchema.js";
import { CoreCoOptionalSchema } from "../schemaTypes/CoOptionalSchema.js";
import { CoreCoValueSchema } from "../schemaTypes/CoValueSchema.js";
import { CoreFileStreamSchema } from "../schemaTypes/FileStreamSchema.js";
import { CorePlainTextSchema } from "../schemaTypes/PlainTextSchema.js";
import { CoreRichTextSchema } from "../schemaTypes/RichTextSchema.js";
import { z } from "../zodReExport.js";
import { InstanceOrPrimitiveOfSchema } from "./InstanceOrPrimitiveOfSchema.js";

/**
 * A loaded CoValue whose references are also loaded.
 */
export type InstanceOfSchema<S extends CoValueClass | AnyZodOrCoValueSchema> =
  S extends CoreCoValueSchema
    ? S extends CoreAccountSchema<infer Shape>
      ? {
          readonly [key in keyof Shape]: InstanceOrPrimitiveOfSchema<
            Shape[key]
          >;
        } & Account
      : S extends CoreGroupSchema
        ? Group
        : S extends CoreCoRecordSchema<infer K, infer V>
          ? {
              readonly [key in z.output<K> &
                string]: InstanceOrPrimitiveOfSchema<V>;
            } & CoMap
          : S extends CoreCoMapSchema<infer Shape, infer CatchAll>
            ? {
                readonly [key in keyof Shape]: InstanceOrPrimitiveOfSchema<
                  Shape[key]
                >;
              } & (CatchAll extends AnyZodOrCoValueSchema
                ? {
                    readonly [
                      key: string
                    ]: InstanceOrPrimitiveOfSchema<CatchAll>;
                  }
                : {}) &
                CoMap
            : S extends CoreCoListSchema<infer T>
              ? CoList<InstanceOrPrimitiveOfSchema<T>>
              : S extends CoreCoFeedSchema<infer T>
                ? CoFeed<InstanceOrPrimitiveOfSchema<T>>
                : S extends CorePlainTextSchema
                  ? CoPlainText
                  : S extends CoreRichTextSchema
                    ? CoRichText
                    : S extends CoreFileStreamSchema
                      ? FileStream
                      : S extends CoreCoVectorSchema
                        ? Readonly<CoVector>
                        : S extends CoreSnapshotRefSchema<infer T, any>
                          ? SnapshotRef<InstanceOfSchema<T>>
                          : S extends CoreCoOptionalSchema<infer T>
                            ? InstanceOrPrimitiveOfSchema<T> | undefined
                            : S extends CoDiscriminatedUnionSchema<
                                  infer Members
                                >
                              ? InstanceOrPrimitiveOfSchema<Members[number]>
                              : never
    : S extends CoValueClass
      ? InstanceType<S>
      : never;