#include "common/pch.h"

#include "ConfigUtil.h"
#include <iostream>
#include <fstream>
#include "../common/JsonUtil.h"
#include "../common/TimeUtil.h"


bool fileExists(const std::string& filename) {
	std::ifstream file(filename);
	return file.good(); // Èç¹ûÎļþ³É¹¦´ò¿ª£¬Ôò·µ»Ø true
}

string ConfigUtil::getConfigPath()
{
	wchar_t* appDataPath = nullptr;
	CString path;
	// »ñÈ¡APPDataĿ¼·¾¶
	if (SUCCEEDED(SHGetKnownFolderPath(FOLDERID_RoamingAppData, 0, NULL, &appDataPath))) {
		std::wcout << L"APPDataĿ¼·¾¶: " << appDataPath << std::endl;
		path.Append(appDataPath);
		CoTaskMemFree(appDataPath);  // ÊÍ·ÅÄÚ´æ
	}
	else {
		std::wcerr << L"ÎÞ·¨»ñÈ¡APPDataĿ¼·¾¶" << std::endl;
		return "config.cfg";
	}
	path.Append(L"\\¼ÝÊ»²Õ");
	// ÅжÏÎļþ¼ÐÊÇ·ñ´æÔÚ
	DWORD fileAttributes = GetFileAttributesW(path);
	if (fileAttributes == INVALID_FILE_ATTRIBUTES) {
		CreateDirectoryW(path, NULL);
	}
	string fpath = StringUtil::cstring2String(path);
	fpath.append("\\config.cfg");
	return  fpath.c_str();
}

void ConfigUtil::readConfig(libconfig::Config& config)
{

	string strConfPath = getConfigPath();
	if (!fileExists(strConfPath)) {
		std::ofstream file(strConfPath);
		file.is_open();
	}

	//½â¶ÁÅäÖÃÎļþ
	try {
		config.readFile(strConfPath.c_str());
	}
	catch (const libconfig::FileIOException& fioex) {
		std::cerr << "I/O exception while reading the file." << std::endl;
	}
	catch (const libconfig::ParseException& pex) {
		std::cerr << "Parse error at " << pex.getFile() << ":" << pex.getLine()
			<< " - " << pex.getError() << std::endl;
	}
}

void ConfigUtil::writeConfig(libconfig::Config& config)
{
	config.writeFile(getConfigPath().c_str());
}

int ConfigUtil::readIntConfig(const char* key)
{

	libconfig::Config mConfig;
	readConfig(mConfig);
	libconfig::Setting& root = mConfig.getRoot();
	if (root.exists(key)) {
		int val;
		root.lookupValue(key, val);
		return val;
	}
	throw string("ÉÐδ»ñÈ¡µ½ÄÚÈÝ");
}

string ConfigUtil::readStringConfig(const char* key)
{
	libconfig::Config mConfig;
	readConfig(mConfig);
	libconfig::Setting& root = mConfig.getRoot();
	if (root.exists(key)) {
		//string val;
		const char* val;
		root.lookupValue(key, val);
		return string(val);
	}
	throw string("ÉÐδ»ñÈ¡µ½ÄÚÈÝ");
}



void ConfigUtil::setIntConfig(const char* key, int val) {
	libconfig::Config mConfig;
	readConfig(mConfig);
	libconfig::Setting& root = mConfig.getRoot();
	if (root.exists(key)) {
		root[key] = val;
	}
	else {
		root.add(key, libconfig::Setting::TypeInt) = val;
	}
	writeConfig(mConfig);
}
void ConfigUtil::setStringConfig(const char* key, string val) {
	libconfig::Config mConfig;
	readConfig(mConfig);
	libconfig::Setting& root = mConfig.getRoot();
	if (root.exists(key)) {
		root[key] = val.c_str();
	}
	else {
		root.add(key, libconfig::Setting::TypeString) = val.c_str();
	}
	writeConfig(mConfig);
}

list<string> ConfigUtil::getKeys()
{
	libconfig::Config mConfig;
	readConfig(mConfig);
	libconfig::Setting& root = mConfig.getRoot();
	list<string> keys;
	for (auto iter = root.begin(); iter != root.end(); ++iter) {
		keys.push_back(iter->getName());
	}
	return keys;
}

