#!/usr/bin/env bash
# configure-searxng-safesearch.sh
# Set SearXNG safe-search level; optionally disable categories or engines.
#
# Usage:
#   ./configure-searxng-safesearch.sh [strict|moderate|none] [OPTIONS]
#
# Options:
#   --disable-categories  cat1,cat2   videos  images  news  science  social
#   --disable-engines     eng1,eng2   e.g. duckduckgo,bing,yandex
#   --enable-engines      eng1,eng2   re-enable engines auto-disabled by level
#
# Examples:
#   ./configure-searxng-safesearch.sh strict
#   ./configure-searxng-safesearch.sh moderate --disable-categories videos,images
#   ./configure-searxng-safesearch.sh none --disable-engines bing,duckduckgo
#   ./configure-searxng-safesearch.sh strict --disable-categories videos \
#       --disable-engines bing --enable-engines yandex

set -euo pipefail

# ── Helpers ───────────────────────────────────────────────────────────────────
red()  { printf '\e[31m%s\e[0m\n' "$*"; }
grn()  { printf '\e[32m%s\e[0m\n' "$*"; }
blu()  { printf '\e[34m%s\e[0m\n' "$*"; }
yel()  { printf '\e[33m%s\e[0m\n' "$*"; }
die()  { red "ERROR: $*"; exit 1; }
ok()   { grn "  ✓  $*"; }
info() { blu "  →  $*"; }
warn() { yel "  !  $*"; }

# ── Defaults ──────────────────────────────────────────────────────────────────
LEVEL="moderate"
DISABLE_CATS=""
DISABLE_ENGINES_EXTRA=""
ENABLE_ENGINES_EXTRA=""
BASE="${BASE:-$HOME/docker/ai-stack}"

# ── Parse arguments ───────────────────────────────────────────────────────────
while [[ $# -gt 0 ]]; do
  case "$1" in
    strict|moderate|none)   LEVEL="$1";                         shift ;;
    --disable-categories)   DISABLE_CATS="$2";                  shift 2 ;;
    --disable-engines)      DISABLE_ENGINES_EXTRA="$2";         shift 2 ;;
    --enable-engines)       ENABLE_ENGINES_EXTRA="$2";          shift 2 ;;
    --base)                 BASE="$2";                          shift 2 ;;
    -h|--help)
      sed -n '2,20p' "$0" | sed 's/^# \?//'
      exit 0
      ;;
    *) die "Unknown argument: $1  (run with --help)" ;;
  esac
done

SETTINGS="$BASE/searxng/settings.yml"
COMPOSE="$BASE/docker-compose.yml"

# Ensure settings directory exists
mkdir -p "$(dirname "$SETTINGS")"

# ── Level → integer ───────────────────────────────────────────────────────────
case "$LEVEL" in
  none)     SAFE_INT=0 ;;
  moderate) SAFE_INT=1 ;;
  strict)   SAFE_INT=2 ;;
esac

info "Safe search: $LEVEL (${SAFE_INT})"

# ── Engines with no safe-search support ───────────────────────────────────────
# Auto-disabled when level is moderate or strict.
NO_SAFESEARCH_ENGINES=(
  # Torrent / P2P — no filtering possible
  "1337x" "piratebay" "nyaa" "torrentz" "kickass torrents"
  # Web engines without safe-search API
  "mojeek" "naver" "baidu"
  # Yandex: parameter exists but not reliably enforced for non-Russian queries
  "yandex"
  # Video frontends — no safe-search passthrough
  "invidious" "piped" "peertube" "sepiasearch"
)

# ── Category → engine lists ───────────────────────────────────────────────────
VIDEOS_ENGINES=(
  "youtube" "invidious" "piped" "peertube" "sepiasearch"
  "dailymotion" "vimeo"
  "bing videos" "duckduckgo videos" "google videos"
)
IMAGES_ENGINES=(
  "google images" "bing images" "duckduckgo images"
  "brave images" "qwant images"
  "flickr" "unsplash" "imgur" "deviantart" "openverse"
)
NEWS_ENGINES=(
  "google news" "bing news" "duckduckgo news" "brave news" "qwant news"
)
SCIENCE_ENGINES=(
  "arxiv" "semantic scholar" "pubmed" "crossref" "base"
)
SOCIAL_ENGINES=(
  "reddit" "lemmy" "mastodon"
)

# ── Build disable/enable maps ─────────────────────────────────────────────────
declare -A DISABLE_MAP   # engine → 1
declare -A ENABLE_MAP    # engine → 1  (overrides everything)

