diff options
Diffstat (limited to 'pkgs')
| -rw-r--r-- | pkgs/yo.nix | 79 |
1 files changed, 62 insertions, 17 deletions
diff --git a/pkgs/yo.nix b/pkgs/yo.nix index 91f30f7..6846046 100644 --- a/pkgs/yo.nix +++ b/pkgs/yo.nix @@ -1,34 +1,79 @@ {pkgs, ...}: let yo = pkgs.writeText "yo.lua" '' - local verb + local SUDO = os.getenv("SUDO") or "sudo" - local flags = {} - if arg[2] ~= nil then - for i = 2, #arg do - table.insert(flags, arg[i]) + local function collectArgs(start) + local start = start or 2 + + local flags = {} + if arg[start] ~= nil then + for i = start, #arg do + table.insert(flags, arg[i]) + end + end + return flags + end + + local function nixosRebuild(args) + local sudo = (args.sudo or false) == true + local verb = args.verb or "switch" + local flake = args.flake or "/home/collin/nixos" + local extraArgs = table.concat(args.extraArgs or {}, " ") + + cmd = ("nixos-rebuild %s --flake %s %s --log-format internal-json |& ${pkgs.nix-output-monitor}/bin/nom --json"):format(verb, flake, extraArgs) + if sudo then + local sudo_cmd = ("%s su -c '%s'"):format(SUDO, cmd) + print("+ "..sudo_cmd) + os.execute(sudo_cmd) + else + print("+ "..cmd) + os.execute(cmd) end end if arg[1] == "deploy" or arg[1] == "dep" or arg[1] == "d" then - cmd = ("nix run github:serokell/deploy-rs -- --skip-checks %s -- --log-format internal-json |& ${pkgs.nix-output-monitor}/bin/nom --json"):format(table.concat(flags, " ")) + local deploy_args = {} + local nix_args = {} + + local found_dashes = false + if arg[2] ~= nil then + for i = 2, #arg do + if arg[i] == "--" then + found_dashes = true + elseif found_dashes then + table.insert(nix_args, arg[i]) + else + table.insert(deploy_args, arg[i]) + end + end + end + + cmd = ("nix run github:serokell/deploy-rs -- --skip-checks %s -- --log-format internal-json %s |& ${pkgs.nix-output-monitor}/bin/nom --json"):format(table.concat(deploy_args, " "), table.concat(nix_args, " ")) print("+ "..cmd) os.execute(cmd) elseif arg[1] == "switch" or arg[1] == "sw" or arg[1] == "s" then - verb = "switch" + local target = "/home/collin/nixos" + if arg[2] ~= nil and arg[2]:match("#") then + target = arg[2] + end + + nixosRebuild{ + sudo = true, + verb = "switch", + flake = target, + extraArgs = collectArgs(3) + } elseif arg[1] == "boot" or arg[1] == "bo" then - verb = "boot" + nixosRebuild{sudo = true, verb = "boot", extraArgs = collectArgs()} elseif arg[1] == "test" or arg[1] == "t" then - verb = "test" + nixosRebuild{sudo = true, verb = "test", extraArgs = collectArgs()} elseif arg[1] == "build" or arg[1] == "bu" then - cmd = ("nixos-rebuild build --flake /home/collin/nixos %s --log-format internal-json |& ${pkgs.nix-output-monitor}/bin/nom --json"):format(table.concat(flags, " ")) - print("+ "..cmd) - os.execute(cmd) - end + local target = "/home/collin/nixos" + if arg[2] ~= nil and arg[2]:match("#") then + target = arg[2] + end - if verb then - cmd = ("run0 --background= sh -c 'nixos-rebuild %s --flake /home/collin/nixos %s --log-format internal-json |& ${pkgs.nix-output-monitor}/bin/nom --json'"):format(verb, table.concat(flags, " ")) - print("+ "..cmd) - os.execute(cmd) + nixosRebuild{verb = "build", flake = target, extraArgs = collectArgs(3)} end ''; in |