135 lines
3.3 KiB
JavaScript
135 lines
3.3 KiB
JavaScript
export default async function runMigration(db) {
|
|
function getValue(node) {
|
|
return node.expr != null ? node.expr : node.cachedValue;
|
|
}
|
|
|
|
db.execQuery(`
|
|
CREATE TABLE zero_budget_months
|
|
(id TEXT PRIMARY KEY,
|
|
buffered INTEGER DEFAULT 0);
|
|
|
|
CREATE TABLE zero_budgets
|
|
(id TEXT PRIMARY KEY,
|
|
month INTEGER,
|
|
category TEXT,
|
|
amount INTEGER DEFAULT 0,
|
|
carryover INTEGER DEFAULT 0);
|
|
|
|
CREATE TABLE reflect_budgets
|
|
(id TEXT PRIMARY KEY,
|
|
month INTEGER,
|
|
category TEXT,
|
|
amount INTEGER DEFAULT 0,
|
|
carryover INTEGER DEFAULT 0);
|
|
|
|
CREATE TABLE notes
|
|
(id TEXT PRIMARY KEY,
|
|
note TEXT);
|
|
|
|
CREATE TABLE kvcache (key TEXT PRIMARY KEY, value TEXT);
|
|
CREATE TABLE kvcache_key (id INTEGER PRIMARY KEY, key REAL);
|
|
`);
|
|
|
|
// Migrate budget amounts and carryover
|
|
let budget = db.runQuery(
|
|
`SELECT * FROM spreadsheet_cells WHERE name LIKE 'budget%!budget-%'`,
|
|
[],
|
|
true
|
|
);
|
|
db.transaction(() => {
|
|
budget.map((monthBudget) => {
|
|
let match = monthBudget.name.match(
|
|
/^(budget-report|budget)(\d+)!budget-(.+)$/
|
|
);
|
|
if (match == null) {
|
|
console.log('Warning: invalid budget month name', monthBudget.name);
|
|
return;
|
|
}
|
|
|
|
let type = match[1];
|
|
let month = match[2].slice(0, 4) + '-' + match[2].slice(4);
|
|
let dbmonth = parseInt(match[2]);
|
|
let cat = match[3];
|
|
|
|
let amount = parseInt(getValue(monthBudget));
|
|
if (isNaN(amount)) {
|
|
amount = 0;
|
|
}
|
|
|
|
let sheetName = monthBudget.name.split('!')[0];
|
|
let carryover = db.runQuery(
|
|
'SELECT * FROM spreadsheet_cells WHERE name = ?',
|
|
[`${sheetName}!carryover-${cat}`],
|
|
true
|
|
);
|
|
|
|
let table = type === 'budget-report' ? 'reflect_budgets' : 'zero_budgets';
|
|
db.runQuery(
|
|
`INSERT INTO ${table} (id, month, category, amount, carryover) VALUES (?, ?, ?, ?, ?)`,
|
|
[
|
|
`${month}-${cat}`,
|
|
dbmonth,
|
|
cat,
|
|
amount,
|
|
carryover.length > 0 && getValue(carryover[0]) === 'true' ? 1 : 0
|
|
]
|
|
);
|
|
});
|
|
});
|
|
|
|
// Migrate buffers
|
|
let buffers = db.runQuery(
|
|
`SELECT * FROM spreadsheet_cells WHERE name LIKE 'budget%!buffered'`,
|
|
[],
|
|
true
|
|
);
|
|
db.transaction(() => {
|
|
buffers.map((buffer) => {
|
|
let match = buffer.name.match(/^budget(\d+)!buffered$/);
|
|
if (match) {
|
|
let month = match[1].slice(0, 4) + '-' + match[1].slice(4);
|
|
let amount = parseInt(getValue(buffer));
|
|
if (isNaN(amount)) {
|
|
amount = 0;
|
|
}
|
|
|
|
db.runQuery(
|
|
`INSERT INTO zero_budget_months (id, buffered) VALUES (?, ?)`,
|
|
[month, amount]
|
|
);
|
|
}
|
|
});
|
|
});
|
|
|
|
// Migrate notes
|
|
let notes = db.runQuery(
|
|
`SELECT * FROM spreadsheet_cells WHERE name LIKE 'notes!%'`,
|
|
[],
|
|
true
|
|
);
|
|
|
|
let parseNote = (str) => {
|
|
try {
|
|
let value = JSON.parse(str);
|
|
return value && value !== '' ? value : null;
|
|
} catch (e) {
|
|
return null;
|
|
}
|
|
};
|
|
|
|
db.transaction(() => {
|
|
notes.forEach((note) => {
|
|
let parsed = parseNote(getValue(note));
|
|
if (parsed) {
|
|
let [, id] = note.name.split('!');
|
|
db.runQuery(`INSERT INTO notes (id, note) VALUES (?, ?)`, [id, parsed]);
|
|
}
|
|
});
|
|
});
|
|
|
|
db.execQuery(`
|
|
DROP TABLE spreadsheet_cells;
|
|
ANALYZE;
|
|
VACUUM;
|
|
`);
|
|
}
|