#!/bin/zsh
set -u

TARGET_DIR="${1:-$HOME/Documents/PrivacyGuard/v0.1.0}"
FAILURES=0
WARNINGS=0

ok() { print "[OK] $1"; }
warn() { WARNINGS=$((WARNINGS + 1)); print "[WARN] $1"; }
fail() { FAILURES=$((FAILURES + 1)); print "[FAIL] $1"; }
has_command() { command -v "$1" >/dev/null 2>&1; }

print "PrivacyGuard standard offline preflight (macOS)"
print "Target: $TARGET_DIR"
print "------------------------------------------------"

[[ "$(uname -s)" == "Darwin" ]] || fail "This package requires macOS."
[[ "$(uname -m)" == "arm64" ]] && ok "Architecture: Apple Silicon arm64" || fail "The v0.1.0 macOS package does not support Intel Macs."

MAC_VERSION="$(sw_vers -productVersion 2>/dev/null || print 0)"
MAC_MAJOR="${MAC_VERSION%%.*}"
if [[ "$MAC_MAJOR" == <-> ]] && (( MAC_MAJOR >= 14 )); then
  ok "macOS: $MAC_VERSION"
else
  fail "macOS 14 or newer is required by the bundled arm64 wheels; found $MAC_VERSION."
fi

RAM_BYTES="$(sysctl -n hw.memsize 2>/dev/null || print 0)"
RAM_GB=$((RAM_BYTES / 1024 / 1024 / 1024))
if (( RAM_GB >= 16 )); then ok "Memory: ${RAM_GB} GB"; elif (( RAM_GB >= 8 )); then warn "Memory: ${RAM_GB} GB; 16 GB is recommended and OCR may be slow."; else fail "Memory: ${RAM_GB} GB; at least 8 GB is required."; fi

FREE_KB="$(df -Pk "$HOME" | awk 'NR==2 {print $4}')"
FREE_GB=$((FREE_KB / 1024 / 1024))
if (( FREE_GB >= 15 )); then ok "Free disk: ${FREE_GB} GB"; elif (( FREE_GB >= 10 )); then warn "Free disk: ${FREE_GB} GB; 15 GB is recommended."; else fail "Free disk: ${FREE_GB} GB; at least 10 GB is required."; fi

for cmd in curl tar shasum; do
  has_command "$cmd" && ok "$cmd is available" || fail "$cmd is missing from the system."
done

if has_command python3.11; then
  ok "Python 3.11 is already installed; the bundled installer is not needed."
else
  warn "Python 3.11 is not installed; use app/prerequisites/python-3.11.9-macos11.pkg after extracting the runtime package."
fi

for port in 8000 8080 8082; do
  owner="$(lsof -nP -iTCP:$port -sTCP:LISTEN 2>/dev/null | awk 'NR==2 {print $1" (PID "$2")"}')"
  [[ -n "$owner" ]] && warn "Port $port is already in use by $owner." || ok "Port $port is available"
done

print "------------------------------------------------"
print "Bundled: frontend, Python wheels, Python 3.11 installer, llama.cpp, HaS and PaddleOCR model packages."
print "Not required: Node.js, npm, Git, Homebrew, WSL, Docker, CUDA."
print "Result: $FAILURES failure(s), $WARNINGS warning(s)"
(( FAILURES == 0 )) || exit 2