From 18e7b168e982c3582d817ef1815403abf0f7c496 Mon Sep 17 00:00:00 2001
From: Duncan Beevers <duncan@dweebd.com>
Date: Tue, 28 May 2024 18:21:01 -0700
Subject: [PATCH] fixup! feat: Support updates with readonly array values

Closes #828
---
 src/__tests__/model.test.ts | 12 +++++++----
 src/model.ts                |  4 ++--
 src/mongodbTypes.ts         | 42 ++++++++++++++++++++++++-------------
 3 files changed, 37 insertions(+), 21 deletions(-)

diff --git a/src/__tests__/model.test.ts b/src/__tests__/model.test.ts
index 614e15594..2d20c7b2c 100644
--- a/src/__tests__/model.test.ts
+++ b/src/__tests__/model.test.ts
@@ -239,7 +239,7 @@ describe('model', () => {
 
   describe('bulkWrite', () => {
     test('simple schema', async () => {
-      const operations: PaprBulkWriteOperation<SimpleDocument, SimpleOptions>[] = [
+      const operations = [
         {
           insertOne: {
             document: {
@@ -283,7 +283,9 @@ describe('model', () => {
             filter: { foo: 'foo' },
           },
         },
-      ];
+      ] as const;
+
+      expectType<readonly PaprBulkWriteOperation<SimpleDocument, SimpleOptions>[]>(operations);
 
       await simpleModel.bulkWrite(operations);
 
@@ -293,7 +295,7 @@ describe('model', () => {
     });
 
     test('schema with defaults', async () => {
-      const operations: PaprBulkWriteOperation<SimpleDocument, SimpleOptions>[] = [
+      const operations = [
         {
           insertOne: {
             document: {
@@ -367,7 +369,9 @@ describe('model', () => {
             upsert: true,
           },
         },
-      ];
+      ] as const;
+
+      expectType<readonly PaprBulkWriteOperation<SimpleDocument, SimpleOptions>[]>(operations);
 
       await simpleModel.bulkWrite(operations);
 
diff --git a/src/model.ts b/src/model.ts
index 9a4a4b90c..2512f822d 100644
--- a/src/model.ts
+++ b/src/model.ts
@@ -57,7 +57,7 @@ export interface Model<TSchema extends BaseSchema, TOptions extends SchemaOption
   ) => Promise<TResult[]>;
 
   bulkWrite: (
-    operations: PaprBulkWriteOperation<TSchema, TOptions>[],
+    operations: readonly PaprBulkWriteOperation<TSchema, TOptions>[],
     options?: BulkWriteOptions
   ) => Promise<BulkWriteResult | void>;
 
@@ -376,7 +376,7 @@ export function build<TSchema extends BaseSchema, TOptions extends SchemaOptions
   model.bulkWrite = wrap(
     model,
     async function bulkWrite(
-      operations: PaprBulkWriteOperation<TSchema, TOptions>[],
+      operations: readonly PaprBulkWriteOperation<TSchema, TOptions>[],
       options?: BulkWriteOptions
     ): Promise<BulkWriteResult | void> {
       if (operations.length === 0) {
diff --git a/src/mongodbTypes.ts b/src/mongodbTypes.ts
index 7ba9c4768..86ca8b49d 100644
--- a/src/mongodbTypes.ts
+++ b/src/mongodbTypes.ts
@@ -42,7 +42,7 @@ import { DocumentForInsert, NestedPaths, PropertyType } from './utils';
 // We've adopted these types in this repository and made some improvements to them.
 // See: https://github.com/plexinc/papr/issues/410
 
-// These buik operation types need our own `PaprFilter` and `PaprUpdateFilter` in their definition
+// These bulk operation types need our own `PaprFilter` and `PaprUpdateFilter` in their definition
 export type PaprBulkWriteOperation<TSchema, TOptions extends SchemaOptions<TSchema>> =
   | {
       // @ts-expect-error Type expects a Document extended type, but Document is too generic
@@ -171,7 +171,7 @@ export type PaprAllProperties<TSchema> = {
  * }
  */
 export type PaprArrayElementsProperties<TSchema> = {
-  [Property in `${KeysOfAType<PaprAllProperties<TSchema>, any[]>}.$${
+  [Property in `${KeysOfAType<PaprAllProperties<TSchema>, readonly any[]>}.$${
     | ''
     | `[${string}]`}`]?: ArrayElement<
     PropertyType<TSchema, Property extends `${infer Key}.$${string}` ? Key : never>
@@ -204,6 +204,18 @@ export type PaprMatchKeysAndValues<TSchema> = PaprAllProperties<TSchema> &
   PaprArrayElementsProperties<TSchema> &
   PaprArrayNestedProperties<TSchema>;
 
+type DeepReadonly<TSchema> = TSchema extends (infer Member)[]
+  ? DeepReadonlyArray<Member>
+  : TSchema extends object
+    ? DeepReadonlyObject<TSchema>
+    : TSchema;
+
+interface DeepReadonlyArray<TSchema> extends ReadonlyArray<DeepReadonly<TSchema>> {}
+
+type DeepReadonlyObject<TSchema> = {
+  readonly [Prop in keyof TSchema]: DeepReadonly<TSchema[Prop]>;
+};
+
 export interface PaprUpdateFilter<TSchema> {
   $currentDate?: OnlyFieldsOfType<
     TSchema,
@@ -213,21 +225,21 @@ export interface PaprUpdateFilter<TSchema> {
         $type: 'date' | 'timestamp';
       }
   >;
-  $inc?: OnlyFieldsOfType<TSchema, NumericType | undefined>;
-  $min?: PaprMatchKeysAndValues<TSchema>;
-  $max?: PaprMatchKeysAndValues<TSchema>;
-  $mul?: OnlyFieldsOfType<TSchema, NumericType | undefined>;
+  $inc?: OnlyFieldsOfType<DeepReadonly<TSchema>, NumericType | undefined>;
+  $min?: PaprMatchKeysAndValues<DeepReadonly<TSchema>>;
+  $max?: PaprMatchKeysAndValues<DeepReadonly<TSchema>>;
+  $mul?: OnlyFieldsOfType<DeepReadonly<TSchema>, NumericType | undefined>;
   $rename?: Record<string, string>;
-  $set?: PaprMatchKeysAndValues<TSchema>;
-  $setOnInsert?: PaprMatchKeysAndValues<TSchema>;
-  $unset?: OnlyFieldsOfType<TSchema, any, '' | 1 | true>;
-  $addToSet?: SetFields<TSchema>;
-  $pop?: OnlyFieldsOfType<TSchema, readonly any[], -1 | 1>;
-  $pull?: PullOperator<TSchema>;
-  $push?: PushOperator<TSchema>;
-  $pullAll?: PullAllOperator<TSchema>;
+  $set?: PaprMatchKeysAndValues<DeepReadonly<TSchema>>;
+  $setOnInsert?: PaprMatchKeysAndValues<DeepReadonly<TSchema>>;
+  $unset?: OnlyFieldsOfType<DeepReadonly<TSchema>, any, '' | 1 | true>;
+  $addToSet?: SetFields<DeepReadonly<TSchema>>;
+  $pop?: OnlyFieldsOfType<DeepReadonly<TSchema>, readonly any[], -1 | 1>;
+  $pull?: PullOperator<DeepReadonly<TSchema>>;
+  $push?: PushOperator<DeepReadonly<TSchema>>;
+  $pullAll?: PullAllOperator<DeepReadonly<TSchema>>;
   $bit?: OnlyFieldsOfType<
-    TSchema,
+    DeepReadonly<TSchema>,
     NumericType | undefined,
     | {
         and: IntegerType;