From 9037f8265998576138a8a87ac021b4479f041705 Mon Sep 17 00:00:00 2001
From: LordMZTE <lord@mzte.de>
Date: Sat, 26 Aug 2023 15:04:30 +0200
Subject: [PATCH] mzteinit: add support for launch commands

---
 scripts/mzteinit/src/main.zig | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/scripts/mzteinit/src/main.zig b/scripts/mzteinit/src/main.zig
index aa99d48..ec5d509 100644
--- a/scripts/mzteinit/src/main.zig
+++ b/scripts/mzteinit/src/main.zig
@@ -4,6 +4,8 @@ const env = @import("env.zig");
 const run = @import("run.zig");
 const util = @import("util.zig");
 
+const msg = @import("message.zig").msg;
+
 pub const std_options = struct {
     pub const log_level = .debug;
     pub fn logFn(
@@ -54,13 +56,26 @@ fn tryMain() !void {
     defer _ = gpa.deinit();
     const alloc = gpa.allocator();
 
+    var launch_cmd: ?[][]const u8 = null;
+    defer if (launch_cmd) |cmd| alloc.free(cmd);
+
+    if (std.os.argv.len >= 2) {
+        if (!std.mem.eql(u8, std.mem.span(std.os.argv[1]), "cmd") or std.os.argv.len < 3)
+            return error.InvalidCommand;
+
+        launch_cmd = try alloc.alloc([]const u8, std.os.argv[2..].len);
+        for (launch_cmd.?, std.os.argv[2..]) |*arg, arg_in| {
+            arg.* = std.mem.span(arg_in);
+        }
+    }
+    
     var env_map = try std.process.getEnvMap(alloc);
     defer env_map.deinit();
 
     if (env_map.get("MZTEINIT")) |_| {
         try stdout.writer().writeAll("mzteinit running already, starting shell\n");
         try stdout.flush();
-        var child = std.ChildProcess.init(&.{"fish"}, alloc);
+        var child = std.ChildProcess.init(launch_cmd orelse &.{"fish"}, alloc);
         _ = try child.spawnAndWait();
         return;
     } else {
@@ -71,6 +86,13 @@ fn tryMain() !void {
         try env.populateSysdaemonEnvironment(&env_map);
     }
 
+    if (launch_cmd) |cmd| {
+        try msg("using launch command", .{});
+        var child = std.ChildProcess.init(cmd, alloc);
+        _ = try child.spawnAndWait();
+        return;
+    }
+
     while (true) {
         try stdout.writer().writeAll(util.ansi_clear);