#!/usr/bin/env nu

use ../common.nu *

def main [] {
    print_info "Available generations:"
    darwin-rebuild --list-generations
    
    let gen_num = input "Enter generation number to rollback to: "
    
    if ($gen_num | is-empty) {
        print_error "No generation number provided"
        exit 1
    }
    
    print_warning $"Rolling back to generation ($gen_num)..."
    darwin-rebuild switch --switch-generation $gen_num
    
    print_success $"Rollback to generation ($gen_num) complete"
}