026.(娱乐)魔改浏览器-任务栏图标右上角加提示徽章

news2024/9/29 20:22:18

一、目标：

windows中，打开chromium，任务栏中会出现一个chromium的图标。
我们的目标是给这个图标的右上角，加上"有1条新消息"的小提示图标，也叫徽章(badge)
注意：本章节纯属娱乐，有需要的集帅可以学习模仿。

具体效果如下：

在这里插入图片描述

二、修改源码：

打开：\ui\views\view.cc

1.头部追加：

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

2.找到：

bool View::OnMousePressed(const ui::MouseEvent& event) {
  return false;
}

OnMousePressed()函数是可以点击事件，每次点击浏览器头部时都会触发这个函数。

3.替换为：

void UpdateTaskbarIcon(HWND hwnd, HICON hIcon) {
    ITaskbarList3* pTaskbarList = nullptr;
	HRESULT hr = CoCreateInstance(CLSID_TaskbarList, nullptr, CLSCTX_INPROC_SERVER, IID_PPV_ARGS(&pTaskbarList));
	if (SUCCEEDED(hr)) {
		pTaskbarList->SetOverlayIcon(hwnd, hIcon, L"有1条新消息");
		pTaskbarList->Release();  
		LOG(ERROR) << "SetOverlayIcon成功调用"; 
	}else{
		LOG(ERROR) << "ERRORERRORERROR"; 
	}
}

void SetTaskbarIconOverlay(HWND hwnd) {
	wchar_t className[256];
	GetClassName(hwnd, className, sizeof(className) / sizeof(wchar_t));
	LOG(ERROR) << "窗口类名"; 
	LOG(ERROR) << className; 
    LPCWSTR iconPath = L"C:/Users/Administrator/Desktop/favicon.ico";
	HICON hIcon = (HICON)LoadImage(NULL, iconPath, IMAGE_ICON, 16, 16, LR_LOADFROMFILE);
    if (!hIcon) {
		MessageBox(hwnd, L"无法加载图标。", L"错误", MB_OK | MB_ICONERROR);
    } else {
        UpdateTaskbarIcon(hwnd, hIcon);
    }
}


bool View::OnMousePressed(const ui::MouseEvent& event) {
	CoInitializeEx(nullptr, COINIT_APARTMENTTHREADED);
	
    LPCWSTR className = L"Chrome_WidgetWin_1";
    LPCWSTR windowName = nullptr; // 如果你不知道窗口的标题，可以设置为nullptr

    // 获取窗口句柄
    HWND hwnd = FindWindow(className, windowName);
	if (hwnd != NULL) {
		HWND parentHwnd = GetParent(hwnd);
		if (parentHwnd == NULL) {
			LOG(ERROR) << "hwnd 是一个顶级窗口"; 
		} else {
			LOG(ERROR) << "hwnd 不是一个顶级窗口"; 
		}
	}
	LOG(ERROR) << hwnd;
	
	wchar_t windowTitle[256];
	GetWindowText(hwnd, windowTitle, sizeof(windowTitle) / sizeof(wchar_t));
	LOG(ERROR) << "窗口标题"; 
	LOG(ERROR) << windowTitle; 
	
	bool isVisible = IsWindowVisible(hwnd);
	LOG(ERROR) << "isVisible"; 
	LOG(ERROR) << isVisible; 
	
	DWORD processId;
	GetWindowThreadProcessId(hwnd, &processId);
	LOG(ERROR) << "processId"; 
	LOG(ERROR) << processId; 

	SetTaskbarIconOverlay(hwnd);
	CoUninitialize();
  return false;
}

注意：

将ico图标位置(变量iconPath )替换成你图标的位置，必须是ico其他格式不行。
LOG(ERROR)是用来打印错误日志的，可以忽略
最终实现原理是调用win32编程api里的SetOverlayIcon()函数。

4.编译

ninja  -C  out/Default chrome

编译完成后，打开浏览器，一旦点击浏览器头部，图标就出现啦！

三、代码生成数字ico

有的同学想到要右上角希望是数字图标，我们总不能准备99张ico图标吧。
于是我们用代码在内存中生成ico

将上面的代码改成：

