59 lines
2.0 KiB
JavaScript
59 lines
2.0 KiB
JavaScript
|
// Migration: Add ivanti_findings_cache and ivanti_finding_notes tables
|
||
|
|
const sqlite3 = require('sqlite3').verbose();
|
||
|
|
const path = require('path');
|
||
|
|
|
||
|
|
const dbPath = path.join(__dirname, '..', 'cve_database.db');
|
||
|
|
const db = new sqlite3.Database(dbPath);
|
||
|
|
|
||
|
|
console.log('Starting Ivanti findings tables migration...');
|
||
|
|
|
||
|
|
db.serialize(() => {
|
||
|
|
// Cache table — single row holding the latest sync result
|
||
|
|
db.run(`
|
||
|
|
CREATE TABLE IF NOT EXISTS ivanti_findings_cache (
|
||
|
|
id INTEGER PRIMARY KEY CHECK (id = 1),
|
||
|
|
total INTEGER DEFAULT 0,
|
||
|
|
findings_json TEXT DEFAULT '[]',
|
||
|
|
synced_at DATETIME,
|
||
|
|
sync_status TEXT DEFAULT 'never',
|
||
|
|
error_message TEXT
|
||
|
|
)
|
||
|
|
`, (err) => {
|
||
|
|
if (err) console.error('Error creating findings cache table:', err);
|
||
|
|
else console.log('✓ ivanti_findings_cache table created');
|
||
|
|
});
|
||
|
|
|
||
|
|
db.run(`
|
||
|
|
INSERT OR IGNORE INTO ivanti_findings_cache (id, total, findings_json, sync_status)
|
||
|
|
VALUES (1, 0, '[]', 'never')
|
||
|
|
`, (err) => {
|
||
|
|
if (err) console.error('Error seeding findings cache row:', err);
|
||
|
|
else console.log('✓ ivanti_findings_cache row seeded');
|
||
|
|
});
|
||
|
|
|
||
|
|
// Notes table — one row per finding, persists across cache refreshes
|
||
|
|
db.run(`
|
||
|
|
CREATE TABLE IF NOT EXISTS ivanti_finding_notes (
|
||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
|
|
finding_id TEXT NOT NULL UNIQUE,
|
||
|
|
note TEXT NOT NULL DEFAULT '',
|
||
|
|
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||
|
|
)
|
||
|
|
`, (err) => {
|
||
|
|
if (err) console.error('Error creating finding notes table:', err);
|
||
|
|
else console.log('✓ ivanti_finding_notes table created');
|
||
|
|
});
|
||
|
|
|
||
|
|
db.run(`
|
||
|
|
CREATE INDEX IF NOT EXISTS idx_finding_notes_finding_id
|
||
|
|
ON ivanti_finding_notes(finding_id)
|
||
|
|
`, (err) => {
|
||
|
|
if (err) console.error('Error creating notes index:', err);
|
||
|
|
else console.log('✓ finding_id index created');
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
db.close(() => {
|
||
|
|
console.log('Migration complete!');
|
||
|
|
});
|