From 8c6ac409b5425c0f54a1d42c9f3092fb1b1a9d22 Mon Sep 17 00:00:00 2001
From: Skycoder42 <Skycoder42@users.noreply.github.com>
Date: Thu, 12 Jul 2018 11:33:18 +0200
Subject: [PATCH] added support for (same class) imports

---
 tools/settingsgenerator/main.cpp              |  4 +-
 .../settingsgenerator/qsettingsgenerator.xsd  | 17 ++---
 tools/settingsgenerator/settingsgenerator.cpp | 68 +++++++++++++++++--
 tools/settingsgenerator/settingsgenerator.h   |  3 +
 4 files changed, 76 insertions(+), 16 deletions(-)

diff --git a/tools/settingsgenerator/main.cpp b/tools/settingsgenerator/main.cpp
index df1226c..cebd1b6 100644
--- a/tools/settingsgenerator/main.cpp
+++ b/tools/settingsgenerator/main.cpp
@@ -49,8 +49,8 @@ int main(int argc, char *argv[])
 		};
 		generator.process(parser.value(QStringLiteral("in")));
 		return EXIT_SUCCESS;
-	} catch (const QString &str) {
-		qCritical() << qUtf8Printable(str);
+	} catch (const QException &e) {
+		qCritical() << e.what();
 		return EXIT_FAILURE;
 	}
 }
diff --git a/tools/settingsgenerator/qsettingsgenerator.xsd b/tools/settingsgenerator/qsettingsgenerator.xsd
index b7d844c..9740ad3 100644
--- a/tools/settingsgenerator/qsettingsgenerator.xsd
+++ b/tools/settingsgenerator/qsettingsgenerator.xsd
@@ -7,24 +7,25 @@
 		<qxg:include>QtCore/QHash</qxg:include>
 	</qxg:config>
 	<qxg:method name="read_type_mapping" type="QHash&lt;QString, QString&gt;" asGroup="true"/>
+	<qxg:method name="read_included_file" type="NodeContentGroup"/>
 
 	<!-- Type definitions -->
 	<xs:complexType name="IncludeType">
 		<xs:simpleContent>
-			<xs:extension base="xs:string" qxg:member="include">
+			<xs:extension base="xs:string" qxg:member="includePath">
 				<xs:attribute default="false" name="local" type="xs:boolean" use="optional"/>
 			</xs:extension>
 		</xs:simpleContent>
 	</xs:complexType>
 
-	<!--xs:complexType name="Import">
+	<xs:complexType name="ImportType">
 		<xs:simpleContent>
-			<xs:extension base="xs:string">
+			<xs:extension base="xs:string" qxg:member="importPath">
 				<xs:attribute default="true" name="required" type="xs:boolean" use="optional"/>
 				<xs:attribute name="rootNode" type="xs:string" use="optional"/>
 			</xs:extension>
 		</xs:simpleContent>
-	</xs:complexType-->
+	</xs:complexType>
 
 	<xs:complexType name="TypeMappingEntry">
 		<xs:attribute name="key" type="xs:string" use="required"/>
@@ -56,10 +57,10 @@
 
 	<xs:group name="NodeContentGroup">
 		<xs:sequence>
-			<xs:choice maxOccurs="unbounded" minOccurs="0" qxg:unordered="true">
-				<xs:element name="Node" type="NodeType" qxg:member="nodes"/>
-				<xs:element name="Entry" type="EntryType" qxg:member="entries"/>
-				<!--xs:element name="Import" type="Import"/-->
+			<xs:choice maxOccurs="unbounded" minOccurs="0" qxg:member="contentNodes">
+				<xs:element name="Node" type="NodeType"/>
+				<xs:element name="Entry" type="EntryType"/>
+				<xs:element name="Import" type="ImportType" qxg:method="read_included_file" qxg:methodType="NodeContentGroup"/>
 			</xs:choice>
 		</xs:sequence>
 	</xs:group>
diff --git a/tools/settingsgenerator/settingsgenerator.cpp b/tools/settingsgenerator/settingsgenerator.cpp
index a3a1540..9a1946e 100644
--- a/tools/settingsgenerator/settingsgenerator.cpp
+++ b/tools/settingsgenerator/settingsgenerator.cpp
@@ -37,6 +37,58 @@ bool SettingsGenerator::read_type_mapping(QXmlStreamReader &reader, QHash<QStrin
 	return hasNext;
 }
 
