nodejs-schema-rules
Version:
The CLI tool automatically generates basic validation rules for popular libraries such as JOI, ValidatorJS and @vinejs/vine based on your database table schema!
115 lines (114 loc) • 5.7 kB
JavaScript
;
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());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.SchemaOperationForPostgres = void 0;
const messages_1 = require("../utils/messages");
const PostgresDatabase_1 = require("../databases/PostgresDatabase");
class SchemaOperationForPostgres {
constructor(table, databaseConfig, selectedColumns, skipColumns) {
this.integerTypes = {
smallint: { min: '-32768', max: '32767' },
integer: { min: '-2147483648', max: '2147483647' },
bigint: { min: '-9223372036854775808', max: '9223372036854775807' },
};
this.table = table;
this.databaseConfig = databaseConfig;
this.database = new PostgresDatabase_1.PostgresDatabase(this.databaseConfig);
this.selectedColumns = selectedColumns;
this.skipColumns = skipColumns;
}
getTableSchema() {
var _a, _b, _c;
return __awaiter(this, void 0, void 0, function* () {
yield this.database.connect();
let schema;
try {
const tableExist = yield this.database.query(`SELECT COUNT(table_name) as total_table FROM information_schema.tables WHERE table_name = '${this.table}';`);
if (((_a = tableExist.rows[0]) === null || _a === void 0 ? void 0 : _a.total_table) == undefined || ((_b = tableExist.rows[0]) === null || _b === void 0 ? void 0 : _b.total_table) == '0') {
throw new Error((0, messages_1.errorMessage)(`The ${this.table} table is not exist!`));
}
schema = yield this.database.query(`
SELECT table_name,column_name, data_type, character_maximum_length, is_nullable, column_default
FROM
information_schema.columns
WHERE
table_name = '${this.table}'
ORDER BY ordinal_position ASC;
`);
}
catch (error) {
console.error(error.message);
}
finally {
// Close the database connection
yield this.database.end();
}
return (_c = schema === null || schema === void 0 ? void 0 : schema.rows) !== null && _c !== void 0 ? _c : [];
});
}
generateColumnRules() {
return __awaiter(this, void 0, void 0, function* () {
const rules = {};
let tableSchema = yield this.getTableSchema();
if (this.skipColumns.length || this.selectedColumns.length) {
tableSchema = tableSchema.filter(({ column_name }) => {
return this.selectedColumns.length
? this.selectedColumns.includes(column_name)
: !this.skipColumns.includes(column_name);
});
}
tableSchema.forEach(({ table_name, column_name, data_type, character_maximum_length, is_nullable, column_default }) => {
var _a;
if (column_default && column_default.includes('nextval')) {
return;
}
let columnRules = [];
let type = data_type;
switch (true) {
case type === 'boolean':
columnRules.push('boolean');
break;
case type.includes('char'):
columnRules.push('string');
columnRules.push((_a = 'max:' + character_maximum_length) !== null && _a !== void 0 ? _a : '255');
break;
case type === 'text':
columnRules.push('string');
break;
case type.includes('int'):
columnRules.push('integer');
columnRules.push('min:' + this.integerTypes.integer.min.toString());
columnRules.push('max:' + this.integerTypes.integer.max.toString());
break;
case type.includes('double') ||
type.includes('decimal') ||
type.includes('numeric') ||
type.includes('real'):
columnRules.push('numeric');
break;
case type.includes('date') || type.includes('time'):
columnRules.push('date');
break;
case type.includes('json'):
columnRules.push('json');
break;
default:
// Skip for other type
break;
}
columnRules.push(is_nullable === 'YES' ? 'nullable' : 'required');
rules[column_name] = columnRules;
});
return rules;
});
}
}
exports.SchemaOperationForPostgres = SchemaOperationForPostgres;