@rowengine/common 1.1.2 → 1.1.3

package/build/models/transaction.model.js

@@ -0,0 +1,61 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TransactionModel = void 0;
+const mongoose_1 = __importDefault(require("mongoose"));
+const interfaces_1 = require("../interfaces");
+const transactionSchema = new mongoose_1.default.Schema({
+    company: {
+        type: mongoose_1.default.Schema.Types.ObjectId,
+        ref: 'Company',
+        required: true,
+    },
+    userEmail: {
+        type: String,
+        required: true,
+    },
+    project: {
+        type: Object,
+        required: false,
+    },
+    currentBalance: {
+        type: Number,
+        required: true,
+    },
+    value: {
+        type: Number,
+        required: true,
+    },
+    type: {
+        type: String,
+        enum: Object.values(interfaces_1.TransactionType),
+        required: true,
+    },
+    createdAt: {
+        type: Date,
+        required: true,
+    },
+}, {
+    toJSON: {
+        transform(doc, ret) {
+            ret.id = ret._id;
+            delete ret._id;
+            delete ret.__v;
+        },
+    },
+});
+const autoPopulate = function (next) {
+    // @ts-ignore
+    this.populate('company');
+    next();
+};
+transactionSchema.pre('findOne', autoPopulate);
+transactionSchema.pre('find', autoPopulate);
+// Pass the ts interface.
+transactionSchema.statics.build = (attrs) => {
+    return new TransactionModel(attrs);
+};
+const TransactionModel = mongoose_1.default.model('Transaction', transactionSchema);
+exports.TransactionModel = TransactionModel;

package/build/models/user.model.d.ts

@@ -0,0 +1,19 @@
+import mongoose from 'mongoose';
+import { CompanyDoc } from './company.model';
+import { User } from '../interfaces';
+import { UserLevel } from '../interfaces';
+interface UserAttrs {
+    email: string;
+    company: CompanyDoc;
+    name: string;
+    level: String;
+    password: string;
+    createdAt: Date;
+    createdBy: string;
+}
+type UserDoc = User & mongoose.Document;
+interface UserModel extends mongoose.Model<UserDoc> {
+    build(attrs: UserAttrs): UserDoc;
+}
+declare const UserModel: UserModel;
+export { UserModel, UserDoc, UserLevel };

package/build/models/user.model.js

@@ -0,0 +1,119 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+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 __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UserLevel = exports.UserModel = void 0;
+const mongoose_1 = __importStar(require("mongoose"));
+const bcrypt_1 = __importDefault(require("bcrypt"));
+const interfaces_1 = require("../interfaces");
+Object.defineProperty(exports, "UserLevel", { enumerable: true, get: function () { return interfaces_1.UserLevel; } });
+const userSchema = new mongoose_1.default.Schema({
+    email: {
+        type: String,
+        required: true,
+    },
+    company: {
+        type: mongoose_1.Schema.Types.ObjectId,
+        ref: 'Company',
+        required: true,
+    },
+    name: {
+        type: String,
+        required: true,
+    },
+    level: {
+        type: String,
+        required: true,
+        enum: Object.values(interfaces_1.UserLevel),
+    },
+    blocked: {
+        type: Boolean,
+        required: false,
+        default: false,
+    },
+    password: {
+        type: String,
+        required: true,
+    },
+    createdAt: {
+        type: Date,
+        required: true,
+    },
+    createdBy: {
+        type: String,
+        required: true,
+    },
+}, {
+    toJSON: {
+        transform(_, ret) {
+            ret.id = ret._id;
+            delete ret._id;
+            delete ret.password;
+        },
+    },
+});
+const autoPopulate = function (next) {
+    //@ts-ignore
+    this.populate('company');
+    next();
+};
+userSchema.pre('findOne', autoPopulate).pre('find', autoPopulate);
+// Hash the user password before saving it to the db.
+userSchema.pre('save', function (done) {
+    return __awaiter(this, void 0, void 0, function* () {
+        if (this.isModified('password')) {
+            const hashedPass = bcrypt_1.default.hashSync(this.get('password'), 12);
+            this.set('password', hashedPass);
+        }
+        done();
+    });
+});
+// Create function to pass ts interface.
+userSchema.statics.build = (attrs) => {
+    return new UserModel(attrs);
+};
+const UserModel = mongoose_1.default.model('User', userSchema);
+exports.UserModel = UserModel;

package/build/utils/gc-storage.d.ts

@@ -0,0 +1,3 @@
+import { Bucket } from '@google-cloud/storage';
+declare const _default: Bucket;
+export default _default;

package/build/utils/gc-storage.js

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const storage_1 = require("@google-cloud/storage");
+class GoogleCloudStorage {
+    constructor() {
+        // Create a new Storage object using the environment variables for credentials and project ID
+        this.storage = new storage_1.Storage({
+            projectId: process.env.GCS_PROJECT,
+            credentials: {
+                client_email: process.env.GCS_EMAIL,
+                private_key: process.env.NODE_ENV === 'test'
+                    ? 'fake_key'
+                    : process.env.GCS_KEY
+                        .split(String.raw `\n`)
+                        .join('\n'),
+            },
+        });
+    }
+    static getInstance() {
+        // Create a new instance of GoogleCloudStorage if one does not already exist
+        if (!GoogleCloudStorage.instance) {
+            GoogleCloudStorage.instance = new GoogleCloudStorage();
+        }
+        return GoogleCloudStorage.instance;
+    }
+    getStorage() {
+        // Return the Storage object for the Google Cloud Storage connection
+        return this.storage.bucket(process.env.NODE_ENV === 'test'
+            ? 'fake_bucket'
+            : process.env.GCS_BUCKET);
+    }
+}
+exports.default = GoogleCloudStorage.getInstance().getStorage();

