// GCMDSToolsDlg.cpp : ±¸Çö ÆÄÀÏ
//

#include "stdafx.h"
#include "GCMDSTools.h"
#include "GCMDSToolsDlg.h"

#include "IniFile.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


INT GetProgramDir(char* buf, INT len ) 
{
	char* ProgDir;
	INT result;

	result = GetModuleFileName( theApp.m_hInstance, buf, len - 1 );
	if( result == 0 )
		return 0;

	char pgmPtr[1024];
	strcpy( pgmPtr, buf );

	ProgDir = buf;
	ProgDir += strlen( pgmPtr ) - 1;
	while( *ProgDir != '\\' && ProgDir != pgmPtr )
		ProgDir--;

	*ProgDir = 0;
	return (INT)strlen( buf );
}

BOOL BrowserFolder(char* szFolder)
{
	strcpy(szFolder, "");

	BROWSEINFO bi;

	bi.hwndOwner		= ::GetActiveWindow();
	bi.pidlRoot			= NULL;
	bi.pszDisplayName	= NULL;
	bi.lpszTitle        = "¸®½ºÆ®¿¡¼­ Æú´õ¸¦ ¼±ÅÃÇϼ¼¿ä";
	bi.ulFlags		    = 0;
	bi.lpfn				= NULL;
	bi.lParam			= 0;

	LPITEMIDLIST pidl = SHBrowseForFolder( &bi );

	if( pidl == NULL )
		return FALSE;

	SHGetPathFromIDList( pidl, szFolder);

	LPMALLOC pMalloc;

	if( SHGetMalloc( &pMalloc ) != NOERROR )
		return FALSE;

	pMalloc->Free( pidl );
	pMalloc->Release();

	return TRUE;
}

// ÀÀ¿ë ÇÁ·Î±×·¥ Á¤º¸¿¡ »ç¿ëµÇ´Â CAboutDlg ´ëÈ­ »óÀÚÀÔ´Ï´Ù.

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// ´ëÈ­ »óÀÚ µ¥ÀÌÅÍÀÔ´Ï´Ù.
	enum { IDD = IDD_ABOUTBOX };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV Áö¿øÀÔ´Ï´Ù.

// ±¸ÇöÀÔ´Ï´Ù.
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()


// CGCMDSToolsDlg ´ëÈ­ »óÀÚ




CGCMDSToolsDlg::CGCMDSToolsDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CGCMDSToolsDlg::IDD, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CGCMDSToolsDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_KEY, m_Key);
	DDX_Control(pDX, IDC_INPUT, m_InDir);
	DDX_Control(pDX, IDC_OUTPUT, m_OutDir);
	DDX_Control(pDX, IDC_FILE, m_File);
	DDX_Control(pDX, IDC_CHECKMASK, m_2TypeMask);
}

BEGIN_MESSAGE_MAP(CGCMDSToolsDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(ID_INURL, &CGCMDSToolsDlg::OnBnClickedInurl)
	ON_BN_CLICKED(ID_OUTURL, &CGCMDSToolsDlg::OnBnClickedOuturl)
	ON_BN_CLICKED(ID_START, &CGCMDSToolsDlg::OnBnClickedStart)
	ON_CBN_SELCHANGE(IDC_KEY, &CGCMDSToolsDlg::OnCbnSelchangeKey)
	ON_BN_CLICKED(ID_START2, &CGCMDSToolsDlg::OnBnClickedStart2)
END_MESSAGE_MAP()


// CGCMDSToolsDlg ¸Þ½ÃÁö 󸮱â

BOOL CGCMDSToolsDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// ½Ã½ºÅÛ ¸Þ´º¿¡ "Á¤º¸..." ¸Þ´º Ç׸ñÀ» Ãß°¡ÇÕ´Ï´Ù.

	// IDM_ABOUTBOX´Â ½Ã½ºÅÛ ¸í·É ¹üÀ§¿¡ ÀÖ¾î¾ß ÇÕ´Ï´Ù.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// ÀÌ ´ëÈ­ »óÀÚÀÇ ¾ÆÀÌÄÜÀ» ¼³Á¤ÇÕ´Ï´Ù. ÀÀ¿ë ÇÁ·Î±×·¥ÀÇ ÁÖ Ã¢ÀÌ ´ëÈ­ »óÀÚ°¡ ¾Æ´Ò °æ¿ì¿¡´Â
	//  ÇÁ·¹ÀÓ¿öÅ©°¡ ÀÌ ÀÛ¾÷À» ÀÚµ¿À¸·Î ¼öÇàÇÕ´Ï´Ù.
	SetIcon(m_hIcon, TRUE);			// Å« ¾ÆÀÌÄÜÀ» ¼³Á¤ÇÕ´Ï´Ù.
	SetIcon(m_hIcon, FALSE);		// ÀÛÀº ¾ÆÀÌÄÜÀ» ¼³Á¤ÇÕ´Ï´Ù.

	// TODO: ¿©±â¿¡ Ãß°¡ ÃʱâÈ­ ÀÛ¾÷À» Ãß°¡ÇÕ´Ï´Ù.
	m_Key.AddString("¾Æ¹«°Å³ª");
	m_Key.AddString("ROWGlobalKey");
	m_Key.AddString("Z_all_A_3D");
	m_Key.SetCurSel(0);

	m_File.AddString("GCMDS");
	m_File.AddString("skey");
	m_File.AddString("EDS");
	m_File.SetCurSel(0);
	

	LoadIni();

	return TRUE;  // Æ÷Ä¿½º¸¦ ÄÁÆ®·Ñ¿¡ ¼³Á¤ÇÏÁö ¾ÊÀ¸¸é TRUE¸¦ ¹ÝȯÇÕ´Ï´Ù.
}

