#!/usr/bin/env bash
# verify.sh — Verify memory-continuity installation (3-layer check)
#
# Usage:
#   bash scripts/verify.sh [--workspace PATH] [--sample]
#
# Options:
#   --workspace PATH   Override default workspace (~/.openclaw/workspace/main)
#   --sample           Show a sample high-importance CURRENT_STATE.md

set -euo pipefail

# ---------------------------------------------------------------------------
# Colors
# ---------------------------------------------------------------------------
RED='\033[0;31m'
YELLOW='\033[1;33m'
GREEN='\033[0;32m'
CYAN='\033[0;36m'
BOLD='\033[1m'
RESET='\033[0m'

ok()   { echo -e "  ${GREEN}✅ $*${RESET}"; }
warn() { echo -e "  ${YELLOW}⚠️  $*${RESET}"; }
fail() { echo -e "  ${RED}❌ $*${RESET}"; }
info() { echo -e "  ${CYAN}ℹ️  $*${RESET}"; }
header() { echo -e "\n${BOLD}$*${RESET}"; }

# ---------------------------------------------------------------------------
# Argument parsing
# ---------------------------------------------------------------------------
WORKSPACE="${HOME}/.openclaw/workspace/main"
SHOW_SAMPLE=false

while [[ $# -gt 0 ]]; do
  case "$1" in
    --workspace)
      WORKSPACE="$2"; shift 2 ;;
    --sample)
      SHOW_SAMPLE=true; shift ;;
    *)
      echo "Unknown argument: $1" >&2; exit 1 ;;
  esac
done

# Expand ~ in workspace path
WORKSPACE="${WORKSPACE/#\~/$HOME}"

# ---------------------------------------------------------------------------
# --sample: print example state and exit
# ---------------------------------------------------------------------------
if $SHOW_SAMPLE; then
  echo -e "\n${BOLD}Sample CURRENT_STATE.md (high-importance production task):${RESET}\n"
  cat <<'EOF'
# Current State
> Last updated: 2026-03-19T11:42:00Z

## Objective
Complete production database migration v2→v3 for user activity table (sharded, ~80M rows).

## Current Step
Migration script written and reviewed. Dry-run on staging passed.
**Not yet tested against production replica.**

## Key Decisions
- Using online schema change (pt-online-schema-change) to avoid table lock
- Backfill batch size: 5,000 rows / 500ms to stay under replication lag threshold
- Rollback plan: swap back via feature flag, no destructive drop until T+48h

## Next Action
Schedule test run against prod replica — waiting for Tao to confirm migration
window (UTC+10, Thursday 23:00). Do NOT proceed without explicit sign-off.

## Blockers
- ⚠️  BLOCKED: Tao hasn't confirmed migration window (UTC+10 Thu 23:00)
- ⚠️  Prod replica test not yet scheduled

## Risk
- Skipping production replica test → potential data loss on edge cases not covered by staging schema
- Migration window must be off-peak; violating this risks exceeding replication lag SLA (>30s triggers alert)

## Unsurfaced Results
- Staging dry-run log: /tmp/pt-osc-staging-2026-03-19.log (not yet reviewed for warnings)
EOF
  exit 0
fi

# ---------------------------------------------------------------------------
# Header
# ---------------------------------------------------------------------------
echo -e "\n${BOLD}memory-continuity — Installation Verifier${RESET}"
echo "────────────────────────────────────────────"

ERRORS=0
WARNINGS=0

# ===========================================================================
# Layer 1: Required files exist
# ===========================================================================
header "Layer 1: Required files"

PLUGIN_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"

check_file() {
  local rel="$1"
  local path="${PLUGIN_DIR}/${rel}"
  if [[ -f "$path" ]]; then
    ok "$rel"
  else
    fail "$rel — NOT FOUND (expected at ${path})"
    ERRORS=$((ERRORS + 1))
  fi
}

check_file "SKILL.md"
check_file "scripts/continuity_doctor.py"
check_file "openclaw.plugin.json"

# ===========================================================================
# Layer 2: continuity_doctor.py runs without error
# ===========================================================================
header "Layer 2: Tool availability"

