main.cpp

// C++ Windows Tray Screenshot App - Step 1: Tray icon with Exit


/*
#include <windows.h>
#include <shellapi.h>
#include <gdiplus.h>
#include <sstream>
#include <iomanip>
#include <direct.h> // for _mkdir

#pragma comment(lib, "Gdiplus.lib")

#define WM_TRAYICON (WM_USER + 1)
#define ID_TRAY_EXIT 1001

using namespace Gdiplus;

HHOOK keyboardHook;
HWND hWnd;
NOTIFYICONDATA nid;
ULONG_PTR gdiplusToken;

// Forward declarations
void AddTrayIcon(HWND hWnd);
void RemoveTrayIcon();
void ShowTrayMenu(HWND hWnd);
void CaptureScreenToPNG();
std::string GetTimestampString();
void InitScreenshotFolder();
int GetEncoderClsid(const WCHAR* format, CLSID* pClsid);




// Keyboard hook callback
LRESULT CALLBACK LowLevelKeyboardProc(int nCode, WPARAM wParam, LPARAM lParam) {
    if (nCode == HC_ACTION) {
        KBDLLHOOKSTRUCT* p = (KBDLLHOOKSTRUCT*)lParam;
        if (wParam == WM_KEYDOWN && p->vkCode == VK_SNAPSHOT) {
            CaptureScreenToPNG();
            return 1; // block default behavior
        }
    }
    return CallNextHookEx(NULL, nCode, wParam, lParam);
}

// Get a timestamp-based filename
std::string GetTimestampString() {
    SYSTEMTIME st;
    GetLocalTime(&st);
    std::ostringstream oss;
    oss << st.wYear << "-"
        << std::setw(2) << std::setfill('0') << st.wMonth << "-"
        << std::setw(2) << std::setfill('0') << st.wDay << "-"
        << std::setw(2) << std::setfill('0') << st.wHour << "-"
        << std::setw(2) << std::setfill('0') << st.wMinute << "-"
        << std::setw(2) << std::setfill('0') << st.wSecond << "-"
        << std::setw(3) << std::setfill('0') << st.wMilliseconds;
    return oss.str();
}

// Create C:\Screenshots folder if not exists
void InitScreenshotFolder() {
    _mkdir("C:\\Screenshots");
}

void CaptureScreenToPNG() {
    // Logical resolution
    int width  = GetSystemMetrics(SM_CXSCREEN);
    int height = GetSystemMetrics(SM_CYSCREEN);

    // Manual DPI scaling to get real pixel resolution
    HDC hScreen = GetDC(NULL);
    int dpiX = GetDeviceCaps(hScreen, LOGPIXELSX);
    int dpiY = GetDeviceCaps(hScreen, LOGPIXELSY);
    float scaleX = dpiX / 96.0f;
    float scaleY = dpiY / 96.0f;
    int actualWidth  = (int)(width  * scaleX);
    int actualHeight = (int)(height * scaleY);

    HDC hMem = CreateCompatibleDC(hScreen);
    HBITMAP hBmp = CreateCompatibleBitmap(hScreen, actualWidth, actualHeight);
    SelectObject(hMem, hBmp);
    BitBlt(hMem, 0, 0, actualWidth, actualHeight, hScreen, 0, 0, SRCCOPY);

    // Convert HBITMAP to GDI+ Bitmap
    Bitmap* bmp = Bitmap::FromHBITMAP(hBmp, NULL);

    std::string ts = GetTimestampString();
    std::string filePath = "C:\\Screenshots\\" + ts + ".png";

    // CLSID for PNG encoder
    CLSID pngClsid;
    GetEncoderClsid(L"image/png", &pngClsid);

    // Convert file path to wide string
    wchar_t wFile[260];
    MultiByteToWideChar(CP_UTF8, 0, filePath.c_str(), -1, wFile, 260);

    bmp->Save(wFile, &pngClsid, NULL);

    delete bmp;
    DeleteObject(hBmp);
    DeleteDC(hMem);
    ReleaseDC(NULL, hScreen);

    // Optional confirmation:
    // MessageBox(NULL, "Saved PNG", "OK", MB_OK);
}

// Get CLSID of PNG encoder
int GetEncoderClsid(const WCHAR* format, CLSID* pClsid) {
    UINT num = 0, size = 0;
    GetImageEncodersSize(&num, &size);
    if (size == 0) return -1;

    ImageCodecInfo* pImageCodecInfo = (ImageCodecInfo*)malloc(size);
    GetImageEncoders(num, size, pImageCodecInfo);

    for (UINT j = 0; j < num; ++j) {
        if (wcscmp(pImageCodecInfo[j].MimeType, format) == 0) {
            *pClsid = pImageCodecInfo[j].Clsid;
            free(pImageCodecInfo);
            return 0;
        }
    }
    free(pImageCodecInfo);
    return -1;
}

// Tray icon management
void AddTrayIcon(HWND hWnd) {
    ZeroMemory(&nid, sizeof(nid));
    nid.cbSize = sizeof(nid);
    nid.hWnd   = hWnd;
    nid.uID    = 1;
    nid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
    nid.uCallbackMessage = WM_TRAYICON;
    nid.hIcon = LoadIcon(NULL, IDI_APPLICATION);
    strcpy(nid.szTip, "Auto Screenshot App");
    Shell_NotifyIcon(NIM_ADD, &nid);
}

void RemoveTrayIcon() {
    Shell_NotifyIcon(NIM_DELETE, &nid);
}

void ShowTrayMenu(HWND hWnd) {
    HMENU menu = CreatePopupMenu();
    InsertMenu(menu, -1, MF_BYPOSITION, ID_TRAY_EXIT, "Exit");
    POINT pt;
    GetCursorPos(&pt);
    SetForegroundWindow(hWnd);
    TrackPopupMenu(menu, TPM_BOTTOMALIGN | TPM_LEFTALIGN, pt.x, pt.y, 0, hWnd, NULL);
    DestroyMenu(menu);
}

// Window Procedure
LRESULT CALLBACK WndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {
    switch (msg) {
    case WM_TRAYICON:
        if (lParam == WM_RBUTTONUP)
            ShowTrayMenu(hwnd);
        break;
    case WM_COMMAND:
        if (LOWORD(wParam) == ID_TRAY_EXIT) {
            RemoveTrayIcon();
            PostQuitMessage(0);
        }
        break;
    case WM_DESTROY:
        RemoveTrayIcon();
        PostQuitMessage(0);
        break;
    default:
        return DefWindowProc(hwnd, msg, wParam, lParam);
    }
    return 0;
}

// WinMain
int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE, LPSTR, int) {
    // Initialize GDI+
    GdiplusStartupInput gdiplusStartupInput;
    GdiplusStartup(&gdiplusToken, &gdiplusStartupInput, NULL);

    InitScreenshotFolder();

    const char CLASS_NAME[] = "HiddenTrayApp";
    WNDCLASS wc = { 0 };
    wc.lpfnWndProc = WndProc;
    wc.hInstance = hInstance;
    wc.lpszClassName = CLASS_NAME;
    RegisterClass(&wc);

    hWnd = CreateWindowEx(0, CLASS_NAME, "TrayApp", 0, 0, 0, 0, 0, NULL, NULL, hInstance, NULL);

    AddTrayIcon(hWnd);

    keyboardHook = SetWindowsHookEx(WH_KEYBOARD_LL, LowLevelKeyboardProc, NULL, 0);

    MSG msg;
    while (GetMessage(&msg, NULL, 0, 0)) {
        TranslateMessage(&msg);
        DispatchMessage(&msg);
    }

    UnhookWindowsHookEx(keyboardHook);

    // Cleanup GDI+
    GdiplusShutdown(gdiplusToken);
    return 0;
}
*/

