1 files changed, 153 insertions, 0 deletions
diff --git a/NSCmpts/TempActions.cpp b/NSCmpts/TempActions.cpp
new file mode 100644
index 0000000..24c8a91
--- /dev/null
+++ b/NSCmpts/TempActions.cpp
@@ -0,0 +1,153 @@
+// TempActions.cpp: implementation of the CEncryptActions class.
+//
+//////////////////////////////////////////////////////////////////////
+
+#include "stdafx.h"
+#include "TempActions.h"
+
+#include "resource.h"
+#include <appmisc.h>
+
+
+//////////////////////////////////////////////////////////////////////
+// Construction/Destruction
+//////////////////////////////////////////////////////////////////////
+
+bool CTempActions::DeletePrepare::Do(CActionEngine* pEngine, 
+	  								     HWND hwndUpdates)
+{
+	ASSERT(pEngine != NULL);
+
+	// Need to create another array for the folders
+	// since they need to pushed on backwards after the files
+	// so we don't get "Folder not empty" errors
+	action_array aFolders;
+
+	Update(hwndUpdates, _T("Preparing to clear Temp Folder..."));
+
+	file_iterator fileIter(m_tempFolder, file_iterator::sub_folders | file_iterator::full_paths), end;
+	for( ; fileIter != end; fileIter++)
+	{
+		file_path srcFile(fileIter->cFileName);
+
+		CAction* pAction;
+		if(srcFile.attributes() & FILE_ATTRIBUTE_DIRECTORY)
+		{
+			pAction = new DeleteFolder(srcFile);
+			aFolders.push_back(pAction);
+		}
+		else
+		{
+			pAction = new DeleteFile(srcFile);
+			pEngine->Add(pAction);
+		}
+	}
+
+	while(!aFolders.empty())
+	{
+		pEngine->Add(aFolders.back());
+		aFolders.pop_back();
+	}
+
+	// Error from file_iterator (listing folders)
+	if(fileIter.state() != ERROR_SUCCESS)
+		throw CActionError(HRESULT_FROM_WIN32(fileIter.state()), IDS_TEMP_ERR_LIST, m_tempFolder);
+	
+	return false;
+}
+
+HRESULT CTempActions::DeletePrepare::Initialize(const file_path& temp_folder)
+{
+	m_tempFolder = temp_folder;
+	return S_OK;
+}
+
+bool CTempActions::DeleteFile::Do(CActionEngine* pEngine, 
+	  							  HWND hwndUpdates)
+{
+	// Update any monitors
+	Update(hwndUpdates, m_file);
+
+	::SetFileAttributes(m_file, FILE_ATTRIBUTE_NORMAL);
+
+	try
+	{
+		if(!::DeleteFile(m_file))
+			throw CActionError(HRESULT_FROM_WIN32(::GetLastError()), IDS_TEMP_ERR_DELETE, m_file);
+	}
+	catch(CActionError& err)
+	{
+		// These errors are okay. The file is gone
+		if(err.m_hRes != HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) &&
+		   err.m_hRes != HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND))
+			throw;
+	}
+
+	return false;
+
+}
+bool CTempActions::DeleteFile::IsFixable(HRESULT hr) const
+{
+	return (hr == HRESULT_FROM_WIN32(ERROR_SHARING_VIOLATION) ||
+		   hr == HRESULT_FROM_WIN32(ERROR_LOCK_VIOLATION) ||
+		   hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) ||
+		   hr == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND));
+}
+
+void CTempActions::DeleteFile::Fix(HRESULT hr)
+{
+	bool bFirst = true;
+
+start:
+
+	if(IsFixable(hr))
+	{
+		try
+		{
+			// Try again
+			Do(NULL, NULL);
+			return;
+
+		}
+		catch(CActionError& err)
+		{
+			// If it's the first around and the error is the same...
+			if(bFirst && IsFixable(err.m_hRes))
+			{
+				// ... Prompt user to close programs
+				MessageBox(NULL, _T("Many programs keep files in the temp folder. These files cannot be deleted while the program is open. Close all other programs and then click the OK button."), _T("Empty Temp Folder"), MB_OK | MB_ICONINFORMATION);
+				bFirst = false;
+			}
+			// Otherwise out with the error
+			else
+				throw;
+		}
+	}
+
+	goto start;
+}
+
+bool CTempActions::DeleteFolder::Do(CActionEngine* pEngine, 
+	  								 HWND hwndUpdates)
+{
+	// Update any monitors
+	Update(hwndUpdates, m_folder);
+
+	if(!::RemoveDirectory(m_folder))
+	{
+		DWORD dwError = ::GetLastError();
+
+		// This error will be accompanied by another file error
+		// so we don't need to show it
+		if(dwError != ERROR_DIR_NOT_EMPTY)
+			throw CActionError(HRESULT_FROM_WIN32(::GetLastError()), IDS_TEMP_ERR_DELETE, m_folder);
+	}
+
+	return false;
+
+}
+
+void CTempActions::DeleteFolder::Fix(HRESULT hr)
+{
+
+}