75 lines
2.6 KiB
JavaScript
75 lines
2.6 KiB
JavaScript
import db from './src/lib/db.js';
|
|
|
|
console.log('Starting migration to add team_lead role to users table constraint...');
|
|
|
|
try {
|
|
// Disable foreign key constraints temporarily
|
|
db.pragma('foreign_keys = OFF');
|
|
console.log('Disabled foreign key constraints');
|
|
|
|
// Since SQLite doesn't support modifying CHECK constraints directly,
|
|
// we need to recreate the table with the new constraint
|
|
|
|
// First, create a backup table with current data
|
|
db.exec('CREATE TABLE users_backup AS SELECT * FROM users');
|
|
console.log('Created backup table');
|
|
|
|
// Drop the original table
|
|
db.exec('DROP TABLE users');
|
|
console.log('Dropped original table');
|
|
|
|
// Recreate the table with the updated constraint
|
|
db.exec(`
|
|
CREATE TABLE users (
|
|
id TEXT PRIMARY KEY DEFAULT (lower(hex(randomblob(16)))),
|
|
name TEXT NOT NULL,
|
|
username TEXT UNIQUE NOT NULL,
|
|
password_hash TEXT NOT NULL,
|
|
role TEXT CHECK(role IN ('admin', 'team_lead', 'project_manager', 'user', 'read_only')) DEFAULT 'user',
|
|
created_at TEXT DEFAULT CURRENT_TIMESTAMP,
|
|
updated_at TEXT DEFAULT CURRENT_TIMESTAMP,
|
|
is_active INTEGER DEFAULT 1,
|
|
last_login TEXT,
|
|
failed_login_attempts INTEGER DEFAULT 0,
|
|
locked_until TEXT,
|
|
can_be_assigned INTEGER DEFAULT 1,
|
|
initial TEXT
|
|
)
|
|
`);
|
|
console.log('Created new table with updated constraint');
|
|
|
|
// Copy data back from backup
|
|
db.exec(`
|
|
INSERT INTO users (
|
|
id, name, username, password_hash, role, created_at, updated_at,
|
|
is_active, last_login, failed_login_attempts, locked_until,
|
|
can_be_assigned, initial
|
|
)
|
|
SELECT
|
|
id, name, username, password_hash, role, created_at, updated_at,
|
|
is_active, last_login, failed_login_attempts, locked_until,
|
|
can_be_assigned, initial
|
|
FROM users_backup
|
|
`);
|
|
console.log('Copied data back from backup');
|
|
|
|
// Drop the backup table
|
|
db.exec('DROP TABLE users_backup');
|
|
console.log('Dropped backup table');
|
|
|
|
// Re-enable foreign key constraints
|
|
db.pragma('foreign_keys = ON');
|
|
console.log('Re-enabled foreign key constraints');
|
|
|
|
// Verify the migration
|
|
const userCount = db.prepare('SELECT COUNT(*) as count FROM users').get();
|
|
console.log(`✅ Migration completed successfully! Users table now has ${userCount.count} records`);
|
|
|
|
// Verify the constraint allows the new role
|
|
console.log('✅ CHECK constraint now includes: admin, team_lead, project_manager, user, read_only');
|
|
|
|
} catch (error) {
|
|
console.error('❌ Migration failed:', error.message);
|
|
console.error('You may need to restore from backup manually');
|
|
process.exit(1);
|
|
} |