From 7ec87469faa01dd904641f9ddb9daf44e442c5a1 Mon Sep 17 00:00:00 2001
From: Cosimo Matteini <dev.matteini@gmail.com>
Date: Mon, 11 Nov 2024 16:00:34 +0100
Subject: [PATCH] Collection+DocumentCollection: add replaceOne

---
 packages/effect-mongodb/src/Collection.ts     | 38 +++++++++++++++++++
 .../effect-mongodb/src/DocumentCollection.ts  | 32 +++++++++++++++-
 2 files changed, 69 insertions(+), 1 deletion(-)

diff --git a/packages/effect-mongodb/src/Collection.ts b/packages/effect-mongodb/src/Collection.ts
index df348f9..ca19e9c 100644
--- a/packages/effect-mongodb/src/Collection.ts
+++ b/packages/effect-mongodb/src/Collection.ts
@@ -17,6 +17,7 @@ import type {
   InsertManyResult,
   InsertOneOptions,
   InsertOneResult,
+  ReplaceOptions,
   UpdateFilter,
   UpdateOptions,
   UpdateResult
@@ -207,4 +208,41 @@ export const updateMany: {
     )
 )
 
+export const replaceOne: {
+  <A extends Document, I extends Document>(
+    filter: Filter<I>,
+    // TODO: should we put WithoutId<A> here like the driver signature?
+    replacement: A,
+    options?: ReplaceOptions
+  ): <R>(
+    collection: Collection<A, I, R>
+  ) => Effect.Effect<UpdateResult | Document, MongoError.MongoError | ParseResult.ParseError, R>
+  <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>,
+    filter: Filter<I>,
+    replacement: A,
+    options?: ReplaceOptions
+  ): Effect.Effect<UpdateResult | Document, MongoError.MongoError | ParseResult.ParseError, R>
+} = F.dual(
+  (args) => isCollection(args[0]),
+  <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>,
+    filter: Filter<I>,
+    replacement: A,
+    options?: ReplaceOptions
+  ): Effect.Effect<
+    UpdateResult | Document,
+    MongoError.MongoError | ParseResult.ParseError,
+    R
+  > =>
+    F.pipe(
+      // TODO: extract function in Collection
+      Schema.encode(collection.schema)(replacement),
+      Effect.flatMap((replacement) =>
+        Effect.promise(() => collection.collection.replaceOne(filter, replacement, options))
+      ),
+      Effect.catchAllDefect(MongoError.mongoErrorDie<UpdateResult | Document>("replaceOne error"))
+    )
+)
+
 const isCollection = (x: unknown) => x instanceof Collection
diff --git a/packages/effect-mongodb/src/DocumentCollection.ts b/packages/effect-mongodb/src/DocumentCollection.ts
index 391954c..551005c 100644
--- a/packages/effect-mongodb/src/DocumentCollection.ts
+++ b/packages/effect-mongodb/src/DocumentCollection.ts
@@ -18,9 +18,11 @@ import type {
   InsertOneOptions,
   InsertOneResult,
   OptionalUnlessRequiredId,
+  ReplaceOptions,
   UpdateFilter,
   UpdateOptions,
-  UpdateResult
+  UpdateResult,
+  WithoutId
 } from "mongodb"
 import * as Collection from "./Collection.js"
 import * as DocumentFindCursor from "./DocumentFindCursor.js"
@@ -195,6 +197,34 @@ export const updateMany: {
     )
 )
 
+export const replaceOne: {
+  (
+    filter: Filter<Document>,
+    replacement: WithoutId<Document>,
+    options?: ReplaceOptions
+  ): (
+    collection: DocumentCollection
+  ) => Effect.Effect<UpdateResult | Document, MongoError.MongoError>
+  (
+    collection: DocumentCollection,
+    filter: Filter<Document>,
+    replacement: WithoutId<Document>,
+    options?: ReplaceOptions
+  ): Effect.Effect<UpdateResult | Document, MongoError.MongoError>
+} = F.dual(
+  (args) => isDocumentCollection(args[0]),
+  (
+    collection: DocumentCollection,
+    filter: Filter<Document>,
+    replacement: WithoutId<Document>,
+    options?: ReplaceOptions
+  ): Effect.Effect<UpdateResult | Document, MongoError.MongoError> =>
+    F.pipe(
+      Effect.promise(() => collection.collection.replaceOne(filter, replacement, options)),
+      Effect.catchAllDefect(MongoError.mongoErrorDie<UpdateResult | Document>("replaceOne error"))
+    )
+)
+
 export const typed: {
   <A extends Document, I extends Document = A, R = never>(
     schema: Schema.Schema<A, I, R>