drapcode-utility 1.1.7 → 1.1.8

package/build/encryption/file.d.ts

@@ -0,0 +1,4 @@
+import { Encryption } from "./model";
+export declare const processFileEncryptionDecryption: (data: string, encryption: Encryption, decrypt: boolean) => Promise<string>;
+export declare const encryptFile: (filePath: string, key: string) => Promise<string>;
+export declare const decryptFile: (encryptedFilePath: string, key: string) => Promise<string>;

package/build/encryption/file.js

@@ -0,0 +1,134 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __generator = (this && this.__generator) || function (thisArg, body) {
+    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
+    return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
+    function verb(n) { return function (v) { return step([n, v]); }; }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [op[0] & 2, t.value];
+            switch (op[0]) {
+                case 0: case 1: t = op; break;
+                case 4: _.label++; return { value: op[1], done: false };
+                case 5: _.label++; y = op[1]; op = [0]; continue;
+                case 7: op = _.ops.pop(); _.trys.pop(); continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
+                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
+                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
+                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop(); continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
+        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+    }
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.decryptFile = exports.encryptFile = exports.processFileEncryptionDecryption = void 0;
+var fs_1 = __importDefault(require("fs"));
+var crypto_1 = __importDefault(require("crypto"));
+var defaultAlgorithm = "aes-256-cbc";
+var processFileEncryptionDecryption = function (data, encryption, decrypt) { return __awaiter(void 0, void 0, void 0, function () {
+    var encryptionType, dataKey, _a, result, _b;
+    return __generator(this, function (_c) {
+        switch (_c.label) {
+            case 0:
+                encryptionType = encryption.encryptionType, dataKey = encryption.dataKey;
+                if (!data)
+                    return [2 /*return*/, data];
+                _a = encryptionType;
+                switch (_a) {
+                    case "KMS": return [3 /*break*/, 1];
+                }
+                return [3 /*break*/, 6];
+            case 1:
+                if (!decrypt) return [3 /*break*/, 3];
+                return [4 /*yield*/, (0, exports.decryptFile)(data, dataKey)];
+            case 2:
+                _b = _c.sent();
+                return [3 /*break*/, 5];
+            case 3: return [4 /*yield*/, (0, exports.encryptFile)(data, dataKey)];
+            case 4:
+                _b = _c.sent();
+                _c.label = 5;
+            case 5:
+                result = _b;
+                return [2 /*return*/, result];
+            case 6: return [2 /*return*/, data];
+        }
+    });
+}); };
+exports.processFileEncryptionDecryption = processFileEncryptionDecryption;
+var encryptFile = function (filePath, key) { return __awaiter(void 0, void 0, void 0, function () {
+    var iv, keyBuffer, fileContent, cipher, encryptedDataBuffer, encryptedFilePath, error_1;
+    return __generator(this, function (_a) {
+        switch (_a.label) {
+            case 0:
+                _a.trys.push([0, 3, , 4]);
+                iv = Buffer.from("i4mboZDwaNEC38YCzi77lw==", "base64");
+                keyBuffer = Buffer.from(key, "base64");
+                return [4 /*yield*/, fs_1.default.promises.readFile(filePath)];
+            case 1:
+                fileContent = _a.sent();
+                cipher = crypto_1.default.createCipheriv(defaultAlgorithm, keyBuffer, iv);
+                encryptedDataBuffer = cipher.update(fileContent);
+                encryptedDataBuffer = Buffer.concat([encryptedDataBuffer, cipher.final()]);
+                encryptedFilePath = filePath + ".enc";
+                return [4 /*yield*/, fs_1.default.promises.writeFile(encryptedFilePath, encryptedDataBuffer)];
+            case 2:
+                _a.sent();
+                console.log("File encrypted successfully.");
+                return [2 /*return*/, encryptedFilePath];
+            case 3:
+                error_1 = _a.sent();
+                console.log("\n Error: ", error_1);
+                throw error_1;
+            case 4: return [2 /*return*/];
+        }
+    });
+}); };
+exports.encryptFile = encryptFile;
+var decryptFile = function (encryptedFilePath, key) { return __awaiter(void 0, void 0, void 0, function () {
+    var iv, keyBuffer, encryptedData, decipher, decryptedBuffer, decryptedFilePath, error_2;
+    return __generator(this, function (_a) {
+        switch (_a.label) {
+            case 0:
+                _a.trys.push([0, 3, , 4]);
+                iv = Buffer.from("i4mboZDwaNEC38YCzi77lw==", "base64");
+                keyBuffer = Buffer.from(key, "base64");
+                return [4 /*yield*/, fs_1.default.promises.readFile(encryptedFilePath)];
+            case 1:
+                encryptedData = _a.sent();
+                decipher = crypto_1.default.createDecipheriv(defaultAlgorithm, keyBuffer, iv);
+                decryptedBuffer = decipher.update(encryptedData);
+                decryptedBuffer = Buffer.concat([decryptedBuffer, decipher.final()]);
+                decryptedFilePath = "".concat(encryptedFilePath.slice(0, -4), ".dec");
+                return [4 /*yield*/, fs_1.default.promises.writeFile(decryptedFilePath, decryptedBuffer)];
+            case 2:
+                _a.sent();
+                console.log("File decrypted successfully.");
+                return [2 /*return*/, decryptedFilePath];
+            case 3:
+                error_2 = _a.sent();
+                console.log("Error:", error_2);
+                throw error_2;
+            case 4: return [2 /*return*/];
+        }
+    });
+}); };
+exports.decryptFile = decryptFile;

package/build/encryption/index.d.ts

@@ -46,3 +46,8 @@ export declare const drapcodeEncryptDecrypt: (data: string, encrypt: boolean) =>
     data: string;
     message: string;
 }>;
