index_codebase() {
  local path="${1:-.}"

  echo "🏗️  Indexing codebase at: $path"

  # Find all code files
  files=$(find "$path" -type f \
    \( -name "*.ts" -o -name "*.tsx" -o -name "*.js" -o -name "*.jsx" \
      -o -name "*.py" -o -name "*.go" -o -name "*.rs" -o -name "*.java" \) \
    | grep -v node_modules | grep -v ".next" | grep -v "dist/")

  total=$(echo "$files" | wc -l)
  echo "📊 Found $total files to index"

  # Create index directory
  mkdir -p .claude/siftcoder-state/vector-index

  # Process files in batches
  batch_size=50
  batch=()

  echo "$files" | while read file; do
    batch+=("$file")

    if [ ${#batch[@]} -eq $batch_size ]; then
      index_batch "${batch[@]}"
      batch=()
    fi
  done

  # Process remaining files
  if [ ${#batch[@]} -gt 0 ]; then
    index_batch "${batch[@]}"
  fi

  # Save metadata
  cat > .claude/siftcoder-state/vector-index/metadata.json <<EOF
{
  "created_at": "$(date -u +"%Y-%m-%dT%H:%M:%SZ")",
  "files_indexed": $total,
  "path": "$path",
  "embedding_model": "nomic-embed-text"
}
EOF

  echo "✅ Index complete"
}

search_vectors() {
  local query="$1"
  local limit="${2:-10}"

  # Generate query embedding
  query_emb=$(ollama embed nomic-embed-text "$query" | jq '.embedding')

  # Search LanceDB
  results=$(python3 <<EOF
import lancedb
import json

db = lancedb.connect(".claude/siftcoder-state/vector-index")
table = db.open("codebase")

results = table.search($query_emb).limit($limit).to_df()

for _, row in results.iterrows():
    print(f"{row['file']}:{row['line']}")
    print(f"  Score: {row['_score']:.2f}")
    print(f"  Code: {row['code'][:100]}...")
    print()
EOF
)

  echo "$results"
}

update_index() {
  # Check for modified files
  last_index=$(jq '.created_at' .claude/siftcoder-state/vector-index/metadata.json)

  changed_files=$(find . -type f \
    \( -name "*.ts" -o -name "*.js" -o -name "*.py" \) \
    -newermt "$last_index" \
    | grep -v node_modules)

  if [ -n "$changed_files" ]; then
    echo "📝 Updating index for $(echo "$changed_files" | wc -l) files"

    echo "$changed_files" | while read file; do
      # Remove old entries
      remove_from_index "$file"

      # Add new entries
      index_file "$file"
    done

    # Update timestamp
    jq '.created_at = "'$(date -u +"%Y-%m-%dT%H:%M:%SZ")'"' \
      .claude/siftcoder-state/vector-index/metadata.json > .tmp
    mv .tmp .claude/siftcoder-state/vector-index/metadata.json
  fi
}

expand_query() {
  local query="$1"

  # Extract key concepts
  concepts=$(echo "$query" | grep -oE '\w{3,}' | tr '\n' '|')

  # Find related terms
  synonyms=$(get_synonyms "$query")

  # Build expanded query
  expanded="$query"

  for synonym in $synonyms; do
    expanded="$expanded OR $synonym"
  done

  echo "$expanded"
}

# Initialize index
/siftcoder:search "build index"

# Search for code
/siftcoder:search "user authentication logic"

# Update index
/siftcoder:search "update index"