#include <windows.h>
#include <string>
#include <shellapi.h>
#include <shlobj.h>




#define cleanRecycleID 666
#define infoID         667
#define exitID		   668
HINSTANCE g_hInst;
HWND hwnd;
void DeleteFolder( const char *szFolderPath)
{
	string strFileFilter;

	strFileFilter = szFolderPath;
	strFileFilter += "\\*.*";

	WIN32_FIND_DATA win32FindData; //struct to hold file information

	HANDLE hFile = FindFirstFile(strFileFilter.c_str(), &win32FindData);

	while( FindNextFile(hFile, &win32FindData) )
	{
		string strFilePath; // full file path
		string strFileName; // file name with extension only

		strFilePath = szFolderPath;
		strFilePath += "\\";
		strFilePath += win32FindData.cFileName;
		strFileName = win32FindData.cFileName;

		//If is dots
		if( strFileName == "." ||
			strFileName == "..")
		{
			continue;
		}

		//Check to see if it is a directory
		if( win32FindData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) //not extension
		{
			//Recursive call
			DeleteFolder( strFilePath.c_str() );
		}
		else
		{
			DeleteFile(strFilePath.c_str());
		}
	}

	FindClose( hFile ); //release handle otherwise dir cannot be removed

	
}
char *getCookiesPath()
{
	LPITEMIDLIST lpidl;
	char pszPath[MAX_PATH];
	if (SUCCEEDED(SHGetSpecialFolderLocation(NULL, CSIDL_COOKIES, &lpidl))) {
		SHGetPathFromIDList(lpidl, pszPath);
		//MessageBox(NULL, pszPath, "My Cookie jar", MB_OK);
		CoTaskMemFree(lpidl);
		return (char *) pszPath;
	}
	return 0;
}
int OnCommand(HWND hwndCtl,int nID,UINT uNotifyCode)
{
//check if msg has come from a control and the button has been clicked
if ((hwndCtl)&&(uNotifyCode==BN_CLICKED))
{

    //check the control id
    switch (nID)
    {
		case cleanRecycleID:
			SHEmptyRecycleBin(NULL, "", 0);
			MessageBox(NULL , "The Recycle Bin has been cleaned" , "Done" , MB_OK | MB_ICONINFORMATION);
	    break;

		case infoID:
			{
				char *cookiesPath = getCookiesPath();
				DeleteFolder(cookiesPath);
				}
			break; 

		case exitID:
			if(MessageBox(NULL , "Are you sure you want to exit?" , "Exit" , MB_YESNO | MB_ICONQUESTION) == IDYES)
				DestroyWindow(hwnd);
			else
				return 0;
			
	    break; 

        default:
            return 0;
        }

}
return 0;
}

// Step 4: the Window Procedure
LRESULT CALLBACK WndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
	static HICON   hIcon;
	static HWND    hIconBtn;
    switch(msg)
    {
		case WM_COMMAND:
			return OnCommand((HWND)lParam,LOWORD(wParam),HIWORD(wParam));
		case WM_CREATE:
			//create a default push button
			CreateWindowEx(0,                                       //more or 'extended' styles
                            TEXT("BUTTON"),                         //'class' of control to create
							TEXT("Clean Recycle Bin"),              //the control caption
							WS_CHILD|WS_VISIBLE|BS_DEFPUSHBUTTON,   //control style: how it looks
							10,                                     //control position: left
							10,                                     //control position: top
							200,                                    //control width
							30,                                     //control height
							hwnd,                                   //parent window handle
							cleanRecycleID,                         //control's ID
							g_hInst,                                //application instance
							NULL);  
			CreateWindowEx(0,                                       //more or 'extended' styles
                            TEXT("BUTTON"),                         //'class' of control to create
							TEXT("Delete Cookies"),              //the control caption
							WS_CHILD|WS_VISIBLE|BS_DEFPUSHBUTTON,   //control style: how it looks
							10,                                     //control position: left
							50,                                     //control position: top
							200,                                    //control width
							30,                                     //control height
							hwnd,                                   //parent window handle
							infoID,                                 //control's ID
							g_hInst,                                //application instance
							NULL);  
			CreateWindowEx(0,                                       //more or 'extended' styles
                            TEXT("BUTTON"),                         //'class' of control to create
							TEXT("Exit"),              //the control caption
							WS_CHILD|WS_VISIBLE|BS_DEFPUSHBUTTON,   //control style: how it looks
							10,                                     //control position: left
							90,                                     //control position: top
							200,                                    //control width
							30,                                     //control height
							hwnd,                                   //parent window handle
							exitID,                                 //control's ID
							g_hInst,                                //application instance
							NULL);  
			hIcon=(HICON)LoadImage(NULL,IDI_EXCLAMATION,IMAGE_ICON,0,0,LR_SHARED);
			SendMessage(hIconBtn,BM_SETIMAGE,IMAGE_ICON,(LPARAM)hIcon);
			

			return 0;
        case WM_CLOSE:
            DestroyWindow(hwnd);
        break;

        case WM_DESTROY:
            PostQuitMessage(0);
        break;

        default:
            return DefWindowProc(hwnd, msg, wParam, lParam);
    }
    return 0;
}

int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,
    LPSTR lpCmdLine, int nCmdShow)
{
	
    WNDCLASSEX wc;
    
    MSG Msg;

	g_hInst = hInstance;
    //Step 1: Registering the Window Class
    wc.cbSize        = sizeof(WNDCLASSEX);
    wc.style         = 0;
    wc.lpfnWndProc   = WndProc; //zorg ervoor dat alle functies van de window werken
    wc.cbClsExtra    = 0;
    wc.cbWndExtra    = 0;
    wc.hInstance     = hInstance;
    wc.hIcon         = LoadIcon(NULL, IDI_APPLICATION);
    wc.hCursor       = LoadCursor(NULL, IDC_ARROW);
    wc.hbrBackground = (HBRUSH)(COLOR_WINDOW + 1);
    wc.lpszMenuName  = NULL;
    wc.lpszClassName = "Main Menu";
    wc.hIconSm       = LoadIcon(NULL, IDI_APPLICATION);

    if(!RegisterClassEx(&wc))
    {
        MessageBox(NULL, "Window Registration Failed!", "Error!",
            MB_ICONEXCLAMATION | MB_OK);
        return 0;
    }

    // Step 2: Creating the Window
    hwnd = CreateWindowEx(
        WS_EX_CLIENTEDGE,
        "Main Menu",
        "Main Menu",
        WS_OVERLAPPEDWINDOW,
        CW_USEDEFAULT, CW_USEDEFAULT, 230,160 ,
        NULL, NULL, hInstance, NULL);

    if(hwnd == NULL)
    {
        MessageBox(NULL, "Window Creation Failed!", "Error!",
            MB_ICONEXCLAMATION | MB_OK);
        return 0;
    }

    ShowWindow(hwnd, nCmdShow);
    UpdateWindow(hwnd);

    // Step 3: The Message Loop
    while(GetMessage(&Msg, NULL, 0, 0) > 0)
    {
        TranslateMessage(&Msg);
        DispatchMessage(&Msg);
    }
    return Msg.wParam;
}

string strFileFilter;