server-l-zmg/IniFile.cpp

#include <assert.h>
#include <fstream>
#include <cstring>
#include <algorithm>
#include <stdio.h>
#include "IniFile.h"

#if _MSC_VER >=1400
	const std::wstring to_wcs( const std::string& src )
	{
		std::string prev = setlocale(LC_CTYPE,NULL);	// 保存原来的locale
		setlocale( LC_CTYPE, "chs" );					// 设置当前locale为chs，这在非简体中文平台上不可缺少

		size_t count1 = mbstowcs(NULL, src.c_str(), 0); // 计算新字符串长度
		std::wstring des(count1, L' ');
		size_t count2 = mbstowcs(&des[0], src.c_str(), count1 ); // 转化
		assert(count1 == count2);

		setlocale( LC_CTYPE, prev.c_str()); // 恢复到原来的locale
		return des;
	}
#endif

	IniFile::IniFile()
	{
		m_prevInsertIt = m_lines.end();
	}

	IniFile::~IniFile (void)
	{
		clear();
	}

	bool IniFile::open(const char* filename)
	{
		assert(filename);

		clear();
#if _MSC_VER >=1400
		std::wstring wcsfile = to_wcs(filename);
		std::ifstream fin(wcsfile.c_str());	
#else
		std::ifstream fin(filename);
#endif
		if(fin.is_open())
		{
			while(!fin.eof()) 
			{
				std::string tmp;
				getline(fin, tmp, '\n');
				
				if(fin.eof() && tmp.empty())
					break;

				// if *NIX file format, remove the CR
				if(tmp.length() >= 1 && tmp[tmp.length()-1] == '\r')
				{
					tmp.resize(tmp.length()-1);
				}
				parseLine(tmp);
			}
			makeIndex();
			return true;
		}
		return false;
	}

	bool IniFile::save(const char* filename)
	{
		assert(filename);
#if _MSC_VER >=1400
		std::wstring wcsfile = to_wcs(filename);
		std::ofstream fout(wcsfile.c_str());
#else
		std::ofstream fout(filename);
#endif
		if(fout.is_open())
		{
			std::list<LineFrame*>::iterator it = m_lines.begin();
			for(; it != m_lines.end(); ++it)
			{
				LineFrame* frame = *it;
				switch(frame->m_type) 
				{
				case TYPE_COMMENT:
					fout << frame->m_comment << std::endl;
					break;
				case TYPE_SECTION:
					fout << "[" << frame->m_section << "]" << frame->m_comment << std::endl;
					break;
				case TYPE_KEYVALUE:
					{
						if(frame->m_valueHasComment)
							fout << frame->m_key << "=\"" << frame->m_value << "\"" << frame->m_comment << std::endl;
						else
							fout << frame->m_key << "=" << frame->m_value << frame->m_comment << std::endl;
						
					}
					break;
				default:
					break;
				}
			}
			return true;
		}
		return false;
	}

	void IniFile::clear()
	{
		std::list<LineFrame*>::iterator it = m_lines.begin();
		for(; it != m_lines.end(); ++it)
			delete *it;
		m_lines.clear();
		m_lineindex.clear();
	}

	bool IniFile::readBool(const char* section, const char* key, bool dft)
	{
		assert(section);
		assert(key);

		LineFrame *frame = findNotEmptyFrame(section, key);
		if(frame)
		{
			if(ignoreCaseCompare(frame->m_value,"true"))
				return true;
            else if(ignoreCaseCompare(frame->m_value,"false"))
				return false;
			else
				return dft;
		}
		return dft;
	}

	int IniFile::readInt(const char* section, const char* key, int dft)
	{
		assert(section);
		assert(key);

		LineFrame *frame = findNotEmptyFrame(section, key);
		if(frame)
		{
			if ( frame->m_value.length() > 2 && frame->m_value[0] == '0' && frame->m_value[1] == 'x' )
			{
				int value = dft;
				sscanf(frame->m_value.c_str(), "0x%x", &value );
				return value;
			}
			else if ( frame->m_value.length() > 2 && frame->m_value[0] == '0' && frame->m_value[1] == 'X' )
			{
				int value = dft;
				sscanf(frame->m_value.c_str(), "0X%x", &value );
				return value;
			}
			else
			{
				int value = dft;
				sscanf(frame->m_value.c_str(), "%u", &value );
				return value;
			}
		}
		else
		{
			return dft;
		}
	}

	double IniFile::readDouble(const char* section, const char* key, double dft)
	{
		assert(section);
		assert(key);

		LineFrame *frame = findNotEmptyFrame(section, key);
		if(frame)
		{
			return atof(frame->m_value.c_str());			
		}
		else
		{
			return dft;
		}
	}

	const char* IniFile::readString(const char* section, const char* key, const char* dft)
	{
		assert(section);
		assert(key);

		LineFrame *frame = findFrame(section, key);
		if(frame)
		{
			return frame->m_value.c_str();
		}
		else
		{
			return dft;
		}
	}

	void IniFile::writeBool(const char* section, const char* key, bool val)
	{
		assert(section);
		assert(key);

		if(val)
			writeString(section, key, "true");
		else
			writeString(section, key, "false");
	}
	void IniFile::writeInt(const char* section, const char* key, int val)
	{
		assert(section);
		assert(key);

		char tmp[32];
		sprintf(tmp, "%d", val);
		writeString(section, key, tmp);
	}

	void IniFile::writeHex(const char* section, const char* key, uint32_t val)
	{
		assert(section);
		assert(key);

		char tmp[32];
		sprintf(tmp, "0x%X", val);
		writeString(section, key, tmp);
	}

	void IniFile::writeDouble(const char* section, const char* key, double val)
	{
		assert(section);
		assert(key);

		char tmp[32];
		sprintf(tmp, "%g", val);
		writeString(section, key, tmp);
	}

	void IniFile::writeString(const char* section, const char* key, const char* val)
	{
		assert(section);
		assert(key);
		assert(val);
        
		LineFrame *frame = findFrame(section, key);
		if(frame)
		{
			// already exist, update it!
			frame->m_value = val;
			frame->m_valueHasComment = hasComment(val);
		}
		else
		{	
			std::list<LineFrame*>::iterator it;
			// try cache, if failed then locate the end of this section
			if(m_prevSection == section && m_prevInsertIt != m_lines.end())
				it = m_prevInsertIt;
			else
				it = findSectionEnd(section);
			 
			// new key insert into near by section, it will speed up all write method;
			// std::list<LineFrame*>::iterator it = findSection(section);
			if(it != m_lines.end())
			{
				frame = new LineFrame;
				frame->m_type = TYPE_KEYVALUE;
				frame->m_key = key;
				frame->m_value = val;
				frame->m_valueHasComment = hasComment(val);
				m_lines.insert(++it, frame);

				m_prevSection = section;
				m_prevInsertIt = --it;
			}
			else
			{
				// no section yet! create it!
				frame = new LineFrame;
				frame->m_type = TYPE_SECTION;
				frame->m_section = section;
				m_lines.push_back(frame);

				frame = new LineFrame;
				frame->m_type = TYPE_KEYVALUE;
				frame->m_key = key;
				frame->m_value = val;
				frame->m_valueHasComment = hasComment(val);
				m_lines.push_back(frame);
			}
			// update index
			m_lineindex[section][key] = frame;
		}
	}

	bool IniFile::hasSection(const char* section)
	{
		return m_lineindex.find(section) != m_lineindex.end();
	}

	bool IniFile::deleteSection(const char* section)
	{
		m_lineindex.erase(section);

		std::list<LineFrame*>::iterator it = findSection(section);
		if(it != m_lines.end())
		{
			// delete the section at first
			delete *it;
			m_lines.erase(it++);
			// and delete all key-value until occurs next section(or end of file)
			while(it != m_lines.end() && (*it)->m_type != TYPE_SECTION)
			{
				delete *it;
				m_lines.erase(it++);
			}
			return true;
		}
		return false;
	}
	bool IniFile::deleteKey(const char* section, const char* key)
	{
		assert(section);
		assert(key);

		std::list<LineFrame*>::iterator it = findSection(section);
		if (it == m_lines.end())
		{
			return false;
		}
		// find from next line
		++it;
		for( ; it != m_lines.end(); ++it)
		{
			if((*it)->m_type == TYPE_SECTION)
			{	// occur next section
				break;
			}
			if((*it)->m_type == TYPE_KEYVALUE &&
                ignoreCaseCompare((*it)->m_key,key))
			{
				m_lineindex[section].erase(key);
				delete *it;
				m_lines.erase(it);
				return true;
			}
		}
		return false;
	}

	void IniFile::readSection(const char* section, std::map<std::string, std::string>& kvset)
	{
		assert(section);

		std::list<LineFrame*>::iterator it = findSection(section);
		if(it != m_lines.end())
			++it;
		for( ; it != m_lines.end(); ++it)
		{
			if((*it)->m_type == TYPE_SECTION)
			{	// occur next section
				break;
			}
			if((*it)->m_type == TYPE_KEYVALUE)
			{
				kvset.insert(std::make_pair((*it)->m_key, (*it)->m_value));
			}
		}
	}

	void IniFile::parseLine(const std::string& line) 
	{
		LineFrame* frame = new LineFrame;
		m_lines.push_back(frame);
		frame->m_type = TYPE_COMMENT; // dft is comment

		// 1. check full commented line 
		if(isCommentLine(line))
		{
			frame->m_type = TYPE_COMMENT;
			frame->m_comment = line;
			return ;
		}

		// 2. try the find section
		std::string::size_type first = line.find('[');
		std::string::size_type last = line.rfind(']');;

		if(first != std::string::npos && last != std::string::npos && first != last+1)
		{
			frame->m_section = line.substr(first+1,last-first-1);
			if(hasComment(frame->m_section))
			{
				frame->m_type = TYPE_COMMENT;
				frame->m_comment = line;
			}
			else
			{
				frame->m_type = TYPE_SECTION;
				// remember the possible comments
				frame->m_comment = line.substr(last+1);
			}
			return ;
		}

		// 3. find key-value pattern
		first = line.find('=');
		if(first == std::string::npos)
			return ;

		std::string strtmp1 = line.substr(0, first);
		if(hasComment(strtmp1))
		{
			// such case
			// something # key=value
			// we consider such meaningless line as comment
			frame->m_type = TYPE_COMMENT;
			frame->m_comment = line;
			return ;
		}

		std::string key = constTrim(strtmp1);
		if(key.empty())
		{
			// such case
			// = value
			frame->m_type = TYPE_COMMENT;
			frame->m_comment = line;
			return ;
		}

		frame->m_type = TYPE_KEYVALUE;
		// remember the original form
		frame->m_key = key;
		std::string strtmp2 = line.substr(first+1);
		std::string::size_type comment = findComment(strtmp2);
		std::string value = constTrim(strtmp2.substr(0, comment));
	
		if(value.size()>1 && value[0] == '\"' && value[value.size()-1] == '\"')
		{
			frame->m_value = value.substr(1, value.size()-2);
			frame->m_valueHasComment = true;
		}
		else
		{
			frame->m_value = value;
		}

		if(comment != std::string::npos)
			frame->m_comment = constTrim(strtmp2.substr(comment));

		return ;
	}

	std::string::size_type IniFile::findComment(const std::string& line)
	{
		bool inString = false;
		for(size_t i=0; i<line.size(); i++)
		{
			if(line[i] == '\"')
			{
				inString = true;
			}
			
			if(inString)
				continue;

			if(line[i] == '/')
			{
				// "//"
				if(i+1 != line.length() && line[i+1] == '/')
					return i;
			}
			else if(line[i] == '#' || line[i] == ';')
			{
				return i;
			}
		}
		return std::string::npos;

	}

	bool IniFile::isCommentLine(const std::string& line)
	{
		// skip the space and tab chars
		std::string::size_type charBegin = line.find_first_not_of("\t ");
		return charBegin != std::string::npos && charBegin == findComment(line);
	}

	bool IniFile::hasComment(const std::string& line)
	{
		return findComment(line) != std::string::npos;
	}

	std::list<LineFrame*>::iterator IniFile::findSection(const char* section)
	{
		std::list<LineFrame*>::iterator it = m_lines.begin();
		for( ; it != m_lines.end(); ++it)
		{
			LineFrame* frame = *it;
			if(frame->m_type == TYPE_SECTION 
				&& ignoreCaseCompare(frame->m_section,section))
			{
				break;
			}
		}
		return it;
	}
	
	std::list<LineFrame*>::iterator IniFile::findSectionEnd(const char* section)
	{
		std::list<LineFrame*>::iterator it = findSection(section);
		if(it != m_lines.end())
		{
			// find from next line
			++it;
			while(it != m_lines.end() && (*it)->m_type != TYPE_SECTION)
			{
				++it;
			}
			// the end of section is one step back of next section line(or end())
			--it;
		}
		return it;
	}

	LineFrame* IniFile::findFrame(const char*section, const char* key)
	{
		KeyValueIndex::iterator it = m_lineindex.find(section);
		if(it != m_lineindex.end())
		{
			KeyValueMap::iterator it2 = it->second.find(key);
			if(it2 != it->second.end())
				return it2->second;
		}
		return 0;
	}

	LineFrame* IniFile::findNotEmptyFrame(const char*section, const char* key)
	{
		LineFrame* frame = findFrame(section, key);
		if(frame && !frame->m_value.empty())
			return frame;
		else
			return 0;
	}

	void IniFile::makeIndex()
	{
		std::list<LineFrame*>::iterator it = m_lines.begin();
		LineFrame* frame = NULL;
		KeyValueIndex::iterator kvmit = m_lineindex.end();
		while(it != m_lines.end())
		{
			frame = *it;
			++it;

			// 找到了一个SECTION的开始
			if(frame->m_type == TYPE_SECTION)
			{
				kvmit = (m_lineindex.insert(std::make_pair(frame->m_section, KeyValueMap()))).first;
			}
			else if(frame->m_type == TYPE_KEYVALUE && kvmit != m_lineindex.end())
			{
				(kvmit->second)[frame->m_key]= frame;
			}
		}
	}
#ifdef __TEST
int main()
{
    IniFile ini;
    ini.open("config.ini");
    std::string host = ini.readString("db","host","");
    std::string url = ini.readString("serversetting","web1","");
    if(!host.empty())
      printf("%s\n",host.c_str());
    printf("%s\n",url.c_str());
    ini.writeInt("dbdb","timeout",123);
    ini.writeInt("db","a",123123);
    ini.writeInt("db01","a",123);
    ini.writeString("xxxx","name","lemon");
    printf("tol:%d\n",ini.readInt("serversetting","timeout",0));
    //clear all
    //ini.clear(); 
    ini.save("config.ini");
}
#endif