274 lines
13 KiB
Text
274 lines
13 KiB
Text
|
#!/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"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|