ledger/objects.h

Go to the documentation of this file.

#pragma once
 
#include <utility>
#include <set>
#include <string>
#include <vector>
#include <stdio.h>
#include <stdint.h>
#include "core/time_utils.h"
 
namespace teditor {
namespace ledger {
 
typedef std::set<std::string> Aliases;
 
class Account {
public:
  Account(const std::string& n, const std::string& d="",
          const Aliases& a=Aliases()): nam(n), descr(d), alias(a), bal(0) {}
 
  const std::string& name() const { return nam; }
  const std::string& desc() const { return descr; }
  int64_t rawBalance() const { return bal; }
  double balance() const { return rawBalance() / 100.0; }
 
  bool isAnAlias(const std::string& a) const {
    return alias.find(a) != alias.end();
  }
  void insertAlias(const std::string& a) { alias.insert(a); }
 
  Account& operator+=(double val) {
    bal += (int64_t)(val * 100.0);
    return *this;
  }
  Account& operator+=(int64_t val) { bal += val; return *this; }
 
private:
  std::string nam, descr;
  Aliases alias;
  int64_t bal;
};
 
class Accounts: public std::vector<Account> {
public:
  Account& find(const std::string& a) {
    for(size_t i=0;i<size();++i) {
      auto& itr = at(i);
      if(itr.name() == a) return itr;
      if(itr.isAnAlias(a)) return itr;
    }
    push_back(Account(a));
    return back();
  }
};
 
typedef std::pair<std::string, int64_t> AccInfo;
 
typedef std::vector<AccInfo> AccountsInfo;
 
class Transaction {
public:
  Transaction(const std::string& dateStr, const std::string& n):
    nam(n), dat(timeFromStr(dateStr)), accts() {}
 
  void add(const std::string& acct, double val) {
    accts.push_back(AccInfo(acct, int64_t(val * 100.0)));
  }
 
  void add(const std::string& acct) {
    int64_t val = 0;
    for(const auto& a : accts) val += a.second;
    accts.push_back(AccInfo(acct, -val));
  }
 
  void updateAccounts(Accounts& acc) {
    for(const auto& a : accts) acc.find(a.first) += a.second;
  }
 
  const std::string& name() const { return nam; }
  const TimePoint& date() const { return dat; }
 
  int64_t rawBalance() const {
    int64_t val = 0;
    for(const auto& a : accts) val += a.second;
    return val;
  }
 
  double balance() const { return rawBalance() / 100.0; }
 
private:
  std::string nam;
  TimePoint dat;
  AccountsInfo accts;
};
 
typedef std::vector<Transaction> Transactions;
 
} // end namespace ledger
} // end namespace teditor