+void SettingsGenerator::read_included_file(QXmlStreamReader &reader, NodeContentGroup &data)
+{
+	ImportType import;
+	read_ImportType(reader, import);
+
+	//make the path relative if possbile
+	if(dynamic_cast<QFileDevice*>(reader.device())) {
+		QFileInfo docInfo{static_cast<QFileDevice*>(reader.device())->fileName()};
+		import.importPath = docInfo.dir().absoluteFilePath(import.importPath);
+	}
+
+	// read the document
+	try {
+		auto settings = readDocument(import.importPath);
+		NodeContentGroup *cGrp = &settings;
+		if(import.rootNode) {
+			for(const auto &key : import.rootNode.value().split(QLatin1Char('/'), QString::SkipEmptyParts)) {
+				cGrp = findContentGroup(cGrp, key);
+				if(!cGrp)
+					return;
+			}
+		}
+		data = *cGrp;
+	} catch(QException &e) {
+		if(import.required)
+			throw;
+		else {
+			qWarning() << e.what();
+			return;
+		}
+	}
+}
+
+SettingsGeneratorBase::NodeContentGroup *SettingsGenerator::findContentGroup(SettingsGeneratorBase::NodeContentGroup *cGrp, const QString &key)
+{
+	for(const auto &cNode : cGrp->contentNodes) {
+		if(nonstd::holds_alternative<NodeType>(cNode)) {
+			if(std::get<NodeType>(cNode).key == key)
+				return const_cast<NodeType*>(&(std::get<NodeType>(cNode)));
+		} else if(nonstd::holds_alternative<EntryType>(cNode)) {
+			if(std::get<EntryType>(cNode).key == key)
+				return const_cast<EntryType*>(&(std::get<EntryType>(cNode)));
+		} else if(nonstd::holds_alternative<NodeContentGroup>(cNode)) {
+			auto res = findContentGroup(const_cast<NodeContentGroup*>(&(std::get<NodeContentGroup>(cNode))), key);
+			if(res)
+				return res;
+		}
+	}
+
+	return nullptr;
+}
+
 void SettingsGenerator::writeHeader(const SettingsType &settings)
 {
 	QString incGuard = QFileInfo{_hdrFile.fileName()}
@@ -49,9 +101,9 @@ void SettingsGenerator::writeHeader(const SettingsType &settings)
 	// write the includes
 	for(const auto &inc : settings.includes) {
 		if(inc.local)
-			_hdr << "#include \"" << inc.include << "\"\n";
+			_hdr << "#include \"" << inc.includePath << "\"\n";
 		else
-			_hdr << "#include <" << inc.include << ">\n";
+			_hdr << "#include <" << inc.includePath << ">\n";
 	}
 	_hdr << "\n";
 
@@ -77,10 +129,14 @@ void SettingsGenerator::writeHeader(const SettingsType &settings)
 
 void SettingsGenerator::writeNodeElements(const NodeContentGroup &node, int intendent)
 {
-	for(const auto &cNode : node.nodes)
-		writeNode(cNode, intendent);
-	for(const auto &cEntry : node.entries)
-		writeEntry(cEntry, intendent);
+	for(const auto &cNode : node.contentNodes) {
+		if(nonstd::holds_alternative<NodeType>(cNode))
+			writeNode(nonstd::get<NodeType>(cNode), intendent);
+		else if(nonstd::holds_alternative<EntryType>(cNode))
+			writeEntry(nonstd::get<EntryType>(cNode), intendent);
+		else if(nonstd::holds_alternative<NodeContentGroup>(cNode))
+			writeNodeElements(nonstd::get<NodeContentGroup>(cNode), intendent);
+	}
 }
 
 void SettingsGenerator::writeNode(const NodeType &node, int intendent)
diff --git a/tools/settingsgenerator/settingsgenerator.h b/tools/settingsgenerator/settingsgenerator.h
index 1d5c5b6..9d897b5 100644
--- a/tools/settingsgenerator/settingsgenerator.h
+++ b/tools/settingsgenerator/settingsgenerator.h
@@ -16,6 +16,7 @@ public:
 
 protected:
 	bool read_type_mapping(QXmlStreamReader &reader, QHash<QString, QString> &data, bool hasNext) override;
+	void read_included_file(QXmlStreamReader &reader, NodeContentGroup &data) override;
 
 private:
 	QFile _hdrFile;
@@ -24,6 +25,8 @@ private:
 	QTextStream _hdr;
 	QTextStream _src;
 
+	NodeContentGroup *findContentGroup(NodeContentGroup *cGrp, const QString &key);
+
 	void writeHeader(const SettingsType &settings);
 	void writeNodeElements(const NodeContentGroup &node, int intendent = 1);
 	void writeNode(const NodeType &node, int intendent = 1);