#!/usr/bin/env bash

# ============================================
# Lifecycle
# ============================================

function cmd::config::on_load() {
  flag::register --name
  flag::register --type
  flag::register --force
  flag::register --dry-run
}

# ============================================
# Help
# ============================================

function cmd::config::help() {
  cat <<EOF
Usage: wgctl config --name <name>

Show the WireGuard config file for a client.

Options:
  --name <name>     Client name (e.g. phone-nuno)

Examples:
  wgctl config --name phone-nuno
  wgctl config --name laptop-nuno
EOF
}

# ============================================
# Run
# ============================================

function cmd::config::run() {
  local subcmd="${1:-show}"

  # If first arg is a flag, treat as 'show' subcommand
  if [[ "$subcmd" == --* ]]; then
    subcmd="show"
  else
    shift || true
  fi

  case "$subcmd" in
    show)    cmd::config::_show    "$@" ;;
    migrate) cmd::config::migrate  "$@" ;;
    help)    cmd::config::help          ;;
    *)
      log::error "Unknown subcommand: '${subcmd}'"
      cmd::config::help
      return 1
      ;;
  esac
}

# ============================================
# Show
# ============================================

function cmd::config::_show() {
  local name="" type=""
  while [[ $# -gt 0 ]]; do
    case "$1" in
      --name)  name="$2";         shift 2 ;;
      --type)  type="$2";         shift 2 ;;
      --help)  cmd::config::help; return  ;;
      *) log::error "Unknown flag: $1"; return 1 ;;
    esac
  done

  [[ -z "$name" ]] && log::error "Missing required flag: --name" && return 1
  name=$(peers::resolve_and_require "$name" "$type") || return 1

  local conf
  conf="$(ctx::clients)/${name}.conf"

  log::section "Client Config: ${name}"
  cat "$conf"
}

# ============================================
# Migrate
# ============================================

function cmd::config::migrate() {
  local force=false dry_run=false
 
  while [[ $# -gt 0 ]]; do
    case "$1" in
      --force)   force=true;   shift ;;
      --dry-run) dry_run=true; shift ;;
      --help)    cmd::config::help; return ;;
      *) log::error "Unknown flag: $1"; return 1 ;;
    esac
  done
 
  local wgctl_dir
  wgctl_dir="$(ctx::wgctl)"
  local config_dir="${wgctl_dir}/config"
  local data_dir="${wgctl_dir}/data"
  local legacy_conf="${wgctl_dir}/wgctl.conf"
  local json_conf="${config_dir}/wgctl.json"
 
  # Check if already migrated
  if [[ -f "$json_conf" && ! -f "$legacy_conf" ]]; then
    log::wg_warning "Already migrated to new config structure"
    return 0
  fi
 
  log::section "wgctl Config Migration"
  printf "\n"
  printf "  This will:\n"
  printf "  1. Create %s/config/ and %s/data/\n" "$wgctl_dir" "$wgctl_dir"
  printf "  2. Convert wgctl.conf → wgctl.json\n"
  printf "  3. Move data files to data/\n\n"
 
  if ! $force && ! $dry_run; then
    read -r -p "  Proceed? [y/N] " confirm
    case "$confirm" in [yY]*) ;; *) log::info "Aborted"; return 0 ;; esac
  fi
 
  local do=""
  $dry_run && do="echo [dry-run]"
 
  # 1. Create directories
  $dry_run || mkdir -p "$config_dir" "$data_dir"
  $dry_run && printf "  Would create: %s/config/\n" "$wgctl_dir"
  $dry_run && printf "  Would create: %s/data/\n"   "$wgctl_dir"
 
  # 2. Convert wgctl.conf → wgctl.json
  if [[ -f "$legacy_conf" ]]; then
    if ! $dry_run; then
      config::_convert_to_json "$legacy_conf" "$json_conf"
    fi
    printf "  %s wgctl.conf → config/wgctl.json\n" "$($dry_run && echo '[dry-run]' || echo '✓')"
  else
    log::wg_warning "No wgctl.conf found — creating default wgctl.json"
    $dry_run || config::_write_default_json "$json_conf"
  fi
 
  # 3. Move data files
  local -a data_files=(
    "hosts.json"
    "services.json"
    "subnets.json"
    "policies.json"
  )
  local -a data_dirs=(
    "rules"
    "identities"
    "groups"
    "blocks"
    "meta"
    "peer-history"
  )
 
  for f in "${data_files[@]}"; do
    if [[ -f "${wgctl_dir}/${f}" ]]; then
      $dry_run || mv "${wgctl_dir}/${f}" "${data_dir}/${f}"
      printf "  %s %s → data/%s\n" \
        "$($dry_run && echo '[dry-run]' || echo '✓')" "$f" "$f"
    fi
  done
 
  for d in "${data_dirs[@]}"; do
    if [[ -d "${wgctl_dir}/${d}" ]]; then
      $dry_run || mv "${wgctl_dir}/${d}" "${data_dir}/${d}"
      printf "  %s %s/ → data/%s/\n" \
        "$($dry_run && echo '[dry-run]' || echo '✓')" "$d" "$d"
    fi
  done
 
  # 4. Remove legacy conf after successful migration
  if ! $dry_run && [[ -f "$legacy_conf" ]]; then
    mv "$legacy_conf" "${legacy_conf}.bak"
    printf "  ✓ wgctl.conf → wgctl.conf.bak (backup)\n"
  fi
 
  printf "\n"
  $dry_run && log::wg_warning "Dry run — no changes made" \
           || log::wg_success "Migration complete"
}
 
