Nia Public Search

#!/usr/bin/env bash
# Shared library for Nia Public skill — no API key required
# Source this file: source "$(dirname "$0")/lib.sh"

BASE_URL="${NIA_PUBLIC_URL:-https://apigcp.trynia.ai}/public"

urlencode() {
  jq -nr --arg v "$1" '$v|@uri'
}

# Generic curl wrapper: nia_curl METHOD URL [DATA]
nia_curl() {
  local method="$1" url="$2" data="${3:-}"
  local args=(-s -w '\n__HTTP_STATUS:%{http_code}' -X "$method" "$url")
  if [ -n "$data" ]; then
    args+=(-H "Content-Type: application/json" -d "$data")
  fi
  local response
  response=$(curl "${args[@]}" 2>/dev/null) || true
  local http_status="${response##*__HTTP_STATUS:}"
  local body="${response%__HTTP_STATUS:*}"
  http_status="${http_status//[!0-9]/}"
  : "${http_status:=0}"

  if [ "$http_status" = "429" ]; then
    echo ""
    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
    echo "  Rate limit reached (20 queries/hour free tier)"
    echo ""
    echo "  Get unlimited access:"
    echo "  1. Sign up at https://trynia.ai"
    echo "  2. Get your API key"
    echo "  3. Install the full Nia skill: npx nia-wizard@latest"
    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
    echo ""
    exit 1
  fi

  if echo "$body" | jq '.' >/dev/null 2>&1; then
    echo "$body"
  else
    jq -n --arg msg "$body" --argjson code "$http_status" '{error: $msg, http_status: $code}' 2>/dev/null \
      || printf '{"error":"request failed","http_status":%d}\n' "$http_status"
  fi
}

nia_get()  { nia_curl GET  "$1" | jq '.'; }
nia_post() { nia_curl POST "$1" "$2" | jq '.'; }

#!/usr/bin/env bash
# Nia Public Search — query any indexed public source, no API key needed
# Usage: search.sh <command> [args...]
set -e
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
source "$SCRIPT_DIR/lib.sh"

# ─── query — AI-powered search against a specific source or all public sources
cmd_query() {
  if [ -z "$1" ]; then
    echo "Usage: search.sh query <query> [source]"
    echo ""
    echo "  source: repo slug (vercel/ai), URL (https://docs.stripe.com), or canonical_id"
    echo "  Env: TOP_K (default 10)"
    echo ""
    echo "Examples:"
    echo "  search.sh query 'how to use hooks' vercel/next.js"
    echo "  search.sh query 'authentication' https://docs.supabase.com"
    echo "  search.sh query 'how to build an agent'   # searches all public sources"
    return 1
  fi
  local query="$1" source="${2:-}"
  DATA=$(jq -n \
    --arg q "$query" --arg src "$source" \
    --argjson k "${TOP_K:-10}" \
    '{query: $q, top_k: $k}
    + (if $src != "" then {source: $src} else {} end)')
  nia_post "$BASE_URL/search" "$DATA"
}

# ─── sources — list all searchable public sources
cmd_sources() {
  local type="${1:-}" limit="${LIMIT:-20}" offset="${OFFSET:-0}"
  local url="$BASE_URL/sources?limit=$limit&offset=$offset"
  if [ -n "$type" ]; then url="${url}&type=$type"; fi
  nia_get "$url"
}

# ─── dispatch ─────────────────────────────────────────────────────────────────
case "${1:-}" in
  query)   shift; cmd_query "$@" ;;
  sources) shift; cmd_sources "$@" ;;
  *)
    echo "Usage: $(basename "$0") <command> [args...]"
    echo ""
    echo "Commands:"
    echo "  query    Search any public source with AI (no API key needed)"
    echo "  sources  List all searchable public sources"
    echo ""
    echo "Examples:"
    echo "  search.sh query 'how to use hooks' vercel/next.js"
    echo "  search.sh query 'webhooks' https://docs.stripe.com"
    echo "  search.sh sources repository"
    exit 1
    ;;
esac