648 lines
16 KiB
JavaScript
648 lines
16 KiB
JavaScript
module.exports = {
|
|
/**
|
|
* @param {QueryInterface} queryInterface
|
|
* @param {Sequelize} Sequelize
|
|
* @returns {Promise<void>}
|
|
*/
|
|
async up(queryInterface, Sequelize) {
|
|
/**
|
|
* @type {Transaction}
|
|
*/
|
|
const transaction = await queryInterface.sequelize.transaction();
|
|
try {
|
|
await queryInterface.createTable(
|
|
'users',
|
|
{
|
|
id: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
defaultValue: Sequelize.DataTypes.UUIDV4,
|
|
primaryKey: true,
|
|
},
|
|
createdById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
updatedById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
createdAt: { type: Sequelize.DataTypes.DATE },
|
|
updatedAt: { type: Sequelize.DataTypes.DATE },
|
|
deletedAt: { type: Sequelize.DataTypes.DATE },
|
|
importHash: {
|
|
type: Sequelize.DataTypes.STRING(255),
|
|
allowNull: true,
|
|
unique: true,
|
|
},
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.createTable(
|
|
'reports',
|
|
{
|
|
id: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
defaultValue: Sequelize.DataTypes.UUIDV4,
|
|
primaryKey: true,
|
|
},
|
|
createdById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
updatedById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
createdAt: { type: Sequelize.DataTypes.DATE },
|
|
updatedAt: { type: Sequelize.DataTypes.DATE },
|
|
deletedAt: { type: Sequelize.DataTypes.DATE },
|
|
importHash: {
|
|
type: Sequelize.DataTypes.STRING(255),
|
|
allowNull: true,
|
|
unique: true,
|
|
},
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.createTable(
|
|
'targets',
|
|
{
|
|
id: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
defaultValue: Sequelize.DataTypes.UUIDV4,
|
|
primaryKey: true,
|
|
},
|
|
createdById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
updatedById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
createdAt: { type: Sequelize.DataTypes.DATE },
|
|
updatedAt: { type: Sequelize.DataTypes.DATE },
|
|
deletedAt: { type: Sequelize.DataTypes.DATE },
|
|
importHash: {
|
|
type: Sequelize.DataTypes.STRING(255),
|
|
allowNull: true,
|
|
unique: true,
|
|
},
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.createTable(
|
|
'tools',
|
|
{
|
|
id: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
defaultValue: Sequelize.DataTypes.UUIDV4,
|
|
primaryKey: true,
|
|
},
|
|
createdById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
updatedById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
createdAt: { type: Sequelize.DataTypes.DATE },
|
|
updatedAt: { type: Sequelize.DataTypes.DATE },
|
|
deletedAt: { type: Sequelize.DataTypes.DATE },
|
|
importHash: {
|
|
type: Sequelize.DataTypes.STRING(255),
|
|
allowNull: true,
|
|
unique: true,
|
|
},
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.createTable(
|
|
'vulnerabilities',
|
|
{
|
|
id: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
defaultValue: Sequelize.DataTypes.UUIDV4,
|
|
primaryKey: true,
|
|
},
|
|
createdById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
updatedById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
createdAt: { type: Sequelize.DataTypes.DATE },
|
|
updatedAt: { type: Sequelize.DataTypes.DATE },
|
|
deletedAt: { type: Sequelize.DataTypes.DATE },
|
|
importHash: {
|
|
type: Sequelize.DataTypes.STRING(255),
|
|
allowNull: true,
|
|
unique: true,
|
|
},
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.createTable(
|
|
'roles',
|
|
{
|
|
id: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
defaultValue: Sequelize.DataTypes.UUIDV4,
|
|
primaryKey: true,
|
|
},
|
|
createdById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
updatedById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
createdAt: { type: Sequelize.DataTypes.DATE },
|
|
updatedAt: { type: Sequelize.DataTypes.DATE },
|
|
deletedAt: { type: Sequelize.DataTypes.DATE },
|
|
importHash: {
|
|
type: Sequelize.DataTypes.STRING(255),
|
|
allowNull: true,
|
|
unique: true,
|
|
},
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.createTable(
|
|
'permissions',
|
|
{
|
|
id: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
defaultValue: Sequelize.DataTypes.UUIDV4,
|
|
primaryKey: true,
|
|
},
|
|
createdById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
updatedById: {
|
|
type: Sequelize.DataTypes.UUID,
|
|
references: {
|
|
key: 'id',
|
|
model: 'users',
|
|
},
|
|
},
|
|
createdAt: { type: Sequelize.DataTypes.DATE },
|
|
updatedAt: { type: Sequelize.DataTypes.DATE },
|
|
deletedAt: { type: Sequelize.DataTypes.DATE },
|
|
importHash: {
|
|
type: Sequelize.DataTypes.STRING(255),
|
|
allowNull: true,
|
|
unique: true,
|
|
},
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'users',
|
|
'firstName',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'users',
|
|
'lastName',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'users',
|
|
'phoneNumber',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'users',
|
|
'email',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'users',
|
|
'disabled',
|
|
{
|
|
type: Sequelize.DataTypes.BOOLEAN,
|
|
|
|
defaultValue: false,
|
|
allowNull: false,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'users',
|
|
'password',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'users',
|
|
'emailVerified',
|
|
{
|
|
type: Sequelize.DataTypes.BOOLEAN,
|
|
|
|
defaultValue: false,
|
|
allowNull: false,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'users',
|
|
'emailVerificationToken',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'users',
|
|
'emailVerificationTokenExpiresAt',
|
|
{
|
|
type: Sequelize.DataTypes.DATE,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'users',
|
|
'passwordResetToken',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'users',
|
|
'passwordResetTokenExpiresAt',
|
|
{
|
|
type: Sequelize.DataTypes.DATE,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'users',
|
|
'provider',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'reports',
|
|
'title',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'reports',
|
|
'generated_at',
|
|
{
|
|
type: Sequelize.DataTypes.DATE,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'reports',
|
|
'generated_byId',
|
|
{
|
|
type: Sequelize.DataTypes.UUID,
|
|
|
|
references: {
|
|
model: 'users',
|
|
key: 'id',
|
|
},
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'targets',
|
|
'ip_address',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'targets',
|
|
'domain',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'tools',
|
|
'name',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'tools',
|
|
'description',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'tools',
|
|
'category',
|
|
{
|
|
type: Sequelize.DataTypes.ENUM,
|
|
|
|
values: [
|
|
'recon',
|
|
'vulnerability_assessment',
|
|
'exploitation',
|
|
'forensic_analysis',
|
|
'network_monitoring',
|
|
'reporting',
|
|
],
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'vulnerabilities',
|
|
'name',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'vulnerabilities',
|
|
'cvss_score',
|
|
{
|
|
type: Sequelize.DataTypes.DECIMAL,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'vulnerabilities',
|
|
'reported_byId',
|
|
{
|
|
type: Sequelize.DataTypes.UUID,
|
|
|
|
references: {
|
|
model: 'users',
|
|
key: 'id',
|
|
},
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'permissions',
|
|
'name',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'roles',
|
|
'name',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'roles',
|
|
'role_customization',
|
|
{
|
|
type: Sequelize.DataTypes.TEXT,
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.addColumn(
|
|
'users',
|
|
'app_roleId',
|
|
{
|
|
type: Sequelize.DataTypes.UUID,
|
|
|
|
references: {
|
|
model: 'roles',
|
|
key: 'id',
|
|
},
|
|
},
|
|
{ transaction },
|
|
);
|
|
|
|
await transaction.commit();
|
|
} catch (err) {
|
|
await transaction.rollback();
|
|
throw err;
|
|
}
|
|
},
|
|
/**
|
|
* @param {QueryInterface} queryInterface
|
|
* @param {Sequelize} Sequelize
|
|
* @returns {Promise<void>}
|
|
*/
|
|
async down(queryInterface, Sequelize) {
|
|
/**
|
|
* @type {Transaction}
|
|
*/
|
|
const transaction = await queryInterface.sequelize.transaction();
|
|
try {
|
|
await queryInterface.removeColumn('users', 'app_roleId', { transaction });
|
|
|
|
await queryInterface.removeColumn('roles', 'role_customization', {
|
|
transaction,
|
|
});
|
|
|
|
await queryInterface.removeColumn('roles', 'name', { transaction });
|
|
|
|
await queryInterface.removeColumn('permissions', 'name', { transaction });
|
|
|
|
await queryInterface.removeColumn('vulnerabilities', 'reported_byId', {
|
|
transaction,
|
|
});
|
|
|
|
await queryInterface.removeColumn('vulnerabilities', 'cvss_score', {
|
|
transaction,
|
|
});
|
|
|
|
await queryInterface.removeColumn('vulnerabilities', 'name', {
|
|
transaction,
|
|
});
|
|
|
|
await queryInterface.removeColumn('tools', 'category', { transaction });
|
|
|
|
await queryInterface.removeColumn('tools', 'description', {
|
|
transaction,
|
|
});
|
|
|
|
await queryInterface.removeColumn('tools', 'name', { transaction });
|
|
|
|
await queryInterface.removeColumn('targets', 'domain', { transaction });
|
|
|
|
await queryInterface.removeColumn('targets', 'ip_address', {
|
|
transaction,
|
|
});
|
|
|
|
await queryInterface.removeColumn('reports', 'generated_byId', {
|
|
transaction,
|
|
});
|
|
|
|
await queryInterface.removeColumn('reports', 'generated_at', {
|
|
transaction,
|
|
});
|
|
|
|
await queryInterface.removeColumn('reports', 'title', { transaction });
|
|
|
|
await queryInterface.removeColumn('users', 'provider', { transaction });
|
|
|
|
await queryInterface.removeColumn(
|
|
'users',
|
|
'passwordResetTokenExpiresAt',
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.removeColumn('users', 'passwordResetToken', {
|
|
transaction,
|
|
});
|
|
|
|
await queryInterface.removeColumn(
|
|
'users',
|
|
'emailVerificationTokenExpiresAt',
|
|
{ transaction },
|
|
);
|
|
|
|
await queryInterface.removeColumn('users', 'emailVerificationToken', {
|
|
transaction,
|
|
});
|
|
|
|
await queryInterface.removeColumn('users', 'emailVerified', {
|
|
transaction,
|
|
});
|
|
|
|
await queryInterface.removeColumn('users', 'password', { transaction });
|
|
|
|
await queryInterface.removeColumn('users', 'disabled', { transaction });
|
|
|
|
await queryInterface.removeColumn('users', 'email', { transaction });
|
|
|
|
await queryInterface.removeColumn('users', 'phoneNumber', {
|
|
transaction,
|
|
});
|
|
|
|
await queryInterface.removeColumn('users', 'lastName', { transaction });
|
|
|
|
await queryInterface.removeColumn('users', 'firstName', { transaction });
|
|
|
|
await queryInterface.dropTable('permissions', { transaction });
|
|
|
|
await queryInterface.dropTable('roles', { transaction });
|
|
|
|
await queryInterface.dropTable('vulnerabilities', { transaction });
|
|
|
|
await queryInterface.dropTable('tools', { transaction });
|
|
|
|
await queryInterface.dropTable('targets', { transaction });
|
|
|
|
await queryInterface.dropTable('reports', { transaction });
|
|
|
|
await queryInterface.dropTable('users', { transaction });
|
|
|
|
await transaction.commit();
|
|
} catch (err) {
|
|
await transaction.rollback();
|
|
throw err;
|
|
}
|
|
},
|
|
};
|