#include <windows.h>
#include <shellapi.h>
#include <gdiplus.h>
#include <sstream>
#include <iomanip>
#include <shlobj.h>
#include <direct.h>
#include <fstream>

#pragma comment(lib, "Gdiplus.lib")

#define WM_TRAYICON (WM_USER + 1)
#define ID_TRAY_SETFOLDER 1000
#define ID_TRAY_EXIT 1001

using namespace Gdiplus;

HHOOK keyboardHook;
HWND hWnd;
NOTIFYICONDATA nid;
ULONG_PTR gdiplusToken;

std::string currentFolder = "C:\\Screenshots"; // default if no config

// ---------------- Forward Declarations
void AddTrayIcon(HWND hWnd);
void RemoveTrayIcon();
void ShowTrayMenu(HWND hWnd);
void CaptureScreenToPNG();
std::string GetTimestampString();
int GetEncoderClsid(const WCHAR* format, CLSID* pClsid);
void LoadFolderFromConfig();
void SaveFolderToConfig();
std::string BrowseForFolder(HWND owner);

// ------------------- File config
void LoadFolderFromConfig() {
    std::ifstream fin("folder.ini");
    if (fin) {
        std::getline(fin, currentFolder);
        fin.close();
    }
}

void SaveFolderToConfig() {
    std::ofstream fout("folder.ini");
    fout << currentFolder;
    fout.close();
}

