Split way and node validation for XMLFeature.

1 files changed, 16 insertions, 3 deletions

diff --git a/editor/xml_feature.cpp b/editor/xml_feature.cpp
index 610311f64b..3705e82a7b 100644
--- a/editor/xml_feature.cpp
+++ b/editor/xml_feature.cpp
@@ -57,6 +57,16 @@ void ValidateNode(pugi::xml_node const & node)
   if (!node.attribute(kTimestamp))
     MYTHROW(editor::XMLFeatureNoTimestampError, ("Node has no timestamp attribute"));
 }
+
+void ValidateWay(pugi::xml_node const & way)
+{
+  if (!way)
+    MYTHROW(editor::XMLFeatureNoNodeError, ("Document has no node"));
+
+  if (!way.attribute(kTimestamp))
+    MYTHROW(editor::XMLFeatureNoTimestampError, ("Way has no timestamp attribute"));
+}
+
 } // namespace
 
 namespace editor
@@ -73,20 +83,23 @@ XMLFeature::XMLFeature(Type const type)
 XMLFeature::XMLFeature(string const & xml)
 {
   m_document.load(xml.data());
-  ValidateNode(GetRootNode());
+  auto const r = GetRootNode();
+  r.name() == kNodeType ? ValidateNode(r) : ValidateWay(r);
 }
 
 XMLFeature::XMLFeature(pugi::xml_document const & xml)
 {
   m_document.reset(xml);
-  ValidateNode(GetRootNode());
+  auto const r = GetRootNode();
+  r.name() == kNodeType ? ValidateNode(r) : ValidateWay(r);
 }
 
 XMLFeature::XMLFeature(pugi::xml_node const & xml)
 {
   m_document.reset();
   m_document.append_copy(xml);
-  ValidateNode(GetRootNode());
+  auto const r = GetRootNode();
+  r.name() == kNodeType ? ValidateNode(r) : ValidateWay(r);
 }
 
 XMLFeature::Type XMLFeature::GetType() const