Collection+DocumentCollection: add rename

doubleloop-io · Nov 11, 2024 · a1c9463 · a1c9463
1 parent 7ec8746
commit a1c9463
Show file tree

Hide file tree

Showing 2 changed files with 53 additions and 0 deletions.
diff --git a/packages/effect-mongodb/src/Collection.ts b/packages/effect-mongodb/src/Collection.ts
@@ -17,6 +17,7 @@ import type {
   InsertManyResult,
   InsertOneOptions,
   InsertOneResult,
+  RenameOptions,
   ReplaceOptions,
   UpdateFilter,
   UpdateOptions,
@@ -245,4 +246,30 @@ export const replaceOne: {
     )
 )
 
+export const rename: {
+  (
+    newName: string,
+    options?: RenameOptions
+  ): <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>
+  ) => Effect.Effect<Collection<A, I, R>, MongoError.MongoError, R>
+  <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>,
+    newName: string,
+    options?: RenameOptions
+  ): Effect.Effect<Collection<A, I, R>, MongoError.MongoError, R>
+} = F.dual(
+  (args) => isCollection(args[0]),
+  <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>,
+    newName: string,
+    options?: RenameOptions
+  ): Effect.Effect<Collection<A, I, R>, MongoError.MongoError, R> =>
+    F.pipe(
+      Effect.promise(() => collection.collection.rename(newName, options)),
+      Effect.map((newCollection) => new Collection<A, I, R>({ collection: newCollection, schema: collection.schema })),
+      Effect.catchAllDefect(MongoError.mongoErrorDie<Collection<A, I, R>>("rename error"))
+    )
+)
+
 const isCollection = (x: unknown) => x instanceof Collection
diff --git a/packages/effect-mongodb/src/DocumentCollection.ts b/packages/effect-mongodb/src/DocumentCollection.ts
@@ -18,6 +18,7 @@ import type {
   InsertOneOptions,
   InsertOneResult,
   OptionalUnlessRequiredId,
+  RenameOptions,
   ReplaceOptions,
   UpdateFilter,
   UpdateOptions,
@@ -225,6 +226,31 @@ export const replaceOne: {
     )
 )
 
+export const rename: {
+  (
+    newName: string,
+    options?: RenameOptions
+  ): (
+    collection: DocumentCollection
+  ) => Effect.Effect<MongoCollection, MongoError.MongoError>
+  (
+    collection: DocumentCollection,
+    newName: string,
+    options?: RenameOptions
+  ): Effect.Effect<MongoCollection, MongoError.MongoError>
+} = F.dual(
+  (args) => isDocumentCollection(args[0]),
+  (
+    collection: DocumentCollection,
+    newName: string,
+    options?: RenameOptions
+  ): Effect.Effect<MongoCollection, MongoError.MongoError> =>
+    F.pipe(
+      Effect.promise(() => collection.collection.rename(newName, options)),
+      Effect.catchAllDefect(MongoError.mongoErrorDie<MongoCollection>("rename error"))
+    )
+)
+
 export const typed: {
   <A extends Document, I extends Document = A, R = never>(
     schema: Schema.Schema<A, I, R>