123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
import { Collection } from "./collection/mod.ts";import { CommandCursor } from "./protocol/mod.ts";import { CreateUserOptions, Document } from "./types.ts";import { Cluster } from "./cluster.ts";
interface ListCollectionsReponse {  cursor: {    id: bigint;    ns: string;    firstBatch: [      {        name: string;        type: "collection";      },    ];  };  ok: 1;}
export interface ListCollectionsResult {  name: string;  type: "collection";}
export class Database {  #cluster: Cluster;
  constructor(cluster: Cluster, readonly name: string) {    this.#cluster = cluster;  }
  collection<T>(name: string): Collection<T> {    return new Collection(this.#cluster.protocol, this.name, name);  }
  listCollections(options?: {    filter?: Document;    nameOnly?: boolean;    authorizedCollections?: boolean;    comment?: Document;  }): CommandCursor<ListCollectionsResult> {    if (!options) {      options = {};    }    return new CommandCursor<ListCollectionsResult>(      this.#cluster.protocol,      async () => {        const { cursor } = await this.#cluster.protocol.commandSingle<          ListCollectionsReponse        >(this.name, {          listCollections: 1,          ...options,          batchSize: 1,        });        return {          id: cursor.id,          ns: cursor.ns,          firstBatch: cursor.firstBatch,        };      },    );  }
  async listCollectionNames(options?: {    filter?: Document;    authorizedCollections?: boolean;    comment?: Document;  }): Promise<string[]> {    const cursor = this.listCollections({      ...options,      nameOnly: true,      authorizedCollections: true,    });    const names: string[] = [];    for await (const item of cursor) {      names.push(item!.name);    }    return names;  }
  async createUser(    username: string,    password: string,    options?: CreateUserOptions,  ) {    await this.#cluster.protocol.commandSingle(this.name, {      createUser: options?.username ?? username,      pwd: options?.password ?? password,      customData: options?.customData,      roles: options?.roles ?? [],      writeConcern: options?.writeConcern,      authenticationRestrictions: options?.authenticationRestrictions,      mechanisms: options?.mechanisms,      digestPassword: options?.digestPassword,      comment: options?.comment,    });  }
  async dropUser(username: string, options?: {    writeConcern?: Document;    comment?: Document;  }) {    await this.#cluster.protocol.commandSingle(this.name, {      dropUser: username,      writeConcern: options?.writeConcern,      comment: options?.comment,    });  }}