add estimetedDocumentCount and countDocuments in Collection and DocumentCollection

Author: VenomAV

.changeset/red-ways-check.md

@@ -0,0 +1,5 @@
+---
+"effect-mongodb": patch
+---
+
+Add estimetedDocumentCount and countDocuments in Collection and DocumentCollection

packages/effect-mongodb/src/Collection.ts

@@ -11,12 +11,14 @@ import type {
   AggregateOptions,
   BulkWriteOptions,
   Collection as MongoCollection,
+  CountDocumentsOptions,
   CreateIndexesOptions,
   DeleteOptions,
   DeleteResult,
   Document,
   DropCollectionOptions,
   DropIndexesOptions,
+  EstimatedDocumentCountOptions,
   FindOneAndReplaceOptions,
   FindOptions as MongoFindOptions,
   IndexDescription,
@@ -493,4 +495,51 @@ export const aggregate: {
     })
 )
 
+export const estimatedDocumentCount: {
+  (
+    options?: EstimatedDocumentCountOptions
+  ): <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>
+  ) => Effect.Effect<number, MongoError.MongoError, R>
+  <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>,
+    options?: EstimatedDocumentCountOptions
+  ): Effect.Effect<number, MongoError.MongoError, R>
+} = F.dual(
+  (args) => isCollection(args[0]),
+  <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>,
+    options?: EstimatedDocumentCountOptions
+  ): Effect.Effect<number, MongoError.MongoError, R> =>
+    F.pipe(
+      Effect.promise(() => collection.collection.estimatedDocumentCount(options)),
+      Effect.catchAllDefect(MongoError.mongoErrorDie<number>("estimatedDocumentCount error"))
+    )
+)
+
+export const countDocuments: {
+  <I extends Document>(
+    filter?: Filter<I>,
+    options?: CountDocumentsOptions
+  ): <A extends Document, R>(
+    collection: Collection<A, I, R>
+  ) => Effect.Effect<number, MongoError.MongoError, R>
+  <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>,
+    filter?: Filter<I>,
+    options?: CountDocumentsOptions
+  ): Effect.Effect<number, MongoError.MongoError, R>
+} = F.dual(
+  (args) => isCollection(args[0]),
+  <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>,
+    filter?: Filter<I>,
+    options?: CountDocumentsOptions
+  ): Effect.Effect<number, MongoError.MongoError, R> =>
+    F.pipe(
+      Effect.promise(() => collection.collection.countDocuments(filter, options)),
+      Effect.catchAllDefect(MongoError.mongoErrorDie<number>("countDocuments error"))
+    )
+)
+
 const isCollection = (x: unknown) => x instanceof Collection

packages/effect-mongodb/src/DocumentCollection.ts

@@ -10,12 +10,14 @@ import type {
   AggregateOptions,
   BulkWriteOptions,
   Collection as MongoCollection,
+  CountDocumentsOptions,
   CreateIndexesOptions,
   DeleteOptions,
   DeleteResult,
   Document,
   DropCollectionOptions,
   DropIndexesOptions,
+  EstimatedDocumentCountOptions,
   Filter,
   FindOneAndReplaceOptions,
   FindOptions,
@@ -449,6 +451,53 @@ export const aggregate: {
     })
 )
 
+export const estimatedDocumentCount: {
+  (
+    options?: EstimatedDocumentCountOptions
+  ): (
+    collection: DocumentCollection
+  ) => Effect.Effect<number, MongoError.MongoError>
+  (
+    collection: DocumentCollection,
+    options?: EstimatedDocumentCountOptions
+  ): Effect.Effect<number, MongoError.MongoError>
+} = F.dual(
+  (args) => isDocumentCollection(args[0]),
+  (
+    collection: DocumentCollection,
+    options?: EstimatedDocumentCountOptions
+  ): Effect.Effect<number, MongoError.MongoError> =>
+    F.pipe(
+      Effect.promise(() => collection.collection.estimatedDocumentCount(options)),
+      Effect.catchAllDefect(MongoError.mongoErrorDie<number>("estimatedDocumentCount error"))
+    )
+)
+
+export const countDocuments: {
+  (
+    filter?: Filter<Document>,
+    options?: CountDocumentsOptions
+  ): (
+    collection: DocumentCollection
+  ) => Effect.Effect<number, MongoError.MongoError>
+  (
+    collection: DocumentCollection,
+    filter?: Filter<Document>,
+    options?: CountDocumentsOptions
+  ): Effect.Effect<number, MongoError.MongoError>
+} = F.dual(
+  (args) => isDocumentCollection(args[0]),
+  (
+    collection: DocumentCollection,
+    filter?: Filter<Document>,
+    options?: CountDocumentsOptions
+  ): Effect.Effect<number, MongoError.MongoError> =>
+    F.pipe(
+      Effect.promise(() => collection.collection.countDocuments(filter, options)),
+      Effect.catchAllDefect(MongoError.mongoErrorDie<number>("countDocuments error"))
+    )
+)
+
 export const typed: {
   <A extends Document, I extends Document = A, R = never>(
     schema: Schema.Schema<A, I, R>