fix(FN-4493): harden seeded prompt updater db access
Fusion-Task-Id: FN-4493 Fusion-Task-Lineage: 966c7c55-fd51-4263-9d4e-87e9d06bfb6f
This commit is contained in:
@@ -7,7 +7,7 @@
|
|||||||
* To capture production rollback text before applying:
|
* To capture production rollback text before applying:
|
||||||
* sqlite3 <db> "SELECT id, prompt FROM workflow_steps WHERE id IN ('WS-004','WS-006') ORDER BY id;"
|
* sqlite3 <db> "SELECT id, prompt FROM workflow_steps WHERE id IN ('WS-004','WS-006') ORDER BY id;"
|
||||||
*/
|
*/
|
||||||
import Database from "better-sqlite3";
|
import { execFileSync } from "node:child_process";
|
||||||
import { resolve } from "node:path";
|
import { resolve } from "node:path";
|
||||||
|
|
||||||
const dbPath = (process.argv.find((arg) => arg.startsWith("--db=")) || "--db=.fusion/fusion.db").slice(5);
|
const dbPath = (process.argv.find((arg) => arg.startsWith("--db=")) || "--db=.fusion/fusion.db").slice(5);
|
||||||
@@ -27,13 +27,34 @@ const seededMappings = [
|
|||||||
{ workflowStepId: "WS-006", templateId: "frontend-ux-design" },
|
{ workflowStepId: "WS-006", templateId: "frontend-ux-design" },
|
||||||
];
|
];
|
||||||
|
|
||||||
const db = new Database(resolve(dbPath));
|
function createCliDatabase(dbFile) {
|
||||||
const getRow = db.prepare("SELECT id, name, prompt FROM workflow_steps WHERE id = ?");
|
const runSql = (sql) => execFileSync("sqlite3", [dbFile, sql], { encoding: "utf8" });
|
||||||
const updateRow = db.prepare("UPDATE workflow_steps SET prompt = ?, updatedAt = ? WHERE id = ?");
|
const runSqlJson = (sql) => execFileSync("sqlite3", ["-json", dbFile, sql], { encoding: "utf8" });
|
||||||
|
return {
|
||||||
|
getRow(id) {
|
||||||
|
const escapedId = String(id).replace(/'/g, "''");
|
||||||
|
const output = runSqlJson(`SELECT id, name, prompt FROM workflow_steps WHERE id = '${escapedId}';`).trim();
|
||||||
|
if (!output) return null;
|
||||||
|
const rows = JSON.parse(output);
|
||||||
|
if (!Array.isArray(rows) || rows.length === 0) return null;
|
||||||
|
const row = rows[0];
|
||||||
|
return { id: row.id, name: row.name, prompt: row.prompt ?? "" };
|
||||||
|
},
|
||||||
|
updateRow(id, prompt, updatedAt) {
|
||||||
|
const escapedId = String(id).replace(/'/g, "''");
|
||||||
|
const escapedPrompt = String(prompt).replace(/'/g, "''");
|
||||||
|
const escapedUpdatedAt = String(updatedAt).replace(/'/g, "''");
|
||||||
|
runSql(`UPDATE workflow_steps SET prompt = '${escapedPrompt}', updatedAt = '${escapedUpdatedAt}' WHERE id = '${escapedId}';`);
|
||||||
|
},
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
const db = createCliDatabase(resolve(dbPath));
|
||||||
|
|
||||||
let updated = 0;
|
let updated = 0;
|
||||||
let skipped = 0;
|
let skipped = 0;
|
||||||
let noOp = 0;
|
let noOp = 0;
|
||||||
|
let differs = 0;
|
||||||
|
|
||||||
for (const { workflowStepId, templateId } of seededMappings) {
|
for (const { workflowStepId, templateId } of seededMappings) {
|
||||||
const nextPrompt = templateMap.get(templateId);
|
const nextPrompt = templateMap.get(templateId);
|
||||||
@@ -42,7 +63,7 @@ for (const { workflowStepId, templateId } of seededMappings) {
|
|||||||
process.exit(1);
|
process.exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
const row = getRow.get(workflowStepId);
|
const row = db.getRow(workflowStepId);
|
||||||
if (!row) {
|
if (!row) {
|
||||||
skipped += 1;
|
skipped += 1;
|
||||||
console.log(`${workflowStepId}: not present, skipping`);
|
console.log(`${workflowStepId}: not present, skipping`);
|
||||||
@@ -57,14 +78,18 @@ for (const { workflowStepId, templateId } of seededMappings) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if (checkOnly) {
|
if (checkOnly) {
|
||||||
updated += 1;
|
differs += 1;
|
||||||
console.log(`${workflowStepId}: differs (check-only)`);
|
console.log(`${workflowStepId}: differs (check-only)`);
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
updateRow.run(nextPrompt, new Date().toISOString(), workflowStepId);
|
db.updateRow(workflowStepId, nextPrompt, new Date().toISOString());
|
||||||
updated += 1;
|
updated += 1;
|
||||||
console.log(`${workflowStepId}: updated (${row.name})`);
|
console.log(`${workflowStepId}: updated (${row.name})`);
|
||||||
}
|
}
|
||||||
|
|
||||||
console.log(`Updated ${updated} row(s), skipped ${skipped}, no-op ${noOp}`);
|
console.log(
|
||||||
|
checkOnly
|
||||||
|
? `Updated ${updated} row(s), skipped ${skipped}, no-op ${noOp}, differs ${differs}`
|
||||||
|
: `Updated ${updated} row(s), skipped ${skipped}, no-op ${noOp}`,
|
||||||
|
);
|
||||||
|
|||||||
Reference in New Issue
Block a user