// Folder browse dialog
std::string BrowseForFolder(HWND owner) {
    BROWSEINFO bi = { 0 };
    bi.lpszTitle = "Select Folder for Screenshots";
    LPITEMIDLIST pidl = SHBrowseForFolder(&bi);
    if (pidl != 0) {
        char path[MAX_PATH];
        if (SHGetPathFromIDList(pidl, path)) {
            return std::string(path);
        }
    }
    return "";
}

// Tray icon
void AddTrayIcon(HWND hWnd) {
    ZeroMemory(&nid, sizeof(nid));
    nid.cbSize = sizeof(nid);
    nid.hWnd = hWnd;
    nid.uID = 1;
    nid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
    nid.uCallbackMessage = WM_TRAYICON;
   // nid.hIcon = LoadIcon(NULL, IDI_APPLICATION);
    nid.hIcon = (HICON)LoadImage(NULL, "AutoCap.ico", IMAGE_ICON, 0, 0, LR_LOADFROMFILE);
    strcpy(nid.szTip, "Auto Screenshot App");
    Shell_NotifyIcon(NIM_ADD, &nid);
}

void RemoveTrayIcon() {
    Shell_NotifyIcon(NIM_DELETE, &nid);
}

// Time string function
std::string GetTimestampString() {
    SYSTEMTIME st;
    GetLocalTime(&st);
    std::ostringstream oss;
    oss << st.wYear << "-"
        << std::setw(2) << std::setfill('0') << st.wMonth << "-"
        << std::setw(2) << std::setfill('0') << st.wDay << "-"
        << std::setw(2) << std::setfill('0') << st.wHour << "-"
        << std::setw(2) << std::setfill('0') << st.wMinute << "-"
        << std::setw(2) << std::setfill('0') << st.wSecond << "-"
        << std::setw(3) << std::setfill('0') << st.wMilliseconds;
    return oss.str();
}

// Save screenshot as PNG
void CaptureScreenToPNG() {
    int width = GetSystemMetrics(SM_CXSCREEN);
    int height = GetSystemMetrics(SM_CYSCREEN);

    HDC hScreen = GetDC(NULL);
    int dpiX = GetDeviceCaps(hScreen, LOGPIXELSX);
    int dpiY = GetDeviceCaps(hScreen, LOGPIXELSY);
    float scaleX = dpiX / 96.0f;
    float scaleY = dpiY / 96.0f;
    int actualWidth = (int)(width * scaleX);
    int actualHeight = (int)(height * scaleY);

    HDC hMem = CreateCompatibleDC(hScreen);
    HBITMAP hBmp = CreateCompatibleBitmap(hScreen, actualWidth, actualHeight);
    SelectObject(hMem, hBmp);
    BitBlt(hMem, 0, 0, actualWidth, actualHeight, hScreen, 0, 0, SRCCOPY);

    Bitmap* bmp = Bitmap::FromHBITMAP(hBmp, NULL);

    // make sure folder exists
    _mkdir(currentFolder.c_str());

    std::string ts = GetTimestampString();
    std::string filePath = currentFolder + "\\" + ts + ".png";

    CLSID pngClsid;
    GetEncoderClsid(L"image/png", &pngClsid);
    wchar_t wFile[260];
    MultiByteToWideChar(CP_UTF8, 0, filePath.c_str(), -1, wFile, 260);
    bmp->Save(wFile, &pngClsid, NULL);

    delete bmp;
    DeleteObject(hBmp);
    DeleteDC(hMem);
    ReleaseDC(NULL, hScreen);
}

