libhv/utils/iniparser.h

#ifndef HV_INI_PARSER_H_
#define HV_INI_PARSER_H_

#include <list>
#include <string>
using std::string;

#define DEFAULT_INI_COMMENT "#"
#define DEFAULT_INI_DELIM   "="

/**********************************
# div

[section]

key = value # span

# div
***********************************/

class IniNode {
 public:
    enum Type {
        INI_NODE_TYPE_UNKNOWN,
        INI_NODE_TYPE_ROOT,
        INI_NODE_TYPE_SECTION,
        INI_NODE_TYPE_KEY_VALUE,
        INI_NODE_TYPE_DIV,
        INI_NODE_TYPE_SPAN,
    } type;
    string  label; // section|key|comment
    string  value;
    std::list<IniNode*>    children;

    virtual ~IniNode() {
        for (auto pNode : children) {
            if (pNode) {
                delete pNode;
            }
        }
        children.clear();
    }

    void Add(IniNode* pNode) {
        children.push_back(pNode);
    }

    void Del(IniNode* pNode) {
        for (auto iter = children.begin(); iter != children.end(); ++iter) {
            if ((*iter) == pNode) {
                delete (*iter);
                children.erase(iter);
                return;
            }
        }
    }

    IniNode* Get(const string& label, Type type = INI_NODE_TYPE_KEY_VALUE) {
        for (auto pNode : children) {
            if (pNode->type == type && pNode->label == label) {
                return pNode;
            }
        }
        return NULL;
    }
};

class IniSection : public IniNode {
public:
    IniSection() : IniNode(), section(label) {
        type = INI_NODE_TYPE_SECTION;
    }
    string &section;
};

class IniKeyValue : public IniNode {
public:
    IniKeyValue() : IniNode(), key(label) {
        type = INI_NODE_TYPE_KEY_VALUE;
    }
    string &key;
};

class IniComment : public IniNode {
public:
    IniComment() : IniNode(), comment(label) {
    }
    string &comment;
};

class IniParser {
 public:
    IniParser();
    ~IniParser();

    void SetIniComment(const string& comment) {_comment = comment;}
    void SetIniDilim(const string& delim) {_delim = delim;}

    int LoadFromFile(const char* filepath);
    int LoadFromMem(const char* data);
    int Unload();

    void DumpString(IniNode* pNode, string& str);
    string DumpString();
    int Save();
    int SaveAs(const char* filepath);

    string GetValue(const string& key, const string& section = "");
    void   SetValue(const string& key, const string& value, const string& section = "");

    // T = [bool, int, float]
    template<typename T>
    T Get(const string& key, const string& section = "", T defvalue = 0);

    // T = [bool, int, float]
    template<typename T>
    void Set(const string& key, const T& value, const string& section = "");

 private:
    string  _comment;
    string  _delim;
    string  _filepath;

    IniNode* root_;
};

#endif // HV_INI_PARSER_H_