actual-scripts/calc-schedule-budget.js

97 lines
2.8 KiB
JavaScript

import { format } from 'date-fns';
import chalk from 'chalk';
import { checkEnv } from './lib/env.js';
import { formatAmount, formatSchedule } from './lib/format.js';
import * as actual from './lib/actual.js';
import { addFromFrequency } from './lib/dates.js';
checkEnv([
'BUDGET_INCOME_NAME',
'BUDGET_OUTCOME_NAME',
]);
const categories = await actual.getCategories();
const incomeCategory = categories.find(c => c.name === process.env.BUDGET_INCOME_NAME);
const outcomeCategory = categories.find(c => c.name === process.env.BUDGET_OUTCOME_NAME);
console.log();
if (process.env.DRY_RUN) {
console.log(chalk.grey.bold('Running in dry run mode\n'));
}
const [month = format(new Date(), 'yyyy-MM')] = process.argv.slice(2);
const schedules = await actual.getSchedulesSQL();
const categoriesTotal = new Map();
let totalOutcome = 0;
let totalIncome = 0;
for (const s of schedules) {
if (s.completed) {
console.log(chalk.grey(`- ${formatSchedule(s)}: completed`));
continue;
}
const start = (typeof s._date === 'string' ? s._date : s._date.start).slice(0, 7);
if (start > month) {
console.log(chalk.grey(`- ${formatSchedule(s)}: not in range`));
continue;
}
let date = start;
if (typeof s._date === 'object') {
let occurrences = 0;
while (date < month && occurrences <= s._date.endOccurrences) {
if (s._date.endMode !== 'never') {
occurrences += 1;
}
date = addFromFrequency(date, s._date.frequency, s._date.interval);
}
}
if (date !== month) {
console.log(chalk.grey(`- ${formatSchedule(s)}: not the specified month`));
continue;
}
const amount = Math.abs(s._amount);
let categoryId = s._actions.find((a) => a.field === 'category' && a.op === 'set')?.value;
if (!categoryId) {
console.log(chalk.yellow(` ${formatSchedule(s)}: falling back on default`));
categoryId = (s._amount > 0 ? incomeCategory?.id : outcomeCategory?.id);
}
if (!categoryId) {
console.log(chalk.red(`- ${formatSchedule(s)}: no category found`));
continue;
}
if (s._amount > 0) {
totalIncome += amount;
} else {
totalOutcome += amount;
}
categoriesTotal.set(categoryId, (categoriesTotal.get(categoryId) || 0) + amount);
console.log(chalk.green(`+ ${formatSchedule(s, categories.find((c) => c.id === categoryId))}`));
}
console.log(chalk.bgBlue(`\nTotal income: +${formatAmount(totalIncome)}\nTotal outcome: -${formatAmount(totalOutcome)}\n`));
console.log(`Budgets for ${month}: `);
for (const [id, total] of categoriesTotal) {
const category = categories.find((c) => c.id === id);
if (!category) {
console.log(chalk.red(`Category ${id} not found`));
continue;
}
if (!process.env.DRY_RUN) {
actual.setBudgetAmount(month, id, total);
}
console.log(chalk.blue(`${chalk.underline(category.name)} -> ${formatAmount(total)}`));
}
console.log();
await actual.shutdown();