#!/bin/sh -e

GREEN='\033[1;32m'
YELLOW='\033[1;33m'
RED='\033[1;31m'
NC='\033[0m'

if [ $# -gt 0 ]; then
    MACHINE_NAME="$1"
    shift
else
    MACHINE_NAME=$(hostname | cut -d'.' -f1)
fi

if [ ! -d "hosts/darwin/$MACHINE_NAME" ]; then
    echo "${RED}Error: Machine configuration '$MACHINE_NAME' not found in hosts/darwin/${NC}"
    echo "${YELLOW}Available machines:${NC}"
    ls -1 hosts/darwin/ | grep -v "default.nix\|shared.nix" || echo "  No machine configurations found"
    echo ""
    echo "${YELLOW}Usage: $0 [machine-name] [additional-nix-args]${NC}"
    echo "  If no machine name is provided, uses current hostname: $(hostname | cut -d'.' -f1)"
    exit 1
fi

FLAKE_SYSTEM="darwinConfigurations.${MACHINE_NAME}.system"


echo "${YELLOW}Building configuration for machine: ${MACHINE_NAME}${NC}"
NIXPKGS_ALLOW_UNFREE=1 nix --extra-experimental-features 'nix-command flakes' build --impure .#$FLAKE_SYSTEM "$@"

echo "${YELLOW}Switching to new generation...${NC}"
sudo NIXPKGS_ALLOW_UNFREE=1 ./result/sw/bin/darwin-rebuild switch --impure --flake .#${MACHINE_NAME}

echo "${YELLOW}Cleaning up...${NC}"
unlink ./result

echo "${GREEN}Switch to new generation complete for ${MACHINE_NAME}!${NC}"