VOID CGCMDSToolsDlg::LoadIni()
{
	char strBuff[512];
	GetProgramDir(strBuff, 512);

	strcat(strBuff, "\\Data.ini");

	CIniFile ini;
	if(!ini.Open(strBuff))
	{
		SaveIni();
		return;
	}

	DWORD lLen = 0;
	DWORD dwID = -1;
	char Data[512];
	ZeroMemory(Data, 512);
	if(ini.GetValue("GCMDS", "Key", &dwID))
		m_Key.SetCurSel(dwID);

	lLen = 0;
	ZeroMemory(Data, 512);
	if(ini.GetValue("GCMDS", "InputFolder", Data, &lLen) && lLen > 0)
		m_InDir.SetWindowText(Data);

	lLen = 0;
	ZeroMemory(Data, 512);
	if(ini.GetValue("GCMDS", "OutFolder", Data, &lLen) && lLen > 0)
		m_OutDir.SetWindowText(Data);

	ini.Close();
}

VOID CGCMDSToolsDlg::SaveIni()
{
	char strBuff[512];
	GetProgramDir(strBuff, 512);

	strcat(strBuff, "\\Data.ini");

	CIniFile ini;
	ini.Open(strBuff);

	char Data[512];
	DWORD dwID = m_Key.GetCurSel();
	ini.SetValue("GCMDS", "Key", dwID);

	m_InDir.GetWindowText(Data, 512);
	ini.SetValue("GCMDS", "InputFolder", Data);

	m_OutDir.GetWindowText(Data, 512);
	ini.SetValue("GCMDS", "OutFolder", Data);

	ini.Close();
}

void CGCMDSToolsDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// ´ëÈ­ »óÀÚ¿¡ ÃÖ¼ÒÈ­ ´ÜÃ߸¦ Ãß°¡ÇÒ °æ¿ì ¾ÆÀÌÄÜÀ» ±×¸®·Á¸é
//  ¾Æ·¡ Äڵ尡 ÇÊ¿äÇÕ´Ï´Ù. ¹®¼­/ºä ¸ðµ¨À» »ç¿ëÇÏ´Â MFC ÀÀ¿ë ÇÁ·Î±×·¥ÀÇ °æ¿ì¿¡´Â
//  ÇÁ·¹ÀÓ¿öÅ©¿¡¼­ ÀÌ ÀÛ¾÷À» ÀÚµ¿À¸·Î ¼öÇàÇÕ´Ï´Ù.

void CGCMDSToolsDlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // ±×¸®±â¸¦ À§ÇÑ µð¹ÙÀ̽º ÄÁÅؽºÆ®

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// Ŭ¶óÀ̾ðÆ® »ç°¢Çü¿¡¼­ ¾ÆÀÌÄÜÀ» °¡¿îµ¥¿¡ ¸ÂÃä´Ï´Ù.
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// ¾ÆÀÌÄÜÀ» ±×¸³´Ï´Ù.
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// »ç¿ëÀÚ°¡ ÃÖ¼ÒÈ­µÈ âÀ» ²ô´Â µ¿¾È¿¡ Ä¿¼­°¡ Ç¥½ÃµÇµµ·Ï ½Ã½ºÅÛ¿¡¼­
//  ÀÌ ÇÔ¼ö¸¦ È£ÃâÇÕ´Ï´Ù.
HCURSOR CGCMDSToolsDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}


void CGCMDSToolsDlg::OnBnClickedInurl()
{
	// TODO: ¿©±â¿¡ ÄÁÆ®·Ñ ¾Ë¸² 󸮱â Äڵ带 Ãß°¡ÇÕ´Ï´Ù.
	// ÀÔ·Â °æ·Î ÁöÁ¤
	char strBuff[512];
	BrowserFolder(strBuff);

	m_InDir.SetWindowText(strBuff);

	SaveIni();
}

void CGCMDSToolsDlg::OnBnClickedOuturl()
{
	// TODO: ¿©±â¿¡ ÄÁÆ®·Ñ ¾Ë¸² 󸮱â Äڵ带 Ãß°¡ÇÕ´Ï´Ù.
	// Ãâ·Â °æ·Î ÁöÁ¤
	char strBuff[512];
	BrowserFolder(strBuff);

	m_OutDir.SetWindowText(strBuff);

	SaveIni();
}


void CGCMDSToolsDlg::OnCbnSelchangeKey()
{
	// TODO: ¿©±â¿¡ ÄÁÆ®·Ñ ¾Ë¸² 󸮱â Äڵ带 Ãß°¡ÇÕ´Ï´Ù.
	SaveIni();
}

