nuno / gistfile1.txt

#!/usr/bin/env bash function cmd::net::on_load() { flag::register --name flag::register --ip flag::register --port flag::register --desc flag::register --tag flag::register --detailed flag::register --force } function cmd::net::help() { cat <<EOF Usage: wgctl net <subcommand> [options] Manage named network services for use with block/allow rules. Services map names to IPs and ports, making rules more readable. Subcommands: list List all services show --name <name> Show service details add --name <name> --ip <ip> Add a service add --name <svc:port-name> --port <port:proto> Add a port to a service rm --name <name> Remove service or port rm --name <svc:ports> Remove all ports from service Options for add (service): --name <name> Service name (e.g. proxmox) --ip <ip> Service IP address --desc <description> Optional description --tag <tag> Optional tag (repeatable) Options for add (port): --name <svc:port-name> Service:port-name (e.g. proxmox:web-ui) --port <port:proto> Port and protocol (e.g. 8006:tcp) --desc <description> Optional description Options for list: --detailed Show ports for each service --tag <tag> Filter by tag Examples: wgctl net list wgctl net list --detailed wgctl net list --tag admin wgctl net show --name proxmox wgctl net add --name proxmox --ip 10.0.0.100 --desc "Proxmox VE" wgctl net add --name proxmox:web-ui --port 8006:tcp --desc "Web UI" wgctl net add --name proxmox:ssh --port 22:tcp wgctl net rm --name proxmox:web-ui wgctl net rm --name proxmox:ports wgctl net rm --name proxmox EOF } function cmd::net::run() { local subcmd="${1:-list}" shift || true case "$subcmd" in list) cmd::net::list "$@" ;; show) cmd::net::show "$@" ;; add) cmd::net::add "$@" ;; rm|remove|del) cmd::net::rm "$@" ;; help) cmd::net::help ;; *) log::error "Unknown subcommand: '${subcmd}'" cmd::net::help return 1 ;; esac } # ============================================ # List # ============================================ function cmd::net::list() { local detailed=false filter_tag="" while [[ $# -gt 0 ]]; do case "$1" in --detailed) detailed=true; shift ;; --tag) filter_tag="$2"; shift 2 ;; --help) cmd::net::help; return ;; *) log::error "Unknown flag: $1"; return 1 ;; esac done local net_file net_file="$(ctx::net)" if [[ ! -f "$net_file" ]]; then log::wg_warning "No services configured. Use 'wgctl net add' to add one." return 0 fi log::section "Network Services" printf "\n %-20s %-16s %-6s %s\n" "NAME" "IP" "PORTS" "DESCRIPTION" local divider divider=$(printf '─%.0s' {1..72}) printf " %s\n" "$divider" local found=false while IFS="|" read -r name ip desc tags ports; do [[ -z "$name" ]] && continue # Tag filter if [[ -n "$filter_tag" ]]; then [[ "$tags" != *"$filter_tag"* ]] && continue fi found=true local tag_display="" [[ -n "$tags" ]] && tag_display=" \033[0;37m[${tags}]\033[0m" printf " %-20s %-16s %-6s %s%b\n" \ "$name" "$ip" "${ports}p" "${desc:-—}" "$tag_display" if $detailed; then local has_ports=false # Show ports inline while IFS="|" read -r ptype pname pport pproto pdesc; do [[ "$ptype" != "port" ]] && continue has_ports=true local ann ann=$(net::annotation "$ip" "$pport" "$pproto") printf " \033[0;37m%-18s %s:%s%s\033[0m\n" \ "${pname}" "$pport" "$pproto" \ "${pdesc:+ # $pdesc}" done < <(json::net_show "$net_file" "$name") $has_ports && printf "\n" # newline after each service with ports fi done < <(json::net_list "$net_file") if ! $found; then [[ -n "$filter_tag" ]] && \ log::wg_warning "No services with tag: ${filter_tag}" || \ log::wg_warning "No services configured" fi printf "\n" return 0 } # ============================================ # Show # ============================================ function cmd::net::show() { local name="" while [[ $# -gt 0 ]]; do case "$1" in --name) util::require_flag "--name" "${2:-}" || return 1 name="$2"; shift 2 ;; --help) cmd::net::help; return ;; *) log::error "Unknown flag: $1"; return 1 ;; esac done [[ -z "$name" ]] && log::error "Missing required flag: --name" && return 1 net::require_exists "$name" || return 1 log::section "Service: ${name}" printf "\n" while IFS="|" read -r key val1 val2 val3 val4; do case "$key" in name) ui::row "Name" "$val1" ;; ip) ui::row "IP" "$val1" ;; desc) ui::row "Description" "${val1:-—}" ;; tags) ui::row "Tags" "${val1:-—}" ;; port) # val1=port_name val2=port val3=proto val4=desc local ann ann=$(net::annotation "$(json::net_resolve "$(ctx::net)" "$name")" \ "$val2" "$val3" 2>/dev/null || true) printf " %-20s \033[0;36m%s\033[0m %s:%s%s\n" \ "${val1}:" "" "$val2" "$val3" \ "${val4:+ # $val4}" ;; esac done < <(json::net_show "$(ctx::net)" "$name") printf "\n" return 0 } # ============================================ # Add # ============================================ function cmd::net::add() { local name="" ip="" port="" desc="" tags=() while [[ $# -gt 0 ]]; do case "$1" in --name) util::require_flag "--name" "${2:-}" || return 1 name="$2"; shift 2 ;; --ip) ip="$2"; shift 2 ;; --port) port="$2"; shift 2 ;; --desc) desc="$2"; shift 2 ;; --tag) tags+=("$2"); shift 2 ;; --help) cmd::net::help; return ;; *) log::error "Unknown flag: $1"; return 1 ;; esac done [[ -z "$name" ]] && log::error "Missing required flag: --name" && return 1 if [[ "$name" == *:* ]]; then # Port mode: proxmox:web-ui local svc_name="${name%%:*}" local port_name="${name##*:}" [[ -z "$port" ]] && log::error "Missing required flag: --port" && return 1 net::require_exists "$svc_name" || return 1 local port_num proto if [[ "$port" == *:* ]]; then port_num="${port%%:*}" proto="${port##*:}" else port_num="$port" proto="tcp" fi json::net_add_port "$(ctx::net)" "$svc_name" "$port_name" \ "$port_num" "$proto" "$desc" log::wg_success "Added port: ${svc_name}:${port_name} → ${port_num}/${proto}" else # Service mode: proxmox [[ -z "$ip" ]] && log::error "Missing required flag: --ip" && return 1 local tags_str tags_str=$(IFS=','; echo "${tags[*]}") json::net_add_service "$(ctx::net)" "$name" "$ip" "$desc" "$tags_str" log::wg_success "Service added: ${name} → ${ip}" fi return 0 } # ============================================ # Remove # ============================================ function cmd::net::rm() { local name="" force=false while [[ $# -gt 0 ]]; do case "$1" in --name) util::require_flag "--name" "${2:-}" || return 1 name="$2"; shift 2 ;; --force) force=true; shift ;; --help) cmd::net::help; return ;; *) log::error "Unknown flag: $1"; return 1 ;; esac done [[ -z "$name" ]] && log::error "Missing required flag: --name" && return 1 # Validate existence if [[ "$name" == *:* ]]; then local svc_name="${name%%:*}" local port_name="${name##*:}" if [[ "$port_name" != "ports" ]]; then # Check specific port exists local exists exists=$(json::net_exists "$(ctx::net)" "$name") if [[ "$exists" != "true" ]]; then log::error "Port not found: ${name}" return 1 fi else net::require_exists "$svc_name" || return 1 fi else net::require_exists "$name" || return 1 fi if ! $force; then local what="service '${name}'" [[ "$name" == *:ports ]] && what="all ports from '${name%%:*}'" [[ "$name" == *:* && "$name" != *:ports ]] && what="port '${name}'" read -r -p "Remove ${what}? [y/N] " confirm case "$confirm" in [yY]*) ;; *) log::info "Aborted"; return 0 ;; esac fi json::net_remove "$(ctx::net)" "$name" log::wg_success "Removed: ${name}" return 0 }