123const std = @import("std");4const builtin = @import("builtin");5const types = @import("types.zig");6const errors = @import("errors.zig");7const config = @import("config.zig");8const output = @import("output.zig");9const exec = @import("exec.zig");10const api = @import("api.zig");11const search_tui = @import("search.zig");12const version = @import("version.zig");1314const UPDATE_URL = "https://kepr.uk/api/nina/releases/latest";1516pub const Ctx = struct {17 gpa: std.mem.Allocator,18 io: std.Io,19 environ: *const std.process.Environ.Map,20 machine: types.Machine,21 cfg: types.NinaConfig,22 args: []const []const u8, 23 sub: []const u8, 24 dry: bool,25 check: bool,26 no_apply: bool,27 all: bool,28 last: u32,29 json: bool,30};3132pub fn apply(ctx: Ctx) !void {33 if (ctx.dry) {34 output.applyDry(ctx.machine.name);35 const code = try exec.nixosRebuildDryActivate(ctx.io, &ctx.machine, ctx.gpa);36 if (code != 0) return error.BuildFailed;37 output.applyDryDone();38 return;39 }40 if (ctx.check) {41 return check(ctx);42 }4344 output.applyStart(ctx.machine.name);45 const start = std.Io.Clock.awake.now(ctx.io);4647 const code = exec.nixosRebuildSwitch(ctx.io, &ctx.machine, ctx.gpa) catch |e| {48 exec.markApplyFailed(ctx.io, ctx.environ);49 output.applyFailed("");50 return e;51 };5253 if (code != 0) {54 exec.markApplyFailed(ctx.io, ctx.environ);55 output.applyFailed("");56 return error.BuildFailed;57 }5859 const elapsed_ns = @max(0, start.untilNow(ctx.io, .awake).toNanoseconds());60 const elapsed: u64 = @intCast(@divFloor(elapsed_ns, 1_000_000));61 const gen = exec.getCurrentGeneration(ctx.gpa, ctx.io, &ctx.machine) catch 0;62 const after_failure = exec.takeApplyFailed(ctx.io, ctx.environ);63 output.applyDone(gen, elapsed, after_failure);64 logAction(ctx, "apply", gen);65}6667pub fn check(ctx: Ctx) !void {68 output.checkStart(ctx.machine.name);69 const code = try exec.nixosRebuildBuild(ctx.io, &ctx.machine, ctx.gpa);70 if (code != 0) {71 output.checkFailed("");72 return error.BuildFailed;73 }74 output.checkOk();75}7677pub fn back(ctx: Ctx) !void {78 const gens = try exec.getGenerations(ctx.gpa, ctx.io, &ctx.machine);79 defer ctx.gpa.free(gens);8081 var current: u32 = 0;82 for (gens) |g| if (g.current) { current = g.number; };8384 85 86 var prev: u32 = 0;87 var has_prev = false;88 for (gens) |g| {89 if (g.number < current and (!has_prev or g.number > prev)) {90 prev = g.number;91 has_prev = true;92 }93 }94 if (!has_prev) {95 output.alreadyOldest();96 return;97 }9899 output.backStart(ctx.machine.name, current, prev);100 if (ctx.cfg.confirm and !output.confirm("")) return;101102 const code = try exec.rollback(ctx.io, &ctx.machine, ctx.gpa);103 if (code != 0) return error.BuildFailed;104 output.backDone(prev);105}106107pub fn go(ctx: Ctx) !void {108 if (ctx.args.len == 0) return error.Internal;109 const target = std.fmt.parseInt(u32, ctx.args[0], 10) catch return error.Internal;110 const current = try exec.getCurrentGeneration(ctx.gpa, ctx.io, &ctx.machine);111112 output.goStart(ctx.machine.name, current, target);113 if (ctx.cfg.confirm and !output.confirm("")) return;114115 const code = try exec.switchGeneration(ctx.io, &ctx.machine, ctx.gpa, target);116 if (code != 0) return error.ProcessFailed;117 output.generationDone(target);118}119120pub fn history(ctx: Ctx) !void {121 const gens = try exec.getGenerations(ctx.gpa, ctx.io, &ctx.machine);122 defer ctx.gpa.free(gens);123 output.historyHeader(ctx.machine.name);124 for (gens) |g| output.historyRow(g);125}126127pub fn genCmd(ctx: Ctx) !void {128 if (std.mem.eql(u8, ctx.sub, "list")) return history(ctx);129 if (std.mem.eql(u8, ctx.sub, "current")) {130 const n = try exec.getCurrentGeneration(ctx.gpa, ctx.io, &ctx.machine);131 output.genCurrent(n);132 return;133 }134 if (std.mem.eql(u8, ctx.sub, "delete")) {135 if (ctx.args.len == 0) return error.Internal;136 if (std.mem.eql(u8, ctx.args[0], "old")) {137 if (ctx.cfg.confirm and !output.confirm("")) return;138 _ = try exec.deleteOldGenerations(ctx.io, &ctx.machine, ctx.gpa);139 } else {140 const n = std.fmt.parseInt(u32, ctx.args[0], 10) catch return error.Internal;141 if (ctx.cfg.confirm and !output.confirm("")) return;142 _ = try exec.deleteGeneration(ctx.io, &ctx.machine, ctx.gpa, n);143 }144 output.done();145 return;146 }147 return error.Internal;148}149150pub fn clean(ctx: Ctx) !void {151 const gens = try exec.getGenerations(ctx.gpa, ctx.io, &ctx.machine);152 defer ctx.gpa.free(gens);153 const all = ctx.all or std.mem.eql(u8, ctx.sub, "all");154 const total: u32 = @intCast(gens.len);155 156 157 const keep: u32 = if (all) 1 else ctx.cfg.generations;158 if (total <= keep) {159 output.cleanNothing(ctx.machine.name, total);160 return;161 }162 output.cleanStart(ctx.machine.name, keep, total);163 if (ctx.cfg.confirm and !output.confirm("removing — cannot be undone")) return;164 _ = try exec.collectGarbage(ctx.io, &ctx.machine, ctx.gpa, all, ctx.cfg.generations);165 output.cleanDone("?");166}167168pub fn statusCmd(ctx: Ctx) !void {169 output.statusHeader();170 171 172 if (ctx.all and ctx.cfg.machines.len > 0) {173 for (ctx.cfg.machines) |m| statusRowFor(ctx, &m);174 return;175 }176 statusRowFor(ctx, &ctx.machine);177}178179fn statusRowFor(ctx: Ctx, machine: *const types.Machine) void {180 181 182 const gen = exec.getCurrentGeneration(ctx.gpa, ctx.io, machine) catch {183 output.statusRow(machine.name, 0, "", "", false);184 return;185 };186 const st = exec.systemStatus(ctx.gpa, ctx.io, machine);187 defer ctx.gpa.free(st.size);188 defer ctx.gpa.free(st.age);189 output.statusRow(machine.name, gen, st.age, st.size, true);190}191192pub fn diff(ctx: Ctx) !void {193 const gens = try exec.getGenerations(ctx.gpa, ctx.io, &ctx.machine);194 defer ctx.gpa.free(gens);195 var current: u32 = 0;196 var prev: u32 = 0;197 for (gens) |g| {198 if (g.current) current = g.number;199 if (!g.current and g.number > prev and g.number < current) prev = g.number;200 }201 if (ctx.args.len >= 2) {202 const a = std.fmt.parseInt(u32, ctx.args[0], 10) catch current - 1;203 const b = std.fmt.parseInt(u32, ctx.args[1], 10) catch current;204 output.diffHeader(a, b, ctx.machine.name);205 if (try exec.diffClosures(ctx.io, &ctx.machine, ctx.gpa, a, b) != 0) return error.ProcessFailed;206 } else {207 output.diffHeader(if (prev > 0) prev else current - 1, current, ctx.machine.name);208 if (try exec.diffClosures(ctx.io, &ctx.machine, ctx.gpa, if (prev > 0) prev else current - 1, current) != 0) return error.ProcessFailed;209 }210}211212pub fn mood(ctx: Ctx) !void {213 const gen = exec.getCurrentGeneration(ctx.gpa, ctx.io, &ctx.machine) catch 0;214 const st = exec.systemStatus(ctx.gpa, ctx.io, &ctx.machine);215 defer ctx.gpa.free(st.size);216 defer ctx.gpa.free(st.age);217 const state = exec.systemRunningState(ctx.gpa, ctx.io, &ctx.machine) catch try ctx.gpa.dupe(u8, "unknown");218 defer ctx.gpa.free(state);219 output.mood(ctx.machine.name, state, gen, st.age, st.size);220}221222pub fn doctor(ctx: Ctx) !void {223 output.doctorHeader(ctx.machine.name);224 const checks = try exec.runDoctorChecks(ctx.gpa, ctx.io, &ctx.machine);225 defer ctx.gpa.free(checks);226 var warnings: u32 = 0;227 var failures: u32 = 0;228 for (checks) |ch| {229 output.doctorRow(ch);230 if (ch.status == .warn) warnings += 1;231 if (ch.status == .fail) failures += 1;232 }233 output.doctorSummary(warnings, failures);234}235236237pub fn update(ctx: Ctx) !void {238 output.updateSelfStart();239240 if (comptime std.mem.eql(u8, version.PLATFORM, "unsupported")) {241 output.updateSelfNoPlatform(@tagName(builtin.cpu.arch) ++ "-" ++ @tagName(builtin.os.tag));242 return error.Internal;243 }244245 const env_url = ctx.environ.get("NINA_UPDATE_SOURCE");246 const source_url = env_url orelse UPDATE_URL;247248 const body = exec.fetchUrl(ctx.gpa, ctx.io, source_url) catch |e| switch (e) {249 error.NetworkError, error.HttpError => {250 output.updateSelfNetworkError();251 return;252 },253 else => return e,254 };255 defer ctx.gpa.free(body);256257 const parsed = std.json.parseFromSlice(std.json.Value, ctx.gpa, body, .{}) catch {258 output.printWarning("could not parse release metadata");259 return;260 };261 defer parsed.deinit();262263 const root = parsed.value;264 if (root != .object) { output.printWarning("malformed release metadata"); return; }265266 const remote_ver = blk: {267 const v = root.object.get("version") orelse { output.printWarning("malformed release metadata"); return; };268 break :blk switch (v) { .string => |s| s, else => { output.printWarning("malformed release metadata"); return; } };269 };270271 if (std.mem.eql(u8, remote_ver, version.VERSION)) {272 output.updateSelfCurrent(version.VERSION);273 return;274 }275276 output.updateSelfAvailable(version.VERSION, remote_ver);277 if (ctx.check) return;278279 var exe_buf: [std.Io.Dir.max_path_bytes]u8 = undefined;280 const install_path = exec.selfExePath(ctx.io, &exe_buf) catch {281 errors.error_info.setMessage("could not determine install path", .{});282 return error.Internal;283 };284285 286 287 if (std.mem.startsWith(u8, install_path, "/nix/store/")) {288 output.updateSelfNixManaged(remote_ver);289 if (ctx.cfg.confirm and !output.confirm("")) return;290 const ec = exec.streamLocal(ctx.io, &.{ "nix", "profile", "upgrade", ".*nina.*" }) catch {291 output.updateSelfNixFailed();292 return;293 };294 if (ec != 0) output.updateSelfNixSystemManaged();295 return;296 }297298 const plat_info = blk: {299 const pv = root.object.get("platforms") orelse break :blk null;300 if (pv != .object) break :blk null;301 break :blk pv.object.get(version.PLATFORM);302 };303 if (plat_info == null) { output.updateSelfNoPlatform(version.PLATFORM); return; }304305 const dl_url = blk: {306 const v = plat_info.?.object.get("url") orelse { output.printWarning("malformed release metadata"); return; };307 break :blk switch (v) { .string => |s| s, else => { output.printWarning("malformed release metadata"); return; } };308 };309310 const raw_cksum = blk: {311 if (plat_info.?.object.get("blake3")) |v|312 break :blk switch (v) { .string => |s| s, else => { output.printWarning("malformed release metadata"); return; } };313 if (plat_info.?.object.get("checksum")) |v|314 break :blk switch (v) { .string => |s| s, else => { output.printWarning("malformed release metadata"); return; } };315 output.printWarning("missing checksum in release metadata");316 return;317 };318 const expected_hex = if (std.mem.startsWith(u8, raw_cksum, "blake3:")) raw_cksum["blake3:".len..] else raw_cksum;319320 if (ctx.cfg.confirm and !output.confirm("")) return;321322 output.updateSelfDownloading(remote_ver, version.PLATFORM);323 const tmp_path = exec.downloadToTemp(ctx.gpa, ctx.io, dl_url) catch {324 output.printWarning("download failed — check your connection and try again");325 return;326 };327 defer ctx.gpa.free(tmp_path);328329 output.updateSelfVerifying();330 exec.verifyAndInstall(ctx.gpa, ctx.io, tmp_path, expected_hex, install_path) catch |e| switch (e) {331 error.ChecksumMismatch => { output.updateSelfChecksumFailed(); return; },332 else => return e,333 };334335 output.updateSelfDone(version.VERSION, remote_ver);336}337338339pub fn goodbye(ctx: Ctx) !void {340 output.goodbyeStart();341342 var exe_buf: [std.Io.Dir.max_path_bytes]u8 = undefined;343 const install_path = exec.selfExePath(ctx.io, &exe_buf) catch {344 errors.error_info.setMessage("could not determine install path", .{});345 return error.Internal;346 };347348 if (std.mem.startsWith(u8, install_path, "/nix/store/")) {349 output.goodbyeNix();350 if (ctx.cfg.confirm and !output.confirm("")) return;351 const ec = exec.streamLocal(ctx.io, &.{ "nix", "profile", "remove", ".*nina.*" }) catch {352 output.goodbyeFailed("nix profile remove failed");353 return;354 };355 if (ec != 0) {356 output.goodbyeNixSystemManaged();357 return;358 }359 } else {360 output.goodbyeFile(install_path);361 if (ctx.cfg.confirm and !output.confirm("")) return;362 std.Io.Dir.deleteFileAbsolute(ctx.io, install_path) catch |e| {363 errors.error_info.setMessage("could not remove {s}", .{install_path});364 return e;365 };366 }367368 369 if (ctx.environ.get("HOME")) |home| {370 const nina_nix: ?[]u8 = std.fmt.allocPrint(ctx.gpa, "{s}/nina.nix", .{home}) catch null;371 if (nina_nix) |path| {372 defer ctx.gpa.free(path);373 const exists = blk: {374 const bytes = std.Io.Dir.readFileAlloc(.cwd(), ctx.io, path, ctx.gpa, .unlimited) catch break :blk false;375 ctx.gpa.free(bytes);376 break :blk true;377 };378 if (exists) {379 output.goodbyeNixFileFound(path);380 if (!ctx.cfg.confirm or output.confirm("")) {381 std.Io.Dir.deleteFileAbsolute(ctx.io, path) catch {382 output.printWarning("could not remove ~/nina.nix — remove it manually to clean up");383 };384 }385 }386 }387 }388389 output.goodbyeDone();390}391392fn channelUpdateCmd(ctx: Ctx) !void {393 output.updateStart(ctx.machine.name);394 _ = try exec.channelUpdate(ctx.io, &ctx.machine, ctx.gpa);395 output.updateDone();396}397398pub fn upgrade(ctx: Ctx) !void {399 output.upgradeStart(ctx.machine.name);400 output.printSubstep("updating channels", .{});401 _ = exec.channelUpdate(ctx.io, &ctx.machine, ctx.gpa) catch {};402 output.printSubstep("rebuilding", .{});403 const start = std.Io.Clock.awake.now(ctx.io);404 const code = try exec.nixosRebuildSwitch(ctx.io, &ctx.machine, ctx.gpa);405 if (code != 0) {406 output.applyFailed("");407 return error.BuildFailed;408 }409 const elapsed_ns = @max(0, start.untilNow(ctx.io, .awake).toNanoseconds());410 const elapsed: u64 = @intCast(@divFloor(elapsed_ns, 1_000_000));411 const gen = exec.getCurrentGeneration(ctx.gpa, ctx.io, &ctx.machine) catch 0;412 output.applyDone(gen, elapsed, false);413}414415pub fn log(ctx: Ctx) !void {416 output.logHeader(ctx.machine.name);417 const content = exec.readNinaLog(ctx.gpa, ctx.io, ctx.environ, ctx.last) catch {418 419 return;420 };421 defer ctx.gpa.free(content);422 var lines = std.mem.splitScalar(u8, content, '\n');423 while (lines.next()) |line| {424 const t = std.mem.trim(u8, line, " \t\r");425 if (t.len == 0) continue;426 if (t[0] == '{') {427 renderJsonLogLine(ctx.gpa, t);428 } else {429 430 var parts = std.mem.tokenizeScalar(u8, t, '\t');431 const date = parts.next() orelse continue;432 const time = parts.next() orelse "";433 const action = parts.next() orelse "";434 const detail = parts.next() orelse "";435 output.logRow(date, time, action, detail);436 }437 }438}439440441442fn renderJsonLogLine(gpa: std.mem.Allocator, line: []const u8) void {443 const parsed = std.json.parseFromSlice(std.json.Value, gpa, line, .{}) catch {444 output.logRow(line, "", "", ""); 445 return;446 };447 defer parsed.deinit();448 if (parsed.value != .object) return;449 const o = parsed.value.object;450451 const ts = jsonStr(o, "ts");452 const date = if (ts.len >= 10) ts[0..10] else ts;453 const time = if (ts.len >= 16) ts[11..16] else "";454 const action = jsonStr(o, "command");455 const outcome = jsonStr(o, "outcome");456457 var buf: [160]u8 = undefined;458 var detail: []const u8 = outcome;459 if (o.get("gen_after")) |g| {460 if (g == .integer) {461 detail = std.fmt.bufPrint(&buf, "{s} -> gen {d}", .{ outcome, g.integer }) catch outcome;462 }463 }464 output.logRow(date, time, action, detail);465}466467fn jsonStr(o: std.json.ObjectMap, key: []const u8) []const u8 {468 if (o.get(key)) |v| {469 if (v == .string) return v.string;470 }471 return "";472}473474pub fn edit(ctx: Ctx) !void {475 const hardware = ctx.args.len > 0 and std.mem.eql(u8, ctx.args[0], "hardware");476 const filename = if (hardware) "hardware-configuration.nix" else "configuration.nix";477 const path = try std.fmt.allocPrint(ctx.gpa, "{s}/{s}", .{ ctx.machine.config_path, filename });478 defer ctx.gpa.free(path);479 output.editOpen(path, ctx.cfg.editor);480 _ = try exec.openEditor(ctx.io, ctx.gpa, ctx.cfg.editor, path, null);481}482483pub fn fmt(ctx: Ctx) !void {484 const path = try std.fmt.allocPrint(ctx.gpa, "{s}/configuration.nix", .{ctx.machine.config_path});485 defer ctx.gpa.free(path);486 if (ctx.check) {487 const code = try exec.nixpkgsFmt(ctx.io, &ctx.machine, ctx.gpa, path, true);488 if (code != 0) output.fmtNeedsFormat() else output.fmtDone();489 return;490 }491 output.fmtStart(path);492 _ = try exec.nixpkgsFmt(ctx.io, &ctx.machine, ctx.gpa, path, false);493 output.fmtDone();494}495496pub fn repl(ctx: Ctx) !void {497 output.replStart();498 _ = try exec.nixRepl(ctx.io, &ctx.machine, ctx.gpa);499}500501pub fn develop(ctx: Ctx) !void {502 if (std.mem.eql(u8, ctx.sub, "show")) {503 _ = try exec.nixBuild(ctx.io, &ctx.machine, ctx.gpa, null);504 return;505 }506 const run_cmd: ?[]const u8 = if (ctx.args.len > 0 and std.mem.eql(u8, ctx.sub, "run"))507 ctx.args[0]508 else509 null;510 output.developStart(&.{});511 _ = try exec.nixDevelop(ctx.io, &ctx.machine, ctx.gpa, run_cmd);512 output.developDone();513}514515pub fn info(ctx: Ctx) !void {516 output.infoHeader(ctx.machine.name);517 const nixos_ver = exec.systemInfo(ctx.gpa, ctx.io, &ctx.machine) catch try ctx.gpa.dupe(u8, "unknown");518 defer ctx.gpa.free(nixos_ver);519 output.infoRow("nixos", std.mem.trim(u8, nixos_ver, " \t\n\r"));520521 const kernel = exec.kernelInfo(ctx.gpa, ctx.io, &ctx.machine) catch try ctx.gpa.dupe(u8, "unknown");522 defer ctx.gpa.free(kernel);523 output.infoRow("kernel", std.mem.trim(u8, kernel, " \t\n\r"));524525 const uptime = exec.uptimeInfo(ctx.gpa, ctx.io, &ctx.machine) catch try ctx.gpa.dupe(u8, "unknown");526 defer ctx.gpa.free(uptime);527 output.infoRow("uptime", std.mem.trim(u8, uptime, " \t\n\r"));528}529530pub fn boot(ctx: Ctx) !void {531 output.bootHeader(ctx.machine.name);532 const entries = exec.bootEntries(ctx.gpa, ctx.io, &ctx.machine) catch &.{};533 defer ctx.gpa.free(entries);534 for (entries, 0..) |entry, i| output.bootEntry(entry, i);535}536537pub fn searchCmd(ctx: Ctx) !void {538 const query = if (ctx.args.len > 0) ctx.args[0] else "";539 api.ensureRegistryPinned(ctx.gpa, ctx.io, ctx.environ);540 const result = try search_tui.run(ctx.gpa, ctx.io, query, .packages);541 if (result) |r| {542 installFromResult(ctx, r) catch |e| return e;543 }544}545546pub fn optionCmd(ctx: Ctx) !void {547 const query = if (ctx.args.len > 0) ctx.args[0] else "";548 api.ensureRegistryPinned(ctx.gpa, ctx.io, ctx.environ);549 _ = try search_tui.run(ctx.gpa, ctx.io, query, .options);550}551552pub fn install(ctx: Ctx) !void {553 const pkg_name = if (ctx.args.len > 0) ctx.args[0] else return error.PackageNotFound;554555 api.ensureRegistryPinned(ctx.gpa, ctx.io, ctx.environ);556557 558 const pkgs = api.searchPackages(ctx.gpa, ctx.io, pkg_name) catch {559 errors.error_info.setMessage("'{s}' not found in nixpkgs", .{pkg_name});560 errors.error_info.setSuggestion("nina search {s}", .{pkg_name});561 return error.PackageNotFound;562 };563 defer ctx.gpa.free(pkgs);564565 if (pkgs.len == 0) {566 errors.error_info.setMessage("'{s}' not found in nixpkgs", .{pkg_name});567 errors.error_info.setSuggestion("nina search {s}", .{pkg_name});568 return error.PackageNotFound;569 }570571 572 573 var found = pkgs[0];574 for (pkgs) |p| {575 if (std.mem.eql(u8, p.pname, pkg_name)) {576 found = p;577 break;578 }579 }580 output.installSelectPrompt(found.pname, found.version);581582 var choice_buf: [8]u8 = undefined;583 var choice_len: usize = 0;584 while (choice_len < choice_buf.len) {585 const n = std.posix.read(std.posix.STDIN_FILENO, choice_buf[choice_len .. choice_len + 1]) catch break;586 if (n == 0) break;587 if (choice_buf[choice_len] == '\n') break;588 choice_len += 1;589 }590 const choice = std.mem.trim(u8, choice_buf[0..choice_len], " \t");591592 if (choice.len == 0 or choice[0] == 'i' or choice[0] == 'I') {593 try profileInstallPkg(ctx, found);594 } else if (choice[0] == 's' or choice[0] == 'S') {595 try systemInstallPkg(ctx, found);596 } else if (choice[0] == 't' or choice[0] == 'T') {597 try tryPkg(ctx, found);598 }599}600601fn installFromResult(ctx: Ctx, result: types.SearchResult) !void {602 switch (result.action) {603 .profile => try profileInstallPkg(ctx, result.package),604 .system => try systemInstallPkg(ctx, result.package),605 }606}607608fn profileInstallPkg(ctx: Ctx, npkg: types.NixPackage) !void {609 output.installStart(npkg.pname, .profile);610 output.printSubstep("nix profile install nixpkgs#{s}", .{npkg.attr});611 output.printTeach(try std.fmt.allocPrint(ctx.gpa, "nix profile install nixpkgs#{s}", .{npkg.attr}));612 const code = try exec.profileInstall(ctx.io, &ctx.machine, ctx.gpa, npkg.attr);613 if (code != 0) return error.ProcessFailed;614 output.installDone(npkg.pname);615}616617fn systemInstallPkg(ctx: Ctx, npkg: types.NixPackage) !void {618 const config_path = try std.fmt.allocPrint(ctx.gpa, "{s}/configuration.nix", .{ctx.machine.config_path});619 defer ctx.gpa.free(config_path);620 const line = api.findSystemPackagesLine(ctx.io, config_path) orelse 1;621 output.installEditorPrompt(npkg.pname, config_path, line, ctx.cfg.editor);622 _ = try exec.openEditor(ctx.io, ctx.gpa, ctx.cfg.editor, config_path, line);623 if (ctx.cfg.confirm) {624 if (output.confirm("apply now? ")) {625 try apply(ctx);626 }627 }628}629630fn tryPkg(ctx: Ctx, npkg: types.NixPackage) !void {631 output.tryStart(npkg.pname);632 output.printSubstep("nix shell nixpkgs#{s}", .{npkg.attr});633 _ = try exec.nixShell(ctx.io, &ctx.machine, ctx.gpa, npkg.attr);634 output.tryDone();635}636637pub fn remove(ctx: Ctx) !void {638 const pkg_name = if (ctx.args.len > 0) ctx.args[0] else return error.PackageNotFound;639 output.removeStart(pkg_name);640 const code = try exec.profileRemove(ctx.io, &ctx.machine, ctx.gpa, pkg_name);641 if (code != 0) return error.ProcessFailed;642 output.done();643}644645pub fn list(ctx: Ctx) !void {646 output.listHeader(ctx.machine.name);647648 649 650 651 if (exec.profileListJson(ctx.gpa, ctx.io, &ctx.machine) catch null) |json| {652 defer ctx.gpa.free(json);653 output.listSection("profile");654 renderProfileJson(ctx.gpa, json);655 }656657 658 659 if (exec.systemPackagePaths(ctx.gpa, ctx.io, &ctx.machine) catch null) |paths| {660 defer ctx.gpa.free(paths);661 output.blankLine();662 output.listSection("system");663 var lines = std.mem.splitScalar(u8, paths, '\n');664 while (lines.next()) |line| {665 const t = std.mem.trim(u8, line, " \t\r");666 if (t.len == 0) continue;667 const row = pkgFromStorePath(t);668 669 if (row.version.len == 0) continue;670 output.listPkg(row.name, row.version);671 }672 }673}674675const PkgRow = struct { name: []const u8, version: []const u8 };676677678679680681fn pkgFromStorePath(path: []const u8) PkgRow {682 const base = std.fs.path.basename(path);683 const after_hash = if (base.len > 33 and base[32] == '-') base[33..] else base;684 var i: usize = 0;685 while (i + 1 < after_hash.len) : (i += 1) {686 if (after_hash[i] == '-' and std.ascii.isDigit(after_hash[i + 1])) {687 return .{ .name = after_hash[0..i], .version = stripOutputSuffix(after_hash[i + 1 ..]) };688 }689 }690 return .{ .name = after_hash, .version = "" };691}692693694695696fn stripOutputSuffix(ver: []const u8) []const u8 {697 const suffixes = [_][]const u8{ "-bin", "-man", "-dev", "-doc", "-lib", "-info", "-out", "-static", "-devdoc" };698 for (suffixes) |s| {699 if (std.mem.endsWith(u8, ver, s)) return ver[0 .. ver.len - s.len];700 }701 return ver;702}703704test "pkgFromStorePath splits name and version at first dash-digit" {705 const a = pkgFromStorePath("/nix/store/abcdefghijklmnopqrstuvwxyz012345-ripgrep-15.1.0");706 try std.testing.expectEqualStrings("ripgrep", a.name);707 try std.testing.expectEqualStrings("15.1.0", a.version);708709 const b = pkgFromStorePath("/nix/store/abcdefghijklmnopqrstuvwxyz012345-helix-25.01");710 try std.testing.expectEqualStrings("helix", b.name);711 try std.testing.expectEqualStrings("25.01", b.version);712713 714 const c = pkgFromStorePath("/nix/store/abcdefghijklmnopqrstuvwxyz012345-curl-8.20.0-bin");715 try std.testing.expectEqualStrings("curl", c.name);716 try std.testing.expectEqualStrings("8.20.0", c.version);717}718719fn firstStorePath(el: std.json.Value) ?[]const u8 {720 if (el != .object) return null;721 const sp = el.object.get("storePaths") orelse return null;722 if (sp != .array or sp.array.items.len == 0) return null;723 const first = sp.array.items[0];724 return if (first == .string) first.string else null;725}726727728729fn renderProfileJson(gpa: std.mem.Allocator, json: []const u8) void {730 const parsed = std.json.parseFromSlice(std.json.Value, gpa, json, .{}) catch return;731 defer parsed.deinit();732 if (parsed.value != .object) return;733 const elements = parsed.value.object.get("elements") orelse return;734 switch (elements) {735 .object => |obj| {736 var it = obj.iterator();737 while (it.next()) |entry| {738 const ver = if (firstStorePath(entry.value_ptr.*)) |p| pkgFromStorePath(p).version else "";739 output.listPkg(entry.key_ptr.*, ver);740 }741 },742 .array => |arr| {743 for (arr.items) |el| {744 const path = firstStorePath(el) orelse continue;745 const row = pkgFromStorePath(path);746 output.listPkg(row.name, row.version);747 }748 },749 else => {},750 }751}752753pub fn tryCmd(ctx: Ctx) !void {754 const pkg_name_arg = if (ctx.args.len > 0) ctx.args[0] else return error.PackageNotFound;755 const shell_pkg = types.NixPackage{ .attr = pkg_name_arg, .pname = pkg_name_arg, .version = "", .description = "" };756 try tryPkg(ctx, shell_pkg);757}758759760761pub fn service(ctx: Ctx) !void {762 if (std.mem.eql(u8, ctx.sub, "list") or ctx.sub.len == 0) {763 output.serviceListHeader(ctx.machine.name);764 const svcs = exec.serviceList(ctx.gpa, ctx.io, &ctx.machine) catch &.{};765 defer ctx.gpa.free(svcs);766 for (svcs) |s| output.serviceRow(s);767 return;768 }769 if (std.mem.eql(u8, ctx.sub, "logs")) {770 const name = if (ctx.args.len > 0) ctx.args[0] else return error.Internal;771 const follow = ctx.args.len > 1 and std.mem.eql(u8, ctx.args[1], "-f");772 output.serviceLogsHeader(name, ctx.last, ctx.machine.name);773 _ = try exec.serviceLogs(ctx.io, &ctx.machine, ctx.gpa, name, ctx.last, follow);774 return;775 }776 if (std.mem.eql(u8, ctx.sub, "status")) {777 const name = if (ctx.args.len > 0) ctx.args[0] else return error.Internal;778 output.serviceVerb("status", name, ctx.machine.name);779 _ = try exec.serviceCtl(ctx.io, &ctx.machine, ctx.gpa, "status", name);780 return;781 }782 783 const name = if (ctx.args.len > 0) ctx.args[0] else return error.Internal;784 if (ctx.cfg.confirm and (std.mem.eql(u8, ctx.sub, "stop") or std.mem.eql(u8, ctx.sub, "disable"))) {785 if (!output.confirm("")) return;786 }787 output.serviceVerb(ctx.sub, name, ctx.machine.name);788 _ = try exec.serviceCtl(ctx.io, &ctx.machine, ctx.gpa, ctx.sub, name);789 output.serviceDone();790}791792793794pub fn flake(ctx: Ctx) !void {795 if (std.mem.eql(u8, ctx.sub, "update")) {796 output.flakeUpdateStart();797 const input: ?[]const u8 = if (ctx.args.len > 0) ctx.args[0] else null;798 _ = try exec.flakeUpdate(ctx.io, &ctx.machine, ctx.gpa, input);799 output.flakeLockWritten();800 return;801 }802 if (std.mem.eql(u8, ctx.sub, "check")) {803 output.flakeCheckStart();804 const code = try exec.flakeCheck(ctx.io, &ctx.machine, ctx.gpa);805 if (code == 0) output.flakeValid() else output.checkFailed("");806 return;807 }808 if (std.mem.eql(u8, ctx.sub, "show")) {809 const json_out = exec.flakeShow(ctx.gpa, ctx.io, &ctx.machine) catch null;810 if (json_out) |j| {811 defer ctx.gpa.free(j);812 output.flakeShowHeader("flake outputs");813 814 const parsed = std.json.parseFromSlice(std.json.Value, ctx.gpa, j, .{}) catch return;815 defer parsed.deinit();816 if (parsed.value == .object) {817 var it = parsed.value.object.iterator();818 while (it.next()) |entry| {819 if (entry.value_ptr.* == .object) {820 var inner = entry.value_ptr.object.iterator();821 while (inner.next()) |e2| {822 const desc = if (e2.value_ptr.* == .object)823 if (e2.value_ptr.object.get("description")) |d|824 if (d == .string) d.string else ""825 else ""826 else "";827 output.flakeShowEntry(entry.key_ptr.*, e2.key_ptr.*, desc);828 }829 }830 }831 }832 }833 return;834 }835 if (std.mem.eql(u8, ctx.sub, "init")) {836 _ = try exec.flakeInit(ctx.io, &ctx.machine, ctx.gpa);837 return;838 }839 if (std.mem.eql(u8, ctx.sub, "lock")) {840 _ = try exec.flakeLock(ctx.io, &ctx.machine, ctx.gpa);841 return;842 }843 if (std.mem.eql(u8, ctx.sub, "clone")) {844 const url = if (ctx.args.len > 0) ctx.args[0] else return error.Internal;845 _ = try exec.flakeClone(ctx.io, &ctx.machine, ctx.gpa, url);846 return;847 }848}849850851852pub fn store(ctx: Ctx) !void {853 if (std.mem.eql(u8, ctx.sub, "gc")) {854 _ = try exec.storeGc(ctx.io, &ctx.machine, ctx.gpa);855 return;856 }857 if (std.mem.eql(u8, ctx.sub, "verify")) {858 output.storeVerifyStart();859 _ = try exec.storeVerify(ctx.io, &ctx.machine, ctx.gpa);860 return;861 }862 if (std.mem.eql(u8, ctx.sub, "repair")) {863 if (ctx.cfg.confirm and !output.confirm("")) return;864 output.storeRepairStart();865 _ = try exec.storeRepair(ctx.io, &ctx.machine, ctx.gpa);866 return;867 }868 if (std.mem.eql(u8, ctx.sub, "info")) {869 const si = try exec.storeInfo(ctx.gpa, ctx.io, &ctx.machine);870 output.storeInfo(si);871 return;872 }873 if (std.mem.eql(u8, ctx.sub, "path")) {874 const attr = if (ctx.args.len > 0) ctx.args[0] else return error.Internal;875 const path = try exec.storePath(ctx.gpa, ctx.io, &ctx.machine, attr);876 defer ctx.gpa.free(path);877 output.storePath(path);878 return;879 }880 881 const si = try exec.storeInfo(ctx.gpa, ctx.io, &ctx.machine);882 output.storeInfo(si);883}884885886887pub fn channel(ctx: Ctx) !void {888 if (std.mem.eql(u8, ctx.sub, "list") or ctx.sub.len == 0) {889 output.channelListHeader(ctx.machine.name);890 const chans = try exec.channelList(ctx.gpa, ctx.io, &ctx.machine);891 defer ctx.gpa.free(chans);892 for (chans) |ch| output.channelRow(ch);893 return;894 }895 if (std.mem.eql(u8, ctx.sub, "add")) {896 const url = if (ctx.args.len > 0) ctx.args[0] else return error.Internal;897 const name = if (ctx.args.len > 1) ctx.args[1] else "";898 _ = try exec.channelAdd(ctx.io, &ctx.machine, ctx.gpa, url, name);899 output.channelDone();900 return;901 }902 if (std.mem.eql(u8, ctx.sub, "remove")) {903 const name = if (ctx.args.len > 0) ctx.args[0] else return error.Internal;904 _ = try exec.channelRemove(ctx.io, &ctx.machine, ctx.gpa, name);905 output.channelDone();906 return;907 }908 if (std.mem.eql(u8, ctx.sub, "update")) {909 return channelUpdateCmd(ctx);910 }911}912913914915pub fn profile(ctx: Ctx) !void {916 if (std.mem.eql(u8, ctx.sub, "list") or ctx.sub.len == 0) {917 output.profileListHeader();918 const out = exec.profileList(ctx.gpa, ctx.io, &ctx.machine) catch null;919 if (out) |o| {920 defer ctx.gpa.free(o);921 var lines = std.mem.splitScalar(u8, o, '\n');922 while (lines.next()) |line| {923 const t = std.mem.trim(u8, line, " \t\r");924 if (t.len == 0) continue;925 output.profileRow(t, "", "");926 }927 }928 return;929 }930 if (std.mem.eql(u8, ctx.sub, "install")) {931 const attr = if (ctx.args.len > 0) ctx.args[0] else return error.PackageNotFound;932 _ = try exec.profileInstall(ctx.io, &ctx.machine, ctx.gpa, attr);933 output.done();934 return;935 }936 if (std.mem.eql(u8, ctx.sub, "remove")) {937 const attr = if (ctx.args.len > 0) ctx.args[0] else return error.PackageNotFound;938 _ = try exec.profileRemove(ctx.io, &ctx.machine, ctx.gpa, attr);939 output.done();940 return;941 }942 if (std.mem.eql(u8, ctx.sub, "upgrade")) {943 _ = try exec.profileUpgrade(ctx.io, &ctx.machine, ctx.gpa);944 output.done();945 return;946 }947}948949950951pub fn pkg(ctx: Ctx) !void {952 const name = if (ctx.args.len > 0) ctx.args[0] else return error.PackageNotFound;953 if (std.mem.eql(u8, ctx.sub, "why")) {954 _ = try exec.pkgWhy(ctx.io, &ctx.machine, ctx.gpa, name);955 return;956 }957 if (std.mem.eql(u8, ctx.sub, "deps")) {958 _ = try exec.pkgDeps(ctx.io, &ctx.machine, ctx.gpa, name);959 return;960 }961 if (std.mem.eql(u8, ctx.sub, "size")) {962 const s = exec.pkgSize(ctx.gpa, ctx.io, &ctx.machine, name) catch null;963 if (s) |sz| {964 defer ctx.gpa.free(sz);965 output.pkgResult("size", sz);966 }967 return;968 }969 if (std.mem.eql(u8, ctx.sub, "path")) {970 const p = try exec.storePath(ctx.gpa, ctx.io, &ctx.machine, name);971 defer ctx.gpa.free(p);972 output.pkgResult("path", p);973 return;974 }975 if (std.mem.eql(u8, ctx.sub, "closure")) {976 _ = try exec.pkgClosure(ctx.io, &ctx.machine, ctx.gpa, name);977 return;978 }979}980981982983pub fn pin(ctx: Ctx) !void {984 const input = if (ctx.args.len > 0) ctx.args[0] else return error.Internal;985 const commit = if (ctx.args.len > 1) ctx.args[1] else return error.Internal;986 exec.editFlakeLock(ctx.gpa, input, commit) catch return error.ProcessFailed;987 output.pinDone(input, commit);988}989990pub fn unpin(ctx: Ctx) !void {991 const input = if (ctx.args.len > 0) ctx.args[0] else return error.Internal;992 _ = try exec.flakeUpdate(ctx.io, &ctx.machine, ctx.gpa, input);993 output.unpinDone(input);994}995996997998pub fn hash(ctx: Ctx) !void {999 const path = if (ctx.args.len > 0) ctx.args[0] else return error.Internal;1000 const h = try exec.nixHash(ctx.gpa, ctx.io, &ctx.machine, path);1001 defer ctx.gpa.free(h);1002 output.hashResult(std.mem.trim(u8, h, " \t\n\r"));1003}10041005pub fn fetch(ctx: Ctx) !void {1006 const url = if (ctx.args.len > 0) ctx.args[0] else return error.Internal;1007 const h = try exec.nixFetch(ctx.gpa, ctx.io, &ctx.machine, url);1008 defer ctx.gpa.free(h);1009 output.fetchResult(std.mem.trim(u8, h, " \t\n\r"), url);1010}1011101210131014pub fn build(ctx: Ctx) !void {1015 const attr: ?[]const u8 = if (ctx.args.len > 0) ctx.args[0] else null;1016 output.buildStart(attr orelse "default");1017 const code = try exec.nixBuild(ctx.io, &ctx.machine, ctx.gpa, attr);1018 if (code != 0) return error.BuildFailed;1019 output.buildDone();1020}10211022pub fn runCmd(ctx: Ctx) !void {1023 const pkg_arg = if (ctx.args.len > 0) ctx.args[0] else return error.PackageNotFound;1024 output.runStart(pkg_arg);1025 _ = try exec.nixRun(ctx.io, &ctx.machine, ctx.gpa, pkg_arg);1026}1027102810291030pub fn hello(ctx: Ctx) !void {1031 output.hello(ctx.cfg.machines);1032}10331034pub fn setup(gpa: std.mem.Allocator, io: std.Io, environ: *const std.process.Environ.Map) !void {1035 output.setupCheck();1036 const status = exec.checkNixFeatures(gpa, io, environ);1037 output.setupFeatures(status.nix_command, status.flakes);10381039 if (status.allEnabled()) {1040 output.setupAllSet();1041 api.ensureRegistryPinned(gpa, io, environ);1042 return;1043 }10441045 output.setupEnabling();1046 const path = try exec.enableNixFeatures(gpa, io, environ);1047 defer gpa.free(path);10481049 output.setupDone(path, exec.isNixOS(io));1050 api.ensureRegistryPinned(gpa, io, environ);1051}1052105310541055fn logAction(ctx: Ctx, action: []const u8, gen: u32) void {1056 var ts_buf: [32]u8 = undefined;1057 const ts = nowIso(ctx.io, &ts_buf);1058 var buf: [256]u8 = undefined;1059 const entry = std.fmt.bufPrint(&buf, "{{\"ts\":\"{s}\",\"machine\":\"{s}\",\"command\":\"{s}\",\"outcome\":\"success\",\"gen_before\":null,\"gen_after\":{d},\"duration_ms\":0}}\n", .{1060 ts, ctx.machine.name, action, gen,1061 }) catch return;1062 exec.appendNinaLog(ctx.io, ctx.environ, entry);1063}1064106510661067fn nowIso(io: std.Io, buf: []u8) []const u8 {1068 const fallback = "1970-01-01T00:00:00Z";1069 const ns = std.Io.Clock.real.now(io).toNanoseconds();1070 if (ns <= 0) return fallback;1071 const secs: u64 = @intCast(@divFloor(ns, 1_000_000_000));1072 const es = std.time.epoch.EpochSeconds{ .secs = secs };1073 const ed = es.getEpochDay();1074 const yd = ed.calculateYearDay();1075 const md = yd.calculateMonthDay();1076 const ds = es.getDaySeconds();1077 return std.fmt.bufPrint(buf, "{d:0>4}-{d:0>2}-{d:0>2}T{d:0>2}:{d:0>2}:{d:0>2}Z", .{1078 yd.year, md.month.numeric(), @as(u32, md.day_index) + 1,1079 ds.getHoursIntoDay(), ds.getMinutesIntoHour(), ds.getSecondsIntoMinute(),1080 }) catch return fallback;1081}