#!/usr/bin/env nu
# Thin entry for server commands.
# Loads servers/create.nu directly — bypasses full dispatcher + run_module re-invocation.
# Cuts startup from ~46s to ~3-5s (single Nushell process, no exec re-spawn).

export-env {
    let lib_dirs_raw = ($env.NU_LIB_DIRS? | default "")
    let current_lib_dirs = if ($lib_dirs_raw | type) == "string" {
        if ($lib_dirs_raw | is-empty) { [] } else { ($lib_dirs_raw | split row ":") }
    } else {
        $lib_dirs_raw
    }
    let dynamic = ($env.PROVISIONING? | default "" | path join "core" "nulib")
    $env.NU_LIB_DIRS = ([
        "/opt/provisioning/core/nulib"
        "/usr/local/provisioning/core/nulib"
    ] | append $current_lib_dirs | append (if ($dynamic | is-not-empty) { [$dynamic] } else { [] }))

    # Strip leading "server"/"s" token so get-provisioning-args returns the sub-command
    # e.g. "server create --infra x" → "create --infra x"
    let args_raw = ($env.PROVISIONING_ARGS? | default "")
    $env.PROVISIONING_ARGS = ($args_raw | str replace --regex '^(server|s)\s+' '')

    # Bash exports booleans as strings — normalize before any module code runs
    let _coerce = {|raw| $raw == "true" or $raw == "1" }
    let raw_no_titles = ($env.PROVISIONING_NO_TITLES? | default "")
    if ($raw_no_titles | describe) == "string" and ($raw_no_titles | is-not-empty) {
        $env.PROVISIONING_NO_TITLES = (do $_coerce $raw_no_titles)
    }
    let raw_no_terminal = ($env.PROVISIONING_NO_TERMINAL? | default "")
    if ($raw_no_terminal | describe) == "string" and ($raw_no_terminal | is-not-empty) {
        $env.PROVISIONING_NO_TERMINAL = (do $_coerce $raw_no_terminal)
    }
    let raw_titles_shown = ($env.PROVISIONING_TITLES_SHOWN? | default "")
    if ($raw_titles_shown | describe) == "string" and ($raw_titles_shown | is-not-empty) {
        $env.PROVISIONING_TITLES_SHOWN = (do $_coerce $raw_titles_shown)
    }
    let raw_debug = ($env.PROVISIONING_DEBUG? | default "")
    if ($raw_debug | describe) == "string" and ($raw_debug | is-not-empty) {
        $env.PROVISIONING_DEBUG = (do $_coerce $raw_debug)
    }
}

use orchestration/servers/create.nu *
use orchestration/servers/delete.nu *
use orchestration/servers/ssh.nu *
use orchestration/servers/list.nu *
use orchestration/servers/upgrade.nu *
use orchestration/servers/status.nu ["main status" "main show"]
use orchestration/servers/generate.nu [on_generate_storage_server]
use orchestration/servers/formula.nu [on_server_formula]

