add track-investments script

track-investments.js

@@ -0,0 +1,128 @@
+const api = require('@actual-app/api');
+const fs = require('fs');
+const readline = require('readline-sync');
+const { closeBudget, ensureCategory, ensurePayee, getAccountBalance, getAccountNote, getTransactions, openBudget } = require('./utils');
+require("dotenv").config();
+
+
+const getCredentials = async () => {
+  const token = readline.question('Enter your SimpleFIN setup token: ');
+  const url = atob(token.trim());
+
+  const response = await fetch(url, { method: 'post' });
+  const api_url = await response.text();
+
+  const rest = api_url.split('//', 2)[1];
+  const auth = rest.split('@', 1)[0];
+  const username = auth.split(':')[0];
+  const pw = auth.split(':')[1];
+
+  const data = `${username}:${pw}`;
+  fs.writeFileSync('simplefin.credentials', data);
+  return data;
+};
+
+const loadCredentials = () => {
+  try {
+    return fs.readFileSync('simplefin.credentials', 'utf8');
+  } catch (err) {
+    return undefined;
+  }
+};
+
+const getSimplefinBalances = async () => {
+  let credentials = loadCredentials();
+  if (!credentials) {
+    credentials = await getCredentials();
+  }
+  const username = credentials.split(':')[0];
+  const pw = credentials.split(':')[1];
+
+  const url = `https://beta-bridge.simplefin.org/simplefin/accounts?start-date=${new Date().getTime()}&end-date=${new Date().getTime()}`;
+  const response = await fetch(url, {
+    headers: {
+      'Authorization': `Basic ${btoa(`${username}:${pw}`)}`
+    }
+  });
+  const data = await response.json();
+  const accounts = data.accounts;
+  const balances = {};
+  accounts.forEach(a => balances[a.name] = parseFloat(a.balance));
+  return balances;
+};
+
+const shouldDrop = (payment) => {
+  const note = payment.notes;
+  return note && (note.indexOf('YOU BOUGHT ') > -1 || note == 'Buy Other' || note == 'Sell Other');
+};
+
+const zeroTransaction = async (payment) => {
+  await api.updateTransaction(
+    payment.id,
+    { 'amount': 0 }
+  );
+}
+
+(async () => {
+  await openBudget();
+
+  const payeeId = await ensurePayee(process.env.IMPORTER_INVESTMENT_PAYEE_NAME || 'Investment');
+
+  const categoryId = await ensureCategory(process.env.IMPORTER_INVESTMENT_CATEGORY_NAME || 'Investment');
+
+  const simplefinBalances = await getSimplefinBalances();
+
+  const accounts = await api.getAccounts();
+  for (const account of accounts) {
+    if (account.closed) {
+      continue;
+    }
+
+    const note = await getAccountNote(account);
+
+    if (note) {
+      const data = await getTransactions(account);
+
+      if (note.indexOf('zeroSmall') > -1) {
+        const payments = data.filter(payment => payment.amount > -10000 && payment.amount < 10000 && payment.amount != 0 && payment.category == categoryId)
+        for (const payment of payments) {
+          if (shouldDrop(payment)) {
+            await zeroTransaction(payment);
+          }
+        }
+      }
+
+      if (note.indexOf('dropPayments') > -1) {
+        const payments = data.filter(payment => payment.amount < 0)
+        for (const payment of payments) {
+          if (shouldDrop(payment)) {
+            await zeroTransaction(payment);
+          }
+        }
+      }
+
+      if (note.indexOf('calcInvestment') > -1) {
+        const currentBalance = await getAccountBalance(account);
+        const simplefinBalance = parseInt(simplefinBalances[account.name] * 100);
+        const diff = simplefinBalance - currentBalance;
+
+        if (diff != 0) {
+          console.log('Account:', account.name);
+          console.log('Simplefin Balance:', simplefinBalance);
+          console.log('Current Balance:', currentBalance);
+          console.log('Difference:', diff);
+
+            await api.importTransactions(account.id, [{
+            date: new Date(),
+            payee: payeeId,
+            amount: diff,
+            category: categoryId,
+            notes: `Update investment balance to ${simplefinBalance / 100}`,
+          }]);
+        }
+      }
+    }
+  }
+
+  await closeBudget();
+})();