diff options
Diffstat (limited to 'xml.cc')
| -rw-r--r-- | xml.cc | 443 |
1 files changed, 0 insertions, 443 deletions
diff --git a/xml.cc b/xml.cc deleted file mode 100644 index 63043fbe..00000000 --- a/xml.cc +++ /dev/null @@ -1,443 +0,0 @@ -#include "xml.h" -#include "journal.h" -#include "datetime.h" -#include "error.h" - -#include <iostream> -#include <sstream> -#include <cstring> - -extern "C" { -#if defined(HAVE_EXPAT) -#include <expat.h> // expat XML parser -#elif defined(HAVE_XMLPARSE) -#include <xmlparse.h> // expat XML parser -#else -#error "No XML parser library defined." -#endif -} - -namespace ledger { - -static XML_Parser current_parser; -static unsigned int count; - -static journal_t * curr_journal; -static entry_t * curr_entry; -static commodity_t * curr_comm; -static std::string comm_flags; - -static std::string data; -static bool ignore; -static std::string have_error; - -static void startElement(void *userData, const char *name, const char **attrs) -{ - if (ignore) - return; - - if (std::strcmp(name, "entry") == 0) { - assert(! curr_entry); - curr_entry = new entry_t; - } - else if (std::strcmp(name, "transaction") == 0) { - assert(curr_entry); - curr_entry->add_transaction(new transaction_t); - } - else if (std::strcmp(name, "commodity") == 0) { - if (std::string(attrs[0]) == "flags") - comm_flags = attrs[1]; - } - else if (std::strcmp(name, "total") == 0) { - ignore = true; - } -} - -static void endElement(void *userData, const char *name) -{ - if (ignore) { - if (std::strcmp(name, "total") == 0) - ignore = false; - return; - } - - if (std::strcmp(name, "entry") == 0) { - assert(curr_entry); - if (curr_journal->add_entry(curr_entry)) { - count++; - } else { - account_t * acct = curr_journal->find_account("<Unknown>"); - curr_entry->add_transaction(new transaction_t(acct)); - if (curr_journal->add_entry(curr_entry)) { - count++; - } else { - delete curr_entry; - have_error = "Entry cannot be balanced"; - } - } - curr_entry = NULL; - } - else if (std::strcmp(name, "en:date") == 0) { - quick_parse_date(data.c_str(), &curr_entry->date); - } - else if (std::strcmp(name, "en:cleared") == 0) { - curr_entry->state = entry_t::CLEARED; - } - else if (std::strcmp(name, "en:pending") == 0) { - curr_entry->state = entry_t::PENDING; - } - else if (std::strcmp(name, "en:code") == 0) { - curr_entry->code = data; - } - else if (std::strcmp(name, "en:payee") == 0) { - curr_entry->payee = data; - } - else if (std::strcmp(name, "tr:account") == 0) { - curr_entry->transactions.back()->account = curr_journal->find_account(data); - } - else if (std::strcmp(name, "tr:virtual") == 0) { - curr_entry->transactions.back()->flags |= TRANSACTION_VIRTUAL; - } - else if (std::strcmp(name, "tr:generated") == 0) { - curr_entry->transactions.back()->flags |= TRANSACTION_AUTO; - } - else if (std::strcmp(name, "commodity") == 0) { - assert(! curr_comm); - curr_comm = commodity_t::find_commodity(data, true); - curr_comm->flags |= COMMODITY_STYLE_SUFFIXED; - if (! comm_flags.empty()) - for (std::string::size_type i = 0, l = comm_flags.length(); i < l; i++) - switch (comm_flags[i]) { - case 'P': curr_comm->flags &= ~COMMODITY_STYLE_SUFFIXED; break; - case 'S': curr_comm->flags |= COMMODITY_STYLE_SEPARATED; break; - case 'T': curr_comm->flags |= COMMODITY_STYLE_THOUSANDS; break; - case 'E': curr_comm->flags |= COMMODITY_STYLE_EUROPEAN; break; - } - } - else if (std::strcmp(name, "quantity") == 0) { - curr_entry->transactions.back()->amount.parse(data); - if (curr_comm) { - std::string::size_type i = data.find('.'); - if (i != std::string::npos) { - int precision = data.length() - i - 1; - if (precision > curr_comm->precision) - curr_comm->precision = precision; - } - curr_entry->transactions.back()->amount.set_commodity(*curr_comm); - curr_comm = NULL; - } - } - else if (std::strcmp(name, "tr:amount") == 0) { - curr_comm = NULL; - } -} - -static void dataHandler(void *userData, const char *s, int len) -{ - if (! ignore) - data = std::string(s, len); -} - -bool xml_parser_t::test(std::istream& in) const -{ - char buf[80]; - - in.getline(buf, 79); - if (std::strncmp(buf, "<?xml", 5) != 0) { - in.seekg(0, std::ios::beg); - return false; - } - - in.getline(buf, 79); - if (! std::strstr(buf, "<ledger")) { - in.seekg(0, std::ios::beg); - return false; - } - - in.seekg(0, std::ios::beg); - return true; -} - -unsigned int xml_parser_t::parse(std::istream& in, - journal_t * journal, - account_t * master, - const std::string * original_file) -{ - char buf[BUFSIZ]; - - count = 0; - curr_journal = journal; - curr_entry = NULL; - curr_comm = NULL; - ignore = false; - - unsigned int offset = 2; - XML_Parser parser = XML_ParserCreate(NULL); - current_parser = parser; - - XML_SetElementHandler(parser, startElement, endElement); - XML_SetCharacterDataHandler(parser, dataHandler); - - while (! in.eof()) { - in.getline(buf, BUFSIZ - 1); - std::strcat(buf, "\n"); - bool result; - try { - result = XML_Parse(parser, buf, std::strlen(buf), in.eof()); - } - catch (const std::exception& err) { - unsigned long line = XML_GetCurrentLineNumber(parser) - offset++; - XML_ParserFree(parser); - throw parse_error(original_file ? *original_file : "<xml>", line, - err.what()); - } - - if (! have_error.empty()) { - unsigned long line = XML_GetCurrentLineNumber(parser) - offset++; - parse_error err(original_file ? *original_file : "<xml>", line, - have_error); - std::cerr << "Error: " << err.what() << std::endl; - have_error = ""; - } - - if (! result) { - unsigned long line = XML_GetCurrentLineNumber(parser) - offset++; - const char * err = XML_ErrorString(XML_GetErrorCode(parser)); - XML_ParserFree(parser); - throw parse_error(original_file ? *original_file : "<xml>", line, err); - } - } - - XML_ParserFree(parser); - - return count; -} - -void xml_write_amount(std::ostream& out, const amount_t& amount, - const int depth = 0) -{ - for (int i = 0; i < depth; i++) out << ' '; - out << "<amount>\n"; - - commodity_t& c = amount.commodity(); - for (int i = 0; i < depth + 2; i++) out << ' '; - out << "<commodity flags=\""; - if (! (c.flags & COMMODITY_STYLE_SUFFIXED)) out << 'P'; - if (c.flags & COMMODITY_STYLE_SEPARATED) out << 'S'; - if (c.flags & COMMODITY_STYLE_THOUSANDS) out << 'T'; - if (c.flags & COMMODITY_STYLE_EUROPEAN) out << 'E'; - out << "\">" << c.symbol << "</commodity>\n"; - - for (int i = 0; i < depth + 2; i++) out << ' '; - out << "<quantity>"; - out << amount.quantity_string() << "</quantity>\n"; - - for (int i = 0; i < depth; i++) out << ' '; - out << "</amount>\n"; -} - -void xml_write_value(std::ostream& out, const value_t& value, - const int depth = 0) -{ - balance_t * bal = NULL; - - for (int i = 0; i < depth; i++) out << ' '; - out << "<value type=\""; - switch (value.type) { - case value_t::BOOLEAN: out << "boolean"; break; - case value_t::INTEGER: out << "integer"; break; - case value_t::AMOUNT: out << "amount"; break; - case value_t::BALANCE: - case value_t::BALANCE_PAIR: out << "balance"; break; - } - out << "\">\n"; - - switch (value.type) { - case value_t::BOOLEAN: - for (int i = 0; i < depth + 2; i++) out << ' '; - out << "<boolean>" << *((bool *) value.data) << "</boolean>\n"; - break; - - case value_t::INTEGER: - for (int i = 0; i < depth + 2; i++) out << ' '; - out << "<integer>" << *((long *) value.data) << "</integer>\n"; - break; - - case value_t::AMOUNT: - xml_write_amount(out, *((amount_t *) value.data), depth + 2); - break; - - case value_t::BALANCE: - bal = (balance_t *) value.data; - // fall through... - - case value_t::BALANCE_PAIR: - if (! bal) - bal = &((balance_pair_t *) value.data)->quantity; - - for (int i = 0; i < depth + 2; i++) out << ' '; - out << "<balance>\n"; - - for (amounts_map::const_iterator i = bal->amounts.begin(); - i != bal->amounts.end(); - i++) - xml_write_amount(out, (*i).second, depth + 4); - - for (int i = 0; i < depth + 2; i++) out << ' '; - out << "</balance>\n"; - break; - - default: - assert(0); - break; - } - - for (int i = 0; i < depth; i++) out << ' '; - out << "</value>\n"; -} - -void output_xml_string(std::ostream& out, const std::string& str) -{ - for (const char * s = str.c_str(); *s; s++) { - switch (*s) { - case '<': - out << "<"; - break; - case '>': - out << "&rt;"; - break; - case '&': - out << "&"; - break; - default: - out << *s; - break; - } - } -} - -void format_xml_entries::format_last_entry() -{ - char buf[256]; - std::strftime(buf, 255, format_t::date_format.c_str(), - std::localtime(&last_entry->date)); - - output_stream << " <entry>\n" - << " <en:date>" << buf << "</en:date>\n"; - - if (last_entry->state == entry_t::CLEARED) - output_stream << " <en:cleared/>\n"; - else if (last_entry->state == entry_t::PENDING) - output_stream << " <en:pending/>\n"; - - if (! last_entry->code.empty()) { - output_stream << " <en:code>"; - output_xml_string(output_stream, last_entry->code); - output_stream << "</en:code>\n"; - } - - if (! last_entry->payee.empty()) { - output_stream << " <en:payee>"; - output_xml_string(output_stream, last_entry->payee); - output_stream << "</en:payee>\n"; - } - - bool first = true; - for (transactions_list::const_iterator i = last_entry->transactions.begin(); - i != last_entry->transactions.end(); - i++) { - if (transaction_has_xdata(**i) && - transaction_xdata_(**i).dflags & TRANSACTION_TO_DISPLAY) { - if (first) { - output_stream << " <en:transactions>\n"; - first = false; - } - - output_stream << " <transaction>\n"; - - if ((*i)->flags & TRANSACTION_VIRTUAL) - output_stream << " <tr:virtual/>\n"; - if ((*i)->flags & TRANSACTION_AUTO) - output_stream << " <tr:generated/>\n"; - - if ((*i)->account) { - std::string name = (*i)->account->fullname(); - if (name == "<Total>") - name = "[TOTAL]"; - else if (name == "<Unknown>") - name = "[UNKNOWN]"; - - output_stream << " <tr:account>"; - output_xml_string(output_stream, name); - output_stream << "</tr:account>\n"; - } - - output_stream << " <tr:amount>\n"; - if (transaction_xdata_(**i).dflags & TRANSACTION_COMPOSITE) - xml_write_value(output_stream, - transaction_xdata_(**i).composite_amount, 10); - else - xml_write_value(output_stream, value_t((*i)->amount), 10); - output_stream << " </tr:amount>\n"; - - if ((*i)->cost) { - output_stream << " <tr:cost>\n"; - xml_write_value(output_stream, value_t(*(*i)->cost), 10); - output_stream << " </tr:cost>\n"; - } - - if (! (*i)->note.empty()) { - output_stream << " <tr:note>"; - output_xml_string(output_stream, (*i)->note); - output_stream << "</tr:note>\n"; - } - - if (show_totals) { - output_stream << " <total>\n"; - xml_write_value(output_stream, transaction_xdata_(**i).total, 10); - output_stream << " </total>\n"; - } - - output_stream << " </transaction>\n"; - - transaction_xdata_(**i).dflags |= TRANSACTION_DISPLAYED; - } - } - - if (! first) - output_stream << " </en:transactions>\n"; - - output_stream << " </entry>\n"; -} - -} // namespace ledger - -#ifdef USE_BOOST_PYTHON - -#include <boost/python.hpp> - -using namespace boost::python; -using namespace ledger; - -BOOST_PYTHON_MEMBER_FUNCTION_OVERLOADS(xml_parse_overloads, - xml_parser_t::parse, 2, 4) - -void export_xml() { - class_< xml_parser_t, bases<parser_t> > ("XmlParser") - .def("test", &xml_parser_t::test) - .def("parse", &xml_parser_t::parse, xml_parse_overloads()) - ; - - typedef - pystream_handler_wrap<format_xml_entries, transaction_t, bool> - format_xml_entries_wrap; - - class_< format_xml_entries_wrap, bases<item_handler<transaction_t> > > - ("FormatXmlEntries", - init<PyObject *, bool>()[with_custodian_and_ward<1, 2>()]) - .def("flush", &format_xml_entries_wrap::flush) - .def("__call__", &format_xml_entries_wrap::operator()) - ; -} - -#endif // USE_BOOST_PYTHON |