Compare commits

...

8 Commits

8 changed files with 946 additions and 36 deletions

58
.github/workflows/release.yml vendored Normal file
View File

@ -0,0 +1,58 @@
name: Build and Publish Release
on:
push:
tags:
- "v[0-9]+.[0-9]+.[0-9]+"
paths-ignore:
- "**.md"
- .github/workflows/build-and-release.yml
- .gitignore
- LICENSE
- img/**
jobs:
build-and-publish:
name: Build and Publish Release
permissions:
contents: write
runs-on: windows-latest
steps:
- name: Checkout source code
uses: actions/checkout@v4.1.0
- name: Setup workflow cache
uses: actions/cache@v4.1.0
with:
path: |
~/.cargo/bin/
~/.cargo/registry/index/
~/.cargo/registry/cache/
~/.cargo/git/db/
target/
key: windows-cargo-${{ runner.os }}-${{ hashFiles('**/Cargo.lock') }}
- name: Setup Rust stable toolchain
uses: actions-rs/toolchain@v1
with:
profile: minimal
toolchain: stable
target: x86_64-pc-windows-msvc
- name: Build
run: cargo build --release --target x86_64-pc-windows-msvc
- name: Upload workflow artifact
uses: actions/upload-artifact@v4.1.0
with:
name: razer-battery-report
path: ./target/x86_64-pc-windows-msvc/release/razer-battery-report.exe
if-no-files-found: error
- name: Publish Release
uses: softprops/action-gh-release@v2.1.0
with:
files: ./target/x86_64-pc-windows-msvc/release/razer-battery-report.exe
draft: true
fail_on_unmatched_files: true

775
Cargo.lock generated

File diff suppressed because it is too large Load Diff

View File

@ -1,6 +1,6 @@
[package]
name = "razer-battery-report"
version = "0.2.3"
version = "0.3.0"
authors = ["xzeldon <contact@zeldon.ru>"]
edition = "2021"
description = "Razer Battery Level Tray Indicator"
@ -38,3 +38,6 @@ winapi = { version = "0.3.9", features = ["winuser", "wincon", "consoleapi"] }
# Efficient synchronization primitives (e.g. Mutex, RwLock and etc.)
parking_lot = "0.12"
# Desktop notifications
notify-rust = "4.11.4"

View File

