123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221
import { Document, EJSON } from "./deps.ts";
export interface MongoClientConstructorOptions {  appId: string;  dataSource: string;  apiKey: string;  endpoint?: string;  fetch?: typeof fetch;}
export class MongoClient {  appId: string;  dataSource: string;  apiKey: string;  endpoint = "https://data.mongodb-api.com";  fetch = fetch;
  constructor(    { appId, dataSource, apiKey, endpoint, fetch }:      MongoClientConstructorOptions,  ) {    this.appId = appId;    this.dataSource = dataSource;    this.apiKey = apiKey;    if (endpoint) {      this.endpoint = endpoint;    }
    if (fetch) {      this.fetch = fetch;    }  }
  database(name: string) {    return new Database(name, this);  }}
class Database {  name: string;  client: MongoClient;
  constructor(name: string, client: MongoClient) {    this.name = name;    this.client = client;  }
  collection<T = Document>(name: string) {    return new Collection<T>(name, this);  }}
class Collection<T> {  name: string;  database: Database;  client: MongoClient;
  constructor(name: string, database: Database) {    this.name = name;    this.database = database;    this.client = database.client;  }
  insertOne(doc: T): Promise<{ insertedId: string }> {    return this.callApi("insertOne", { document: doc });  }
  insertMany(docs: T[]): Promise<{ insertedIds: string[] }> {    return this.callApi("insertMany", { documents: docs });  }
  async findOne(    filter: Document,    { projection }: { projection?: Document } = {},  ): Promise<T> {    const result = await this.callApi("findOne", {      filter,      projection,    });    return result.document;  }
  async find(    filter?: Document,    { projection, sort, limit, skip }: {      projection?: Document;      sort?: Document;      limit?: number;      skip?: number;    } = {},  ): Promise<T[]> {    const result = await this.callApi("find", {      filter,      projection,      sort,      limit,      skip,    });    return result.documents;  }
  updateOne(    filter: Document,    update: Document,    { upsert }: { upsert?: boolean },  ): Promise<    { matchedCount: number; modifiedCount: number; upsertedId?: string }  > {    return this.callApi("updateOne", {      filter,      update,      upsert,    });  }
  updateMany(    filter: Document,    update: Document,    { upsert }: { upsert?: boolean },  ): Promise<    { matchedCount: number; modifiedCount: number; upsertedId?: string }  > {    return this.callApi("updateMany", {      filter,      update,      upsert,    });  }
  replaceOne(    filter: Document,    replacement: Document,    { upsert }: { upsert?: boolean },  ): Promise<    { matchedCount: number; modifiedCount: number; upsertedId?: string }  > {    return this.callApi("replaceOne", {      filter,      replacement,      upsert,    });  }
  deleteOne(filter: Document): Promise<{ deletedCount: number }> {    return this.callApi("deleteOne", { filter });  }
  deleteMany(filter: Document): Promise<{ deletedCount: number }> {    return this.callApi("deleteMany", { filter });  }
  async aggregate<T = Document>(pipeline: Document[]): Promise<T[]> {    const result = await this.callApi("aggregate", { pipeline });    return result.documents;  }
  async countDocuments(    filter?: Document,    options?: { limit?: number; skip?: number },  ): Promise<number> {    const pipeline: Document[] = [];    if (filter) {      pipeline.push({ $match: filter });    }
    if (typeof options?.skip === "number") {      pipeline.push({ $skip: options.limit });    }
    if (typeof options?.limit === "number") {      pipeline.push({ $limit: options.limit });    }
    pipeline.push({ $group: { _id: 1, n: { $sum: 1 } } });
    const [result] = await this.aggregate<{ n: number }>(pipeline);    if (result) return result.n;    return 0;  }
  async estimatedDocumentCount(): Promise<number> {    const pipeline = [      { $collStats: { count: {} } },      { $group: { _id: 1, n: { $sum: "$count" } } },    ];
    const [result] = await this.aggregate<{ n: number }>(pipeline);    if (result) return result.n;    return 0;  }
  // deno-lint-ignore no-explicit-any  async callApi(method: string, extra: Document): Promise<any> {    const { endpoint, appId, apiKey, dataSource } = this.client;    const url = `${endpoint}/app/${appId}/endpoint/data/v1/action/${method}`;
    const response = await this.client.fetch(url, {      method: "POST",      headers: {        "Content-Type": "application/ejson",        "Accept": "application/ejson",        "api-key": apiKey,      },      body: EJSON.stringify({        collection: this.name,        database: this.database.name,        dataSource: dataSource,        ...extra,      }),    });
    const body = await response.text();
    if (!response.ok) {      throw new Error(`${response.statusText}: ${body}`);    }
    return EJSON.parse(body);  }}