Initial ImportHEADmaster

1 files changed, 300 insertions, 0 deletions

diff --git a/DropDlg.cpp b/DropDlg.cpp
new file mode 100644
index 0000000..e3517f7
--- /dev/null
+++ b/DropDlg.cpp
@@ -0,0 +1,300 @@
+// DropDlg.cpp : Implementation of CDropDlg
+#include "stdafx.h"
+#include "DropDlg.h"
+
+/////////////////////////////////////////////////////////////////////////////
+// CDropDlg
+
+const TCHAR* kLastBatch = _T("Batch");
+
+CDropDlg::CDropDlg() :
+	m_settings(HKEY_CURRENT_USER, _T("SOFTWARE\\WS\\Dropper\\")), 
+	CPersistPosWindow<CDropDlg>("Window")
+{
+	m_isRunning = false;
+	m_isCancelled = false;
+}
+
+void AddBatches(const string& wildcard, CComboBox& ctlBatch)
+{
+	WIN32_FIND_DATA findData;
+
+	HANDLE hFindFile = FindFirstFile(wildcard, &findData);
+
+	if(hFindFile != INVALID_HANDLE_VALUE)
+	{
+		do
+		{
+			// Get first part of the file
+			string name = findData.cFileName;
+			name = name.substr(0, name.find_last_of('.'));
+
+			ctlBatch.AddString(name);
+		}
+		while(FindNextFile(hFindFile, &findData));
+	}
+
+	FindClose(hFindFile);
+}
+
+
+
+LRESULT CDropDlg::OnInitDialog(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
+{
+	m_ctlLog = GetDlgItem(IDC_LOG);
+
+	m_baseDir = GetModuleFolder(_Module.GetModuleInstance());
+	CleanFolder(m_baseDir);
+
+	CComboBox ctlBatch = GetDlgItem(IDC_BATCH);
+
+	AddBatches(m_baseDir + _T("*.cmd"), ctlBatch);
+	AddBatches(m_baseDir + _T("*.bat"), ctlBatch);
+
+	SetMin(IDC_MINSIZE);
+	SetControlSizing(IDC_LOG, SD_HORZ | SD_VERT | SD_SIZING);
+	SetControlSizing(IDC_RUN, SD_HORZ | SD_MOVING);
+	SetControlSizing(IDC_BATCH, SD_HORZ | SD_SIZING);
+
+	// Select last used batch
+	ctlBatch.SelectString(-1, m_settings.GetString(kLastBatch));
+	
+	SetRunMode(false);
+
+	// Center it
+	CenterWindow();
+
+	LoadPosition(m_settings);
+ 
+	SetIcon((HICON)LoadImage(_Module.GetResourceInstance(), MAKEINTRESOURCE(IDI_MAIN), IMAGE_ICON, 32, 32, LR_SHARED), TRUE);
+	SetIcon((HICON)LoadImage(_Module.GetResourceInstance(), MAKEINTRESOURCE(IDI_MAIN), IMAGE_ICON, 16, 16, LR_SHARED), FALSE);
+
+	return 1;  // Let the system set the focus
+}
+
+LRESULT CDropDlg::OnDestroy(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
+{
+	SavePosition(m_settings);
+	return 1;
+}
+
+
+void CDropDlg::SetRunMode(bool isRunning)
+{
+	m_isRunning = isRunning;
+	
+	if(isRunning)
+		m_isCancelled = false;
+
+	DragAcceptFiles(isRunning ? FALSE : TRUE);
+	::EnableWindow(GetDlgItem(IDC_BATCH), isRunning ? FALSE : TRUE);
+	::EnableWindow(GetDlgItem(IDC_RUN), TRUE);
+
+	SetDlgItemText(IDC_RUN, isRunning ? _T("Cancel") : _T("Run"));
+}
+
+void CDropDlg::ReportError(HRESULT hr)
+{
+	MessageBox(FormatHR(hr), _T("Dropper"), MB_ICONSTOP | MB_OK);
+}
+
+
+string CDropDlg::CleanFolder(const string& folderName)
+{
+	if(folderName.at(folderName.length() - 1) != _T('\\'))
+		return folderName + _T("\\");
+
+	return folderName;
+}
+
+bool CDropDlg::IsDots(const string& fileName)
+{
+	return fileName == _T(".") || fileName == _T("..");
+}
+	
+
+LRESULT CDropDlg::OnDropFiles(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
+{
+	HDROP hDrop = (HDROP)wParam; 
+	HRESULT hr = S_OK;
+	string fileName;
+	string batchName = GetCurBatch();
+
+	SetRunMode(true);
+
+	SetDlgItemText(IDC_LOG, _T(""));
+
+	UINT numFiles = DragQueryFile(hDrop, 0xFFFFFFFF, NULL, 0);
+
+	for(int i = 0; i < numFiles; i++)
+	{
+		if(DragQueryFile(hDrop, i, fileName.get_buffer(MAX_PATH), MAX_PATH))
+		{
+			fileName.release_buffer();
+			hr = RunWithFile(batchName, fileName);
+			if(FAILED(hr)) 
+			{
+				ReportError(hr);
+				break;
+			}
+		}
+	}
+
+	DragFinish(hDrop);
+
+	SetRunMode(false);
+
+	return 0;
+}
+
+LRESULT CDropDlg::OnClose(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
+{
+	if(!m_isRunning)
+	{
+		string batchName;
+		GetDlgItemText(IDC_BATCH, batchName.get_buffer(MAX_PATH), MAX_PATH);
+		batchName.release_buffer();
+
+		m_settings.WriteString(kLastBatch, batchName);
+
+		EndDialog(wID);
+	}
+	else
+	{
+		m_isCancelled = true;
+	}
+
+	return 0;
+}
+
+
+LRESULT CDropDlg::OnRun(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
+{
+	if(m_isRunning)
+	{
+		// It's a cancel button now
+		m_isCancelled = true;
+		::EnableWindow(GetDlgItem(IDC_RUN), FALSE);
+	}
+	else
+	{
+		// It's a run button
+
+		SetRunMode(true);
+
+		SetDlgItemText(IDC_LOG, _T(""));
+
+		HRESULT hr = RunWithFile(GetCurBatch(), _T(""));
+		if(FAILED(hr)) 
+			ReportError(hr);
+
+		SetRunMode(false);
+	}
+
+	return 0;
+}
+
+string CDropDlg::GetCurBatch()
+{
+	string sBatch;
+	GetDlgItemText(IDC_BATCH, sBatch.get_buffer(MAX_PATH), MAX_PATH);
+	sBatch.release_buffer();
+	return sBatch;
+}
+
+HRESULT CDropDlg::RunWithFolder(const string& batch, const string& folderName)
+{
+	string wildcard = folderName + _T("*.*");
+	WIN32_FIND_DATA findData;
+	HRESULT hr = S_OK;
+
+	HANDLE hFindFile = FindFirstFile(wildcard, &findData);
+
+	if(hFindFile != INVALID_HANDLE_VALUE)
+	{
+		do
+		{
+			if(!IsDots(findData.cFileName))
+			{
+				string fileName = folderName + findData.cFileName;
+				hr = RunWithFile(batch, fileName);
+				if(FAILED(hr)) return hr;
+			}
+
+			if(m_isCancelled)
+			{
+				::SetLastError(ERROR_SUCCESS);
+				break;
+			}
+		}
+		while(FindNextFile(hFindFile, &findData));
+	}
+
+	if(GetLastError() != 0 && GetLastError() != ERROR_NO_MORE_FILES)
+		return HRESULT_FROM_WIN32(::GetLastError());
+
+	return S_OK;
+}
+
+HRESULT CDropDlg::RunWithFile(const string& batch, const string& path)
+{
+	if(path.size() > 0)
+	{
+		DWORD fileAttrib = GetFileAttributes(path);
+		if(fileAttrib & FILE_ATTRIBUTE_DIRECTORY)
+		{
+			return RunWithFolder(batch, CleanFolder(path));
+		}
+	}
+
+
+	string cmdLine = _T("\"") + m_baseDir + batch + 
+					 _T(".bat\"");
+
+	HRESULT hr = S_OK;
+
+	if(path.size() > 0)
+	{
+		 cmdLine += _T(" \"") + path + _T("\"");
+
+		size_t pos = path.find_last_of(_T("\\/"));
+		string file, dir;
+
+		if(pos != string::npos)
+		{
+			pos++;
+			file = path.substr(pos);
+			dir = path.substr(0, pos);
+		}
+		else
+		{
+			file = path;
+		}
+		
+		SetEnvironmentVariable(_T("DROPPED_FILE"), file);
+		SetEnvironmentVariable(_T("DROPPED_DIR"), dir);
+		SetEnvironmentVariable(_T("DROPPED_PATH"), path);
+
+	}
+
+	if(!CreateProcessOutput(NULL, (LPTSTR)(LPCTSTR)cmdLine, NULL, NULL, TRUE,
+							0, NULL, m_baseDir, OutputString, (LPVOID)this))
+	{
+		hr = HRESULT_FROM_WIN32(::GetLastError());
+	}
+
+	// Do an extra line
+	OutputString(_T("\r\n"), (LPVOID)this);
+
+	return S_OK;
+}
+
+BOOL CDropDlg::OutputString(LPCTSTR data, LPVOID param)
+{
+	CDropDlg* pDlg = (CDropDlg*)param;
+	pDlg->m_ctlLog.AppendText(data);
+
+	// TODO: Provide some way to cancel
+	return TRUE;
+}
+