myvtm/Module/mod_localmediaplay/AdvertManage/MediaManage.cpp

#include "stdafx.h"
#include "mediaManage.h"
#include <set>
#include <algorithm>
#include "SpBase.h"



#define DEFAULT_DOWNLOAD_PATH	"D:/rvc/Downloads/package"
#define DEFAULT_RESOURSE_PATH	"D:/rvc/addata/Video"
#define DEFAULT_DOWNLOAD_PATH_C	"C:/rvc/Downloads/package"
#define DEFAULT_RESOURSE_PATH_C	"C:/rvc/addata/Video"
#define DEFAULT_HEAD_CONTAIN	"银行"
#define DEFAULT_FEN_CONTAIN		"分行"
#define DEFAULT_NETWORK_CONTAIN	"支行"
#define DEFAULT_NETWORK2_CONTAIN "网点"


mediaManage::mediaManage()
{
	m_curParse.clear();
	if (checkDirExist(DEFAULT_RESOURSE_PATH))
	{
		Dbg("Set defaultAdvertPath:%s, defaultDownloadPath:%s", DEFAULT_RESOURSE_PATH, DEFAULT_DOWNLOAD_PATH);
		setDefaultAddvertPath(string(DEFAULT_RESOURSE_PATH));
		setDefaultDownloadPath(string(DEFAULT_DOWNLOAD_PATH));
	}
	else
	{
		Dbg("Set defaultAdvertPath:%s, defaultDownloadPath:%s", DEFAULT_RESOURSE_PATH_C, DEFAULT_DOWNLOAD_PATH_C);
		setDefaultAddvertPath(string(DEFAULT_RESOURSE_PATH_C));
		setDefaultDownloadPath(string(DEFAULT_DOWNLOAD_PATH_C));
	}
}

void mediaManage::setDefaultAddvertPath(string addvertPath){ m_addvertPath = addvertPath; }
void mediaManage::setDefaultDownloadPath(string downloadPath){ m_downloadPath = downloadPath; }

void mediaManage::GetPlayList(vector<ResourceParse> &resourceList)
{
	resourceList.clear();
	for (vector<ParseInfo>::iterator i = m_curParse.begin(); i != m_curParse.end(); i++)
	{
		for (vector<ResourceParse>::iterator j = i->m_mediaList.begin(); j != i->m_mediaList.end(); j++)
		{
			if (checkInVaildTime(j->vaildTime, true) && checkInPlayTime(j->playTime, true))
				resourceList.push_back(*j);
		}
	}
}

void mediaManage::GetPlayListByLocal(vector<ResourceParse>& resourceList)
{
	resourceList.clear();

#ifdef RVC_OS_WIN
	for each (auto i in m_localList)
	{
		if (checkInVaildTime(i.vaildTime, true) && checkInPlayTime(i.playTime, true))
			resourceList.push_back(i);
	}
#else
	for (vector<ResourceParse>::iterator it = m_localList.begin(); it < m_localList.end(); it++)
	{
		resourceList.push_back(*it);
	}
#endif // _WIN32
}

bool mediaManage::InitResourseList(vector<string> resourceList)
{
	/*
	（1）原resourseList不存在，初始化资源
	（2）原resourseList存在，删除失效资源，添加新资源
	
	求差集
	*/
	vector<string> needDelete(10), needAdd(10);
	set<string> oldResourceList, newResourseList;
	//get the old resource List
	for (vector<ParseInfo>::iterator j = m_curParse.begin(); j != m_curParse.end(); j++)
		oldResourceList.insert(j->m_resourceName);
	for (vector<string>::iterator i = resourceList.begin(); i != resourceList.end(); i++)
		newResourseList.insert(*i);

	auto deletePos = set_difference(oldResourceList.begin(), oldResourceList.end(), newResourseList.begin(), newResourseList.end(), needDelete.begin());//need to delete
	needDelete.resize(deletePos - needDelete.begin());//重新确定needDelete大小 
	auto addPos = set_difference(newResourseList.begin(), newResourseList.end(), oldResourceList.begin(), oldResourceList.end(), needAdd.begin());//need to add
	needAdd.resize(addPos - needAdd.begin());//重新确定needAdd大小 

	DeleteResourceList(needDelete);
	AddResourceList(needAdd);

	return true;
}