+export declare const cryptFile: (filePath: any, encryption: Encryption, decrypt: boolean) => Promise<string | {
+    status: string;
+    data: string;
+    message: string;
+}>;

package/build/encryption/index.js

@@ -36,8 +36,9 @@ var __generator = (this && this.__generator) || function (thisArg, body) {
     }
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.drapcodeEncryptDecrypt = exports.decryptDataWithKMS = exports.encryptDataWithKMS = exports.createKMSDataKey = exports.processDataEncryptionDecryption = exports.processItemEncryptDecrypt = exports.crypt = void 0;
+exports.cryptFile = exports.drapcodeEncryptDecrypt = exports.decryptDataWithKMS = exports.encryptDataWithKMS = exports.createKMSDataKey = exports.processDataEncryptionDecryption = exports.processItemEncryptDecrypt = exports.crypt = void 0;
 var crypt_1 = require("./crypt");
+var file_1 = require("./file");
 var KMS_1 = require("./KMS");
 var crypt = function (data, fields, encryption, decrypt) { return __awaiter(void 0, void 0, void 0, function () {
     var _a, accessKeyId, secretAccessKey, region, config, plainTextData, promises;
@@ -264,3 +265,32 @@ var drapcodeEncryptDecrypt = function (data, encrypt) { return __awaiter(void 0,
     });
 }); };
 exports.drapcodeEncryptDecrypt = drapcodeEncryptDecrypt;
+var cryptFile = function (filePath, encryption, decrypt) { return __awaiter(void 0, void 0, void 0, function () {
+    var awsConfig, encryptionType, dataKey, accessKeyId, secretAccessKey, region, config, plainTextData, data;
+    return __generator(this, function (_a) {
+        switch (_a.label) {
+            case 0:
+                awsConfig = encryption.awsConfig, encryptionType = encryption.encryptionType, dataKey = encryption.dataKey;
+                if (!(encryptionType === "KMS")) return [3 /*break*/, 2];
+                accessKeyId = awsConfig.accessKeyId, secretAccessKey = awsConfig.secretAccessKey, region = awsConfig.region;
+                config = {
+                    region: region,
+                    accessKeyId: accessKeyId,
+                    secretAccessKey: secretAccessKey,
+                };
+                return [4 /*yield*/, (0, KMS_1.processKMSDecryption)(config, dataKey, {})];
+            case 1:
+                plainTextData = _a.sent();
+                if (plainTextData.status === "FAILED") {
+                    return [2 /*return*/, plainTextData];
+                }
+                encryption.dataKey = plainTextData.data;
+                _a.label = 2;
+            case 2: return [4 /*yield*/, (0, file_1.processFileEncryptionDecryption)(filePath, encryption, decrypt)];
+            case 3:
+                data = _a.sent();
+                return [2 /*return*/, data];
+        }
+    });
+}); };
+exports.cryptFile = cryptFile;

package/build/index.d.ts

