125 lines
3.6 KiB
JavaScript
125 lines
3.6 KiB
JavaScript
const mysql = require('mysql2/promise');
|
|
const path = require('path');
|
|
const dotenv = require('dotenv');
|
|
const { spawn } = require('child_process');
|
|
|
|
dotenv.config({ path: path.join(__dirname, '../.env') });
|
|
|
|
const dbConfig = {
|
|
host: process.env.DB_HOST,
|
|
user: process.env.DB_USER,
|
|
password: process.env.DB_PASSWORD,
|
|
database: process.env.DB_NAME,
|
|
multipleStatements: true
|
|
};
|
|
|
|
// Helper function to output progress in JSON format
|
|
function outputProgress(data) {
|
|
if (!data.status) {
|
|
data = {
|
|
status: 'running',
|
|
...data
|
|
};
|
|
}
|
|
console.log(JSON.stringify(data));
|
|
}
|
|
|
|
async function resetDatabase() {
|
|
outputProgress({
|
|
operation: 'Starting database reset',
|
|
message: 'Connecting to database...'
|
|
});
|
|
|
|
const connection = await mysql.createConnection(dbConfig);
|
|
|
|
try {
|
|
// Get list of all tables
|
|
outputProgress({
|
|
operation: 'Getting table list',
|
|
message: 'Retrieving all table names...'
|
|
});
|
|
|
|
const [tables] = await connection.query(
|
|
'SELECT table_name FROM information_schema.tables WHERE table_schema = ?',
|
|
[dbConfig.database]
|
|
);
|
|
|
|
if (tables.length === 0) {
|
|
outputProgress({
|
|
operation: 'No tables found',
|
|
message: 'Database is already empty'
|
|
});
|
|
} else {
|
|
// Disable foreign key checks to allow dropping tables with dependencies
|
|
await connection.query('SET FOREIGN_KEY_CHECKS = 0');
|
|
|
|
// Drop each table
|
|
for (let i = 0; i < tables.length; i++) {
|
|
const tableName = tables[i].TABLE_NAME;
|
|
outputProgress({
|
|
operation: 'Dropping tables',
|
|
message: `Dropping table: ${tableName}`,
|
|
current: i + 1,
|
|
total: tables.length,
|
|
percentage: (((i + 1) / tables.length) * 100).toFixed(1)
|
|
});
|
|
await connection.query(`DROP TABLE IF EXISTS \`${tableName}\``);
|
|
}
|
|
|
|
// Re-enable foreign key checks
|
|
await connection.query('SET FOREIGN_KEY_CHECKS = 1');
|
|
}
|
|
|
|
// Run setup-db.js
|
|
outputProgress({
|
|
operation: 'Running database setup',
|
|
message: 'Creating new tables...'
|
|
});
|
|
|
|
const setupScript = path.join(__dirname, 'setup-db.js');
|
|
const setupProcess = spawn('node', [setupScript]);
|
|
|
|
setupProcess.stdout.on('data', (data) => {
|
|
const output = data.toString().trim();
|
|
outputProgress({
|
|
operation: 'Database setup',
|
|
message: output
|
|
});
|
|
});
|
|
|
|
setupProcess.stderr.on('data', (data) => {
|
|
const error = data.toString().trim();
|
|
outputProgress({
|
|
status: 'error',
|
|
error
|
|
});
|
|
});
|
|
|
|
await new Promise((resolve, reject) => {
|
|
setupProcess.on('close', (code) => {
|
|
if (code === 0) {
|
|
resolve();
|
|
} else {
|
|
reject(new Error(`Setup process exited with code ${code}`));
|
|
}
|
|
});
|
|
});
|
|
|
|
outputProgress({
|
|
status: 'complete',
|
|
operation: 'Database reset complete',
|
|
message: 'Database has been reset and tables recreated'
|
|
});
|
|
} catch (error) {
|
|
outputProgress({
|
|
status: 'error',
|
|
error: error.message
|
|
});
|
|
process.exit(1);
|
|
} finally {
|
|
await connection.end();
|
|
}
|
|
}
|
|
|
|
// Run the reset
|
|
resetDatabase();
|