void ConfigUtil::delKey(string key)
{
	libconfig::Config mConfig;
	readConfig(mConfig);
	libconfig::Setting& root = mConfig.getRoot();
	root.remove(key.c_str());
	writeConfig(mConfig);
}

void ConfigUtil::setBuildServer(ServerInfo* server)
{
	rapidjson::StringBuffer buf;
	rapidjson::PrettyWriter<rapidjson::StringBuffer> writer(buf);
	writer.StartObject();
	writer.Key("ip");
	writer.String(server->ip.c_str());
	writer.Key("port");
	writer.Int(server->port);
	writer.Key("userName");
	writer.String(server->userName.c_str());
	writer.Key("pwd");
	writer.String(server->pwd.c_str());
	writer.EndObject();
	const char* json_content = buf.GetString();
	setStringConfig("build_server_info", json_content);
}

ServerInfo* ConfigUtil::getBuildServer()
{
	try {
		string result = readStringConfig("build_server_info");
		auto doc = JsonUtil::parseUTF8(result);
		ServerInfo* server = new ServerInfo();
		server->ip = doc["ip"].GetString();
		server->port = doc["port"].GetInt();
		server->userName = doc["userName"].GetString();
		server->pwd = doc["pwd"].GetString();
		return server;
	}
	catch (...) {
		return nullptr;
	}
	
}

void ConfigUtil::setProjectBuildInfo(ProjectBuildInfo* server, int index)
{
	string key = string("project_build_info-").append(to_string(index));
	rapidjson::StringBuffer buf;
	rapidjson::PrettyWriter<rapidjson::StringBuffer> writer(buf);
	writer.StartObject();
	writer.Key("name");
	writer.String(server->name.c_str());
	writer.Key("dir");
	writer.String(server->dir.c_str());
	writer.Key("scriptFile");
	writer.String(server->scriptFile.c_str());
	writer.Key("outputDir");
	writer.String(server->outputDir.c_str());
	writer.Key("outputFile");
	writer.String(server->outputFile.c_str());
	writer.Key("targetServer");
	writer.StartObject();
	writer.Key("ip");
	writer.String(server->targetServer.ip.c_str());
	writer.Key("port");
	writer.Int(server->targetServer.port);
	writer.Key("userName");
	writer.String(server->targetServer.userName.c_str());
	writer.Key("pwd");
	writer.String(server->targetServer.pwd.c_str());

	writer.EndObject();
	writer.Key("targetDir");
	writer.String(server->targetDir.c_str());

	writer.Key("targetAppFile");
	writer.String(server->targetAppFile.c_str());

	writer.Key("targetAppRestartFile");
	writer.String(server->targetAppRestartFile.c_str());
	

	writer.EndObject();
	const char* json_content = buf.GetString();
	setStringConfig(key.c_str(), json_content);
}

ProjectBuildInfo* ConfigUtil::getProjectBuildInfo(int index)
{

	try {
		string key = string("project_build_info-").append(to_string(index));
		string result = readStringConfig(key.c_str());
		auto doc = JsonUtil::parseUTF8(result);
		ProjectBuildInfo* server = new ProjectBuildInfo();
		if (doc.HasMember("name")) {
			server->name = doc["name"].GetString();
		}

		server->dir = doc["dir"].GetString();
		server->scriptFile = doc["scriptFile"].GetString();
		server->outputDir = doc["outputDir"].GetString();
		server->outputFile = doc["outputFile"].GetString();
		auto targetServer =  doc["targetServer"].GetObjectW();
		server->targetServer.ip = targetServer["ip"].GetString();
		server->targetServer.port = targetServer["port"].GetInt();
		server->targetServer.userName = targetServer["userName"].GetString();
		server->targetServer.pwd = targetServer["pwd"].GetString();
		server->targetDir = doc["targetDir"].GetString();
		if (doc.HasMember("targetAppFile")) {
            server->targetAppFile = doc["targetAppFile"].GetString();
		}
		if (doc.HasMember("targetAppRestartFile")) {
			server->targetAppRestartFile = doc["targetAppRestartFile"].GetString();
		}

		return server;
	}
	catch (...) {
		return nullptr;
	}

	return nullptr;
}

void ConfigUtil::setLocalCacheDir(string dir)
{
	setStringConfig("local_cache_dir", dir);

}

string ConfigUtil::getLocalCacheDir()
{
	try {
		string result = readStringConfig("local_cache_dir");
	    return result;
	}
	catch (...) {
		return "";
	}

}