@@ -13,5 +13,6 @@ export * from "./utils/util";
 export * from "./utils/uuid-generator";
 export * from "./utils/check-error";
 export * from "./utils/prepare-query";
+export * from "./utils/s3-util";
 export * from "./encryption/index";
 export * from "./format-fields/index";

package/build/index.js

@@ -29,5 +29,6 @@ __exportStar(require("./utils/util"), exports);
 __exportStar(require("./utils/uuid-generator"), exports);
 __exportStar(require("./utils/check-error"), exports);
 __exportStar(require("./utils/prepare-query"), exports);
+__exportStar(require("./utils/s3-util"), exports);
 __exportStar(require("./encryption/index"), exports);
 __exportStar(require("./format-fields/index"), exports);

package/build/utils/s3-util.d.ts

@@ -0,0 +1,28 @@
+import { S3Client, ObjectCannedACL } from "@aws-sdk/client-s3";
+import { Encryption } from "../encryption/model";
+interface AWSConfig {
+    region: string;
+    accessKey: string;
+    accessSecret: string;
+}
+interface S3Config {
+    acl: ObjectCannedACL;
+    key: string;
+    bucket: string;
+    append: boolean;
+}
+export declare const createS3Client: (awsConfig: AWSConfig) => S3Client | null;
+export declare const fileUploadToS3: (files: any, s3Client: S3Client, s3Config: S3Config, encryption: Encryption, options?: {}) => Promise<{
+    originalname: any;
+    mimetype: any;
+    size: any;
+    key: string | undefined;
+    contentType: string;
+} | {
+    originalname: any;
+    mimetype: any;
+    size: any;
+    key: string | undefined;
+    contentType: string;
+}[] | null>;
+export {};

package/build/utils/s3-util.js

