73 lines
2.8 KiB
JavaScript
73 lines
2.8 KiB
JavaScript
|
|
import db from "../src/inc/sql.mjs";
|
|
import { promises as fs } from "fs";
|
|
|
|
(async () => {
|
|
console.log("Starting migration...");
|
|
|
|
// 1. Ensure column exists
|
|
try {
|
|
await db`select is_deleted from items limit 1`;
|
|
console.log("Column 'is_deleted' already exists.");
|
|
} catch (err) {
|
|
if (err.message.includes('column "is_deleted" does not exist')) {
|
|
console.log("Column 'is_deleted' missing. Adding it now...");
|
|
await db`ALTER TABLE items ADD COLUMN is_deleted BOOLEAN DEFAULT FALSE`;
|
|
console.log("Column added successfully.");
|
|
} else {
|
|
console.error("Unexpected error checking column:", err);
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
const items = await db`select id, dest from items where active = false`;
|
|
console.log(`Found ${items.length} inactive items.`);
|
|
|
|
let trashCount = 0;
|
|
let pendingCount = 0;
|
|
let brokenCount = 0;
|
|
|
|
for (const item of items) {
|
|
try {
|
|
await fs.access(`./deleted/b/${item.dest}`);
|
|
// File exists in deleted, mark as is_deleted = true
|
|
await db`update items set is_deleted = true where id = ${item.id}`;
|
|
trashCount++;
|
|
} catch {
|
|
// Not in deleted, check public
|
|
try {
|
|
await fs.access(`./public/b/${item.dest}`);
|
|
// In public, is_deleted = false (default)
|
|
pendingCount++;
|
|
} catch {
|
|
// Not in either? Broken.
|
|
console.log(`Item ${item.id} (${item.dest}) missing from both locations. Cleaning up...`);
|
|
|
|
// 2. Fix FK constraint: Check if this item is used as an avatar
|
|
try {
|
|
// Find a safe fallback avatar (active item)
|
|
const fallback = await db`select id from items where active = true limit 1`;
|
|
if (fallback.length > 0) {
|
|
const safeId = fallback[0].id;
|
|
const users = await db`update "user_options" set avatar = ${safeId} where avatar = ${item.id} returning user_id`;
|
|
if (users.length > 0) {
|
|
console.log(` > Reassigned avatar for ${users.length} users (from ${item.id} to ${safeId})`);
|
|
}
|
|
}
|
|
} catch (fkErr) {
|
|
console.error(` ! Error fixing avatar FK for ${item.id}:`, fkErr.message);
|
|
}
|
|
|
|
await db`delete from items where id = ${item.id}`;
|
|
brokenCount++;
|
|
}
|
|
}
|
|
}
|
|
|
|
console.log(`Migration complete.`);
|
|
console.log(`Trash (soft-deleted): ${trashCount}`);
|
|
console.log(`Pending: ${pendingCount}`);
|
|
console.log(`Broken: ${brokenCount}`);
|
|
process.exit(0);
|
|
})();
|