prefer controlData.json over devices.txt

This commit is contained in:
Emil Lerch 2025-09-20 13:38:44 -07:00
parent aad54b0b93
commit 97aaee50af
Signed by: lobo
GPG key ID: A7B62D657EF764F8

View file

@ -7,7 +7,7 @@ const DeviceAction = enum {
toggle, toggle,
}; };
fn sendWemoCommand(ip: []const u8, action: DeviceAction, allocator: std.mem.Allocator) !void { fn sendWemoCommand(url_base: []const u8, action: DeviceAction, allocator: std.mem.Allocator) !void {
const state = switch (action) { const state = switch (action) {
.on => "1", .on => "1",
.off => "0", .off => "0",
@ -27,13 +27,13 @@ fn sendWemoCommand(ip: []const u8, action: DeviceAction, allocator: std.mem.Allo
, .{state}); , .{state});
defer allocator.free(soap_body); defer allocator.free(soap_body);
const url = try std.fmt.allocPrint(allocator, "http://{s}/upnp/control/basicevent1", .{ip}); const url = try std.fmt.allocPrint(allocator, "{s}/upnp/control/basicevent1", .{url_base});
defer allocator.free(url); defer allocator.free(url);
var client = std.http.Client{ .allocator = allocator }; var client = std.http.Client{ .allocator = allocator };
defer client.deinit(); defer client.deinit();
std.log.debug("sending wemo command to {s}, action={s}", .{ ip, @tagName(action) }); std.log.debug("sending wemo command to {s}, action={s}", .{ url_base, @tagName(action) });
const res = try client.fetch(.{ const res = try client.fetch(.{
.method = .POST, .method = .POST,
.payload = soap_body, .payload = soap_body,
@ -47,18 +47,72 @@ fn sendWemoCommand(ip: []const u8, action: DeviceAction, allocator: std.mem.Allo
if (res.status == .ok) { if (res.status == .ok) {
var stdout_writer = std.fs.File.stdout().writer(&.{}); var stdout_writer = std.fs.File.stdout().writer(&.{});
const stdout = &stdout_writer.interface; const stdout = &stdout_writer.interface;
try stdout.print("WeMo command sent: IP={s}, Action={s}\n", .{ ip, @tagName(action) }); try stdout.print("WeMo command sent: url={s}, Action={s}\n", .{ url_base, @tagName(action) });
} else { } else {
var stderr_writer = std.fs.File.stderr().writer(&.{}); var stderr_writer = std.fs.File.stderr().writer(&.{});
const stderr = &stderr_writer.interface; const stderr = &stderr_writer.interface;
try stderr.print("WeMo command sent: IP={s}, Action={s}\n", .{ ip, @tagName(action) }); try stderr.print("WeMo command sent: url={s}, Action={s}\n", .{ url_base, @tagName(action) });
} }
} }
fn loadDeviceConfig(allocator: std.mem.Allocator) !std.StringHashMap([]const u8) { fn loadDeviceConfig(allocator: std.mem.Allocator) !std.StringHashMap([]const u8) {
var devices = std.StringHashMap([]const u8).init(allocator); var devices = std.StringHashMap([]const u8).init(allocator);
const file = try std.fs.cwd().openFile("devices.txt", .{}); var stderr_writer = std.fs.File.stderr().writer(&.{});
const stderr = &stderr_writer.interface;
// Try controlData.json first
if (std.fs.cwd().openFile("controlData.json", .{})) |file| {
defer file.close();
const content = try file.readToEndAlloc(allocator, 1024 * 1024);
defer allocator.free(content);
const parsed = std.json.parseFromSlice(std.json.Value, allocator, content, .{}) catch |err| {
try stderr.print(
"Failed to parse controlData.json: {}. Ignoring controlData.json, looking for devices.txt",
.{err},
);
return loadDevicesFromTxt(allocator);
};
defer parsed.deinit();
const root = parsed.value.object;
const device_array = root.get("devices").?.array;
for (device_array.items) |device| {
const device_obj = device.object;
const name = device_obj.get("name").?.string;
const url = device_obj.get("url").?.string;
if (name.len > 0) {
const name_copy = try allocator.alloc(u8, name.len);
_ = std.ascii.lowerString(name_copy, name);
try devices.put(name_copy, try allocator.dupe(u8, url));
std.log.debug("Loaded device: '{s}' -> '{s}'\n", .{ name, url });
}
}
return devices;
} else |_| {
return loadDevicesFromTxt(allocator);
}
}
fn loadDevicesFromTxt(allocator: std.mem.Allocator) !std.StringHashMap([]const u8) {
var devices = std.StringHashMap([]const u8).init(allocator);
const file = std.fs.cwd().openFile("devices.txt", .{}) catch |err| switch (err) {
error.FileNotFound => {
var stderr_writer = std.fs.File.stderr().writer(&.{});
const stderr = &stderr_writer.interface;
try stderr.print(
\\Error: could not load configuration. Please make sure controlData.json or
\\devices.txt is available in the directory with this program
, .{});
return error.ConfigurationNotAvailable;
},
else => return err,
};
defer file.close(); defer file.close();
const content = try file.readToEndAlloc(allocator, 1024 * 1024); const content = try file.readToEndAlloc(allocator, 1024 * 1024);
@ -71,9 +125,11 @@ fn loadDeviceConfig(allocator: std.mem.Allocator) !std.StringHashMap([]const u8)
if (std.mem.indexOf(u8, trimmed, "=")) |eq_pos| { if (std.mem.indexOf(u8, trimmed, "=")) |eq_pos| {
const name = std.mem.trim(u8, trimmed[0..eq_pos], " \t"); const name = std.mem.trim(u8, trimmed[0..eq_pos], " \t");
const ip = std.mem.trim(u8, trimmed[eq_pos + 1 ..], " \t"); const url_base = std.mem.trim(u8, trimmed[eq_pos + 1 ..], " \t");
if (name.len > 0 and ip.len > 0) { if (name.len > 0 and url_base.len > 0) {
try devices.put(try allocator.dupe(u8, name), try allocator.dupe(u8, ip)); const name_copy = try allocator.alloc(u8, name.len);
_ = std.ascii.lowerString(name_copy, name);
try devices.put(name, try allocator.dupe(u8, url_base));
} }
} }
} }
@ -92,19 +148,23 @@ fn parseAction(action_words: [][]const u8) ?DeviceAction {
return null; return null;
} }
fn extractDevice(allocator: std.mem.Allocator, object_words: [][]const u8, devices: *std.StringHashMap([]const u8)) ?[]const u8 { fn extractDevice(allocator: std.mem.Allocator, object_words: [][]const u8, devices: *std.StringHashMap([]const u8)) !?[]const u8 {
if (object_words.len == 0) return null; if (object_words.len == 0) return null;
if (object_words.len == 1) { var total_size: usize = 0;
return devices.get(object_words[0]); for (object_words) |word|
} else if (object_words.len == 2) { total_size += word.len + 1;
var device_name = std.ArrayList(u8){}; var allocating_writer = try std.Io.Writer.Allocating.initCapacity(allocator, total_size);
defer device_name.deinit(allocator); defer allocating_writer.deinit();
device_name.appendSlice(allocator, object_words[0]) catch return null; const writer = &allocating_writer.writer;
device_name.append(allocator, ' ') catch return null; for (object_words, 1..) |word, i| {
device_name.appendSlice(allocator, object_words[1]) catch return null; var buf: [256]u8 = undefined;
return devices.get(device_name.items); const lower_string = std.ascii.lowerString(&buf, word);
try writer.print("{s}{s}", .{ lower_string, if (i < object_words.len) " " else "" });
} }
if (devices.get(allocating_writer.written())) |url_base| return url_base;
std.log.warn("Could not find device '{s}'", .{allocating_writer.written()});
return null; return null;
} }
@ -165,9 +225,15 @@ pub fn main() !void {
}; };
defer gpa.allocator().free(object_words); defer gpa.allocator().free(object_words);
std.debug.print("Object words: ", .{});
for (object_words) |word| {
std.debug.print("'{s}' ", .{word});
}
std.debug.print("\n", .{});
if (parseAction(action_words)) |action| { if (parseAction(action_words)) |action| {
if (extractDevice(allocator, object_words, &devices)) |ip| { if (try extractDevice(allocator, object_words, &devices)) |url_base| {
try sendWemoCommand(ip, action, gpa.allocator()); try sendWemoCommand(url_base, action, gpa.allocator());
} else { } else {
try stdout.print("Device not found in configuration\n", .{}); try stdout.print("Device not found in configuration\n", .{});
} }