Status.cpp

#include "Status.h"
#include "Server.h"
#include "resource.h"
#include "Localization.h"

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

#include <cwchar>
#include <string>

extern NOTIFYICONDATA nid;

namespace
{
ServerStatus g_status = ServerStatus::Offline;
ULONGLONG g_startingUntil = 0;
bool g_startingDotVisible = true;


std::wstring GetStatusLabel()
{
    switch (g_status)
    {
    case ServerStatus::Starting:
        return Translate(L"menu.status.starting", L"Server: STARTING");
    case ServerStatus::Online:
        return Translate(L"menu.status.online", L"Server: ONLINE");
    default:
        return Translate(L"menu.status.offline", L"Server: OFFLINE");
    }
}


void ApplyTrayStatus()
{
    if (!nid.hWnd)
        return;

    switch (g_status)
    {
    case ServerStatus::Starting:
        nid.hIcon = LoadIcon(nullptr, IDI_WARNING);
        wcscpy_s(nid.szTip, Translate(L"tray.starting", L"AudioCpp Server - Starting").c_str());
        break;
    case ServerStatus::Online:
        nid.hIcon = LoadIcon(nullptr, IDI_INFORMATION);
        wcscpy_s(nid.szTip, Translate(L"tray.online", L"AudioCpp Server - Online").c_str());
        break;
    default:
        nid.hIcon = LoadIcon(nullptr, IDI_ERROR);
        wcscpy_s(nid.szTip, Translate(L"tray.offline", L"AudioCpp Server - Offline").c_str());
        break;
    }

    Shell_NotifyIconW(NIM_MODIFY, &nid);
}
}


void SetServerStatus(ServerStatus status)
{
    g_status = status;

    if (status == ServerStatus::Starting)
    {
        g_startingUntil = GetTickCount64() + 1500;
        g_startingDotVisible = true;
    }

    ApplyTrayStatus();
}


ServerStatus GetServerStatus()
{
    return g_status;
}


void UpdateTrayStatus()
{
    if (g_status == ServerStatus::Starting)
    {
        if (!IsServerRunning())
            g_status = ServerStatus::Offline;
        else if (GetTickCount64() >= g_startingUntil)
            g_status = ServerStatus::Online;
    }
    else if (g_status == ServerStatus::Online && !IsServerRunning())
    {
        g_status = ServerStatus::Offline;
    }
    else if (g_status == ServerStatus::Offline && IsServerRunning())
    {
        g_status = ServerStatus::Online;
    }

    ApplyTrayStatus();
}


void TickTrayStatus()
{
    if (g_status == ServerStatus::Starting)
        g_startingDotVisible = !g_startingDotVisible;

    UpdateTrayStatus();

    HWND foreground = GetForegroundWindow();
    wchar_t className[32]{};

    if (foreground && GetClassNameW(foreground, className, 32) > 0 &&
        wcscmp(className, L"#32768") == 0)
    {
        InvalidateRect(foreground, nullptr, TRUE);
    }
}


void MeasureStatusMenuItem(MEASUREITEMSTRUCT* item)
{
    if (!item)
        return;

    item->itemWidth = 190;
    item->itemHeight = 26;
}


void DrawStatusMenuItem(const DRAWITEMSTRUCT* item)
{
    if (!item)
        return;

    HDC dc = item->hDC;
    RECT bounds = item->rcItem;
    bool selected = (item->itemState & ODS_SELECTED) != 0;

    FillRect(dc, &bounds, GetSysColorBrush(selected ? COLOR_HIGHLIGHT : COLOR_MENU));

    COLORREF dotColor = RGB(220, 35, 35);

    if (g_status == ServerStatus::Online)
        dotColor = RGB(35, 180, 70);
    else if (g_status == ServerStatus::Starting)
        dotColor = g_startingDotVisible ? RGB(235, 185, 35) : RGB(115, 95, 20);

    RECT dot{
        bounds.left + 9,
        bounds.top + 8,
        bounds.left + 19,
        bounds.top + 18
    };

    HBRUSH dotBrush = CreateSolidBrush(dotColor);
    HBRUSH oldBrush = reinterpret_cast<HBRUSH>(SelectObject(dc, dotBrush));
    HPEN dotPen = CreatePen(PS_SOLID, 1, dotColor);
    HPEN oldPen = reinterpret_cast<HPEN>(SelectObject(dc, dotPen));
    Ellipse(dc, dot.left, dot.top, dot.right, dot.bottom);
    SelectObject(dc, oldPen);
    SelectObject(dc, oldBrush);
    DeleteObject(dotPen);
    DeleteObject(dotBrush);

    SetBkMode(dc, TRANSPARENT);
    SetTextColor(dc, GetSysColor(selected ? COLOR_HIGHLIGHTTEXT : COLOR_GRAYTEXT));

    RECT textBounds = bounds;
    textBounds.left += 28;
    std::wstring label = GetStatusLabel();
    DrawTextW(dc, label.c_str(), -1, &textBounds, DT_SINGLELINE | DT_VCENTER | DT_LEFT);
}


void NotifyServerEvent(const wchar_t* message)
{
    if (!message || !nid.hWnd)
        return;

    NOTIFYICONDATA notification = nid;
    notification.uFlags = NIF_INFO;
    wcscpy_s(notification.szInfoTitle, L"AudioCpp");
    wcscpy_s(notification.szInfo, message);
    notification.dwInfoFlags = NIIF_INFO;
    notification.uTimeout = 4000;

    Shell_NotifyIconW(NIM_MODIFY, &notification);
}