1 files changed, 12 insertions, 0 deletions

@@ -75,6 +75,13 @@ IdMappings ReadIdsFromYaml(const std::string& filename) {
     }
   }
 
+  if (document["endings"]) {
+    for (const auto& ending_it : document["endings"]) {
+      (*result.mutable_endings())[ending_it.first.as<std::string>()] =
+          ending_it.second.as<uint64_t>();
+    }
+  }
+
   if (document["special"]) {
     for (const auto& special_it : document["special"]) {
       (*result.mutable_special())[special_it.first.as<std::string>()] =
@@ -127,6 +134,11 @@ void WriteIdsAsYaml(const IdMappings& ids, const std::string& filename) {
     result["letters"][letter_name] = letter_id;
   });
 
+  OperateOnSortedMap(ids.endings(), [&result](const std::string& ending_name,
+                                              uint64_t ending_id) {
+    result["endings"][ending_name] = ending_id;
+  });
+
   OperateOnSortedMap(ids.special(), [&result](const std::string& special_name,
                                               uint64_t special_id) {
     result["special"][special_name] = special_id;