@smartsoft001/mongo 1.1.91 → 1.3.0

package/src/lib/repositories/attachment.repository.ts

@@ -0,0 +1,113 @@
+import { Injectable } from '@nestjs/common';
+import { IEntity, IAttachmentRepository } from '@smartsoft001/domain-core';
+import { MongoConfig } from '@smartsoft001/mongo';
+import { MongoClient } from 'mongodb';
+import * as mongo from 'mongodb';
+import { Readable, Stream } from 'stream';
+
+import { getMongoUrl } from '../mongo.utils';
+
+@Injectable()
+export class MongoAttachmentRepository<
+  T extends IEntity<string>,
+> extends IAttachmentRepository<T> {
+  constructor(private config: MongoConfig) {
+    super();
+  }
+
+  async upload(
+    data: {
+      id: string;
+      fileName: string;
+      stream: Stream;
+      mimeType: string;
+      encoding: string;
+    },
+    options?: { streamCallback?: (r) => void },
+  ): Promise<void> {
+    const client = await MongoClient.connect(this.getUrl());
+
+    return await new Promise<void>((res, rej) => {
+      const db = client.db(this.config.database);
+      const bucket = new mongo.GridFSBucket(db, {
+        bucketName: this.config.collection,
+      });
+
+      const writeStream = bucket.openUploadStreamWithId(
+        data.id as any,
+        data.fileName,
+        {
+          contentType: data.mimeType,
+        },
+      );
+
+      if (options?.streamCallback) options.streamCallback(writeStream);
+
+      data.stream.pipe(writeStream);
+
+      writeStream.on('error', (error) => {
+        rej(error);
+      });
+
+      writeStream.on('finish', () => {
+        res();
+      });
+    });
+  }
+
+  async getInfo(
+    id: string,
+  ): Promise<{ fileName: string; contentType: string; length: number }> {
+    const client = await MongoClient.connect(this.getUrl());
+
+    const db = client.db(this.config.database);
+
+    const bucket = new mongo.GridFSBucket(db, {
+      bucketName: this.config.collection,
+    });
+
+    const items = await bucket
+      .find({
+        _id: id as any,
+      })
+      .toArray();
+
+    if (!items || items.length === 0) return null;
+
+    return {
+      fileName: items[0].filename,
+      contentType: items[0].contentType,
+      length: items[0].length,
+    };
+  }
+
+  async getStream(
+    id: string,
+    options: { start: number; end: number } | undefined,
+  ): Promise<Readable> {
+    const client = await MongoClient.connect(this.getUrl());
+
+    const db = client.db(this.config.database);
+    const bucket = new mongo.GridFSBucket(db, {
+      bucketName: this.config.collection,
+    });
+
+    return bucket.openDownloadStream(id as any, options);
+  }
+
+  async delete(id: string): Promise<void> {
+    const client = await MongoClient.connect(this.getUrl());
+
+    const db = client.db(this.config.database);
+    const bucket = new mongo.GridFSBucket(db, {
+      bucketName: this.config.collection,
+    });
+
+    await bucket.delete(id as any);
+    await client.close();
+  }
+
+  private getUrl(): string {
+    return getMongoUrl(this.config);
+  }
+}

package/src/lib/repositories/interfaces.ts

@@ -0,0 +1,28 @@
+export interface IItemCreateData {
+  id: string;
+  type: 'create';
+  data: any;
+}
+
+export interface IItemUpdateData {
+  id: string;
+  type: 'update';
+  data: {
+    removedFields: Array<string>;
+    updatedFields: {
+      [key: string]: any;
+    };
+  };
+}
+
+export interface IItemDeleteData {
+  id: string;
+  type: 'delete';
+}
+
+export type ItemChangedData =
+  | IItemCreateData
+  | IItemUpdateData
+  | IItemDeleteData;
+
+export type ItemChangedDataType = 'create' | 'update' | 'delete';