220 lines
No EOL
7 KiB
Nix
220 lines
No EOL
7 KiB
Nix
{ lib, stdenv, writeShellScriptBin, rsync, openssh, ... }:
|
|
|
|
writeShellScriptBin "lab" ''
|
|
#!/usr/bin/env bash
|
|
|
|
# Home-lab administration tools
|
|
# Deploy and manage NixOS configurations across home lab infrastructure
|
|
|
|
set -euo pipefail
|
|
|
|
# Configuration
|
|
HOMELAB_ROOT="/home/geir/Home-lab"
|
|
TEMP_CONFIG_DIR="/tmp/home-lab-config"
|
|
|
|
# Color output
|
|
RED='\033[0;31m'
|
|
GREEN='\033[0;32m'
|
|
YELLOW='\033[1;33m'
|
|
BLUE='\033[0;34m'
|
|
NC='\033[0m' # No Color
|
|
|
|
log() {
|
|
echo -e "''${BLUE}[lab]''${NC} $1"
|
|
}
|
|
|
|
success() {
|
|
echo -e "''${GREEN}[lab]''${NC} $1"
|
|
}
|
|
|
|
warn() {
|
|
echo -e "''${YELLOW}[lab]''${NC} $1"
|
|
}
|
|
|
|
error() {
|
|
echo -e "''${RED}[lab]''${NC} $1" >&2
|
|
}
|
|
|
|
# Deployment function
|
|
deploy_machine() {
|
|
local machine="$1"
|
|
local mode="''${2:-boot}" # boot, test, or switch
|
|
|
|
case "$machine" in
|
|
"congenital-optimist")
|
|
# Local deployment - no SSH needed
|
|
log "Deploying $machine (mode: $mode) locally"
|
|
|
|
# Deploy the configuration locally
|
|
log "Running nixos-rebuild $mode locally..."
|
|
if ! sudo nixos-rebuild $mode --flake "$HOMELAB_ROOT#$machine"; then
|
|
error "Failed to deploy configuration to $machine"
|
|
exit 1
|
|
fi
|
|
|
|
success "Successfully deployed $machine"
|
|
return 0
|
|
;;
|
|
"sleeper-service")
|
|
local target_host="sma@sleeper-service"
|
|
;;
|
|
"grey-area")
|
|
local target_host="sma@grey-area"
|
|
;;
|
|
"reverse-proxy")
|
|
local target_host="sma@reverse-proxy"
|
|
;;
|
|
*)
|
|
error "Unknown machine: $machine"
|
|
error "Available machines: congenital-optimist, sleeper-service, grey-area, reverse-proxy"
|
|
exit 1
|
|
;;
|
|
esac
|
|
|
|
log "Deploying $machine (mode: $mode)"
|
|
|
|
# Sync configuration to target machine
|
|
log "Syncing configuration to $target_host..."
|
|
if ! ${rsync}/bin/rsync -av --delete "$HOMELAB_ROOT/" "$target_host:$TEMP_CONFIG_DIR/"; then
|
|
error "Failed to sync configuration to $machine"
|
|
exit 1
|
|
fi
|
|
|
|
# Deploy the configuration
|
|
log "Running nixos-rebuild $mode on $machine..."
|
|
if ! ${openssh}/bin/ssh "$target_host" "cd $TEMP_CONFIG_DIR && sudo nixos-rebuild $mode --flake .#$machine"; then
|
|
error "Failed to deploy configuration to $machine"
|
|
exit 1
|
|
fi
|
|
|
|
success "Successfully deployed $machine"
|
|
}
|
|
|
|
# Update all machines function
|
|
update_all_machines() {
|
|
local mode="''${1:-boot}" # boot, test, or switch
|
|
local machines=("congenital-optimist" "sleeper-service" "grey-area" "reverse-proxy")
|
|
local failed_machines=()
|
|
|
|
log "Starting update of all machines (mode: $mode)"
|
|
|
|
for machine in "''${machines[@]}"; do
|
|
log "Updating $machine..."
|
|
if deploy_machine "$machine" "$mode"; then
|
|
success "✓ $machine updated successfully"
|
|
else
|
|
error "✗ Failed to update $machine"
|
|
failed_machines+=("$machine")
|
|
fi
|
|
echo "" # Add spacing between machines
|
|
done
|
|
|
|
if [[ ''${#failed_machines[@]} -eq 0 ]]; then
|
|
success "All machines updated successfully!"
|
|
else
|
|
error "Failed to update: ''${failed_machines[*]}"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
# Show deployment status
|
|
show_status() {
|
|
log "Home-lab infrastructure status:"
|
|
|
|
# Check congenital-optimist (local)
|
|
if /run/current-system/sw/bin/systemctl is-active --quiet tailscaled; then
|
|
success " congenital-optimist: ✓ Online (local)"
|
|
else
|
|
warn " congenital-optimist: ⚠ Tailscale inactive"
|
|
fi
|
|
|
|
# Check if -v (verbose) flag is passed
|
|
local verbose=0
|
|
if [[ "''${1:-}" == "-v" ]]; then
|
|
verbose=1
|
|
fi
|
|
|
|
# Check remote machines
|
|
for machine in sleeper-service grey-area reverse-proxy; do
|
|
local ssh_user="sma" # Using sma as the admin user for remote machines
|
|
|
|
# Test SSH connectivity with debug info if in verbose mode
|
|
if [[ $verbose -eq 1 ]]; then
|
|
log "Testing SSH connection to $machine (LAN)..."
|
|
${openssh}/bin/ssh -v -o ConnectTimeout=5 -o BatchMode=yes "$ssh_user@$machine" "echo ✓ SSH connection to $machine successful" 2>&1 | grep -E '(debug1|Authentication|Connection)'
|
|
|
|
log "Testing SSH connection to $machine.tailnet (Tailscale)..."
|
|
${openssh}/bin/ssh -v -o ConnectTimeout=5 -o BatchMode=yes "$ssh_user@$machine.tailnet" "echo ✓ SSH connection to $machine.tailnet successful" 2>&1 | grep -E '(debug1|Authentication|Connection)'
|
|
fi
|
|
|
|
# Try with normal SSH first (for LAN)
|
|
if ${openssh}/bin/ssh -o ConnectTimeout=2 -o BatchMode=yes "$ssh_user@$machine" "echo OK" >/dev/null 2>&1; then
|
|
success " $machine: ✓ Online (LAN)"
|
|
# Try with Tailscale hostname as fallback
|
|
elif ${openssh}/bin/ssh -o ConnectTimeout=2 -o BatchMode=yes "$ssh_user@$machine.tailnet" "echo OK" >/dev/null 2>&1; then
|
|
success " $machine: ✓ Online (Tailscale)"
|
|
else
|
|
warn " $machine: ⚠ Unreachable"
|
|
fi
|
|
done
|
|
}
|
|
|
|
# Main command handling
|
|
case "''${1:-}" in
|
|
"deploy")
|
|
if [[ $# -lt 2 ]]; then
|
|
error "Usage: lab deploy <machine> [mode]"
|
|
error "Machines: congenital-optimist, sleeper-service, grey-area, reverse-proxy"
|
|
error "Modes: boot (default), test, switch"
|
|
exit 1
|
|
fi
|
|
|
|
machine="$2"
|
|
mode="''${3:-boot}"
|
|
|
|
if [[ ! "$mode" =~ ^(boot|test|switch)$ ]]; then
|
|
error "Invalid mode: $mode. Use boot, test, or switch"
|
|
exit 1
|
|
fi
|
|
|
|
deploy_machine "$machine" "$mode"
|
|
;;
|
|
|
|
"status")
|
|
show_status
|
|
;;
|
|
|
|
"update")
|
|
mode="''${2:-boot}"
|
|
|
|
if [[ ! "$mode" =~ ^(boot|test|switch)$ ]]; then
|
|
error "Invalid mode: $mode. Use boot, test, or switch"
|
|
exit 1
|
|
fi
|
|
|
|
update_all_machines "$mode"
|
|
;;
|
|
|
|
*)
|
|
echo "Home-lab Management Tool"
|
|
echo ""
|
|
echo "Usage: lab <command> [options]"
|
|
echo ""
|
|
echo "Available commands:"
|
|
echo " deploy <machine> [mode] - Deploy configuration to a machine"
|
|
echo " Machines: congenital-optimist, sleeper-service, grey-area, reverse-proxy"
|
|
echo " Modes: boot (default), test, switch"
|
|
echo " update [mode] - Update all machines"
|
|
echo " Modes: boot (default), test, switch"
|
|
echo " status - Check infrastructure connectivity"
|
|
echo ""
|
|
echo "Examples:"
|
|
echo " lab deploy congenital-optimist boot # Deploy workstation for next boot"
|
|
echo " lab deploy sleeper-service boot # Deploy and set for next boot"
|
|
echo " lab deploy grey-area switch # Deploy and switch immediately"
|
|
echo " lab update boot # Update all machines for next boot"
|
|
echo " lab update switch # Update all machines immediately"
|
|
echo " lab status # Check all machines"
|
|
;;
|
|
esac
|
|
'' |