XeNTaX

Code: Select all

#include<windows.h>

#ifndef RC_INVOKED
#include "shlobj.h"
#pragma comment(lib, "shell32.lib")
#endif

#ifndef RC_INVOKED
#include<sstream>
#endif

static const UINT WM_SHELLCHANGENOTIFY = (WM_USER + 0x101);
static HINSTANCE basead = NULL;
static HWND mainhwnd = NULL;

// shell variables
static LPSHELLFOLDER lpShellFolder = NULL;
static LPTSTR lpShellFolderPath = TEXT("c:\\users\\semory\\desktop\\New folder\\");
static LPITEMIDLIST lpShellFolderIDList = NULL;
static IShellItem* lpShellItem = NULL;
static ULONG watchcode = 0;

static LRESULT CALLBACK MainWindowProc(HWND window, UINT message, WPARAM wparam, LPARAM lparam);
static LRESULT EvNCCreate(HWND window, WPARAM wparam, LPARAM lparam);
static LRESULT EvCreate(HWND window, WPARAM wparam, LPARAM lparam);
static LRESULT EvDestroy(HWND window, WPARAM wparam, LPARAM lparam);
static LRESULT EvNCDestroy(HWND window, WPARAM wparam, LPARAM lparam);
static LRESULT EvPaint(HWND window, WPARAM wparam, LPARAM lparam);
static LRESULT EvShellChangeNotify(HWND window, WPARAM wparam, LPARAM lparam);

int run(HINSTANCE handle)
{
 // set base address
 basead = handle;

 // register window
 WNDCLASSEX wcx;
 wcx.cbSize = sizeof(WNDCLASSEX);
 wcx.style = CS_DBLCLKS | CS_HREDRAW | CS_VREDRAW;
 wcx.lpfnWndProc = (WNDPROC)MainWindowProc;
 wcx.cbClsExtra = 0;
 wcx.cbWndExtra = 0;
 wcx.hInstance = basead;
 wcx.hIcon = NULL;
 wcx.hCursor = LoadCursor(NULL, IDC_ARROW);
 wcx.hbrBackground = (HBRUSH)GetStockObject(WHITE_BRUSH);
 wcx.lpszMenuName = NULL;
 wcx.lpszClassName = TEXT("MainWindow");
 wcx.hIconSm = NULL;
 if(RegisterClassEx(&wcx) == FALSE) {
    MessageBox(0, TEXT("Failed to register window class."), TEXT("Error"), MB_ICONSTOP);
    return FALSE;
   }

 // create window
 DWORD styleEx = 0;
 DWORD style = WS_OVERLAPPEDWINDOW | WS_VISIBLE;
 int x = CW_USEDEFAULT; int y = CW_USEDEFAULT;
 int w = CW_USEDEFAULT; int h = CW_USEDEFAULT;
 HWND hwnd = CreateWindowEx(styleEx, TEXT("MainWindow"), TEXT("Hello World!"), style, x, y, w, h, NULL, NULL, basead, NULL);
 if(!hwnd) return -1;

 // create shell item
 HRESULT result = SHGetDesktopFolder(&lpShellFolder);
 if(SUCCEEDED(result)) {
    ULONG eaten = 0;
    result = lpShellFolder->ParseDisplayName(NULL, NULL, lpShellFolderPath, &eaten, &lpShellFolderIDList, NULL);
    if(SUCCEEDED(result)) {
       result = SHCreateItemFromIDList(lpShellFolderIDList, IID_IShellItem, (LPVOID*)&lpShellItem);
       if(SUCCEEDED(result)) SetWindowText(hwnd, lpShellFolderPath);
       else SetWindowText(hwnd, TEXT("SHCreateShellItem failed."));
      }
    else SetWindowText(hwnd, TEXT("ParseDisplayName failed."));
   }
 else
    SetWindowText(hwnd, TEXT("SHGetDesktopFolder failed."));

 // watch shell item
 PIDLIST_ABSOLUTE pidlWatch;
 result = SHGetIDListFromObject(lpShellItem, &pidlWatch);
 if(SUCCEEDED(result)) {
    INT sources = SHCNRF_InterruptLevel | SHCNRF_RecursiveInterrupt | SHCNRF_NewDelivery;
    LONG events = SHCNE_ALLEVENTS;
    SHChangeNotifyEntry entry = { pidlWatch, TRUE };
    watchcode = SHChangeNotifyRegister(hwnd, sources, events, WM_SHELLCHANGENOTIFY, 1, &entry);
    if(watchcode != 0) SetWindowText(hwnd, TEXT("SUCCESSFULLY WATCHING!"));
   }

 // message loop
 MSG msg;
 for(;;) {
     BOOL status = GetMessage(&msg, NULL, 0, 0);
     if(status == -1) return -1;
     else if(status != 0) {
        TranslateMessage(&msg);
        DispatchMessage(&msg);
       }
     else
        break;
    }
 return (int)msg.wParam;
}