void CGCMDSToolsDlg::Recurse( CString perfolder )
{
	CString		str;
	CString		folder   = perfolder;

	char strKey[64];
	m_File.GetWindowText(strKey, 64);

	folder += "\\*.";
	folder += strKey;

	CFileFind finder;
	BOOL bWorking = finder.FindFile( folder );

	while( bWorking )
	{
		bWorking = finder.FindNextFile();

		if( finder.IsDots() )
			continue;

		CString path = finder.GetFilePath();
		if( finder.IsDirectory() )
			Recurse( path );
		else
		{
			m_FileList.push_back(path.GetBuffer());
		}

		Sleep( 1 );
	}
}

void CGCMDSToolsDlg::OnBnClickedStart()
{
	// TODO: ¿©±â¿¡ ÄÁÆ®·Ñ ¾Ë¸² 󸮱â Äڵ带 Ãß°¡ÇÕ´Ï´Ù.
	SaveIni();

	char strKey[32];
	m_Key.GetWindowText(strKey, 32);

	if(m_2TypeMask.GetCheck())
	{
		int iLen = strlen(strKey);
		for(int k = 0; k < iLen; ++k)
		{
			strKey[k] = strKey[k]^0xc8;
		}
	}

	m_FileList.clear();

	char strInDir[512];
	m_InDir.GetWindowText(strInDir, 512);

	Recurse(strInDir);

	char strInput[512];
	m_InDir.GetWindowText(strInput, 512);
	int iInputLen = (int)strlen(strInput);

	char strOut[512];
	m_OutDir.GetWindowText(strOut, 512);

	char strTemp[512];

	CString strCmdLine;

	char strExec[512];
	GetProgramDir(strExec, 512);
	strcat(strExec, "\\XORenc.exe");


	// º¯È¯ ½ÃÀÛ
	int iSize = (int)m_FileList.size();
	for(int i = 0; i < iSize; ++i)
	{
		CString sIn = m_FileList[i];
		strcpy(strTemp, sIn);

		CString sOut = strOut;
		sOut += &strTemp[iInputLen];

		strCmdLine = '"'+sIn+'"'+" "+'"'+strKey+'"'+" "+'"'+sOut+'"';

		HINSTANCE nResult = ShellExecute(
		NULL,
		NULL, // do not request elevation unless needed
		strExec,
		strCmdLine, // params 
		NULL, // directory 
		SW_HIDE);
/*
		STARTUPINFO si;
		PROCESS_INFORMATION pi;
		ZeroMemory( &si, sizeof(si) );
		si.cb = sizeof(si);
		ZeroMemory( &pi, sizeof(pi) );
		CreateProcess( "XORenc.exe", (LPSTR)strCmdLine.GetString(), NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi );
*/
	}

	
	return;
}


void CGCMDSToolsDlg::OnBnClickedStart2()
{
	// TODO: ¿©±â¿¡ ÄÁÆ®·Ñ ¾Ë¸² 󸮱â Äڵ带 Ãß°¡ÇÕ´Ï´Ù.
	// TODO: ¿©±â¿¡ ÄÁÆ®·Ñ ¾Ë¸² 󸮱â Äڵ带 Ãß°¡ÇÕ´Ï´Ù.
	SaveIni();

	char strKey[32];
	m_Key.GetWindowText(strKey, 32);

	if(m_2TypeMask.GetCheck())
	{
		int iLen = strlen(strKey);
		for(int k = 0; k < iLen; ++k)
		{
			strKey[k] = strKey[k]^0xc8;
		}
	}

	m_FileList.clear();

	char strInDir[512];
	m_OutDir.GetWindowText(strInDir, 512);

	Recurse(strInDir);

	char strInput[512];
	m_OutDir.GetWindowText(strInput, 512);
	int iInputLen = (int)strlen(strInput);

	char strOut[512];
	m_InDir.GetWindowText(strOut, 512);

	char strTemp[512];

	CString strCmdLine;

	char strExec[512];
	GetProgramDir(strExec, 512);
	strcat(strExec, "\\XORenc.exe");


	// º¯È¯ ½ÃÀÛ
	int iSize = (int)m_FileList.size();
	for(int i = 0; i < iSize; ++i)
	{
		CString sIn = m_FileList[i];
		strcpy(strTemp, sIn);

		CString sOut = strOut;
		sOut += &strTemp[iInputLen];

		strCmdLine = '"'+sIn+'"'+" "+'"'+strKey+'"'+" "+'"'+sOut+'"';

		HINSTANCE nResult = ShellExecute(
		NULL,
		NULL, // do not request elevation unless needed
		strExec,
		strCmdLine, // params 
		NULL, // directory 
		SW_HIDE);
/*
		STARTUPINFO si;
		PROCESS_INFORMATION pi;
		ZeroMemory( &si, sizeof(si) );
		si.cb = sizeof(si);
		ZeroMemory( &pi, sizeof(pi) );
		CreateProcess( "XORenc.exe", (LPSTR)strCmdLine.GetString(), NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi );
*/
	}


}