Add INIReader Sections and Keys methods (#186)

Author: Ishan09811

cpp/INIReader.cpp

@@ -56,7 +56,7 @@ long INIReader::GetInteger(const string& section, const string& name, long defau
     return end > value ? n : default_value;
 }
 
-INI_API int64_t INIReader::GetInteger64(const std::string& section, const std::string& name, int64_t default_value) const
+INI_API int64_t INIReader::GetInteger64(const string& section, const string& name, int64_t default_value) const
 {
     string valstr = Get(section, name, "");
     const char* value = valstr.c_str();
@@ -76,7 +76,7 @@ unsigned long INIReader::GetUnsigned(const string& section, const string& name,
     return end > value ? n : default_value;
 }
 
-INI_API uint64_t INIReader::GetUnsigned64(const std::string& section, const std::string& name, uint64_t default_value) const
+INI_API uint64_t INIReader::GetUnsigned64(const string& section, const string& name, uint64_t default_value) const
 {
     string valstr = Get(section, name, "");
     const char* value = valstr.c_str();
@@ -109,6 +109,30 @@ bool INIReader::GetBoolean(const string& section, const string& name, bool defau
         return default_value;
 }
 
+std::vector<string> INIReader::Sections() const
+{
+    std::set<string> sectionSet;
+    for (std::map<string, string>::const_iterator it = _values.begin(); it != _values.end(); ++it) {
+        size_t pos = it->first.find('=');
+        if (pos != string::npos) {
+            sectionSet.insert(it->first.substr(0, pos));
+        }
+    }
+    return std::vector<string>(sectionSet.begin(), sectionSet.end());
+}
+
+std::vector<string> INIReader::Keys(const string& section) const
+{
+    std::vector<string> keys;
+    string keyPrefix = MakeKey(section, "");
+    for (std::map<string, string>::const_iterator it = _values.begin(); it != _values.end(); ++it) {
+        if (it->first.compare(0, keyPrefix.length(), keyPrefix) == 0) {
+            keys.push_back(it->first.substr(keyPrefix.length()));
+        }
+    }
+    return keys;
+}
+
 bool INIReader::HasSection(const string& section) const
 {
     const string key = MakeKey(section, "");

cpp/INIReader.h

@@ -15,6 +15,8 @@
 #include <map>
 #include <string>
 #include <cstdint>
+#include <vector>
+#include <set>
 
 // Visibility symbols, required for Windows DLLs
 #ifndef INI_API
@@ -89,6 +91,12 @@ class INIReader
     // and valid false values are "false", "no", "off", "0" (not case sensitive).
     INI_API bool GetBoolean(const std::string& section, const std::string& name, bool default_value) const;
 
+    // Return a newly-allocated vector of all section names, in alphabetical order.
+    INI_API std::vector<std::string> Sections() const;
+
+    // Return a newly-allocated vector of keys in the given section, in alphabetical order.
+    INI_API std::vector<std::string> Keys(const std::string& section) const;
+
     // Return true if the given section exists (section must contain at least
     // one name=value pair).
     INI_API bool HasSection(const std::string& section) const;

examples/INIReaderExample.cpp

@@ -24,5 +24,20 @@ int main()
               << ", user.nose=" << reader.HasValue("user", "nose") << "\n";
     std::cout << "Has sections: user=" << reader.HasSection("user")
               << ", fizz=" << reader.HasSection("fizz") << "\n";
+    
+    std::cout << "Sections:\n";
+    std::vector<std::string> sections = reader.Sections();
+    for (std::vector<std::string>::const_iterator it = sections.begin(); it != sections.end(); ++it) {
+        std::cout << "- " << *it << "\n";
+    }
+    
+    for (std::vector<std::string>::const_iterator it = sections.begin(); it != sections.end(); ++it) {
+        std::cout << "Keys in section [" << *it << "]:\n";
+        std::vector<std::string> keys = reader.Keys(*it);
+        for (std::vector<std::string>::const_iterator kit = keys.begin(); kit != keys.end(); ++kit) {
+            std::cout << "- " << *kit << "\n";
+        }
+    }
+    
     return 0;
 }

examples/cpptest.txt

@@ -1,3 +1,14 @@
 Config loaded from 'test.ini': version=6, unsigned version=6, trillion=1000000000000, unsigned trillion=1000000000000, name=Bob Smith, [email protected], pi=3.14159, active=1
 Has values: user.name=1, user.nose=0
 Has sections: user=1, fizz=0
+Sections:
+- protocol
+- user
+Keys in section [protocol]:
+- version
+Keys in section [user]:
+- active
+- email
+- name
+- pi
+- trillion