2656df94d3
- Add bu_teams column to users table (migration + fresh schema) - Create shared KNOWN_TEAMS constant and validateTeams helper - Expose user teams in auth middleware, login, and /me responses - Add bu_teams CRUD to user management routes with audit logging - Make Ivanti FINDINGS_FILTERS configurable via IVANTI_BU_FILTER env var - Add query-time team filtering to GET /findings and /findings/counts - Update AuthContext with teams helpers and admin scope toggle - Create AdminScopeToggle component (My Teams / All BUs) - Scope ReportingPage findings fetch by user teams - Scope CompliancePage team selector by user teams - Scope ExportsPage findings exports by user teams - Add BU teams multi-select to UserManagement create/edit forms - Display team badges in user list table
69 lines
2.2 KiB
JavaScript
69 lines
2.2 KiB
JavaScript
// Migration: Add bu_teams column to users table
|
|
// Stores comma-separated BU team identifiers per user (e.g. 'STEAM,ACCESS-ENG')
|
|
// Existing users get empty string (admin must assign teams post-migration)
|
|
// Idempotent — safe to run multiple times
|
|
const sqlite3 = require('sqlite3').verbose();
|
|
const path = require('path');
|
|
|
|
const DB_FILE = path.join(__dirname, '..', 'cve_database.db');
|
|
|
|
/**
|
|
* Run the migration against the given database instance.
|
|
* Exported for testing with in-memory databases.
|
|
* @param {sqlite3.Database} db
|
|
* @returns {Promise<void>}
|
|
*/
|
|
function runMigration(db) {
|
|
return new Promise((resolve, reject) => {
|
|
db.serialize(() => {
|
|
// Check if bu_teams column already exists
|
|
db.all("PRAGMA table_info(users)", (err, columns) => {
|
|
if (err) {
|
|
reject(err);
|
|
return;
|
|
}
|
|
|
|
const hasBuTeams = columns.some(col => col.name === 'bu_teams');
|
|
|
|
if (hasBuTeams) {
|
|
console.log('✓ bu_teams column already exists — skipping migration');
|
|
resolve();
|
|
return;
|
|
}
|
|
|
|
console.log('Adding bu_teams column to users table...');
|
|
|
|
db.run(
|
|
`ALTER TABLE users ADD COLUMN bu_teams TEXT NOT NULL DEFAULT ''`,
|
|
(err) => {
|
|
if (err) {
|
|
reject(err);
|
|
return;
|
|
}
|
|
console.log('✓ Added bu_teams column (default: empty string)');
|
|
console.log(' Note: Admin must assign teams to existing users via user management UI');
|
|
resolve();
|
|
}
|
|
);
|
|
});
|
|
});
|
|
});
|
|
}
|
|
|
|
// Run directly if executed as a script
|
|
if (require.main === module) {
|
|
const db = new sqlite3.Database(DB_FILE);
|
|
runMigration(db)
|
|
.then(() => {
|
|
console.log('Migration complete.');
|
|
db.close();
|
|
})
|
|
.catch((err) => {
|
|
console.error('Migration failed:', err.message);
|
|
db.close();
|
|
process.exit(1);
|
|
});
|
|
}
|
|
|
|
module.exports = { runMigration };
|