judge if the json_extention is .json

a1henu · May 1, 2024 · fbdb46b · fbdb46b
1 parent ef68dce
commit fbdb46b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/main.cpp b/main.cpp
@@ -9,8 +9,20 @@
 #include "utils.hpp"
 
 // use CitationPtr to represent a shared pointer to Citation
+bool hasJsonExtension(const std::string& filename) {
+    if (filename.size() < 5) {
+        return false;
+    }
+
+    return filename.substr(filename.size() - 5) == ".json";
+}
+
 std::unordered_map<std::string, CitationPtr> loadCitations(const std::string& filename) {
     // FIXME: load citations from file
+    if (!hasJsonExtension(filename)) {
+        std::exit(1);
+    }
+
     std::ifstream file{filename};
     nlohmann::json citationJson;