84 lines
2.2 KiB
C++
84 lines
2.2 KiB
C++
#include <fstream>
|
|
|
|
#include "ledger.h"
|
|
|
|
#include <pcre.h> // Perl regular expression library
|
|
|
|
//////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Command-line parser and top-level logic.
|
|
//
|
|
|
|
namespace ledger {
|
|
extern bool parse_ledger(std::istream& in);
|
|
extern bool parse_gnucash(std::istream& in);
|
|
|
|
extern void report_balances(int argc, char **argv, std::ostream& out);
|
|
extern void print_ledger(int argc, char *argv[], std::ostream& out);
|
|
}
|
|
|
|
using namespace ledger;
|
|
|
|
void show_help(std::ostream& out)
|
|
{
|
|
out << "usage: ledger [options] DATA_FILE COMMAND [ARGS]"
|
|
<< std::endl
|
|
<< "options:" << std::endl
|
|
<< " -s show sub-accounts in balance totals" << std::endl
|
|
<< " -S show empty accounts in balance totals" << std::endl
|
|
<< "commands:" << std::endl
|
|
<< " balance show balance totals" << std::endl
|
|
<< " print print all ledger entries" << std::endl;
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
// Global defaults
|
|
|
|
commodity * usd = new commodity("$", true, false, true, false, 2);
|
|
commodities.insert(commodities_entry("USD", usd));
|
|
|
|
// Parse the command-line options
|
|
|
|
int c;
|
|
while (-1 != (c = getopt(argc, argv, "+hw"))) {
|
|
switch (char(c)) {
|
|
case 'h': show_help(std::cout); break;
|
|
case 'w': use_warnings = true; break;
|
|
}
|
|
}
|
|
|
|
if (optind == argc) {
|
|
std::cerr << "usage: ledger [options] DATA_FILE COMMAND [ARGS]"
|
|
<< std::endl
|
|
<< "options:" << std::endl
|
|
<< " -s show sub-accounts in balance totals" << std::endl
|
|
<< " -S show empty accounts in balance totals" << std::endl
|
|
<< "commands:" << std::endl
|
|
<< " balance show balance totals" << std::endl
|
|
<< " print print all ledger entries" << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
// Parse the ledger
|
|
|
|
std::ifstream file(argv[optind++]);
|
|
|
|
char buf[32];
|
|
file.get(buf, 31);
|
|
file.seekg(0);
|
|
|
|
if (std::strncmp(buf, "<?xml version=\"1.0\"?>", 21) == 0)
|
|
parse_gnucash(file);
|
|
else
|
|
parse_ledger(file);
|
|
|
|
// Process the command
|
|
|
|
const std::string command = argv[optind];
|
|
|
|
if (command == "balance")
|
|
report_balances(argc - optind, &argv[optind], std::cout);
|
|
else if (command == "print")
|
|
print_ledger(argc - optind, &argv[optind], std::cout);
|
|
}
|