#include "Info.h"
#include "Utils.h"
#include "Objects.h"

#include <ctime>

static const char  *c_asInfoItemNames[] =
{
	"CreationDate",
	"ModDate",
	"Author",
	"Creator",
	"Producer",
	"Title",
	"Subject",
	"Keywords",
	NULL
};

namespace PdfWriter
{
	static const char* InfoTypeToName(EInfoType eType)
	{
		unsigned int nIndex = (unsigned int)eType;
		return c_asInfoItemNames[nIndex];
	}
	//----------------------------------------------------------------------------------------
	// CInfoDict
	//----------------------------------------------------------------------------------------
	CInfoDict::CInfoDict(CXref* pXref)
	{
		pXref->Add(this);
	}
	void        CInfoDict::SetInfo(EInfoType eType, const char* sValue)
	{
		const char* sName = InfoTypeToName(eType);

		if (eType <= InfoModaDate)
			return;

		Add(sName, new CStringObject(sValue));
	}
	const char* CInfoDict::GetInfo(EInfoType eType)
	{
		const char* sName = InfoTypeToName(eType);
		CStringObject* pString = (CStringObject*)Get(std::string(sName));

		if (!pString || object_type_STRING != pString->GetType())
			return NULL;

		return (const char*)pString->GetString();
	}
	void        CInfoDict::SetInfo(EInfoType eType, const TDate& oDate)
	{
		char sTemp[DATE_TIME_STR_LEN + 1];
		char* pTemp = NULL;
		const char* sName = InfoTypeToName(eType);

		if (eType > InfoModaDate)
			return;

		MemSet(sTemp, 0, DATE_TIME_STR_LEN + 1);

		if (oDate.nMonth < 1 || 12 < oDate.nMonth ||
			oDate.nDay < 1 ||
			23 < oDate.nHour || 59 < oDate.nMinutes || 59 < oDate.nSeconds ||
			(oDate.nInd != '+'  && oDate.nInd != '-'   && oDate.nInd != 'Z'   && oDate.nInd != ' ') ||
			23 < oDate.nOffHour || 59 < oDate.nOffMinutes)
		{
			return;
		}

		switch (oDate.nMonth)
		{
		case 1:
		case 3:
		case 5:
		case 7:
		case 8:
		case 10:
		case 12:
			if (oDate.nDay > 31)
				return;

			break;
		case 4:
		case 6:
		case 9:
		case 11:
			if (oDate.nDay > 30)
				return;

			break;
		case 2:
			if (oDate.nDay > 29 || (oDate.nDay == 29 && (oDate.nYear % 4 != 0 || (oDate.nYear % 100 == 0 && oDate.nYear % 400 != 0))))
				return;

			break;
		default:
			return;
		}

		pTemp = (char*)MemCpy((BYTE*)sTemp, (BYTE*)"D:", 2);
		pTemp = ItoA2(pTemp, oDate.nYear, 5);
		pTemp = ItoA2(pTemp, oDate.nMonth, 3);
		pTemp = ItoA2(pTemp, oDate.nDay, 3);
		pTemp = ItoA2(pTemp, oDate.nHour, 3);
		pTemp = ItoA2(pTemp, oDate.nMinutes, 3);
		pTemp = ItoA2(pTemp, oDate.nSeconds, 3);

		if (' ' != oDate.nInd)
		{
			*pTemp++ = oDate.nInd;
			pTemp = ItoA2(pTemp, oDate.nOffHour, 3);
			*pTemp++ = '\'';
			pTemp = ItoA2(pTemp, oDate.nOffMinutes, 3);
			*pTemp++ = '\'';
		}
		*pTemp = 0;

		Add(sName, new CStringObject(sTemp));
	}
	void        CInfoDict::SetCreationTime()
	{
		time_t oTime = time(0);
		struct tm* oNow = localtime(&oTime);

		TDate oDate;

		oDate.nYear    = oNow->tm_year + 1900;
		oDate.nMonth   = oNow->tm_mon + 1;
		oDate.nDay     = oNow->tm_mday;
		oDate.nHour    = oNow->tm_hour;
		oDate.nMinutes = oNow->tm_min;
		oDate.nSeconds = oNow->tm_sec;
		oDate.nInd     = ' ';

		SetInfo(InfoCreationDate, oDate);
	}
}