1 files changed, 126 insertions, 0 deletions
diff --git a/Swiften/Client/XMLBeautifier.cpp b/Swiften/Client/XMLBeautifier.cpp
new file mode 100644
index 0000000..b70fc48
--- /dev/null
+++ b/Swiften/Client/XMLBeautifier.cpp
@@ -0,0 +1,126 @@
+/*
+ * Copyright (c) 2011 Tobias Markmann
+ * Licensed under the simplified BSD license.
+ * See Documentation/Licenses/BSD-simplified.txt for more information.
+ */
+
+#include <sstream>
+#include <stack>
+
+#include <Swiften/Base/foreach.h>
+#include <Swiften/Base/Log.h>
+#include <Swiften/Client/XMLBeautifier.h>
+#include <Swiften/Parser/PlatformXMLParserFactory.h>
+
+namespace Swift {
+
+XMLBeautifier::XMLBeautifier(bool indention, bool coloring) : doIndention(indention), doColoring(coloring), intLevel(0), parser(NULL), lastWasStepDown(false) {
+	factory = new PlatformXMLParserFactory();
+}
+
+XMLBeautifier::~XMLBeautifier() {
+	delete factory;
+}
+
+std::string XMLBeautifier::beautify(const std::string &text) {
+	parser = factory->createXMLParser(this);
+	intLevel = 0;
+	buffer.str(std::string());
+	parser->parse(text);
+	delete parser;
+	return buffer.str();
+}
+
+void XMLBeautifier::indent() {
+	for (int i = 0; i < intLevel; ++i) {
+		buffer << " ";
+	}
+}
+
+// all bold but reset
+const char colorBlue[] = "\x1b[01;34m";
+const char colorCyan[] = "\x1b[01;36m";
+const char colorGreen[] = "\x1b[01;32m";
+const char colorMagenta[] = "\x1b[01;35m";
+const char colorRed[] = "\x1b[01;31m";
+const char colorReset[] = "\x1b[0m";
+const char colorYellow[] = "\x1b[01;33m";
+
+
+
+std::string XMLBeautifier::styleTag(const std::string& text) const {
+	std::string result;
+	result += colorYellow;
+	result += text;
+	result += colorReset;
+	return result;
+}
+
+std::string XMLBeautifier::styleNamespace(const std::string& text) const {
+	std::string result;
+	result += colorRed;
+	result += text;
+	result += colorReset;
+	return result;
+}
+
+std::string XMLBeautifier::styleAttribute(const std::string& text) const {
+	std::string result;
+	result += colorGreen;
+	result += text;
+	result += colorReset;
+	return result;
+}
+std::string XMLBeautifier::styleValue(const std::string& text) const {
+	std::string result;
+	result += colorCyan;
+	result += text;
+	result += colorReset;
+	return result;
+}
+
+void XMLBeautifier::handleStartElement(const std::string& element, const std::string& ns, const AttributeMap& attributes) {
+	if (doIndention) {
+		if (intLevel) buffer << std::endl;
+	}
+	indent();
+	buffer << "<" << (doColoring ? styleTag(element) : element);
+	if (!ns.empty() && (!parentNSs.empty() && parentNSs.top() != ns)) {
+		buffer << " ";
+		buffer << (doColoring ? styleAttribute("xmlns") : "xmlns");
+		buffer << "=";
+		buffer << "\"" << (doColoring ? styleNamespace(ns) : ns) << "\"";
+	}
+	if (!attributes.getEntries().empty()) {
+		foreach(AttributeMap::Entry entry, attributes.getEntries()) {
+			buffer << " ";
+			buffer << (doColoring ? styleAttribute(entry.getAttribute().getName()) : entry.getAttribute().getName());
+			buffer << "=";
+			buffer << "\"" << (doColoring ? styleValue(entry.getValue()) : entry.getValue()) << "\"";
+		}
+	}
+	buffer << ">";
+	++intLevel;
+	lastWasStepDown = false;
+	parentNSs.push(ns);
+}
+
+void XMLBeautifier::handleEndElement(const std::string& element, const std::string& /* ns */) {
+	--intLevel;
+	parentNSs.pop();
+	if (/*hadCDATA.top() ||*/ lastWasStepDown) {
+		if (doIndention) {
+			buffer << std::endl;
+		}
+		indent();
+	}
+	buffer << "</" << (doColoring ? styleTag(element) : element) << ">";
+	lastWasStepDown = true;
+}
+
+void XMLBeautifier::handleCharacterData(const std::string& data) {
+	buffer << data;
+	lastWasStepDown = false;
+}
+
+}