diff options
| -rw-r--r-- | src/main.zig | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/src/main.zig b/src/main.zig index f8612f0..2694172 100644 --- a/src/main.zig +++ b/src/main.zig @@ -38,6 +38,7 @@ pub fn main(init: std.process.Init) !void { } = .{}; { + var payload_buf: [4096]u8 = undefined; var i: usize = 1; while (i < args.len) : (i += 1) { if (to_option.get(args[i])) |opt| { @@ -49,7 +50,9 @@ pub fn main(init: std.process.Init) !void { .relay => { i += 1; if (i < args.len) { - flags.relay = args[i]; + var w: Writer = .fixed(&payload_buf); + try w.printBase64(args[i]); + flags.relay = w.buffered(); } else { std.debug.print("-r/--relay requires a string\n", .{}); return error.InvalidArguments; @@ -67,10 +70,11 @@ pub fn main(init: std.process.Init) !void { .connect => { i += 1; if (i < args.len) { - flags.connect = args[i]; + var w: Writer = .fixed(&payload_buf); + try w.printBase64(args[i]); + flags.connect = w.buffered(); } else { - std.debug.print("-c/--connect requires a string\n", .{}); - return error.InvalidArguments; + flags.connect = ""; } }, } |