# Parse --enable-engines
if [[ -n "$ENABLE_ENGINES_EXTRA" ]]; then
  IFS=',' read -ra _engs <<< "$ENABLE_ENGINES_EXTRA"
  for e in "${_engs[@]}"; do
    e="${e#"${e%%[![:space:]]*}"}"; e="${e%"${e##*[![:space:]]}"}"   # trim
    [[ -n "$e" ]] && ENABLE_MAP["$e"]=1
  done
fi

# Helper: add to DISABLE_MAP unless explicitly re-enabled
mark_disabled() {
  local eng="$1"
  [[ -n "${ENABLE_MAP[$eng]+x}" ]] && return   # user said keep it
  DISABLE_MAP["$eng"]=1
}

# Auto-disable no-safesearch engines for moderate/strict
if [[ "$LEVEL" != "none" ]]; then
  for eng in "${NO_SAFESEARCH_ENGINES[@]}"; do
    mark_disabled "$eng"
  done
fi

# Category disables
if [[ -n "$DISABLE_CATS" ]]; then
  IFS=',' read -ra _cats <<< "$DISABLE_CATS"
  for cat in "${_cats[@]}"; do
    cat="${cat#"${cat%%[![:space:]]*}"}"; cat="${cat%"${cat##*[![:space:]]}"}"
    cat="${cat,,}"
    case "$cat" in
      videos)  for e in "${VIDEOS_ENGINES[@]}";  do mark_disabled "$e"; done ;;
      images)  for e in "${IMAGES_ENGINES[@]}";  do mark_disabled "$e"; done ;;
      news)    for e in "${NEWS_ENGINES[@]}";     do mark_disabled "$e"; done ;;
      science) for e in "${SCIENCE_ENGINES[@]}";  do mark_disabled "$e"; done ;;
      social)  for e in "${SOCIAL_ENGINES[@]}";   do mark_disabled "$e"; done ;;
      "")      ;;
      *)       warn "Unknown category '$cat' — valid: videos images news science social" ;;
    esac
  done
fi

# Extra engine disables
if [[ -n "$DISABLE_ENGINES_EXTRA" ]]; then
  IFS=',' read -ra _engs <<< "$DISABLE_ENGINES_EXTRA"
  for e in "${_engs[@]}"; do
    e="${e#"${e%%[![:space:]]*}"}"; e="${e%"${e##*[![:space:]]}"}"
    [[ -n "$e" ]] && mark_disabled "$e"
  done
fi

# ── Preserve existing secret key ─────────────────────────────────────────────
SECRET_KEY=$(grep -oP '(?<=secret_key: ")[^"]+' "$SETTINGS" 2>/dev/null || true)
[[ -z "$SECRET_KEY" ]] && SECRET_KEY=$(openssl rand -hex 32)

# ── Build engine override block ───────────────────────────────────────────────
ENGINE_BLOCK=""
for eng in "${!DISABLE_MAP[@]}"; do
  ENGINE_BLOCK+="  - name: ${eng}\n    disabled: true\n"
done
for eng in "${!ENABLE_MAP[@]}"; do
  ENGINE_BLOCK+="  - name: ${eng}\n    disabled: false\n"
done

# ── Write settings.yml ────────────────────────────────────────────────────────
{
  printf 'use_default_settings: true\n'
  printf 'general:\n  instance_name: "Local Search"\n'
  printf 'server:\n  secret_key: "%s"\n  limiter: false\n' "$SECRET_KEY"
  printf 'search:\n  safe_search: %d\n  default_lang: "en"\n  formats: [html, json]\n' "$SAFE_INT"
  if [[ -n "$ENGINE_BLOCK" ]]; then
    printf 'engines:\n'
    printf '%b' "$ENGINE_BLOCK"
  fi
} > "$SETTINGS"

ok "Updated settings.yml  (safe_search: $SAFE_INT)"
if [[ ${#DISABLE_MAP[@]} -gt 0 ]]; then
  info "Disabled (${#DISABLE_MAP[@]}): $(printf '%s, ' "${!DISABLE_MAP[@]}" | sed 's/, $//')"
fi
if [[ ${#ENABLE_MAP[@]} -gt 0 ]]; then
  info "Re-enabled: $(printf '%s, ' "${!ENABLE_MAP[@]}" | sed 's/, $//')"
fi

# ── Update &safesearch= in SEARXNG_QUERY_URL inside docker-compose.yml ────────
if [[ -f "$COMPOSE" ]]; then
  sed -i -E \
    "s|(SEARXNG_QUERY_URL=http://searxng:[0-9]+/search\?[^&[:space:]]*)(&safesearch=[0-9])?|\1\&safesearch=${SAFE_INT}|g" \
    "$COMPOSE"
  ok "Updated SEARXNG_QUERY_URL  (&safesearch=${SAFE_INT})"
fi

# ── Restart SearXNG ───────────────────────────────────────────────────────────
if docker ps --format '{{.Names}}' 2>/dev/null | grep -q '^searxng$'; then
  info "Restarting SearXNG..."
  docker restart searxng
  ok "SearXNG restarted"
else
  info "SearXNG not running — changes take effect on next start"
fi

echo
grn "Done — safe search: $LEVEL"
[[ "$LEVEL" != "none" ]] && \
  info "Engines skipped (can't enforce '$LEVEL'): ${#DISABLE_MAP[@]} total"