bool mediaManage::AddResourceList(vector<string> resourceList)
{
	if (0 == resourceList.size())
		return true;

	for (vector<string>::iterator i = resourceList.begin(); i != resourceList.end(); i++)
	{
		int pos = i->find(".zip");
		if (-1 == pos)
			continue;
		string dirName = i->substr(0, pos);
		string fileName = m_downloadPath + SPLIT_SLASH_STR + *i;
		string dirPath = m_addvertPath + SPLIT_SLASH_STR + dirName;
		Dbg("dirName:%s, fileName:%s, dirPath:%s", dirName.c_str(), fileName.c_str(), dirPath.c_str());

		wstring wfileName, wdirPath;
		StringToWstring(wfileName, fileName);
		StringToWstring(wdirPath, dirPath);
		//(1)已解压；（2）压缩文件不存在；（3）解压失败

		if (!checkFileExist(dirPath))
		{
			Dbg("dir %s not exist!", dirPath.c_str());
			continue;//压缩文件夹不存在
		}
/*
		if (!checkFileExist(dirPath) && createDir(dirPath) && !Unzip2Folder((BSTR)wfileName.c_str(), (BSTR)wdirPath.c_str()))
		{
			Dbg("unzip %s to %s fail!", fileName.c_str(), dirPath.c_str());
			continue;
		}
		*/
		string configPath = "";
		if (!findVaildConfig(dirPath, configPath))
		{
			Dbg("can not find config file in %s//%s", dirPath.c_str(), configPath.c_str());
			continue;//配置文件不存在
		}

		vector<ResourceParse> mediaList;
		if (!parseResourceIni(configPath.c_str(), mediaList))
		{
			Dbg("parse config %s fail!", configPath.c_str());
			continue;//配置文件读取失败
		}

		string resourcePath = dirPath + SPLIT_SLASH_STR;
/*		if (!getUniqueDir(dirPath, resourcePath))
		{
			Dbg("error:find multi resource dir!");
			continue;//未找到唯一得广告目录
		}
		resourcePath = dirPath + "\\" + resourcePath + "\\";
		*/
		for (vector<ResourceParse>::iterator j = mediaList.begin(); j != mediaList.end(); j++)
			j->resourcePath = resourcePath;

		ParseInfo resourseInfo(*i, mediaList);
		m_curParse.push_back(resourseInfo);//modify resourse list
	}

	return true;
}

void mediaManage::InitResourceListByLocal()
{
	string headFile = m_addvertPath + SPLIT_SLASH_STR + HEADINI_NAME;
	string branchFile = m_addvertPath + SPLIT_SLASH_STR + BRANCHINI_NAME;
	string networkFile = m_addvertPath + SPLIT_SLASH_STR + NETWORKINI_NAME;

	vector<ResourceParse> headList, branchList, networkList, allList;

	if (!checkFileExist(headFile))
		Dbg("head config %s not exist!", headFile.c_str());
	else if (!parseResourceIni(headFile.c_str(), headList))
		Dbg("parse head config %s fail!", headFile.c_str());

	if (!checkFileExist(branchFile))
		Dbg("network  config %s not exist!", networkFile.c_str());
	else if (!parseResourceIni(branchFile.c_str(), branchList))
		Dbg("parse branch config %s fail!", branchFile.c_str());

	if (!checkFileExist(networkFile))
		Dbg("network config %s not exist!", networkFile.c_str());
	else if (!parseResourceIni(networkFile.c_str(), networkList))
		Dbg("parse network config %s fail!", networkFile.c_str());

	m_localList.clear();
#ifdef RVC_OS_WIN
	for each (auto i in headList)
		m_localList.push_back(i);
	for each (auto i in branchList)
		m_localList.push_back(i);
	for each (auto i in networkList)
		m_localList.push_back(i);
#else
	for (vector<ResourceParse>::iterator it = headList.begin(); it < headList.end(); ++it)
	{
		m_localList.push_back(*it);
	}

	for (vector<ResourceParse>::iterator it = branchList.begin(); it < branchList.end(); ++it)
	{
		m_localList.push_back(*it);
	}

	for (vector<ResourceParse>::iterator it = networkList.begin(); it < networkList.end(); ++it)
	{
		m_localList.push_back(*it);
	}
#endif // RVC_OS_WIN

	std::stable_sort(m_localList.begin(), m_localList.end(), [](const ResourceParse& a, const ResourceParse& b) {
		return a.priority > b.priority;
		});//按priority排序


	for (auto i = m_localList.begin(); i != m_localList.end(); i++)
	{
		auto pos = i->videoNames.find(SPLIT_SLASH_STR);
		if (pos > 0)
		{
			i->resourcePath = m_addvertPath + SPLIT_SLASH_STR + i->videoNames.substr(0, pos + 1);
			i->videoNames = i->videoNames.substr(pos + 1, i->videoNames.length() - pos - 1);
		}
		else
			i->resourcePath = m_addvertPath + SPLIT_SLASH_STR;
	}

}