// Get PNG encoder CLSID
int GetEncoderClsid(const WCHAR* format, CLSID* pClsid) {
    UINT num = 0, size = 0;
    GetImageEncodersSize(&num, &size);
    if (size == 0) return -1;
    ImageCodecInfo* pImageCodecInfo = (ImageCodecInfo*)malloc(size);
    GetImageEncoders(num, size, pImageCodecInfo);
    for (UINT j = 0; j < num; ++j) {
        if (wcscmp(pImageCodecInfo[j].MimeType, format) == 0) {
            *pClsid = pImageCodecInfo[j].Clsid;
            free(pImageCodecInfo);
            return j;
        }
    }
    free(pImageCodecInfo);
    return -1;
}

// Hook for PrintScreen
LRESULT CALLBACK LowLevelKeyboardProc(int nCode, WPARAM wParam, LPARAM lParam) {
    if (nCode == HC_ACTION) {
        KBDLLHOOKSTRUCT* p = (KBDLLHOOKSTRUCT*)lParam;
        if (wParam == WM_KEYDOWN && p->vkCode == VK_SNAPSHOT) {
            CaptureScreenToPNG();
            return 1;
        }
    }
    return CallNextHookEx(NULL, nCode, wParam, lParam);
}

// Tray menu
void ShowTrayMenu(HWND hWnd) {
    HMENU menu = CreatePopupMenu();
    InsertMenu(menu, -1, MF_BYPOSITION, ID_TRAY_SETFOLDER, "Set Folder...");
    InsertMenu(menu, -1, MF_BYPOSITION, ID_TRAY_EXIT, "Exit");
    POINT pt;
    GetCursorPos(&pt);
    SetForegroundWindow(hWnd);
    TrackPopupMenu(menu, TPM_BOTTOMALIGN | TPM_LEFTALIGN, pt.x, pt.y, 0, hWnd, NULL);
    DestroyMenu(menu);
}

// Window procedure
LRESULT CALLBACK WndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {
    switch (msg) {
    case WM_TRAYICON:
        if (lParam == WM_RBUTTONUP) {
            ShowTrayMenu(hwnd);
        }
        break;
    case WM_COMMAND:
        if (LOWORD(wParam) == ID_TRAY_EXIT) {
            RemoveTrayIcon();
            PostQuitMessage(0);
        }
        else if (LOWORD(wParam) == ID_TRAY_SETFOLDER) {
            std::string sel = BrowseForFolder(hwnd);
            if (!sel.empty()) {
                currentFolder = sel;
                SaveFolderToConfig();
            }
        }
        break;
    case WM_DESTROY:
        RemoveTrayIcon();
        PostQuitMessage(0);
        break;
    default:
        return DefWindowProc(hwnd, msg, wParam, lParam);
    }
    return 0;
}

// WinMain
int APIENTRY WinMain(HINSTANCE hInst, HINSTANCE, LPSTR, int) {
    // INIT GDI+ once
    GdiplusStartupInput input;
    GdiplusStartup(&gdiplusToken, &input, NULL);

    // Load last used folder
    LoadFolderFromConfig();

    const char CLASS_NAME[] = "TrayCaptureApp";
    WNDCLASS wc = { 0 };
    wc.lpfnWndProc = WndProc;
    wc.hInstance   = hInst;
    wc.lpszClassName = CLASS_NAME;
    RegisterClass(&wc);

    hWnd = CreateWindowEx(0, CLASS_NAME, "TrayApp", 0, 0, 0, 0, 0, NULL, NULL, hInst, NULL);

    AddTrayIcon(hWnd);
    keyboardHook = SetWindowsHookEx(WH_KEYBOARD_LL, LowLevelKeyboardProc, NULL, 0);

    MSG msg;
    while (GetMessage(&msg, NULL, 0, 0)) {
        TranslateMessage(&msg);
        DispatchMessage(&msg);
    }

    UnhookWindowsHookEx(keyboardHook);
    GdiplusShutdown(gdiplusToken);
    return 0;
}