#include "stdafx.h"
#include "uv.h"
#include "HttpLiveServer.h"
#include "HlsWorker.h"
#include "LiveClient.h"

namespace HttpWsServer
{
 static map<string,CHlsWorker*> m_workerMapHls;
 static CriticalSection m_csHls;
 static map<string, uint64_t> m_lives;
	static FILE *m_fall = NULL;

 //////////////////////////////////////////////////////////////////////////

 CHlsWorker::CHlsWorker(string strCode)
 : m_strCode(strCode)
 , m_nType(0)
		, m_nID(0)
 , m_pLive(nullptr){
 m_nLastVistTime = time(NULL);

 m_pLive = LiveClient::GetWorker(strCode);
 if(m_pLive)
 m_pLive->AddHandle(this, HandleType::ts_handle, 1);

		m_fall = fopen("all.ts","wb");
 }

 CHlsWorker::~CHlsWorker()
 {
 if(m_pLive)
 m_pLive->RemoveHandle(this);

 Log::debug("CHttpWorker release");
 }

 bool CHlsWorker::AddConnect(pss_http_ws_live* pss)
 {
 //pss->pss_next = m_pPssList;
 m_pPssList = pss;

 return true;
 }

 bool CHlsWorker::DelConnect(pss_http_ws_live* pss)
 {
 //lws_ll_fwd_remove(pss_http_ws_live, pss_next, pss, m_pPssList);
 return true;
 }

 AV_BUFF CHlsWorker::GetHeader()
 {
 m_nLastVistTime = time(nullptr);
 AV_BUFF ret = {AV_TYPE::TS, NULL, 0};

 if(m_listTs.empty())
 return ret;

 stringstream ss;
 ss << "#EXTM3U\n#EXT-X-TARGETDURATION:1\n";
 int nNum = 0;
 //MutexShareLock lock(&m_rwLock);
 list<TS_BUFF>::reverse_iterator rIt = m_listTs.rbegin();
 while(rIt != m_listTs.rend() && nNum < 10)
 {
 if (nNum == 0)
 {
 ss << "#EXT-X-MEDIA-SEQUENCE:" << rIt->nID << "\n\n";
 }
 ss << "#EXTINF:1\n" << m_strCode << "/" << rIt->nID << ".ts\n";
 rIt++;
 nNum++;
 }
 string strM3U8 = ss.str();
 ret.nLen = strM3U8.size();
 ret.pData = (char*)malloc(ret.nLen);
 memcpy(ret.pData, strM3U8.c_str(), strM3U8.size());

 return ret;
 }

 AV_BUFF CHlsWorker::GetVideo(uint64_t id)
 {
 m_nLastVistTime = time(nullptr);
 //MutexShareLock lock(&m_rwLock);

 for (auto rIt = m_listTs.rbegin(); rIt != m_listTs.rend(); ++rIt)
 {
 if (rIt->nID == id)
 {
 return rIt->buff;
 }
 }

 AV_BUFF ret = {AV_TYPE::NONE,nullptr,0};
 return ret;
 }

 void CHlsWorker::play_answer(int ret, string error_info)
 {
 }

 void CHlsWorker::push_video_stream(AV_BUFF buff)
 {
 TS_BUFF ts;
 ts.nID = m_nID++;
 ts.buff = buff;
 m_listTs.push_back(ts);

		char tmp[100]={0};
		sprintf(tmp, "T%04d.ts", ts.nID);
		FILE* f = fopen(tmp, "wb");
		fwrite(ts.buff.pData, 1, ts.buff.nLen, f);
		fclose(f);
		fwrite(ts.buff.pData, 1, ts.buff.nLen, m_fall);
		fflush(m_fall);

 while (m_listTs.size() > 100)
 {
 TS_BUFF& firstTs = m_listTs.front();
 if (firstTs.buff.pData!=nullptr){
 free(firstTs.buff.pData);
 }
 m_listTs.pop_front();
 }

		//lws_start_foreach_llp_safe(pss_http_ws_live **, ppss, m_pPssList, pss_next) {
		//	lws_callback_on_writable((*ppss)->wsi);
 //} lws_end_foreach_llp_safe(ppss);
		m_pPssList = NULL;
 }

 void CHlsWorker::stop()
 {
 //视频源没有数据并超时
 Log::debug("no data recived any more, stopped");

 //断开所有客户端连接
 //lws_start_foreach_llp_safe(pss_http_ws_live **, ppss, m_pPssList, pss_next) {
 // lws_set_timeout((*ppss)->wsi, PENDING_TIMEOUT_CLOSE_SEND, LWS_TO_KILL_ASYNC);
 //} lws_end_foreach_llp_safe(ppss);
 }

 LiveClient::ClientInfo CHlsWorker::get_clients_info()
 {
 //vector<LiveClient::ClientInfo> ret;
 //lws_start_foreach_llp(pss_http_ws_live **, ppss, m_pPssList) {
 // LiveClient::ClientInfo info;
 // info.devCode = m_strCode;
 // info.connect = "HLS";
 // info.media = "TS";
 // info.clientIP= (*ppss)->clientIP;
 // ret.push_back(info);
 //} lws_end_foreach_llp(ppss, pss_next);
 //return ret;
 LiveClient::ClientInfo info;
 return info;
 }

 //////////////////////////////////////////////////////////////////////////

 CHlsWorker* CreatHlsWorker(string strCode)
 {
 Log::debug("CreatHlsWorker begin");
 CHlsWorker* pNew = new CHlsWorker(strCode);
 MutexLock lock(&m_csHls);
 m_workerMapHls.insert(make_pair(strCode, pNew));
 return pNew;
 }

 CHlsWorker* GetHlsWorker(string strCode)
 {
 //Log::debug("GetWorker begin");
 MutexLock lock(&m_csHls);
 auto itFind = m_workerMapHls.find(strCode);
 if (itFind != m_workerMapHls.end()) {
 return itFind->second;
 }
 //Log::error("GetWorker failed: %s",strCode.c_str());
 return nullptr;
 }

 bool DelHlsWorker(string strCode)
 {
 MutexLock lock(&m_csHls);
 auto itFind = m_workerMapHls.find(strCode);
 if (itFind != m_workerMapHls.end()) {
 SAFE_DELETE(itFind->second);
 m_workerMapHls.erase(itFind);
 return true;
 }
 Log::error("dosn't find worker object");
 return false;
 }

}