void mediaManage::DeleteResourceList(vector<string> resourceList)
{//不删除文件夹中文件，防止占用问题，定期清理
	if (0 == resourceList.size())
		return;

	for (vector<string>::iterator i = resourceList.begin(); i != resourceList.end(); i++)
	{
		for (vector<ParseInfo>::iterator j = m_curParse.begin(); j != m_curParse.end(); j++)
		{
			if (*i == j->m_resourceName)
			{
				m_curParse.erase(j);//迭代器失效
				break;
			}
		}
	}
}

bool mediaManage::findVaildConfig(string dirPath, string &configPath)
{
	string headFile = dirPath + SPLIT_SLASH_STR + HEADINI_NAME;
	string branchFile = dirPath + SPLIT_SLASH_STR + BRANCHINI_NAME;
	string networkFile = dirPath + SPLIT_SLASH_STR + NETWORKINI_NAME;

	if (checkFileExist(headFile))
	{
		configPath = headFile;
		return true;
	}
	else if (checkFileExist(branchFile))
	{
		configPath = branchFile;
		return true;
	}
	else if (checkFileExist(networkFile))
	{
		configPath = networkFile;
		return true;
	}
	else
		return false;
}

void mediaManage::clearOutdataResource()
{
	vector<string> dirInfo, needDelete(10);
	set<string> oldResourceList, newResourseList;

	try
	{
		getDirs(m_addvertPath, dirInfo);

		for (vector<ParseInfo>::iterator i = m_curParse.begin(); i != m_curParse.end(); i++)
		{
			int pos = i->m_resourceName.find(".zip");
			if (-1 == pos)
				continue;
			newResourseList.insert(i->m_resourceName.substr(0, pos));
		}

		for (vector<string>::iterator i = dirInfo.begin(); i != dirInfo.end(); i++)
		{
			if(-1 != i->find(DEFAULT_HEAD_CONTAIN) || -1 != i->find(DEFAULT_FEN_CONTAIN) || -1 != i->find(DEFAULT_NETWORK_CONTAIN) || -1 != i->find(DEFAULT_NETWORK2_CONTAIN))
				oldResourceList.insert(*i);
		}

		auto deletePos = set_difference(oldResourceList.begin(), oldResourceList.end(), newResourseList.begin(), newResourseList.end(), needDelete.begin());//need to delete
		needDelete.resize(deletePos - needDelete.begin());//重新确定needDelete大小 


		for (vector<string>::iterator i = needDelete.begin(); i != needDelete.end(); i++)
		{
			string path = m_addvertPath + SPLIT_SLASH_STR + *i;
			Dbg("remove Dir %s %s!", *i, removeDir(path) ? "success" : "fail");
		}
	}
	catch (exception* e)
	{
		Dbg("remove Dir exception:%s", e->what());
	}

}