69 lines
2.5 KiB
JavaScript
69 lines
2.5 KiB
JavaScript
/**
|
|
* Migration: Add IP restriction fields to tenants
|
|
* Adds ip_whitelist, ip_restriction_enabled, and ip_restriction_message fields
|
|
*/
|
|
|
|
'use strict';
|
|
|
|
module.exports = {
|
|
up: async (queryInterface, Sequelize) => {
|
|
try {
|
|
// Check if tenants table exists first
|
|
const tables = await queryInterface.showAllTables();
|
|
if (!tables.includes('tenants')) {
|
|
console.log('⚠️ Tenants table does not exist yet, skipping IP restrictions migration...');
|
|
return;
|
|
}
|
|
|
|
// Check if the columns already exist
|
|
const tableDescription = await queryInterface.describeTable('tenants');
|
|
|
|
if (!tableDescription.ip_whitelist) {
|
|
await queryInterface.addColumn('tenants', 'ip_whitelist', {
|
|
type: Sequelize.JSONB,
|
|
allowNull: true,
|
|
defaultValue: null,
|
|
comment: 'Array of allowed IP addresses/CIDR blocks for this tenant'
|
|
});
|
|
console.log('✅ Added ip_whitelist column to tenants table');
|
|
} else {
|
|
console.log('⚠️ Column ip_whitelist already exists, skipping...');
|
|
}
|
|
|
|
if (!tableDescription.ip_restriction_enabled) {
|
|
await queryInterface.addColumn('tenants', 'ip_restriction_enabled', {
|
|
type: Sequelize.BOOLEAN,
|
|
defaultValue: false,
|
|
allowNull: false,
|
|
comment: 'Whether IP restrictions are enabled for this tenant'
|
|
});
|
|
console.log('✅ Added ip_restriction_enabled column to tenants table');
|
|
} else {
|
|
console.log('⚠️ Column ip_restriction_enabled already exists, skipping...');
|
|
}
|
|
|
|
if (!tableDescription.ip_restriction_message) {
|
|
await queryInterface.addColumn('tenants', 'ip_restriction_message', {
|
|
type: Sequelize.TEXT,
|
|
allowNull: true,
|
|
defaultValue: 'Access denied. Your IP address is not authorized to access this tenant.',
|
|
comment: 'Custom message shown when IP access is denied'
|
|
});
|
|
console.log('✅ Added ip_restriction_message column to tenants table');
|
|
} else {
|
|
console.log('⚠️ Column ip_restriction_message already exists, skipping...');
|
|
}
|
|
} catch (error) {
|
|
console.log('⚠️ Migration skipped - tables may not exist yet:', error.message);
|
|
// Don't throw error, just skip this migration if tables don't exist
|
|
}
|
|
},
|
|
|
|
down: async (queryInterface, Sequelize) => {
|
|
// Remove the added columns
|
|
await queryInterface.removeColumn('tenants', 'ip_whitelist');
|
|
await queryInterface.removeColumn('tenants', 'ip_restriction_enabled');
|
|
await queryInterface.removeColumn('tenants', 'ip_restriction_message');
|
|
}
|
|
};
|