mirror of https://github.com/electron/electron
94 lines
2.7 KiB
C++
94 lines
2.7 KiB
C++
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
|
|
// Copyright (c) 2015 Felix Rieseberg <feriese@microsoft.com> and
|
|
// Jason Poon <jason.poon@microsoft.com>. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE-CHROMIUM file.
|
|
|
|
#include "shell/browser/notifications/win/notification_presenter_win.h"
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "base/environment.h"
|
|
#include "base/files/file_util.h"
|
|
#include "base/hash/md5.h"
|
|
#include "base/logging.h"
|
|
#include "base/strings/utf_string_conversions.h"
|
|
#include "base/time/time.h"
|
|
#include "shell/browser/notifications/win/windows_toast_notification.h"
|
|
#include "shell/common/thread_restrictions.h"
|
|
#include "third_party/skia/include/core/SkBitmap.h"
|
|
#include "ui/gfx/codec/png_codec.h"
|
|
|
|
#pragma comment(lib, "runtimeobject.lib")
|
|
|
|
namespace electron {
|
|
|
|
namespace {
|
|
|
|
bool IsDebuggingNotifications() {
|
|
return base::Environment::Create()->HasVar("ELECTRON_DEBUG_NOTIFICATIONS");
|
|
}
|
|
|
|
bool SaveIconToPath(const SkBitmap& bitmap, const base::FilePath& path) {
|
|
std::vector<unsigned char> png_data;
|
|
if (!gfx::PNGCodec::EncodeBGRASkBitmap(bitmap, false, &png_data))
|
|
return false;
|
|
|
|
return base::WriteFile(path, png_data);
|
|
}
|
|
|
|
} // namespace
|
|
|
|
// static
|
|
NotificationPresenter* NotificationPresenter::Create() {
|
|
if (!WindowsToastNotification::Initialize())
|
|
return nullptr;
|
|
auto presenter = std::make_unique<NotificationPresenterWin>();
|
|
if (!presenter->Init())
|
|
return nullptr;
|
|
|
|
if (IsDebuggingNotifications())
|
|
LOG(INFO) << "Successfully created Windows notifications presenter";
|
|
|
|
return presenter.release();
|
|
}
|
|
|
|
NotificationPresenterWin::NotificationPresenterWin() = default;
|
|
|
|
NotificationPresenterWin::~NotificationPresenterWin() = default;
|
|
|
|
bool NotificationPresenterWin::Init() {
|
|
ScopedAllowBlockingForElectron allow_blocking;
|
|
return temp_dir_.CreateUniqueTempDir();
|
|
}
|
|
|
|
std::wstring NotificationPresenterWin::SaveIconToFilesystem(
|
|
const SkBitmap& icon,
|
|
const GURL& origin) {
|
|
std::string filename;
|
|
|
|
if (origin.is_valid()) {
|
|
filename = base::MD5String(origin.spec()) + ".png";
|
|
} else {
|
|
const int64_t now_usec = base::Time::Now().since_origin().InMicroseconds();
|
|
filename = base::NumberToString(now_usec) + ".png";
|
|
}
|
|
|
|
ScopedAllowBlockingForElectron allow_blocking;
|
|
base::FilePath path = temp_dir_.GetPath().Append(base::UTF8ToWide(filename));
|
|
if (base::PathExists(path))
|
|
return path.value();
|
|
if (SaveIconToPath(icon, path))
|
|
return path.value();
|
|
return base::UTF8ToWide(origin.spec());
|
|
}
|
|
|
|
Notification* NotificationPresenterWin::CreateNotificationObject(
|
|
NotificationDelegate* delegate) {
|
|
return new WindowsToastNotification(delegate, this);
|
|
}
|
|
|
|
} // namespace electron
|