88 lines
3.4 KiB
Text
88 lines
3.4 KiB
Text
|
#!/usr/bin/env nu
|
||
|
|
# Thin entry for state | st commands.
|
||
|
|
# Loads only workspace/state.nu + accessor (~50ms vs ~49s for the full entry).
|
||
|
|
# Bash wrapper routes this for all state subcommands except sync (which delegates to full runner).
|
||
|
|
|
||
|
|
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 { [] }))
|
||
|
|
}
|
||
|
|
|
||
|
|
use workspace/state.nu *
|
||
|
|
use lib_provisioning/utils/settings.nu [find_get_settings]
|
||
|
|
use lib_provisioning/utils/interface.nu [_print]
|
||
|
|
use lib_provisioning/utils/error.nu [throw-error]
|
||
|
|
|
||
|
|
def main [
|
||
|
|
...args: string # args[0] = "state", args[1] = subcommand
|
||
|
|
--infra (-i): string = ""
|
||
|
|
--server: string = ""
|
||
|
|
--taskserv: string = ""
|
||
|
|
--kubeconfig: string = ""
|
||
|
|
--skip-ssh
|
||
|
|
--debug (-x)
|
||
|
|
--notitles
|
||
|
|
]: nothing -> nothing {
|
||
|
|
if $debug { $env.PROVISIONING_DEBUG = true }
|
||
|
|
|
||
|
|
let workspace_path = if ($env.PROVISIONING_WORKSPACE_PATH? | is-not-empty) {
|
||
|
|
$env.PROVISIONING_WORKSPACE_PATH
|
||
|
|
} else {
|
||
|
|
$env.PWD
|
||
|
|
}
|
||
|
|
|
||
|
|
# args[0] = "state" (domain prefix stripped by bash), args[1] = subcommand
|
||
|
|
let sub = ($args | get 1? | default "show")
|
||
|
|
|
||
|
|
match $sub {
|
||
|
|
"show" | "s" => {
|
||
|
|
state-show $workspace_path --server $server
|
||
|
|
},
|
||
|
|
|
||
|
|
"init" | "i" => {
|
||
|
|
let curr_settings = (find_get_settings --infra $infra)
|
||
|
|
state-init $workspace_path $curr_settings
|
||
|
|
_print $"State initialized at (state-path $workspace_path)"
|
||
|
|
},
|
||
|
|
|
||
|
|
"reset" | "r" => {
|
||
|
|
if ($server | is-empty) or ($taskserv | is-empty) {
|
||
|
|
error make { msg: "state reset requires --server <hostname> --taskserv <name>" }
|
||
|
|
}
|
||
|
|
state-node-reset $workspace_path $server $taskserv --source "cli" --actor ($env.USER? | default "system")
|
||
|
|
_print $"($server)/($taskserv) reset to pending"
|
||
|
|
},
|
||
|
|
|
||
|
|
"migrate" | "m" => {
|
||
|
|
state-migrate-from-json $workspace_path
|
||
|
|
},
|
||
|
|
|
||
|
|
# sync requires lib_provisioning (mw_server_info, mw_get_ip) — delegate to full runner
|
||
|
|
"sync" => {
|
||
|
|
let infra_arg = if ($infra | is-not-empty) { ["--infra" $infra] } else { [] }
|
||
|
|
let kconfig_arg = if ($kubeconfig | is-not-empty) { ["--kubeconfig" $kubeconfig] } else { [] }
|
||
|
|
let ssh_arg = if $skip_ssh { ["--skip-ssh"] } else { [] }
|
||
|
|
exec $"($env.PROVISIONING_NAME)" state sync ...$infra_arg ...$kconfig_arg ...$ssh_arg
|
||
|
|
},
|
||
|
|
|
||
|
|
_ => {
|
||
|
|
_print "Usage: provisioning state <subcommand> [options]"
|
||
|
|
_print ""
|
||
|
|
_print " show [--server <hostname>] — display state table"
|
||
|
|
_print " init [--infra <path>] — bootstrap state from settings"
|
||
|
|
_print " reset --server <hostname> --taskserv <name> — reset node to pending"
|
||
|
|
_print " migrate — migrate .json → .ncl"
|
||
|
|
_print " sync [--infra <path>] [--kubeconfig <path>] [--skip-ssh]"
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|