function config::_convert_to_json() {
  local legacy_file="$1" output_file="$2"
 
  # Read legacy conf into variables
  local wg_interface="wg0" wg_endpoint="" wg_dns="10.0.0.103"
  local wg_dns_fallback="" wg_port="51820" wg_subnet="10.1.0.0/16"
  local wg_lan="10.0.0.0/24" wg_hs_check="300" date_format="eu"
 
  while IFS='=' read -r key value || [[ -n "$key" ]]; do
    [[ "$key" =~ ^[[:space:]]*# ]] && continue
    [[ -z "${key// }" ]] && continue
    key="${key// /}"
    value="${value// /}"
    case "$key" in
      WG_INTERFACE)            wg_interface="$value"    ;;
      WG_ENDPOINT)             wg_endpoint="$value"     ;;
      WG_DNS)                  wg_dns="$value"          ;;
      WG_DNS_FALLBACK)         wg_dns_fallback="$value" ;;
      WG_PORT)                 wg_port="$value"         ;;
      WG_SUBNET)               wg_subnet="$value"       ;;
      WG_LAN)                  wg_lan="$value"          ;;
      WG_HANDSHAKE_CHECK_TIME_SEC) wg_hs_check="$value" ;;
      DATE_FORMAT)             date_format="$value"     ;;
    esac
  done < "$legacy_file"
 
  # Build fallback DNS array
  local dns_fallback_json="[]"
  if [[ -n "$wg_dns_fallback" ]]; then
    local fallback_array
    fallback_array=$(echo "$wg_dns_fallback" | tr ',' '\n' | \
      while IFS= read -r s; do
        s="${s// /}"
        [[ -n "$s" ]] && printf '"%s",' "$s"
      done | sed 's/,$//')
    dns_fallback_json="[${fallback_array}]"
  fi
 
  mkdir -p "$(dirname "$output_file")"
  cat > "$output_file" << JSON
{
  "wireguard": {
    "interface": "${wg_interface}",
    "endpoint":  "${wg_endpoint}",
    "port":      ${wg_port},
    "subnet":    "${wg_subnet}",
    "lan":       "${wg_lan}"
  },
  "dns": {
    "primary":  "${wg_dns}",
    "fallback": ${dns_fallback_json}
  },
  "handshake": {
    "check_interval_sec": ${wg_hs_check}
  },
  "activity": {
    "total":   {"low": 1000000, "medium": 10000000, "high": 100000000},
    "current": {"low": 1000000, "medium": 10000000, "high": 100000000}
  },
  "display": {
    "date_format": "${date_format}"
  }
}
JSON
}
 
function config::_write_default_json() {
  local output_file="$1"
  mkdir -p "$(dirname "$output_file")"
  cat > "$output_file" << 'JSON'
{
  "wireguard": {
    "interface": "wg0",
    "endpoint":  "",
    "port":      51820,
    "subnet":    "10.1.0.0/16",
    "lan":       "10.0.0.0/24"
  },
  "dns": {
    "primary":  "10.0.0.103",
    "fallback": []
  },
  "handshake": {
    "check_interval_sec": 300
  },
  "activity": {
    "total":   {"low": 1000000, "medium": 10000000, "high": 100000000},
    "current": {"low": 1000000, "medium": 10000000, "high": 100000000}
  },
  "display": {
    "date_format": "eu"
  }
}
JSON
}