#include <Core/Core.h>

using namespace Upp;

#define TAG_AddressBook "AddressBook"
#define TAG_person "person"
#define TAG_name "name"
#define TAG_surname "surname"
#define TAG_address "address"
#define TAG_email "email"

String PrintData(const XmlNode& node) {
	const XmlNode& ls = node[0];
	StringBuffer text;

	for (int i = 0, n = ls.GetCount(); i < n; ++i) {
		const XmlNode& ll = ls[i];
		if (ll.IsTag(TAG_person)) {
			text << "- Person:\n";
			const XmlNode& lname = ll[TAG_name];
			if (!lname.IsVoid())
				text << TAG_name << ": " << lname[0].GetText() << '\n';

			const XmlNode& lsurname = ll[TAG_surname];
			if (!lsurname.IsVoid())
				text << TAG_surname << ": " << lsurname[0].GetText() << '\n';

			const XmlNode& laddress = ll[TAG_address];
			if (!laddress.IsVoid())
				text << TAG_address << ": " << laddress[0].GetText() << '\n';

			const XmlNode& lemail = ll[TAG_email];
			if (!lemail.IsVoid())
				text << TAG_email << ": " << lemail[0].GetText() << '\n';

			text << '\n';
		}
	}

	return text;
}

void ChangeXmlText(XmlNode& node, const char *tag, const String& txt) {
	int q = node.FindTag(tag);
	if (q < 0)	// Adding new tag in case of not found
		node.Add(tag).AddText(txt);
	else		// Changing found tag
		node.At(q).At(0).CreateText(txt);
}

CONSOLE_APP_MAIN
{
	const Vector<String>& cmdLine = CommandLine();

	if (cmdLine.GetCount() == 0) {
		Cout() << "Specify XML file to load\n";
		return;
	}

	String fileName(NormalizePath(cmdLine[0]));

	// Loading contents of file to String
	const String data = LoadFile(fileName);
	if (data.IsVoid()) {
		Cout() << "Error, while loading '" << NormalizePath(fileName) << "' file\n";
		SetExitCode(1);
		return;
	}

	XmlNode node;
	try { // Trying to parse XML file
		node = ParseXML(data);
	}
	catch (XmlError e) {
		Cout() << "XmlError " << e << '\n';
		SetExitCode(1);
		return;
	}

	if (node.GetCount() == 0 || node[0].GetTag() != TAG_AddressBook) {
		Cout() << "There is no following tag for XML file: " << TAG_AddressBook << '\n';
		return;
	}

	Cout() << "Existing data:\n" << PrintData(node);
	// The names of persons for changing data
	const String names[] = { "qw", "rer", "new name" };
	const String text("*"); // some changed data
	// Selecting contents of "AddressBook" tag
	XmlNode& ls = node.At(0);

	for (int j = 0, m = __countof(names); j < m; ++j) { // Iterate through names array
		bool found = false;
		for (int i = 0, n = ls.GetCount(); i < n; ++i) { // Iterate through "AddressBook" tags
			XmlNode& ll = ls.At(i);
			if (ll.IsTag(TAG_person)) { // Checking for "person" tag
				int q = ll.FindTag(TAG_name);
				if (q < 0)
					continue;
				// Selecting contents of "name" tag
				XmlNode& lname = ll.At(q).At(0);
				if (lname.GetText() == names[j]) { // Comparing names
					found = true;
					// Changing contents of "person" tag
					lname.CreateText(String().Cat() << names[j] << " *");

					ChangeXmlText(ll, TAG_surname, text);
					ChangeXmlText(ll, TAG_address, text);
					ChangeXmlText(ll, TAG_email, text);

					break; // in case of no duplicates
				}
			}
		}

		if (!found) { // Adding new "person" tag with their contents, if name not found
			XmlNode& ll = ls.Add(TAG_person);
			ll.Add(TAG_name).AddText(names[j]);
			ll.Add(TAG_surname).AddText(text);
			ll.Add(TAG_address).AddText(text);
			ll.Add(TAG_email).AddText(text);
		}
	}

	Cout() << "Changed data:\n" << PrintData(node);

	fileName = AppendFileName(GetFileDirectory(fileName),
		String().Cat() << GetFileTitle(fileName) << " (changed)" << GetFileExt(fileName));
	// Saving changed XmlNode(s) to different file path
	if (SaveFile(fileName, AsXML(node)))
		Cout() << "Changed file '" << fileName << "' written successfully\n";
	else
		Cout() << "Error, while writing '" << fileName << "' file\n";
}