// EmptyTempFolder.cpp : Implementation of CEmptyTempFolder
#include "stdafx.h"
#include "NSCmpts.h"
#include "EmptyTempFolder.h"
#include "..\common\defines.h"
#include "ProgressDlg.h"
#include "TempWarnDlg.h"
#include "TempActions.h"
#include <appmisc.h>

/////////////////////////////////////////////////////////////////////////////
// CEmptyTempFolder

STDMETHODIMP CEmptyTempFolder::InterfaceSupportsErrorInfo(REFIID riid)
{
	static const IID* arr[] = 
	{
		&IID_ISecureShutdownWin,
	};
	for (int i=0;i<sizeof(arr)/sizeof(arr[0]);i++)
	{
		if (InlineIsEqualGUID(*arr[i],riid))
			return S_OK;
	}
	return S_FALSE;
}

STDMETHODIMP CEmptyTempFolder::DoShutdown(long hParent, long lMode)
{
	HWND hwndParent = (HWND)hParent;

	bool bPrompt = lMode & nsNoPrompt ? false : true;
	bool bSilent = lMode & nsQuiet ? true : false;

	DWORD dwRet;

	// Get Windows Temp Folder
	string sWindowsTempPath;
	dwRet = ::GetTempPath(MAX_PATH, sWindowsTempPath.get_buffer(MAX_PATH));
	sWindowsTempPath.release_buffer();
	sWindowsTempPath.make_upper();
	file_path::clean_path(sWindowsTempPath);

	if(dwRet == 0)
		return HRESULT_FROM_WIN32(::GetLastError());

	bool bDisable = false;

	// Get Windows folder
	string sTemp;
	GetWindowsDirectory(sTemp.get_buffer(MAX_PATH), MAX_PATH);
	sTemp.release_buffer();
	sTemp.make_upper();
	file_path::clean_path(sTemp);

	// Compare Temp path against Windows folder
	if(sWindowsTempPath == sTemp)
		bDisable = true;
	
	// Get System folder
	GetSystemDirectory(sTemp.get_buffer(MAX_PATH), MAX_PATH);
	sTemp.release_buffer();
	sTemp.make_upper();
	file_path::clean_path(sTemp);

	// Compare Temp path against System folder
	if(sWindowsTempPath == sTemp)
		bDisable = true;

	// Construct Root folder
	sTemp = sWindowsTempPath.substr(0, 3);
	file_path::clean_path(sTemp);

	// Compare Temp path against root 
	if(sWindowsTempPath == sTemp)
		bDisable = true;

	if(bDisable)
	{
		if(bPrompt)
			MessageBoxTimeout(hwndParent, _T("Can't remove temp files. The temp folder on your computer is the same as your Windows or System folder."), 
							  _T("Empty Temp Folder"), MB_OK | MB_ICONSTOP, 15000, 5000);

		m_Data.WriteInt(ENABLED_KEY, false);
		return E_FAIL;
	}

	if(sWindowsTempPath.find(_T("TEMP")) == string::npos &&
		sWindowsTempPath.find(_T("TMP")) == string::npos)
	{
		if(m_Data.GetInt(_T("Folder Prompt"), true))
		{
			if(!bPrompt)
			{
				bDisable = true;
			}
			else
			{
				CTempWarnDlg dlg(sWindowsTempPath);

				bDisable = (dlg.DoModal(hwndParent) == IDCANCEL);
	
				if(dlg.m_bNotAgain)
					m_Data.WriteInt(_T("Folder Prompt"), false);
			}

		}

	}

	if(bDisable)
	{
		m_Data.WriteInt(ENABLED_KEY, false);
		return E_FAIL;
	}

	// Create the dialog
	CActionProgressDlg dlgProgress(m_spUnkSite, IDD_PROGRESSTEMP);
	dlgProgress.Create(hwndParent);
	dlgProgress.PlayAmination(IDR_AVITEMP);


	CTempActions::DeletePrepare* pAction = new CTempActions::DeletePrepare;
	pAction->Initialize(sWindowsTempPath);
	m_Engine.Add(pAction);

	// Create the update_monitor
	m_Engine.SetUpdates(dlgProgress, dlgProgress);

	// start the backup
	HANDLE hThread = m_Engine.StartThread();

	WaitForAndIdle(1, &hThread, INFINITE);

	HRESULT hr = S_OK;
	GetExitCodeThread(hThread, (LPDWORD)&hr);

	dlgProgress.DestroyWindow();

	return hr;
}

// XError: We should be using file_path here
/*string CEmptyTempFolder::CleanSlashes(const string& sPath)
{
	string sTemp;
	if(sPath.at(sPath.length() - 1) == _T('\\'))
		sTemp = sPath.substr(0, sPath.length() - 1);
	else
		sTemp = sPath;

	int nPos;
	while((nPos = sTemp.find(_T("\\\\"), 2)) != string::npos)
		sTemp.erase(nPos, 1);

	return sTemp;
}
*/
STDMETHODIMP CEmptyTempFolder::get_Info(NightSecInfo nsItem, VARIANT* pvVal)
{
	::VariantClear(pvVal);

	CComBSTR bsRetVal;

	switch(nsItem)
	{
	case nsName:
		pvVal->vt = VT_BSTR;
		bsRetVal.LoadString(IDS_TEMPNAME);
		pvVal->bstrVal = bsRetVal.Detach();
		return S_OK;
	case nsHelpText:
		pvVal->vt = VT_BSTR;
		bsRetVal.LoadString(IDS_TEMPDESC);
		pvVal->bstrVal = bsRetVal.Detach();
		return S_OK;
	case nsCmdLine:
		pvVal->vt = VT_BSTR;
		bsRetVal.LoadString(IDS_TEMPPARAM);
		pvVal->bstrVal = bsRetVal.Detach();
		return S_OK;
	}

	::VariantClear(pvVal);
	return S_FALSE;
}


STDMETHODIMP CEmptyTempFolder::SetData(IUnknown* pUnk)
{
	return m_Data.Initialize(pUnk);
}