93 lines
2.4 KiB
JavaScript
93 lines
2.4 KiB
JavaScript
/**
|
|
* Script to update weapon skills in the database by removing "SC Icon.png" strings
|
|
*/
|
|
|
|
const { Pool } = require('pg');
|
|
const fs = require('fs');
|
|
|
|
// Read database configuration
|
|
const dbConfFile = fs.readFileSync('db.conf', 'utf8');
|
|
const dbConfig = {};
|
|
|
|
// Parse the db.conf file
|
|
dbConfFile.split('\n').forEach(line => {
|
|
if (line.trim() === '') return;
|
|
|
|
const [key, value] = line.split('=');
|
|
if (key && value) {
|
|
// Remove quotes if present
|
|
const cleanValue = value.replace(/^['"]|['"]$/g, '');
|
|
dbConfig[key] = cleanValue;
|
|
}
|
|
});
|
|
|
|
// Configure PostgreSQL connection
|
|
const pool = new Pool({
|
|
user: dbConfig.PSQL_USER,
|
|
host: dbConfig.PSQL_HOST,
|
|
database: dbConfig.PSQL_DBNAME,
|
|
password: dbConfig.PSQL_PASSWORD,
|
|
port: dbConfig.PSQL_PORT,
|
|
});
|
|
|
|
// Function to update weapon skills
|
|
async function updateWeaponSkills() {
|
|
const client = await pool.connect();
|
|
|
|
try {
|
|
await client.query('BEGIN');
|
|
|
|
// Get all trusts with weapon skills
|
|
const result = await client.query('SELECT id, weapon_skills FROM trusts WHERE weapon_skills IS NOT NULL AND weapon_skills != \'\'');
|
|
|
|
let updatedCount = 0;
|
|
|
|
// Update each trust's weapon skills
|
|
for (const row of result.rows) {
|
|
const originalWeaponSkills = row.weapon_skills;
|
|
|
|
// Remove "SC Icon.png" from weapon skills
|
|
const updatedWeaponSkills = originalWeaponSkills.replace(/SC Icon\.png/g, '');
|
|
|
|
// Only update if there was a change
|
|
if (updatedWeaponSkills !== originalWeaponSkills) {
|
|
await client.query(
|
|
'UPDATE trusts SET weapon_skills = $1 WHERE id = $2',
|
|
[updatedWeaponSkills, row.id]
|
|
);
|
|
|
|
updatedCount++;
|
|
console.log(`Updated weapon skills for trust ID ${row.id}`);
|
|
}
|
|
}
|
|
|
|
await client.query('COMMIT');
|
|
console.log(`Updated weapon skills for ${updatedCount} trusts`);
|
|
} catch (e) {
|
|
await client.query('ROLLBACK');
|
|
console.error('Error updating weapon skills:', e);
|
|
throw e;
|
|
} finally {
|
|
client.release();
|
|
}
|
|
}
|
|
|
|
// Main function
|
|
async function main() {
|
|
try {
|
|
// Update weapon skills
|
|
await updateWeaponSkills();
|
|
|
|
// Close the pool
|
|
await pool.end();
|
|
|
|
console.log('Database update completed successfully');
|
|
} catch (e) {
|
|
console.error('Error updating database:', e);
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
// Run the main function
|
|
main();
|