@ -10,15 +10,17 @@
Show your wireless Razer devices battery levels in your system tray.
> This is a work in progress and currently support only **Razer DeathAdder V3 Pro**.
> This is a work in progress and currently support only **Razer DeathAdder V3 Pro** and **Razer DeathAdder V3 HyperSpeed**.
> Currently, this works only on **Windows**, should work on **Linux** if you _add udev rule to get access to usb devices_ (see [here](https://github.com/libusb/hidapi/blob/master/udev/69-hid.rules)) and remove/`cfg(windows)` some platform-specific code. But I haven't tested yet.
> Currently, this works only on **Windows**.
## Usage
### Downloading a Prebuilt Binary
### Installation
> _Todo_
1. Download `razer-battery-report.exe` from [latest release](https://github.com/xzeldon/razer-battery-report/releases/latest)
2. Run `razer-battery-report.exe`
3. If you want a start menu shortcut you can make one yourself! Simply right-click `razer-battery-report.exe` and select "Pin to Start". This will automatically create a shortcut in %appdata%\Microsoft\Windows\Start Menu\Programs.
### Building from Source
@ -41,10 +43,10 @@ To build, you must have [Rust](https://www.rust-lang.org/) and
- [x] Tray Applet
- [ ] Force update devices button in tray menu
- [ ] Colored tray icons for different battery levels
- [x] Colored tray icons for different battery levels
- [x] Show log window button in tray menu
- [x] Further reduce CPU usage by using Event Loop Proxy events (more info [here](https://github.com/tauri-apps/tray-icon/issues/83#issuecomment-1697773065))
- [ ] Prebuilt Binary
- [x] Prebuilt Binary
- [ ] Command Line Arguments for update frequency
- [ ] Support for other Razer Devices (I only have DeathAdder V3 Pro, so I won't be able to test it with other devices)

View File

@ -28,7 +28,9 @@ impl DeviceInfo {
pub const fn transaction_id(&self) -> u8 {
match self.pid {
pid if pid == RAZER_DEATHADDER_V3_PRO_WIRED.pid
|| pid == RAZER_DEATHADDER_V3_PRO_WIRELESS.pid =>
|| pid == RAZER_DEATHADDER_V3_PRO_WIRELESS.pid
|| pid == RAZER_DEATHADDER_V3_HYPERSPEED_WIRED.pid
|| pid == RAZER_DEATHADDER_V3_HYPERSPEED_WIRELESS.pid =>
{
0x1F
}
@ -38,11 +40,18 @@ impl DeviceInfo {
}
pub const RAZER_DEATHADDER_V3_PRO_WIRED: DeviceInfo =
DeviceInfo::new("Razer DeathAdder V3 Pro", 0x00B6, 0, 1, 2);
DeviceInfo::new("Razer DeathAdder V3 Pro (Wired)", 0x00B6, 0, 1, 2);
pub const RAZER_DEATHADDER_V3_PRO_WIRELESS: DeviceInfo =
DeviceInfo::new("Razer DeathAdder V3 Pro", 0x00B7, 0, 1, 2);
DeviceInfo::new("Razer DeathAdder V3 Pro (Wireless)", 0x00B7, 0, 1, 2);
pub const RAZER_DEVICE_LIST: [DeviceInfo; 2] = [
pub const RAZER_DEATHADDER_V3_HYPERSPEED_WIRED: DeviceInfo =
DeviceInfo::new("Razer DeathAdder V3 HyperSpeed (Wired)", 0x00C4, 0, 1, 2);
pub const RAZER_DEATHADDER_V3_HYPERSPEED_WIRELESS: DeviceInfo =
DeviceInfo::new("Razer DeathAdder V3 HyperSpeed (Wireless)", 0x00C5, 0, 1, 2);
pub const RAZER_DEVICE_LIST: [DeviceInfo; 4] = [
RAZER_DEATHADDER_V3_PRO_WIRED,
RAZER_DEATHADDER_V3_PRO_WIRELESS,
RAZER_DEATHADDER_V3_HYPERSPEED_WIRED,
RAZER_DEATHADDER_V3_HYPERSPEED_WIRELESS,
];

View File

@ -7,6 +7,7 @@ mod console;
mod controller;
mod devices;
mod manager;
mod notify;
mod tray;
fn main() {

53
src/notify.rs Normal file
View File

@ -0,0 +1,53 @@
use notify_rust::Notification;
pub struct Notify {
app_name: String,
}
impl Notify {
pub fn new() -> Self {
#[cfg(target_os = "windows")]
Self {
app_name: String::from("Razer Battery Report"),
}
}
pub fn battery_low(
&self,
device_name: &str,
battery_level: i32,
) -> Result<(), Box<dyn std::error::Error>> {
Notification::new()
.summary(&self.app_name)
.body(&format!(
"{}: Battery low ({}%)",
device_name, battery_level
))
.show()?;
Ok(())
}
pub fn battery_full(&self, device_name: &str) -> Result<(), Box<dyn std::error::Error>> {
Notification::new()
.summary(&self.app_name)
.body(&format!("{}: Battery fully charged", device_name))
.show()?;
Ok(())
}
pub fn device_connected(&self, device_name: &str) -> Result<(), Box<dyn std::error::Error>> {
Notification::new()
.summary(&self.app_name)
.body(&format!("{}: Connected", device_name))
.show()?;
Ok(())
}
pub fn device_disconnecred(&self, device_name: &str) -> Result<(), Box<dyn std::error::Error>> {
Notification::new()
.summary(&self.app_name)
.body(&format!("{}: Disconnected", device_name))
.show()?;
Ok(())
}
}

View File

@ -5,7 +5,7 @@ use std::{
time::Duration,
};
use crate::{console::DebugConsole, manager::DeviceManager};
use crate::{console::DebugConsole, manager::DeviceManager, notify::Notify};
use log::{error, info, trace};
use parking_lot::Mutex;
use tao::event_loop::{EventLoopBuilder, EventLoopProxy};
@ -17,6 +17,9 @@ use tray_icon::{
const BATTERY_UPDATE_INTERVAL: Duration = Duration::from_secs(300); // 5 min
const DEVICE_FETCH_INTERVAL: Duration = Duration::from_secs(5);
const BATTERY_CRITICAL_LEVEL: i32 = 5;
const BATTERY_LOW_LEVEL: i32 = 15;
#[derive(Debug)]
pub struct MemoryDevice {
pub name: String,
@ -82,7 +85,7 @@ impl TrayInner {
) {
let tray_builder = TrayIconBuilder::new()
.with_menu(Box::new(tray_menu.clone()))
.with_tooltip("Service is running")
.with_tooltip("Search for devices")
.with_icon(icon)
.build();
@ -97,6 +100,7 @@ pub struct TrayApp {
device_manager: Arc<Mutex<DeviceManager>>,
devices: Arc<Mutex<HashMap<u32, MemoryDevice>>>,
tray_inner: TrayInner,
notify: Arc<Notify>,
}
#[derive(Debug)]
@ -111,6 +115,7 @@ impl TrayApp {
device_manager: Arc::new(Mutex::new(DeviceManager::new())),
devices: Arc::new(Mutex::new(HashMap::new())),
tray_inner: TrayInner::new(Arc::new(debug_console)),
notify: Arc::new(Notify::new()),
}
}
@ -141,6 +146,7 @@ impl TrayApp {
fn spawn_device_fetch_thread(&self, proxy: EventLoopProxy<TrayEvent>) {
let devices = Arc::clone(&self.devices);
let device_manager = Arc::clone(&self.device_manager);
let notify = Arc::clone(&self.notify);
thread::spawn(move || {
let mut last_devices = HashSet::new();
@ -154,6 +160,7 @@ impl TrayApp {
for id in removed_devices {
if let Some(device) = devices.remove(&id) {
info!("Device removed: {}", device.name);
let _ = notify.device_disconnecred(&device.name);
}
}
@ -162,6 +169,7 @@ impl TrayApp {
if let Some(name) = device_manager.lock().get_device_name(id) {
devices.insert(id, MemoryDevice::new(name.clone(), id));
info!("New device: {}", name);
let _ = notify.device_connected(&name);
} else {
error!("Failed to get device name for id: {}", id);
}
@ -201,6 +209,7 @@ impl TrayApp {
let tray_icon = Arc::clone(&self.tray_inner.tray_icon);
let debug_console = Arc::clone(&self.tray_inner.debug_console);
let menu_items = Arc::clone(&self.tray_inner.menu_items);
let notify = Arc::clone(&self.notify);
let menu_channel = MenuEvent::receiver();
@ -212,7 +221,7 @@ impl TrayApp {
TrayInner::build_tray(&tray_icon, &tray_menu, icon.clone());
}
tao::event::Event::UserEvent(TrayEvent::DeviceUpdate(device_ids)) => {
Self::update(&devices, &device_manager, &device_ids, &tray_icon);
Self::update(&devices, &device_manager, &device_ids, &tray_icon, &notify);
}
tao::event::Event::UserEvent(TrayEvent::MenuEvent(event)) => {
let menu_items = menu_items.lock();
@ -241,11 +250,33 @@ impl TrayApp {
});
}
fn get_battery_icon(battery_level: i32, is_charging: bool) -> tray_icon::Icon {
let icon = match (battery_level, is_charging) {
(lvl, _) if lvl <= BATTERY_CRITICAL_LEVEL && !is_charging => {
include_bytes!("../assets/mouse_red.png").to_vec()
}
(lvl, _) if lvl <= BATTERY_LOW_LEVEL && !is_charging => {
include_bytes!("../assets/mouse_yellow.png").to_vec()
}
_ => include_bytes!("../assets/mouse_white.png").to_vec(),
};
let image = image::load_from_memory(&icon)
.expect("Failed to open icon")
.into_rgba8();
let (width, height) = image.dimensions();
let rgba = image.into_raw();
tray_icon::Icon::from_rgba(rgba, width, height).expect("Failed to create icon")
}
fn update(
devices: &Arc<Mutex<HashMap<u32, MemoryDevice>>>,
manager: &Arc<Mutex<DeviceManager>>,
device_ids: &[u32],
tray_icon: &Arc<Mutex<Option<TrayIcon>>>,
notify: &Arc<Notify>,
) {
let mut devices = devices.lock();
let manager = manager.lock();
@ -263,7 +294,18 @@ impl TrayApp {
device.battery_level = battery_level;
device.is_charging = is_charging;
Self::check_notify(device);
Self::check_notify(device, notify);
if device.old_battery_level != battery_level
|| device.is_charging != is_charging
{
let new_icon = Self::get_battery_icon(battery_level, is_charging);
if let Some(tray_icon) = tray_icon.lock().as_mut() {
tray_icon
.set_icon(Some(new_icon))
.expect("Failed to update tray icon");
}
}
if let Some(tray_icon) = tray_icon.lock().as_mut() {
let _ = tray_icon
@ -274,16 +316,18 @@ impl TrayApp {
}
}
fn check_notify(device: &MemoryDevice) {
fn check_notify(device: &MemoryDevice, notify: &Notify) {
if device.battery_level == -1 {
return;
}
if !device.is_charging
&& (device.battery_level <= 5
|| (device.old_battery_level > 15 && device.battery_level <= 15))
&& (device.battery_level <= BATTERY_CRITICAL_LEVEL
|| (device.old_battery_level > BATTERY_LOW_LEVEL
&& device.battery_level <= BATTERY_LOW_LEVEL))
{
info!("{}: Battery low ({}%)", device.name, device.battery_level);
let _ = notify.battery_low(&device.name, device.battery_level);
} else if device.old_battery_level <= 99
&& device.battery_level == 100
&& device.is_charging
@ -292,6 +336,7 @@ impl TrayApp {
"{}: Battery fully charged ({}%)",
device.name, device.battery_level
);
let _ = notify.battery_full(&device.name);
}
}
}