This commit is contained in:
ptrcnull 2023-11-13 02:57:51 +01:00
parent b57cbba805
commit d483eb1bcd
Signed by: ptrcnull
GPG key ID: 411F7B30801DD9CA
2 changed files with 74 additions and 38 deletions

View file

@ -24,6 +24,12 @@ pub fn build(b: *std.Build) void {
.optimize = optimize, .optimize = optimize,
}); });
exe.linkLibC();
exe.addIncludePath(.{ .cwd_relative = "/usr/include/glib-2.0" });
exe.addIncludePath(.{ .cwd_relative = "/usr/lib/glib-2.0/include" });
exe.addIncludePath(.{ .cwd_relative = "/usr/include/gdk-pixbuf-2.0" });
exe.linkSystemLibrary("notify");
// This declares intent for the executable to be installed into the // This declares intent for the executable to be installed into the
// standard location when the user invokes the "install" step (the default // standard location when the user invokes the "install" step (the default
// step when running `zig build`). // step when running `zig build`).

View file

@ -1,10 +1,23 @@
const std = @import("std"); const std = @import("std");
const c = @cImport({
@cInclude("libnotify/notify.h");
});
pub const std_options = struct { pub const std_options = struct {
pub const log_level = .info; pub const log_level = .info;
}; };
const NOTIFICATION_TITLE: [:0]const u8 = "bat-alert";
const NOTIFICATION_BODY: [:0]const u8 = "battery running low!";
const ParseMode = enum { none, target, battery };
pub fn main() !void { pub fn main() !void {
if (c.notify_init("bat-alert") == c.FALSE) {
std.log.err("notify_init failed", .{});
std.process.exit(1);
}
var gpa = std.heap.GeneralPurposeAllocator(.{}){}; var gpa = std.heap.GeneralPurposeAllocator(.{}){};
defer _ = gpa.deinit(); defer _ = gpa.deinit();
const allocator = gpa.allocator(); const allocator = gpa.allocator();
@ -15,48 +28,57 @@ pub fn main() !void {
var target: u8 = 15; var target: u8 = 15;
var battery: []const u8 = "BAT0"; var battery: []const u8 = "BAT0";
var parse_mode = ParseMode.none;
for (args, 0..) |arg, i| { for (args, 0..) |arg, i| {
if (std.mem.eql(u8, arg, "--help")) { _ = i;
const stdout = std.io.getStdOut().writer(); switch (parse_mode) {
try stdout.print( ParseMode.none => {
\\Usage: bat-alert [options] if (std.mem.eql(u8, arg, "--help")) {
\\ const stdout = std.io.getStdOut().writer();
\\Options: try stdout.print(
\\ -t, --target NUM Target battery level (default: 15) \\Usage: bat-alert [options]
\\ --battery NAME Battery name (default: BAT0) \\
\\ \\Options:
, .{}); \\ -t, --target NUM Target battery level (default: 15)
std.process.exit(0); \\ --battery NAME Battery name (default: BAT0)
} \\
, .{});
if (std.mem.eql(u8, arg, "--target") or std.mem.eql(u8, arg, "-t")) { std.process.exit(0);
if (i + 1 == args.len) { }
std.log.err("--target requires an argument", .{}); if (std.mem.eql(u8, arg, "--target") or std.mem.eql(u8, arg, "-t")) {
std.process.exit(1); parse_mode = ParseMode.target;
} }
if (std.mem.eql(u8, arg, "--battery")) {
const parsedTarget = std.fmt.parseInt(u8, args[i + 1], 10) catch |err| { parse_mode = ParseMode.battery;
std.log.err("invalid target '{s}': {}", .{ args[i + 1], err }); }
std.process.exit(1); },
}; ParseMode.battery => {
target = parsedTarget; battery = arg;
} parse_mode = ParseMode.none;
},
if (std.mem.eql(u8, arg, "--battery")) { ParseMode.target => {
if (i + 1 == args.len) { const parsedTarget = std.fmt.parseInt(u8, arg, 10) catch {
std.log.err("--battery requires an argument", .{}); std.log.err("cannot parse target '{s}' as int", .{arg});
std.process.exit(1); std.process.exit(1);
} };
target = parsedTarget;
battery = args[i + 1]; parse_mode = ParseMode.none;
},
} }
} }
if (parse_mode != ParseMode.none) {
std.log.err("option requires an argument: --{s}", .{@tagName(parse_mode)});
std.process.exit(1);
}
var sent = false; var sent = false;
const batteryPath = try std.fmt.allocPrint(allocator, "/sys/class/power_supply/{s}/capacity", .{battery}); const batteryPath = try std.fmt.allocPrint(allocator, "/sys/class/power_supply/{s}/capacity", .{battery});
while (true) { while (true) {
const capacityFile = try std.fs.openFileAbsolute(batteryPath, .{}); const capacityFile = std.fs.openFileAbsolute(batteryPath, .{}) catch |err| {
std.log.err("open {s}: {}", .{ batteryPath, err });
std.process.exit(1);
};
var capacityStr = std.ArrayList(u8).init(allocator); var capacityStr = std.ArrayList(u8).init(allocator);
try capacityFile.reader().streamUntilDelimiter(capacityStr.writer(), '\n', null); try capacityFile.reader().streamUntilDelimiter(capacityStr.writer(), '\n', null);
capacityFile.close(); capacityFile.close();
@ -69,10 +91,18 @@ pub fn main() !void {
if (capacity <= target) { if (capacity <= target) {
if (!sent) { if (!sent) {
sent = true; sent = true;
_ = try std.process.Child.exec(.{ // _ = try std.process.Child.exec(.{
.allocator = allocator, // .allocator = allocator,
.argv = &[_][]const u8{ "notify-send", "-t", "5000", "-u", "critical", "bat-alert", "battery running low!" }, // .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});
}
std.log.info("[NOTIFY] battery running low!", .{}); std.log.info("[NOTIFY] battery running low!", .{});
} }
} else { } else {