Justfile

# List available commands by default
default:
  @just system

# allows extremely fast shell commands with fallbacks to nix
invoke function *args:
  #!/usr/bin/env nu
  def shell_command [cmd: list<string>, cmd_backup: list<string>, args: list<string> = []] {
    # the exernal `which` seems to resolve better...
    let cmd_path = try { (^which ($cmd | first)) | lines | first } catch { "" }
    if ($cmd_path | str length) > 0 {
      sh -c ($args | prepend $cmd | str join " ")
    } else {
      print $"💡 Consider installing (ansi green)($cmd | first)(ansi reset) into your PATH"
      print $"   It will enable faster invocations"
      sh -c ($args | prepend $cmd_backup | str join " ")
    }
  }

  def darwin_rebuild [args: list<string> = []] {
    shell_command [darwin-rebuild] [nix shell nix-darwin --command darwin-rebuild] $args
  }
  def nixos_rebuild [args: list<string> = []] {
    shell_command [nixos-rebuild] [nix shell "nixpkgs#nixos-rebuild" --command nixos-rebuild] $args
  }
  def nom [args: list<string> = []] {
    shell_command [nom] [nix shell "nixpkgs#nix-output-monitor" --command nom] $args
  }

  match "{{function}}" {
    "darwin-rebuild" => { darwin_rebuild [{{args}}] }
    "nixos-rebuild" => { nixos_rebuild [{{args}}] }
    "nom" => { nom [{{args}}] }
    _ => { error make {msg: "Unknown function: {{function}}"} }
  }


system:
  #!/usr/bin/env nu
  def launchctl_list [] {
    launchctl list | split row -r '\n' | skip 1 | split column --regex '\s+' PID Status Label
  }
  def pipefail [] {
    let results = complete
    match $results {
      {exit_code: 0} => $results.stdout,
      _ => (error make { msg: $results.stdout })
    }
  }


  match (uname | get kernel-name) {
    "Darwin" => {
      just invoke darwin-rebuild switch --flake . --show-trace out+err>| tee { just invoke nom } | pipefail
      # TODO: relaunch hm services?
      launchctl_list | where Label =~ "^org.nixos" | each {|e|
        print $"🍃 Relaunching ($e.Label)"
        let agentPath = $"~/Library/LaunchAgents/($e.Label).plist"
        let launchGroup = $"gui/(id -u)"
        try { launchtl bootout $launchGroup $agentPath }
        try { launchtl bootstrap $launchGroup $agentPath }
      }
      null
    }
    "Linux" => {
      just invoke nixos-rebuild switch --flake . --show-trace out+err>| tee { just invoke nom } | pipefail

      print "Checking for bad systemd user units..."
      let bad_settings = (systemctl --user list-unit-files --legend=false |
                          lines |
                          split column -r '\s+' unit state preset |
                          where unit !~ "@\\." |
                          each {|row|
                            let unit = $row.unit
                            let has_bad_setting = systemctl --user status $unit | str contains 'bad-setting'
                            { unit: $unit, bad_setting: $has_bad_setting }
                          } |
                          where bad_setting == true)

      if ($bad_settings | length) > 0 {
        error make {msg: $"Bad settings found: ($bad_settings)"}
      } else {
        print "No bad units found!"
      }
    }
    _ => {
      print "Unknown OS"
    }
  }

init:
  #!/usr/bin/env nu
  nix run home-manager/master -- init --switch

bleed:
  #!/usr/bin/env nu
  nix flake lock --update-input nixpkgs-bleeding

update:
  #!/usr/bin/env nu
  nix flake update

diff:
  #!/usr/bin/env nu
  nix run nixpkgs#nvd -- diff /run/booted-system /run/current-system


# `nix flake check` only works on nixos because of
# https://github.com/NixOS/nix/issues/4265
# The above command basically insists on checking things it does not have to.
# Here is excerpt from `nix flake check --help`:
# Evaluation checks
#     · checks.system.name
#     · defaultPackage.system
#     · devShell.system
#     · devShells.system.name
#     · nixosConfigurations.name.config.system.build.toplevel
#     · packages.system.name
# It would be cool to disable nixosConfigurations, but oh well. Maybe one day :).
check:
  #!/usr/bin/env nu
  nix flake check
  # TODO: use treefmt instead
  nix run nixpkgs#deadnix -- -f # check for dead code, fails if any

check-all:
  #!/usr/bin/env nu
  NIXPKGS_ALLOW_UNSUPPORTED_SYSTEM=1 nix flake check --impure --all-systems