#include <windows.h>

#define TEST MessageBox(MainWindow, "Success.", "Report", MB_OK);
#define DesktopsList 1001
#define DesktopSearch 1002

HINSTANCE hInstance = NULL;

HDESK TempDesk = NULL;

HWND MainWindow     = NULL;
HWND DesktopsListLB = NULL;
HWND DesktopSearchB = NULL;

BOOL CALLBACK DesktopSearchProc(LPTSTR DesktopName, LPARAM lParam);
LRESULT CALLBACK MainWNDProc(HWND MainWindow, UINT Msg, WPARAM wParam, LPARAM lParam);

INT WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, INT nShowCmd)
{

 WNDCLASSEX WNDClass;
 MSG Msg;

 ZeroMemory(&WNDClass, sizeof(WNDClass));

 WNDClass.cbSize = sizeof(WNDClass);
 WNDClass.style = CS_HREDRAW | CS_VREDRAW;
 WNDClass.hbrBackground = GetSysColorBrush(COLOR_3DFACE);
 WNDClass.hCursor = LoadCursor(NULL, IDC_ARROW);
 WNDClass.hIcon = LoadIcon(NULL, IDI_APPLICATION);
 WNDClass.hIconSm = LoadIcon(NULL, IDI_APPLICATION);
 WNDClass.hInstance = hInstance;
 WNDClass.lpfnWndProc = MainWNDProc;
 WNDClass.lpszClassName = "MainWNDClass";

 RegisterClassEx(&WNDClass);

 MainWindow = CreateWindowEx(NULL, WNDClass.lpszClassName, "Welcome", WS_VISIBLE | WS_OVERLAPPED | WS_MINIMIZEBOX | WS_SYSMENU | WS_CAPTION, CW_USEDEFAULT, CW_USEDEFAULT, 233, 480, NULL, NULL, hInstance, NULL);

 ShowWindow(MainWindow, nShowCmd);
 UpdateWindow(MainWindow);

 while(GetMessage(&Msg, NULL, NULL, NULL) > 0)
 {

  TranslateMessage(&Msg);
  DispatchMessage(&Msg);

 }

 return Msg.wParam;

}

BOOL CALLBACK DesktopSearchProc(LPTSTR DesktopName, LPARAM lParam)
{

 SendMessage(DesktopsListLB, LB_ADDSTRING, NULL, (LPARAM)DesktopName);

 return TRUE;

}

LRESULT CALLBACK MainWNDProc(HWND MainWindow, UINT Msg, WPARAM wParam, LPARAM lParam)
{

 switch(Msg)
 {

  case WM_CREATE:

   DesktopsListLB = CreateWindowEx(NULL, "LISTBOX", "DesktopsListLB", WS_CHILD | WS_VISIBLE | LBS_STANDARD, 15, 12, 200, 100, MainWindow, (HMENU)DesktopsList, hInstance, NULL);
   DesktopSearchB = CreateWindowEx(NULL, "BUTTON", "Search for Desktops", WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON, 15, 118, 200, 20, MainWindow, (HMENU)DesktopSearch, hInstance, NULL);

   //TempDesk = CreateDesktop("TempDesk", NULL, NULL, NULL, DESKTOP_CREATEMENU | DESKTOP_CREATEWINDOW | DESKTOP_ENUMERATE | DESKTOP_HOOKCONTROL | DESKTOP_JOURNALPLAYBACK | DESKTOP_JOURNALRECORD | DESKTOP_READOBJECTS | DESKTOP_SWITCHDESKTOP | DESKTOP_WRITEOBJECTS, NULL);

   //SwitchDesktop(TempDesk);

  break;

  case WM_COMMAND:

   switch(LOWORD(wParam))
   {

    case DesktopSearch:

     while(SendMessage(DesktopsListLB, LB_GETCOUNT, NULL, NULL) != 0) SendMessage(DesktopsListLB, LB_DELETESTRING, NULL, NULL);
 
     EnumDesktops(GetProcessWindowStation(), DesktopSearchProc, NULL);

    break;

   }

  break;

  case WM_CLOSE:

   DestroyWindow(MainWindow);

  break;

  case WM_DESTROY:

   PostQuitMessage(WM_QUIT); 

  break;

  default: return DefWindowProc(MainWindow, Msg, wParam, lParam);

 }

 return 0;

}