if ! command -v python3 &>/dev/null; then
  fail "python3 not found in PATH"
  ERRORS=$((ERRORS + 1))
else
  ok "python3 found ($(python3 --version 2>&1))"

  DOCTOR="${PLUGIN_DIR}/scripts/continuity_doctor.py"
  if [[ ! -f "$DOCTOR" ]]; then
    fail "continuity_doctor.py missing — skipping doctor check"
    ERRORS=$((ERRORS + 1))
  else
    info "Running: python3 scripts/continuity_doctor.py --workspace ${WORKSPACE}"
    echo ""
    if python3 "$DOCTOR" --workspace "$WORKSPACE" 2>&1 | sed 's/^/    /'; then
      echo ""
      ok "continuity_doctor.py exited cleanly"
    else
      echo ""
      warn "continuity_doctor.py exited with non-zero status (see output above)"
      WARNINGS=$((WARNINGS + 1))
    fi
  fi
fi

# ===========================================================================
# Layer 3: CURRENT_STATE.md content check
# ===========================================================================
header "Layer 3: Workspace state"

STATE_FILE="${WORKSPACE}/memory/CURRENT_STATE.md"

if [[ ! -f "$STATE_FILE" ]]; then
  warn "CURRENT_STATE.md not found at: ${STATE_FILE}"
  warn "This is normal for a brand-new install — state will be created after your first session."
  WARNINGS=$((WARNINGS + 1))
else
  # Detect placeholder / trivial content:
  # - File is empty
  # - Only contains the header line "# Current State"
  # - Contains the phrase "No active" or "placeholder"
  CONTENT=$(cat "$STATE_FILE")
  LINE_COUNT=$(echo "$CONTENT" | grep -c '[^[:space:]]' || true)

  IS_PLACEHOLDER=false

  if [[ $LINE_COUNT -le 2 ]]; then
    IS_PLACEHOLDER=true
  elif ! echo "$CONTENT" | grep -qE '##\s+(Objective|Current Step|In Flight|Next Action|Blocked)'; then
    # Has some lines but none of the expected structured sections
    IS_PLACEHOLDER=true
  else
    # Check if the Objective section itself is trivial (placeholder text right after the heading)
    OBJECTIVE_VALUE=$(echo "$CONTENT" | awk '/^## Objective/{found=1; next} found && /^##/{exit} found && /[^[:space:]]/{print; exit}')
    if echo "$OBJECTIVE_VALUE" | grep -qiE '^\s*(no active|placeholder|todo|tbd|empty|n\/a|none|untitled)$'; then
      IS_PLACEHOLDER=true
    fi
  fi

  if $IS_PLACEHOLDER; then
    echo ""
    warn "CURRENT_STATE.md exists but appears to be empty or a placeholder."
    warn "The plugin is installed correctly, but your working state hasn't been"
    warn "captured yet. After your first real work session it will be populated"
    warn "automatically."
    echo ""
    warn "To see what a high-importance state entry looks like, run:"
    echo -e "    ${CYAN}bash scripts/verify.sh --sample${RESET}"
    WARNINGS=$((WARNINGS + 1))
  else
    ok "CURRENT_STATE.md found with substantive content (${LINE_COUNT} non-blank lines)"
    # Show last-updated line if present
    UPDATED=$(grep -m1 'Last updated' "$STATE_FILE" || true)
    [[ -n "$UPDATED" ]] && info "$UPDATED"
  fi
fi

# ===========================================================================
# Summary
# ===========================================================================
echo ""
echo "────────────────────────────────────────────"
if [[ $ERRORS -eq 0 && $WARNINGS -eq 0 ]]; then
  echo -e "${GREEN}${BOLD}All checks passed. memory-continuity is correctly installed.${RESET}"
elif [[ $ERRORS -eq 0 ]]; then
  echo -e "${YELLOW}${BOLD}Checks passed with ${WARNINGS} warning(s). Review warnings above.${RESET}"
else
  echo -e "${RED}${BOLD}${ERRORS} error(s), ${WARNINGS} warning(s). Installation may be incomplete.${RESET}"
  echo -e "${RED}Re-run the installer:  bash scripts/post-install.sh${RESET}"
fi
echo ""