package/build/utils/gc.d.ts

@@ -0,0 +1,11 @@
+interface UploadFileToGCS {
+    filePath: string;
+    gcsFilePath: string;
+}
+interface UploadFileToGCSResponse {
+    filename: string;
+    size: number;
+}
+export declare const uploadFileToGCS: ({ filePath, gcsFilePath, }: UploadFileToGCS) => Promise<UploadFileToGCSResponse>;
+export declare const uploadFolderToGCS: (folderPath: string, gcsPath: string) => Promise<void>;
+export {};

package/build/utils/gc.js

@@ -0,0 +1,63 @@
+"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 __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.uploadFolderToGCS = exports.uploadFileToGCS = void 0;
+const gc_storage_1 = __importDefault(require("./gc-storage"));
+const path_1 = __importDefault(require("path"));
+const fs_1 = __importDefault(require("fs"));
+const p_limit_1 = __importDefault(require("p-limit"));
+const uploadFileToGCS = ({ filePath, gcsFilePath, }) => {
+    return new Promise((resolve, reject) => {
+        if (!fs_1.default.existsSync(filePath))
+            throw new Error('File not found');
+        const { size } = fs_1.default.statSync(filePath);
+        const blob = gc_storage_1.default.file(gcsFilePath);
+        const blobStream = blob.createWriteStream({
+            resumable: false,
+        });
+        const readStream = fs_1.default.createReadStream(filePath);
+        readStream.pipe(blobStream);
+        blobStream.on('error', (err) => {
+            reject(err);
+        });
+        blobStream.on('finish', () => {
+            resolve({
+                filename: path_1.default.basename(filePath),
+                size: size / 1024 / 1024,
+            });
+        });
+    });
+};
+exports.uploadFileToGCS = uploadFileToGCS;
+const uploadFolderToGCS = (folderPath, gcsPath) => __awaiter(void 0, void 0, void 0, function* () {
+    const limit = (0, p_limit_1.default)(4);
+    const files = fs_1.default.readdirSync(folderPath);
+    let promises = [];
+    for (const file of files) {
+        // If file is directory, we need to upload it recursively.
+        if (fs_1.default.lstatSync(path_1.default.join(folderPath, file)).isDirectory()) {
+            yield (0, exports.uploadFolderToGCS)(path_1.default.join(folderPath, file), path_1.default.join(gcsPath, file));
+        }
+        else {
+            const filePath = path_1.default.join(folderPath, file);
+            const gcsFilePath = path_1.default.join(gcsPath, file);
+            promises.push(limit(() => {
+                return (0, exports.uploadFileToGCS)({ filePath, gcsFilePath });
+            }));
+        }
+    }
+    const res = yield Promise.all(promises);
+    const totalSize = res.reduce((acc, curr) => acc + curr.size, 0);
+});
+exports.uploadFolderToGCS = uploadFolderToGCS;

package/build/utils/grades.d.ts

@@ -0,0 +1,3 @@
+export declare const gradeClasses: number[];
+export declare const gradeColors: string[];
+export declare const getGradeColor: (value: number) => string;

package/build/utils/grades.js

@@ -0,0 +1,25 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getGradeColor = exports.gradeColors = exports.gradeClasses = void 0;
+exports.gradeClasses = [0, 0.4, 0.6, 0.8];
+exports.gradeColors = [
+    "#f44336", // Red
+    "#ff9800", // Orange
+    "#ffc107", // Yellow
+    "#4caf50", // Green
+];
+const getGradeColor = (value) => {
+    if (value < exports.gradeClasses[0]) {
+        return exports.gradeColors[0]; // Red
+    }
+    else if (value < exports.gradeClasses[1]) {
+        return exports.gradeColors[1]; // Orange
+    }
+    else if (value < exports.gradeClasses[2]) {
+        return exports.gradeColors[2]; // Yellow
+    }
+    else {
+        return exports.gradeColors[3]; // Green
+    }
+};
+exports.getGradeColor = getGradeColor;

package/build/utils/index.d.ts

@@ -0,0 +1,3 @@
+export * from './grades';
+export * from './gc';
+export * from './gc-storage';

package/build/utils/index.js

@@ -0,0 +1,19 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./grades"), exports);
+__exportStar(require("./gc"), exports);
+__exportStar(require("./gc-storage"), exports);

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@rowengine/common",
-    "version": "1.1.2",
+    "version": "1.1.3",
     "description": "",
     "main": "./build/index.js",
     "private": false,
@@ -44,9 +44,9 @@
     ],
     "scripts": {
         "clean": "rm -rf ./build",
-        "build": "npm run clean & tsc",
-        "pub": "git add . & git commit -m \"updated common package\" & npm version patch & npm run build & npm publish",
-        "pub-linux": "npm run build ; git add . ; git commit -m \"updated common package\" ; npm version patch ; npm publish"
+        "build": "npm run clean && tsc",
+        "pub": "npm run build && git add . && git commit -m \"updated common package\" && npm version patch && npm publish",
+        "pub-linux": "npm run build && git add . && git commit -m \"updated common package\" && npm version patch && npm publish"
     },
     "author": "Leonardo Ribeiro",
     "license": "ISC",