#include "WriterContext.h"
#include "../HWPFile.h"
#include "../HWPXFile.h"

#include <sstream>
#include <iomanip>

namespace HWP
{
CWriterContext::CWriterContext()
	: m_eType(EHanType::NONE), m_pHWPFile(nullptr), m_pHWPXFile(nullptr)
{}

CWriterContext::~CWriterContext()
{
	if (nullptr != m_pHWPFile)
		delete m_pHWPFile;

	if (nullptr != m_pHWPXFile)
		delete m_pHWPXFile;
}

void CWriterContext::Clear()
{
	m_eType = EHanType::NONE;

	if (nullptr != m_pHWPFile)
	{
		delete m_pHWPFile;
		m_pHWPFile = nullptr;
	}

	if (nullptr != m_pHWPXFile)
	{
		delete m_pHWPXFile;
		m_pHWPXFile = nullptr;
	}
}

EHanType CWriterContext::GetType()
{
	return m_eType;
}

VECTOR<const CHWPSection*> CWriterContext::GetSections()
{
	switch(m_eType)
	{
		case EHanType::NONE: break;
		case EHanType::HWP:
		{
			if (nullptr != m_pHWPFile)
				return m_pHWPFile->GetSections();

			break;
		}
		case EHanType::HWPX:
		{
			if(nullptr != m_pHWPXFile)
				return m_pHWPXFile->GetSections();

			break;
		}
	}

	return VECTOR<const CHWPSection*>();
}

EHanType CWriterContext::DetectHancom(const HWP_STRING& sPathToFile)
{
	bool bDetected = false;

	CHWPFile* pHwpTemp = new CHWPFile(sPathToFile);
	if (nullptr != pHwpTemp)
	{
		if (pHwpTemp->Detect())
		{
			bDetected = true;
			pHwpTemp->Close();
		}

		delete pHwpTemp;
	}

	if (bDetected)
		return EHanType::HWP;

	CHWPXFile* pHwpxTemp = new CHWPXFile(sPathToFile);
	if (nullptr != pHwpxTemp)
	{
		if (pHwpxTemp->Detect())
		{
			bDetected = true;
			pHwpxTemp->Close();
		}

		delete pHwpxTemp;
	}

	if (bDetected)
		return EHanType::HWPX;

	return EHanType::NONE;
}

bool CWriterContext::Detect()
{
	switch(m_eType)
	{
		case EHanType::HWP:
		{
			if (nullptr == m_pHWPFile)
				return false;

			return m_pHWPFile->Detect();
		}
		case EHanType::HWPX:
		{
			if (nullptr == m_pHWPXFile)
				return false;

			return m_pHWPXFile->Detect();
		}
		case EHanType::NONE:
			return false;
	}
}

bool CWriterContext::Open(const HWP_STRING& sPathToFile, EHanType eHanType)
{
	Clear();

	m_eType = eHanType;

	switch (m_eType)
	{
		case EHanType::HWP:
		{
			m_pHWPFile = new CHWPFile(sPathToFile);

			if (nullptr == m_pHWPFile)
				return false;

			return m_pHWPFile->Open();
		}
		case EHanType::HWPX:
		{
			m_pHWPXFile = new CHWPXFile(sPathToFile);

			if (nullptr == m_pHWPXFile)
				return false;

			return m_pHWPXFile->Open();
		}
		case EHanType::NONE:
			break;
	}

	return false;
}

void CWriterContext::Close()
{
	switch (m_eType)
	{
		case EHanType::HWP:
		{
			if (nullptr != m_pHWPFile)
				m_pHWPFile->Close();
			break;
		}
		case EHanType::HWPX:
		{
			if (nullptr != m_pHWPXFile)
				m_pHWPXFile->Close();
			break;
		}
		case EHanType::NONE:
			break;
	}
}

const CHWPDocInfo* CWriterContext::GetDocInfo()
{
	switch (m_eType)
	{
		case EHanType::HWP:
		{
			if (nullptr == m_pHWPFile)
				return nullptr;

			return m_pHWPFile->GetDocInfo();
		}
		case EHanType::HWPX:
		{
			if (nullptr == m_pHWPXFile)
				return nullptr;

			return m_pHWPXFile->GetDocInfo();
		}
		case EHanType::NONE:
			return nullptr;
	}
}

const CHWPRecordBorderFill* CWriterContext::GetBorderFill(short shId)
{
	const CHWPDocInfo* pDocInfo = GetDocInfo();

	if (nullptr == pDocInfo)
		return nullptr;

	return (CHWPRecordBorderFill*)pDocInfo->GetBorderFill(shId);
}

const CHWPRecordParaShape* CWriterContext::GetParaShape(int nId)
{
	const CHWPDocInfo* pDocInfo = GetDocInfo();

	if (nullptr == pDocInfo)
		return nullptr;

	return (CHWPRecordParaShape*)pDocInfo->GetParaShape(nId);
}

const CHWPRecordStyle* CWriterContext::GetParaStyle(short shId)
{
	const CHWPDocInfo* pDocInfo = GetDocInfo();

	if (nullptr == pDocInfo)
		return nullptr;

	return (CHWPRecordStyle*)pDocInfo->GetStyle(shId);
}

const CHWPRecordCharShape* CWriterContext::GetCharShape(int nId)
{
	const CHWPDocInfo* pDocInfo = GetDocInfo();

	if (nullptr == pDocInfo)
		return nullptr;

	return (CHWPRecordCharShape*)pDocInfo->GetCharShape(nId);
}

const CHWPRecordNumbering* CWriterContext::GetNumbering(short shId)
{
	const CHWPDocInfo* pDocInfo = GetDocInfo();

	if (nullptr == pDocInfo)
		return nullptr;

	return (CHWPRecordNumbering*)pDocInfo->GetNumbering(shId);
}

const CHWPRecordBullet* CWriterContext::GetBullet(short shId)
{
	const CHWPDocInfo* pDocInfo = GetDocInfo();

	if (nullptr == pDocInfo)
		return nullptr;

	return (CHWPRecordBullet*)pDocInfo->GetBullet(shId);
}

const CHwpRecordTabDef* CWriterContext::GetTabDef(short shId)
{
	const CHWPDocInfo* pDocInfo = GetDocInfo();

	if (nullptr == pDocInfo)
		return nullptr;

	return (CHwpRecordTabDef*)pDocInfo->GetTabDef(shId);
}

HWP_STRING CWriterContext::GetBinFilename(const HWP_STRING& sId)
{
	const CHWPDocInfo* pDocInfo = GetDocInfo();

	if (nullptr == pDocInfo)
		return HWP_STRING();

	const CHWPRecordBinData* pBinData = dynamic_cast<const CHWPRecordBinData*>(pDocInfo->GetBinData(sId));

	return (nullptr != pBinData) ? pBinData->GetPath() : HWP_STRING();
}

bool CWriterContext::GetBinBytes(const HWP_STRING& sId, CHWPStream& oBuffer, HWP_STRING& sFileName)
{
	const CHWPDocInfo* pDocInfo = nullptr;

	pDocInfo = GetDocInfo();

	if (nullptr == pDocInfo)
		return false;

	const CHWPRecordBinData* pBinData = dynamic_cast<const CHWPRecordBinData*>(pDocInfo->GetBinData(sId));

	if (nullptr == pBinData)
		return false;

	if (EType::LINK == pBinData->GetType() ||
	    EType::EMBEDDING == pBinData->GetType())
	{
		switch (m_eType)
		{
			case EHanType::HWPX:
			{
				sFileName = NSFile::GetFileName(pBinData->GetPath());
				return m_pHWPXFile->GetChildStream(pBinData->GetPath(), oBuffer);
			}
			default:
				break;
		}
	}

	std::wostringstream oStringStream;

	switch (m_eType)
	{
		case EHanType::HWP:
		{
			oStringStream << L"BIN" << std::setw(4) << std::setfill(L'0') << std::hex << pBinData->GetBinDataID() << L"." << pBinData->GetFormat();
			sFileName = oStringStream.str();
			return m_pHWPFile->GetChildStream(oStringStream.str(), pBinData->GetCompressed(), oBuffer);
		}
		case EHanType::HWPX:
		{
			oStringStream << sId << L"." << pBinData->GetFormat();
			sFileName = oStringStream.str();
			return m_pHWPXFile->GetChildStream(L"BinData/" + oStringStream.str(), oBuffer);
		}
		default:
			break;
	}

	return false;
}

HWP_STRING CWriterContext::GetBinFormat(const HWP_STRING& sId)
{
	//TODO::реализовать
	return HWP_STRING();
}
}