bat-alert/src/main.zig

113 lines
3.8 KiB
Zig
Raw Normal View History

2023-10-10 18:03:39 +00:00
const std = @import("std");
2023-11-13 01:57:51 +00:00
const c = @cImport({
@cInclude("libnotify/notify.h");
});
2023-10-10 18:03:39 +00:00
pub const std_options = struct {
pub const log_level = .info;
};
2023-11-13 01:57:51 +00:00
const NOTIFICATION_TITLE: [:0]const u8 = "bat-alert";
const NOTIFICATION_BODY: [:0]const u8 = "battery running low!";
const ParseMode = enum { none, target, battery };
2023-10-10 18:03:39 +00:00
pub fn main() !void {
2023-11-13 01:57:51 +00:00
if (c.notify_init("bat-alert") == c.FALSE) {
std.log.err("notify_init failed", .{});
std.process.exit(1);
}
2023-10-10 18:03:39 +00:00
var target: u8 = 15;
2023-11-02 20:02:14 +00:00
var battery: []const u8 = "BAT0";
2023-10-10 18:03:39 +00:00
2023-11-13 02:32:18 +00:00
var argsIter = std.process.args();
2023-11-13 01:57:51 +00:00
var parse_mode = ParseMode.none;
2023-11-13 02:32:18 +00:00
while (argsIter.next()) |arg| {
2023-11-13 01:57:51 +00:00
switch (parse_mode) {
ParseMode.none => {
if (std.mem.eql(u8, arg, "--help")) {
const stdout = std.io.getStdOut().writer();
try stdout.print(
\\Usage: bat-alert [options]
\\
\\Options:
\\ -t, --target NUM Target battery level (default: 15)
\\ --battery NAME Battery name (default: BAT0)
\\
, .{});
std.process.exit(0);
}
if (std.mem.eql(u8, arg, "--target") or std.mem.eql(u8, arg, "-t")) {
parse_mode = ParseMode.target;
}
if (std.mem.eql(u8, arg, "--battery")) {
parse_mode = ParseMode.battery;
}
},
ParseMode.battery => {
battery = arg;
parse_mode = ParseMode.none;
},
ParseMode.target => {
const parsedTarget = std.fmt.parseInt(u8, arg, 10) catch {
std.log.err("cannot parse target '{s}' as int", .{arg});
std.process.exit(1);
};
target = parsedTarget;
parse_mode = ParseMode.none;
},
2023-11-02 20:02:14 +00:00
}
2023-10-10 18:03:39 +00:00
}
2023-11-13 01:57:51 +00:00
if (parse_mode != ParseMode.none) {
std.log.err("option requires an argument: --{s}", .{@tagName(parse_mode)});
std.process.exit(1);
}
2023-10-10 18:03:39 +00:00
var sent = false;
2023-11-13 02:32:18 +00:00
var batBuf: [256]u8 = undefined;
var capBuf: [8]u8 = undefined;
const batteryPath = try std.fmt.bufPrint(&batBuf, "/sys/class/power_supply/{s}/capacity", .{battery});
2023-10-10 18:03:39 +00:00
while (true) {
2023-11-13 01:57:51 +00:00
const capacityFile = std.fs.openFileAbsolute(batteryPath, .{}) catch |err| {
std.log.err("open {s}: {}", .{ batteryPath, err });
std.process.exit(1);
};
2023-11-13 02:32:18 +00:00
_ = try capacityFile.reader().readAll(&capBuf);
2023-10-10 18:03:39 +00:00
capacityFile.close();
2023-11-13 02:32:18 +00:00
var splat = std.mem.splitScalar(u8, &capBuf, '\n');
2023-10-10 18:03:39 +00:00
2023-11-13 02:32:18 +00:00
// std.log.info("battery: {s}", .{splat.first()});
2023-10-10 18:03:39 +00:00
2023-11-13 02:32:18 +00:00
const capacity = try std.fmt.parseInt(u8, splat.first(), 10);
2023-10-10 18:03:39 +00:00
// std.log.info("battery (but parsed): {}", .{capacity});
if (capacity <= target) {
if (!sent) {
sent = true;
2023-11-13 01:57:51 +00:00
// _ = try std.process.Child.exec(.{
// .allocator = allocator,
// .argv = &[_][]const u8{ "notify-send", "-t", "5000", "-u", "critical", NOTIFICATION_TITLE, NOTIFICATION_BODY },
// });
const notif = c.notify_notification_new(NOTIFICATION_TITLE, NOTIFICATION_BODY, null);
var g_err: [*c]c.GError = null;
const ret = c.notify_notification_show(notif, &g_err);
if (ret == c.FALSE) {
std.log.err("failed to send a notification: {s}", .{g_err.*.message});
}
2023-11-15 01:33:16 +00:00
c.g_object_unref(c.G_OBJECT(notif));
2023-11-13 01:57:51 +00:00
2023-10-10 18:03:39 +00:00
std.log.info("[NOTIFY] battery running low!", .{});
}
} else {
sent = false;
}
std.time.sleep(10 * 1e9);
}
}