HICON CreateNumberIcon(int number) {
    if (number > 99) {
        number = 99;
    }

    // 创建一个16x16的位图
    HDC hdcScreen = GetDC(NULL);
    HDC hdcMem = CreateCompatibleDC(hdcScreen);
    BITMAPINFO bmi = {};
    bmi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
    bmi.bmiHeader.biWidth = 16;
    bmi.bmiHeader.biHeight = -16; // 负值表示自上而下
    bmi.bmiHeader.biPlanes = 1;
    bmi.bmiHeader.biBitCount = 32; // 32位带透明通道
    bmi.bmiHeader.biCompression = BI_RGB;
    bmi.bmiHeader.biSizeImage = 0;
    bmi.bmiHeader.biXPelsPerMeter = 0;
    bmi.bmiHeader.biYPelsPerMeter = 0;
    bmi.bmiHeader.biClrUsed = 0;
    bmi.bmiHeader.biClrImportant = 0;

    void* pBits;
    HBITMAP hBitmap = CreateDIBSection(hdcMem, &bmi, DIB_RGB_COLORS, &pBits, NULL, 0);
    HBITMAP hOldBitmap = (HBITMAP)SelectObject(hdcMem, hBitmap);

    // 设置背景为透明
    memset(pBits, 0, 16 * 16 * 4); // 初始化位图为透明

    // 设置字体和颜色
    HFONT hFont = CreateFont(14, 0, 0, 0, FW_BOLD, FALSE, FALSE, FALSE, DEFAULT_CHARSET, OUT_DEFAULT_PRECIS, CLIP_DEFAULT_PRECIS, CLEARTYPE_QUALITY, DEFAULT_PITCH | FF_DONTCARE, L"Arial");
    HFONT hOldFont = (HFONT)SelectObject(hdcMem, hFont);
    SetTextColor(hdcMem, RGB(255, 0, 0)); // 设置数字颜色为红色
    SetBkMode(hdcMem, TRANSPARENT);

    // 计算数字的居中位置
    std::wstring text = std::to_wstring(number);
    RECT rect = {0, 0, 16, 16};
    DrawText(hdcMem, text.c_str(), text.length(), &rect, DT_CENTER | DT_VCENTER | DT_SINGLELINE);

    // 清理
    SelectObject(hdcMem, hOldFont);
    DeleteObject(hFont);
    SelectObject(hdcMem, hOldBitmap);
    DeleteDC(hdcMem);
    ReleaseDC(NULL, hdcScreen);

    // 将位图转换为图标
    ICONINFO iconInfo = { TRUE, 0, 0, hBitmap, hBitmap };
    HICON hIcon = CreateIconIndirect(&iconInfo);
    DeleteObject(hBitmap);

    return hIcon;
}

void UpdateTaskbarIcon(HWND hwnd, HICON hIcon) {
    ITaskbarList3* pTaskbarList = nullptr;
	HRESULT hr = CoCreateInstance(CLSID_TaskbarList, nullptr, CLSCTX_INPROC_SERVER, IID_PPV_ARGS(&pTaskbarList));
	if (SUCCEEDED(hr)) {
		pTaskbarList->SetOverlayIcon(hwnd, hIcon, L"新消息");
		pTaskbarList->Release();  
		LOG(ERROR) << "SetOverlayIcon成功调用"; 
	}else{
		LOG(ERROR) << "ERRORERRORERROR"; 
	}
}


void SetTaskbarIconOverlay(HWND hwnd) {
	wchar_t className[256];
	GetClassName(hwnd, className, sizeof(className) / sizeof(wchar_t));
	LOG(ERROR) << "窗口类名"; 
	LOG(ERROR) << className; 
    //LPCWSTR iconPath = L"C:/Users/Administrator/Desktop/favicon.ico";
	//HICON hIcon = (HICON)LoadImage(NULL, iconPath, IMAGE_ICON, 16, 16, LR_LOADFROMFILE);
	HICON hIcon = CreateNumberIcon(72);
    if (!hIcon) {
		MessageBox(hwnd, L"无法加载图标。", L"错误", MB_OK | MB_ICONERROR);
    } else {
        UpdateTaskbarIcon(hwnd, hIcon);
    }
}


bool View::OnMousePressed(const ui::MouseEvent& event) {
	CoInitializeEx(nullptr, COINIT_APARTMENTTHREADED);
	
    LPCWSTR className = L"Chrome_WidgetWin_1";
    LPCWSTR windowName = nullptr; // 如果你不知道窗口的标题，可以设置为nullptr
    // 获取窗口句柄
    HWND hwnd = FindWindow(className, windowName);
	SetTaskbarIconOverlay(hwnd);
	
	bool isVisible = IsWindowVisible(hwnd);
	LOG(ERROR) << "isVisible"; 
	LOG(ERROR) << isVisible; 
	
	CoUninitialize();
  return false;
}