def main [
    ...args: string
    --infra (-i): string    = ""
    --settings (-s): string = ""
    --outfile (-o): string  = ""
    --serverpos (-p): int
    --check (-c)
    --yes (-y)
    --del-volume              # (delete) also delete attached volumes
    --del-fip                 # (delete) also delete assigned floating IPs
    --run (-r)
    --live
    --wait (-w)
    --select: string        = ""
    --debug (-x)
    --xm
    --xc
    --xr
    --xld
    --metadata
    --notitles
    --orchestrated
    --orchestrator: string  = ""
    --out: string           = ""
    --helpinfo (-h)
]: nothing -> nothing {
    if $debug { $env.PROVISIONING_DEBUG = true }

    # CMD_ARGS from bash wrapper includes "server"/"s" as arg[0] — strip it.
    let first = ($args | get 0? | default "")
    let rest = if $first in ["server", "s"] { $args | skip 1 } else { $args }

    let subcmd = ($rest | get 0? | default "")
    let name   = ($rest | get 1? | default "")

    match $subcmd {
        "show" | "sh" | "info" => {
            if ($name | is-not-empty) {
                main show $name --infra $infra --settings $settings --out $out --live=$live --debug=$debug
            } else {
                main show --infra $infra --settings $settings --out $out --live=$live --debug=$debug
            }
        }
        "status" | "st" => {
            if ($name | is-not-empty) {
                main status $name --infra $infra --settings $settings --debug=$debug --live=$live
            } else {
                main status --infra $infra --settings $settings --debug=$debug --live=$live
            }
        }
        "list" | "ls" | "lis" | "l" => {
            if ($infra | is-not-empty) {
                main list --infra $infra --debug=$debug --out=$out --live=$live
            } else {
                main list --debug=$debug --out=$out --live=$live
            }
        }
        "create" | "c" => {
            if ($infra | is-not-empty) {
                if ($name | is-not-empty) {
                    main create $name --infra $infra --wait=$wait --check=$check --outfile $outfile
                } else {
                    main create --infra $infra --wait=$wait --check=$check --outfile $outfile
                }
            } else {
                if ($name | is-not-empty) {
                    main create $name --wait=$wait --check=$check --outfile $outfile
                } else {
                    main create --wait=$wait --check=$check --outfile $outfile
                }
            }
        }
        "sync" => {
            if ($infra | is-not-empty) {
                main sync --infra $infra
            } else {
                main sync
            }
        }
        "upgrade" | "u" => {
            if ($name | is-not-empty) {
                main upgrade $name --infra $infra --settings $settings --check=$check --yes=$yes --debug=$debug
            } else {
                main upgrade --infra $infra --settings $settings --check=$check --yes=$yes --debug=$debug
            }
        }
        "delete" | "d" | "del" => {
            if ($name | is-not-empty) {
                if $del_volume and $del_fip {
                    main delete $name --infra $infra --yes=$yes --del-volume --del-fip
                } else if $del_volume {
                    main delete $name --infra $infra --yes=$yes --del-volume
                } else if $del_fip {
                    main delete $name --infra $infra --yes=$yes --del-fip
                } else {
                    main delete $name --infra $infra --yes=$yes
                }
            } else {
                if $del_volume and $del_fip {
                    main delete --all --infra $infra --yes=$yes --del-volume --del-fip
                } else if $del_volume {
                    main delete --all --infra $infra --yes=$yes --del-volume
                } else if $del_fip {
                    main delete --all --infra $infra --yes=$yes --del-fip
                } else {
                    main delete --all --infra $infra --yes=$yes
                }
            }
        }
        "ssh" => {
            # Only forward non-default flags to avoid polluting the sub-command signature
            let has_infra    = ($infra    | is-not-empty)
            let has_settings = ($settings | is-not-empty)
            let has_name     = ($name     | is-not-empty)
            if $run {
                match [$has_name, $has_infra, $has_settings, $debug] {
                    [true,  true,  true,  true ] => { main ssh $name --infra $infra --settings $settings --debug --run }
                    [true,  true,  true,  false] => { main ssh $name --infra $infra --settings $settings --run }
                    [true,  true,  false, true ] => { main ssh $name --infra $infra --debug --run }
                    [true,  true,  false, false] => { main ssh $name --infra $infra --run }
                    [true,  false, true,  true ] => { main ssh $name --settings $settings --debug --run }
                    [true,  false, true,  false] => { main ssh $name --settings $settings --run }
                    [true,  false, false, true ] => { main ssh $name --debug --run }
                    [true,  false, false, false] => { main ssh $name --run }
                    [false, true,  true,  true ] => { main ssh --infra $infra --settings $settings --debug --run }
                    [false, true,  true,  false] => { main ssh --infra $infra --settings $settings --run }
                    [false, true,  false, true ] => { main ssh --infra $infra --debug --run }
                    [false, true,  false, false] => { main ssh --infra $infra --run }
                    [false, false, true,  true ] => { main ssh --settings $settings --debug --run }
                    [false, false, true,  false] => { main ssh --settings $settings --run }
                    [false, false, false, true ] => { main ssh --debug --run }
                    _                            => { main ssh --run }
                }
            } else {
                match [$has_name, $has_infra, $has_settings, $debug] {
                    [true,  true,  true,  true ] => { main ssh $name --infra $infra --settings $settings --debug }
                    [true,  true,  true,  false] => { main ssh $name --infra $infra --settings $settings }
                    [true,  true,  false, true ] => { main ssh $name --infra $infra --debug }
                    [true,  true,  false, false] => { main ssh $name --infra $infra }
                    [true,  false, true,  true ] => { main ssh $name --settings $settings --debug }
                    [true,  false, true,  false] => { main ssh $name --settings $settings }
                    [true,  false, false, true ] => { main ssh $name --debug }
                    [true,  false, false, false] => { main ssh $name }
                    [false, true,  true,  true ] => { main ssh --infra $infra --settings $settings --debug }
                    [false, true,  true,  false] => { main ssh --infra $infra --settings $settings }
                    [false, true,  false, true ] => { main ssh --infra $infra --debug }
                    [false, true,  false, false] => { main ssh --infra $infra }
                    [false, false, true,  true ] => { main ssh --settings $settings --debug }
                    [false, false, true,  false] => { main ssh --settings $settings }
                    [false, false, false, true ] => { main ssh --debug }
                    _                            => { main ssh }
                }
            }
        }
        "generate" | "g" | "gen" => {
            let template = ($rest | get 2? | default "")
            match $template {
                "storage" | "strg" | "st" => {
                    on_generate_storage_server --infra $infra --settings $settings
                }
                _ => {
                    if ($template | is-not-empty) {
                        print $"❌ Unknown generate template '($template)'."
                        print ""
                    }
                    print "Usage: prvng server generate <template> [--infra <i>]"
                    print ""
                    print "Templates:"
                    print "  storage   Generate a Longhorn storage node definition  (alias: strg, st)"
                    print ""
                    print "Examples:"
                    print "  prvng server generate storage"
                    print "  prvng server generate storage --infra infra/libre-wuji"
                }
            }
        }
        "formula" | "f" | "plan" => {
            if ($name | is-not-empty) {
                on_server_formula $name --infra $infra --settings $settings
            } else {
                on_server_formula --infra $infra --settings $settings
            }
        }
        "volume" | "vol" => {
            print "❌ Volume management is not available — provisioning-volume.nu not found."
        }
        _ => {
            if ($subcmd | is-not-empty) {
                print $"❌ Unknown server subcommand '($subcmd)'."
                print ""
            }
            print "Server Management"
            print "================="
            print ""
            print "Usage: prvng server <subcommand> [options]"
            print ""
            print "Subcommands:"
            print "  status   [<name>] [--infra <i>] [--live]          Server status table  (alias: st)"
            print "  list     [--infra <i>] [--out json|yaml] [--live]  List servers        (alias: ls, l)"
            print "  create   [<name>] [--infra <i>] [--wait]          Create server        (alias: c)"
            print "  delete   <name>  [--infra <i>] [--yes]            Delete server        (alias: d, del)"
            print "  ssh      <name>  [--infra <i>] [--run]            SSH into server"
            print "  sync     [--infra <i>]                            Sync server state"
            print "  upgrade  <name>  [--infra <i>] [--check]          Upgrade server       (alias: u)"
            print "  generate <tmpl>  [--infra <i>]                    Generate server def  (alias: g, gen)"
            print "             storage  — Longhorn storage node        (alias: strg, st)"
            print "  formula  [<name>] [--infra <i>]                   Node chain + settings + commands  (alias: f, plan)"
            print ""
            print "Examples:"
            print "  prvng server list"
            print "  prvng server status"
            print "  prvng server create --infra sgoyol --wait"
            print "  prvng server ssh sgoyol-0"
            print "  prvng server generate storage --infra infra/libre-wuji"
            print "  prvng server formula libre-wuji-strg-0 --infra infra/libre-wuji"
            print "  prvng s l                                          # list short form"
            print "  prvng s st                                         # status short form"
            print "  prvng s f libre-wuji-cp-0                         # formula short form"
        }
    }
}