Files
nix-files/scripts/update

136 lines
3.1 KiB
Plaintext
Executable File

#!/usr/bin/env nix-shell
#!nix-shell -i bash -p findutils -p nix-update
NIX_FILES_TOP=/home/colin/nixos
usage() {
echo "update: update rev/hash for one or more packages"
echo "usage: update [options] [attr-path]"
echo ""
echo "options:"
echo "- --dry-run"
echo "- --verbose"
echo ""
echo "examples:"
echo "- update nixpkgs: update only the nixpkgs input"
echo "- update sane: update every package under the 'sane' attribute"
echo "- update: update everything i know how to update"
exit 1
}
warn() {
echo "$@"
}
info() {
echo "$@"
}
debug() {
if [ -n "$verbose" ]; then
echo "$@"
fi
}
hasEffect() {
if [ -n "$dryRun" ]; then
echo "dry-run: skip $@"
else
eval "$@"
fi
}
# usage: getPkgs outVar prefix
getPkgs() {
local -n attrsArr="$1"
local attrPrefix="$2"
if [ -z "$attrPrefix" ]; then
attrPrefix=sane
fi
# # nix-env doesn't seem to build anything when evaluating queries,
# # but since i use Import From Derivation along paths which i also want to query,
# # then i need to ensure those derivations are available for import.
# debug "creating requisite .drv store paths"
# nix-instantiate -A nix "$NIX_FILES_TOP"
# nix-instantiate -A nixpkgs-bootstrap.master "$NIX_FILES_TOP"
debug "querying packages to update as part of '$attrPrefix'"
local attrs=$(nix eval --raw -f "$NIX_FILES_TOP" 'updateTargets."'"$attrPrefix"'"' --apply 'builtins.concatStringsSep " "')
debug "got: $attrs"
attrsArr+=($attrs)
}
updateOnePkg() {
local attrPath="$1"
local updateScript=$(nix eval --raw -f "$NIX_FILES_TOP" 'updateScripts."'"$attrPath"'"')
if [ -z "$updateScript" ]; then
warn "don't know how to update '$attrPath'"
return
fi
# make sure everything needed to invoke the update script exists in-store
local context=$(nix eval --raw -f "$NIX_FILES_TOP" 'updateScripts."'"$attrPath"'"' --apply 's: builtins.concatStringsSep " " (builtins.attrNames (builtins.getContext s))')
for c in $context; do
debug "realizing updateScript requisite: $context"
nix-store --realize "$c" || true
done
info "updating: '$attrPath'"
debug "$updateScript"
"$updateScript"
}
updatePkgsInParallel() {
debug "updating packages in parallel using xargs"
debug "- $@"
debug "- xargs -n 1 -P 4 $0 ${scriptFlags[*]}"
echo "$@" | xargs -n 1 -P 4 "$0" "${scriptFlags[@]}"
}
scriptFlags=()
dryRun=
toplevelsToUpdate=()
verbose=
parseArgs() {
while [ "$#" -ne 0 ]; do
local arg=$1
shift
case "$arg" in
(--help)
usage
;;
(--dry-run)
scriptFlags+=(--dry-run)
dryRun=1
;;
(--verbose)
scriptFlags+=(--verbose)
verbose=1
;;
(*)
toplevelsToUpdate+=("$arg")
;;
esac
done
if [ "${#toplevelsToUpdate[@]}" -eq 0 ]; then
toplevelsToUpdate=(sane)
fi
}
parseArgs "$@"
pkgsToUpdate=()
for t in "${toplevelsToUpdate[@]}"; do
getPkgs pkgsToUpdate "$t"
done
case "${#pkgsToUpdate[@]}" in
(0)
echo "nothing to do"
;;
(1)
hasEffect updateOnePkg "${pkgsToUpdate[0]}"
;;
(*)
hasEffect updatePkgsInParallel "${pkgsToUpdate[@]}"
;;
esac