#pragma once

#include "resource.h"       // main symbols
#include <atlhost.h>
#include "Methods.h"
#include "Multithreading/Thread.h"

// defining an internal error message id.
// Purpose: Processing thread to popup message box.
#define IM_ERROR WM_USER+21

class Processing : public CDialogImpl<Processing>, public GenLib::Thread
{
public:
	Processing(enum CIPHER cipher, const char *source, const char *dest, const char *pwd);
	~Processing();

	/* 
		Caller incharge for freeing returned string! 
	*/
	static char *generateTempFile();

	enum { IDD = IDD_PROCESSING };

BEGIN_MSG_MAP(Processing)
	MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
	MESSAGE_HANDLER(WM_CLOSE, OnCloseDialog)
	MESSAGE_HANDLER(IM_ERROR, OnIMError);
	COMMAND_HANDLER(ID_CLOSE_BUTTON, BN_CLICKED, OnBnClickedCloseButton)
	COMMAND_HANDLER(ID_OPENFILE_BUTTON, BN_CLICKED, OnBnClickedOpenfileButton)
END_MSG_MAP()

private:
	LRESULT OnInitDialog(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);
	LRESULT OnCloseDialog(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/);
	LRESULT OnIMError(UINT /*uMsg*/, WPARAM wParam, LPARAM /*lParam*/, BOOL& /*bHandled*/);
	LRESULT OnBnClickedCloseButton(WORD /*wNotifyCode*/, WORD /*wID*/, HWND /*hWndCtl*/, BOOL& /*bHandled*/);
	LRESULT OnBnClickedOpenfileButton(WORD /*wNotifyCode*/, WORD /*wID*/, HWND /*hWndCtl*/, BOOL& /*bHandled*/);

	/*
		Back thread methods (run calls stepProgressbar & threadError).
	*/
	int run();
	void stepProgressbar() { SendDlgItemMessage(ID_PROGRESSBAR, PBM_STEPIT, 0, 0); };
	int threadError(const char *error);

	CIPHER _cipher;
	char _readFile[MAX_PATH];
	char _writeFile[MAX_PATH];
    char _password[100];
};