mirror of
https://github.com/fastfetch-cli/fastfetch.git
synced 2026-09-13 02:42:09 +02:00
李通洲
106 lines
4.1 KiB
C++
106 lines
4.1 KiB
C++
extern "C"
|
|
{
|
|
#include "bluetooth.h"
|
|
}
|
|
#include "common/windows/wmi.hpp"
|
|
#include "common/windows/unicode.hpp"
|
|
#include "common/windows/util.hpp"
|
|
|
|
extern "C"
|
|
const char* ffBluetoothDetectBattery(FFlist* devices)
|
|
{
|
|
FFWmiQuery query(L"SELECT __PATH FROM Win32_PnPEntity WHERE Service = 'BthHFEnum'", nullptr, FFWmiNamespace::CIMV2);
|
|
if(!query)
|
|
return "Query WMI service failed";
|
|
|
|
IWbemClassObject* pInParams = nullptr;
|
|
on_scope_exit releaseInParams([&] { pInParams && pInParams->Release(); });
|
|
{
|
|
IWbemClassObject* pnpEntityClass = nullptr;
|
|
|
|
if (FAILED(query.pService->GetObjectW(bstr_t(L"Win32_PnPEntity"), 0, nullptr, &pnpEntityClass, nullptr)))
|
|
return "Failed to get PnP entity class";
|
|
on_scope_exit releasePnpEntityClass([&] { pnpEntityClass && pnpEntityClass->Release(); });
|
|
|
|
if (FAILED(pnpEntityClass->GetMethod(bstr_t(L"GetDeviceProperties"), 0, &pInParams, NULL)))
|
|
return "Failed to get GetDeviceProperties method";
|
|
|
|
FFWmiVariant devicePropertyKeys({ L"{104EA319-6EE2-4701-BD47-8DDBF425BBE5} 2", L"DEVPKEY_Bluetooth_DeviceAddress" });
|
|
if (FAILED(pInParams->Put(L"devicePropertyKeys", 0, &devicePropertyKeys, CIM_FLAG_ARRAY | CIM_STRING)))
|
|
return "Failed to put devicePropertyKeys";
|
|
}
|
|
|
|
while (FFWmiRecord record = query.next())
|
|
{
|
|
IWbemCallResult* pCallResult = nullptr;
|
|
|
|
if (FAILED(query.pService->ExecMethod(record.get(L"__PATH").bstrVal, bstr_t(L"GetDeviceProperties"), 0, nullptr, pInParams, nullptr, &pCallResult)))
|
|
continue;
|
|
on_scope_exit releaseCallResult([&] { pCallResult && pCallResult->Release(); });
|
|
|
|
IWbemClassObject* pResultObject = nullptr;
|
|
if (FAILED(pCallResult->GetResultObject((LONG) WBEM_INFINITE, &pResultObject)))
|
|
continue;
|
|
on_scope_exit releaseResultObject([&] { pResultObject && pResultObject->Release(); });
|
|
|
|
VARIANT propArray;
|
|
if (FAILED(pResultObject->Get(L"deviceProperties", 0, &propArray, nullptr, nullptr)))
|
|
continue;
|
|
on_scope_exit releasePropArray([&] { VariantClear(&propArray); });
|
|
|
|
if (propArray.vt != (VT_ARRAY | VT_UNKNOWN) ||
|
|
(propArray.parray->fFeatures & FADF_UNKNOWN) == 0 ||
|
|
propArray.parray->cDims != 1 ||
|
|
propArray.parray->rgsabound[0].cElements != 2
|
|
)
|
|
continue;
|
|
|
|
uint8_t batt = 0;
|
|
for (LONG i = 0; i < 2; i++)
|
|
{
|
|
IWbemClassObject* object = nullptr;
|
|
if (FAILED(SafeArrayGetElement(propArray.parray, &i, &object)))
|
|
continue;
|
|
|
|
FFWmiRecord rec(object);
|
|
auto data = rec.get(L"Data");
|
|
if (data.vt == VT_EMPTY)
|
|
break;
|
|
|
|
if (i == 0)
|
|
batt = data.get<uint8_t>();
|
|
else
|
|
{
|
|
FF_STRBUF_AUTO_DESTROY addr = ffStrbufCreateWSV(data.get<std::wstring_view>()); // MAC address without colon
|
|
if (__builtin_expect(addr.length != 12, 0))
|
|
continue;
|
|
|
|
FF_LIST_FOR_EACH(FFBluetoothResult, bt, *devices)
|
|
{
|
|
if (bt->address.length != 12 + 5)
|
|
continue;
|
|
|
|
if (addr.chars[0] == bt->address.chars[0] &&
|
|
addr.chars[1] == bt->address.chars[1] &&
|
|
addr.chars[2] == bt->address.chars[3] &&
|
|
addr.chars[3] == bt->address.chars[4] &&
|
|
addr.chars[4] == bt->address.chars[6] &&
|
|
addr.chars[5] == bt->address.chars[7] &&
|
|
addr.chars[6] == bt->address.chars[9] &&
|
|
addr.chars[7] == bt->address.chars[10] &&
|
|
addr.chars[8] == bt->address.chars[12] &&
|
|
addr.chars[9] == bt->address.chars[13] &&
|
|
addr.chars[10] == bt->address.chars[15] &&
|
|
addr.chars[11] == bt->address.chars[16])
|
|
{
|
|
bt->battery = batt;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return NULL;
|
|
}
|