diff options
Diffstat (limited to 'main.cc')
| -rw-r--r-- | main.cc | 51 |
1 files changed, 34 insertions, 17 deletions
@@ -4,11 +4,6 @@ #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); @@ -31,6 +26,11 @@ void show_help(std::ostream& out) << " print print all ledger entries" << std::endl; } +////////////////////////////////////////////////////////////////////// +// +// Command-line parser and top-level logic. +// + int main(int argc, char *argv[]) { // Global defaults @@ -40,38 +40,53 @@ int main(int argc, char *argv[]) // Parse the command-line options + std::istream * file = NULL; + int c; - while (-1 != (c = getopt(argc, argv, "+hw"))) { + while (-1 != (c = getopt(argc, argv, "+hwf:"))) { switch (char(c)) { case 'h': show_help(std::cout); break; case 'w': use_warnings = true; break; + case 'f': file = new std::ifstream(optarg); break; } } if (optind == argc) { - std::cerr << "usage: ledger [options] DATA_FILE COMMAND [ARGS]" + std::cerr << "usage: ledger [options] COMMAND [options] [ARGS]" << std::endl + << std::endl + << "ledger options:" << std::endl + << " -f FILE specify pathname of ledger data file" << std::endl << 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; + << " print print all ledger entries" << std::endl + << std::endl + << "`balance' command options:" << std::endl + << " -s show sub-accounts in balance totals" << std::endl + << " -S show empty accounts in balance totals" << std::endl; return 1; } - // Parse the ledger + // The -f option is required + + if (! file) { + std::cerr << "Please specify the ledger file using the -f option." + << std::endl; + return 1; + } - std::ifstream file(argv[optind++]); + // Parse the ledger char buf[32]; - file.get(buf, 31); - file.seekg(0); + file->get(buf, 31); + file->seekg(0); if (std::strncmp(buf, "<?xml version=\"1.0\"?>", 21) == 0) - parse_gnucash(file); + parse_gnucash(*file); else - parse_ledger(file); + parse_ledger(*file); + + delete file; // Process the command @@ -82,3 +97,5 @@ int main(int argc, char *argv[]) else if (command == "print") print_ledger(argc - optind, &argv[optind], std::cout); } + +// main.cc ends here. |