int WINAPI WinMain(HINSTANCE handle, HINSTANCE, LPSTR, int)
{
 HRESULT result = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
 if(FAILED(result)) return -1;
 int retval = run(handle);
 CoUninitialize();
 return retval;
}

LRESULT CALLBACK MainWindowProc(HWND window, UINT message, WPARAM wparam, LPARAM lparam)
{
 switch(message) {
   case(WM_NCCREATE): return EvNCCreate(window, wparam, lparam);
   case(WM_CREATE): return EvCreate(window, wparam, lparam);
   case(WM_DESTROY): return EvDestroy(window, wparam, lparam);
   case(WM_NCDESTROY): return EvNCDestroy(window, wparam, lparam);
   case(WM_PAINT): return EvPaint(window, wparam, lparam);
   case(WM_SHELLCHANGENOTIFY): return EvShellChangeNotify(window, wparam, lparam);
   default: return DefWindowProc(window, message, wparam, lparam);
  }
 return 0;
}

LRESULT EvNCCreate(HWND window, WPARAM wparam, LPARAM lparam)
{
 mainhwnd = window;
 return DefWindowProc(window, WM_NCCREATE, wparam, lparam);
}

LRESULT EvCreate(HWND window, WPARAM wparam, LPARAM lparam)
{
 LRESULT retval = DefWindowProc(window, WM_CREATE, wparam, lparam);
 if(retval == -1) return -1;
 return retval;
}

LRESULT EvDestroy(HWND window, WPARAM wparam, LPARAM lparam)
{
 // stop watching
 if(watchcode) {
    SHChangeNotifyDeregister(watchcode);
    watchcode = 0;
   }

 // cleanup shell variables
 if(lpShellFolderIDList) {
    CoTaskMemFree((LPVOID)lpShellFolderIDList);
    lpShellFolderIDList = NULL;
   }
 if(lpShellItem) {
    lpShellItem->Release();
    lpShellItem = NULL;
   }
 if(lpShellFolder) {
    lpShellFolder->Release();
    lpShellFolder = NULL;
   }

 // default processing
 return DefWindowProc(window, WM_DESTROY, wparam, lparam);
}

LRESULT EvNCDestroy(HWND window, WPARAM wparam, LPARAM lparam)
{
 LRESULT retval = DefWindowProc(window, WM_NCDESTROY, wparam, lparam);
 mainhwnd = NULL;
 PostQuitMessage(0);
 return retval;
}

LRESULT EvPaint(HWND window, WPARAM wparam, LPARAM lparam)
{
 PAINTSTRUCT ps;
 HDC winDC = BeginPaint(window, &ps);
 EndPaint(window, &ps);
 return 0;
}

LRESULT EvShellChangeNotify(HWND window, WPARAM wparam, LPARAM lparam)
{
 long notification = 0;
 PIDLIST_ABSOLUTE* lppidl = NULL;
 HANDLE lock = SHChangeNotification_Lock((HANDLE)wparam, (DWORD)lparam, &lppidl, &notification);
 if(lock) {
    std::basic_stringstream<TCHAR> ss;
    ss << TEXT("CODE: ");
    if(notification & SHCNE_CREATE) ss << TEXT("SHCNE_CREATE - ");
    else if(notification & SHCNE_DELETE) ss << TEXT("SHCNE_DELETE - ");
    else if(notification & SHCNE_RENAMEITEM) ss << TEXT("SHCNE_RENAMEITEM - ");
    else if(notification & SHCNE_UPDATEITEM) ss << TEXT("SHCNE_UPDATEITEM - ");
    else if(notification & SHCNE_MKDIR) ss << TEXT("SHCNE_MKDIR - ");
    else if(notification & SHCNE_RENAMEFOLDER) ss << TEXT("SHCNE_RENAMEFOLDER - ");
    else if(notification & SHCNE_UPDATEDIR) ss << TEXT("-SHCNE_UPDATEDIR - ");
    else ss << TEXT("Something else happened!");
    SetWindowText(window, ss.str().c_str());
    SHChangeNotification_Unlock(lock);
   }
 return 0;
}