@@ -0,0 +1,173 @@
+"use strict";
+var __assign = (this && this.__assign) || function () {
+    __assign = Object.assign || function(t) {
+        for (var s, i = 1, n = arguments.length; i < n; i++) {
+            s = arguments[i];
+            for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
+                t[p] = s[p];
+        }
+        return t;
+    };
+    return __assign.apply(this, arguments);
+};
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __generator = (this && this.__generator) || function (thisArg, body) {
+    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
+    return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
+    function verb(n) { return function (v) { return step([n, v]); }; }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [op[0] & 2, t.value];
+            switch (op[0]) {
+                case 0: case 1: t = op; break;
+                case 4: _.label++; return { value: op[1], done: false };
+                case 5: _.label++; y = op[1]; op = [0]; continue;
+                case 7: op = _.ops.pop(); _.trys.pop(); continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
+                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
+                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
+                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop(); continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
+        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+    }
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fileUploadToS3 = exports.createS3Client = void 0;
+var client_s3_1 = require("@aws-sdk/client-s3");
+var fs_1 = __importDefault(require("fs"));
+var mime_types_1 = __importDefault(require("mime-types"));
+var lib_storage_1 = require("@aws-sdk/lib-storage");
+var encryption_1 = require("../encryption");
+var createS3Client = function (awsConfig) {
+    if (!awsConfig) {
+        return null;
+    }
+    var region = awsConfig.region, accessKey = awsConfig.accessKey, accessSecret = awsConfig.accessSecret;
+    var s3client = new client_s3_1.S3Client({
+        region: region,
+        credentials: {
+            accessKeyId: accessKey,
+            secretAccessKey: accessSecret,
+        },
+    });
+    return s3client;
+};
+exports.createS3Client = createS3Client;
+var fileUploadToS3 = function (files, s3Client, s3Config, encryption, options) {
+    if (options === void 0) { options = {}; }
+    return __awaiter(void 0, void 0, void 0, function () {
+        var fileOptions, s3Key, _a, _b, _c, _i, index, file, fileObj, fileObj;
+        return __generator(this, function (_d) {
+            switch (_d.label) {
+                case 0:
+                    if (!Array.isArray(files)) return [3 /*break*/, 5];
+                    console.log("I have multiple file");
+                    fileOptions = [];
+                    s3Key = s3Config.key;
+                    _a = files;
+                    _b = [];
+                    for (_c in _a)
+                        _b.push(_c);
+                    _i = 0;
+                    _d.label = 1;
+                case 1:
+                    if (!(_i < _b.length)) return [3 /*break*/, 4];
+                    _c = _b[_i];
+                    if (!(_c in _a)) return [3 /*break*/, 3];
+                    index = _c;
+                    file = files[index];
+                    if (s3Config.append) {
+                        s3Config.key = "".concat(s3Key, "/").concat(file.originalname);
+                    }
+                    return [4 /*yield*/, processFileUploadToS3(file, s3Client, s3Config, encryption, options, encryption ? true : false)];
+                case 2:
+                    fileObj = _d.sent();
+                    if (fileObj) {
+                        fileOptions.push(fileObj);
+                    }
+                    _d.label = 3;
+                case 3:
+                    _i++;
+                    return [3 /*break*/, 1];
+                case 4: return [2 /*return*/, fileOptions];
+                case 5:
+                    console.log("I have single file");
+                    if (s3Config.append) {
+                        s3Config.key = "".concat(s3Config.key, "/").concat(files.originalname);
+                    }
+                    return [4 /*yield*/, processFileUploadToS3(files, s3Client, s3Config, encryption, options, encryption ? true : false)];
+                case 6:
+                    fileObj = _d.sent();
+                    return [2 /*return*/, fileObj];
+            }
+        });
+    });
+};
+exports.fileUploadToS3 = fileUploadToS3;
+var processFileUploadToS3 = function (file, s3Client, s3Config, encryption, options, encrypt) {
+    if (options === void 0) { options = {}; }
+    if (encrypt === void 0) { encrypt = false; }
+    return __awaiter(void 0, void 0, void 0, function () {
+        var contentType, acl, key, bucket, params, encryptedBody, parallelUploads3, data, originalname, mimetype, size, finalObject, error_1;
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    _a.trys.push([0, 5, , 6]);
+                    contentType = mime_types_1.default.lookup(file.originalname) || "";
+                    acl = s3Config.acl, key = s3Config.key, bucket = s3Config.bucket;
+                    params = null;
+                    if (!encrypt) return [3 /*break*/, 2];
+                    return [4 /*yield*/, (0, encryption_1.cryptFile)(file.path, encryption, false)];
+                case 1:
+                    encryptedBody = _a.sent();
+                    params = __assign({ Bucket: bucket, ACL: acl, Key: key, 
+                        //@ts-ignore
+                        Body: fs_1.default.createReadStream(encryptedBody), ContentType: contentType }, options);
+                    return [3 /*break*/, 3];
+                case 2:
+                    params = __assign({ Bucket: bucket, ACL: acl, Key: key, Body: fs_1.default.createReadStream(file.path), ContentType: contentType }, options);
+                    _a.label = 3;
+                case 3:
+                    parallelUploads3 = new lib_storage_1.Upload({
+                        client: s3Client,
+                        params: params,
+                    });
+                    return [4 /*yield*/, parallelUploads3.done()];
+                case 4:
+                    data = _a.sent();
+                    originalname = file.originalname, mimetype = file.mimetype, size = file.size;
+                    finalObject = {
+                        originalname: originalname,
+                        mimetype: mimetype,
+                        size: size,
+                        key: data.Key,
+                        contentType: contentType,
+                    };
+                    return [2 /*return*/, finalObject];
+                case 5:
+                    error_1 = _a.sent();
+                    console.log("error", error_1);
+                    return [2 /*return*/, null];
+                case 6: return [2 /*return*/];
+            }
+        });
+    });
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "drapcode-utility",
-  "version": "1.1.7",
+  "version": "1.1.8",
   "description": "",
   "main": "build/index.js",
   "types": "build/index.d.ts",
@@ -20,11 +20,14 @@
   "license": "ISC",
   "devDependencies": {
     "@types/lodash": "^4.14.179",
+    "@types/mime-types": "^2.1.4",
     "del-cli": "^5.0.0",
     "typescript": "^4.0.2"
   },
   "dependencies": {
     "@aws-sdk/client-kms": "^3.540.0",
+    "@aws-sdk/client-s3": "^3.499.0",
+    "@aws-sdk/lib-storage": "^3.499.0",
     "@types/cookie-session": "^2.0.41",
     "@types/express": "^4.17.7",
     "@types/voca": "^1.4.2",
@@ -34,6 +37,7 @@
     "drapcode-redis": "^1.0.4",
     "express": "^4.17.1",
     "lodash": "^4.17.21",
+    "mime-types": "^2.1.35",
     "moment": "^2.29.0",
     "stringify-object": "^3.3.0",
     "voca": "^1.4.0"