electron/brightray/browser/win/win32_notification.cc

#define WIN32_LEAN_AND_MEAN

#include "brightray/browser/win/win32_notification.h"

#include <windows.h>
#include <string>
#include <vector>

#include "third_party/skia/include/core/SkBitmap.h"

namespace brightray {

void Win32Notification::Show(const NotificationOptions& options) {
    auto presenter = static_cast<NotificationPresenterWin7*>(this->presenter());
    if (!presenter) return;

    HBITMAP image = NULL;

    if (!options.icon.drawsNothing()) {
        if (options.icon.colorType() == kBGRA_8888_SkColorType) {
            options.icon.lockPixels();

            BITMAPINFOHEADER bmi = { sizeof(BITMAPINFOHEADER) };
            bmi.biWidth = options.icon.width();
            bmi.biHeight = -options.icon.height();
            bmi.biPlanes = 1;
            bmi.biBitCount = 32;
            bmi.biCompression = BI_RGB;

            HDC hdcScreen = GetDC(NULL);
            image = CreateDIBitmap(hdcScreen, &bmi, CBM_INIT,
                                   options.icon.getPixels(),
                                   reinterpret_cast<BITMAPINFO*>(&bmi),
                                   DIB_RGB_COLORS);
            ReleaseDC(NULL, hdcScreen);

            options.icon.unlockPixels();
        }
    }

    Win32Notification* existing = nullptr;
    if (!options.tag.empty())
        existing = presenter->GetNotificationObjectByTag(options.tag);

    if (existing) {
        existing->tag_.clear();
        this->notification_ref_ = std::move(existing->notification_ref_);
        this->notification_ref_.Set(options.title, options.msg, image);
    } else {
        this->notification_ref_ = presenter->AddNotification(options.title,
                                                             options.msg,
                                                             image);
    }

    this->tag_ = options.tag;

    if (image) DeleteObject(image);
}

void Win32Notification::Dismiss() {
    notification_ref_.Close();
}

}   // namespace brightray