neon_sakura 0.1.4

data/bin/ai-review-local

@@ -0,0 +1,345 @@
+#!/usr/bin/env bash
+# Local AI code review using Ollama
+# Uses the same prompts and logic as GitHub Actions workflows
+#
+# Usage:
+#   bin/ai-review-local [OPTIONS] [BASE_BRANCH] [OLLAMA_URL]
+#   bin/ai-review-local --help | -h
+#
+# Options:
+#   -m, --model MODEL  Specify the model to use (default: codestral:latest)
+#   -i, --interactive  Interactively select model from available models
+#   -f, --full-review  Review all tracked files in current branch (no comparison)
+#   --help, -h         Show this help message
+#
+# Arguments:
+#   BASE_BRANCH - Branch to compare against (default: main or master, ignored with -f)
+#   OLLAMA_URL  - Ollama server URL (default: $OLLAMA_URL env var or http://localhost:11434)
+#
+# Examples:
+#   bin/ai-review-local                          # Compare current branch to main
+#   bin/ai-review-local develop                  # Compare to develop branch
+#   bin/ai-review-local main http://server:11434 # Use remote Ollama server
+#   bin/ai-review-local -i                       # Interactively select model
+#   bin/ai-review-local -m llama3:latest         # Use specific model
+#   bin/ai-review-local -f                       # Review entire current branch
+#   bin/ai-review-local --help                   # Show this help message
+
+# Load shared library
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+if [ -f "$SCRIPT_DIR/../lib/ai-script-base.sh" ]; then
+  # shellcheck source=../lib/ai-script-base.sh
+  source "$SCRIPT_DIR/../lib/ai-script-base.sh"
+else
+  echo "Error: Shared library not found at $SCRIPT_DIR/../lib/ai-script-base.sh"
+  exit 1
+fi
+
+# Show help
+show_help() {
+  cat << EOF
+Local AI Code Review - Offline code reviews using Ollama
+
+USAGE:
+  bin/ai-review-local [OPTIONS] [BASE_BRANCH] [OLLAMA_URL]
+  bin/ai-review-local --help | -h
+
+OPTIONS:
+  -m, --model MODEL     Specify the model to use (default: codestral:latest)
+  -i, --interactive     Interactively select model from available models
+  -f, --full-review     Review all tracked files in current branch (no comparison)
+  --help, -h            Show this help message
+
+ARGUMENTS:
+  BASE_BRANCH          Branch to compare against (default: auto-detect main/master)
+                       Ignored when using -f/--full-review
+  OLLAMA_URL           Ollama server URL (default: \$OLLAMA_URL env var or http://localhost:11434)
+
+EXAMPLES:
+  # Compare current branch to main
+  bin/ai-review-local
+
+  # Compare to develop branch
+  bin/ai-review-local develop
+
+  # Use remote Ollama server
+  bin/ai-review-local main http://server:11434
+
+  # Interactively select model
+  bin/ai-review-local -i
+
+  # Use specific model
+  bin/ai-review-local -m llama3:latest
+
+  # Review entire current branch (useful for main branch review)
+  bin/ai-review-local -f
+
+  # Combine options
+  bin/ai-review-local -m mistral:latest develop
+
+  # Show help
+  bin/ai-review-local --help
+
+REQUIREMENTS:
+  - Ollama installed and running (https://ollama.ai)
+  - At least one model pulled (e.g., ollama pull codestral:latest)
+  - Git repository with commits to review
+
+ENVIRONMENT VARIABLES:
+  OLLAMA_URL           Ollama server URL (overridden by command line argument)
+                       Example: export OLLAMA_URL=http://192.168.1.100:11434
+
+OUTPUT:
+  Reviews are saved to: docs/ai-reviews/ai-review-YYYYMMDD-HHMMSS.md
+
+MORE INFO:
+  See docs/AI_PR_REVIEWER.md for full documentation
+EOF
+  exit 0
+}
+
+# Parse options
+INTERACTIVE_MODE=false
+FULL_REVIEW=false
+MODEL_NAME="codestral:latest"
+BASE_BRANCH=""
+OLLAMA_URL=""
+
+while [[ $# -gt 0 ]]; do
+  case $1 in
+    --help|-h)
+      show_help
+      ;;
+    -i|--interactive)
+      INTERACTIVE_MODE=true
+      shift
+      ;;
+    -f|--full-review)
+      FULL_REVIEW=true
+      shift
+      ;;
+    -m|--model)
+      if [ -z "${2:-}" ]; then
+        log_error "Option -m/--model requires a model name"
+        exit 1
+      fi
+      MODEL_NAME="$2"
+      shift 2
+      ;;
+    *)
+      # Positional arguments
+      if [ -z "$BASE_BRANCH" ]; then
+        BASE_BRANCH="$1"
+      elif [ -z "$OLLAMA_URL" ]; then
+        OLLAMA_URL="$1"
+      else
+        log_error "Unknown argument: $1"
+        echo "Run 'bin/ai-review-local --help' for usage information"
+        exit 1
+      fi
+      shift
+      ;;
+  esac
+done
+
+# Set defaults for positional arguments
+OLLAMA_URL=$(get_ollama_url "$OLLAMA_URL")
+if [ "$FULL_REVIEW" = false ]; then
+  BASE_BRANCH=$(determine_base_branch "$BASE_BRANCH")
+fi
+
+# Create temp directory
+TEMP_DIR=$(create_temp_dir)
+
+log_info "Starting AI code review..."
+if [ "$FULL_REVIEW" = true ]; then
+  log_info "Mode: ${CYAN}Full review (all tracked files)${NC}"
+else
+  log_info "Base branch: ${CYAN}$BASE_BRANCH${NC}"
+fi
+log_info "Ollama server: ${CYAN}$OLLAMA_URL${NC}"
+echo ""
+
+# Check if Ollama is accessible
+check_ollama_server "$OLLAMA_URL" || exit 1
+echo ""
+
+# Interactive model selection if requested
+if [ "$INTERACTIVE_MODE" = true ]; then
+  MODEL_NAME=$(select_ollama_model "$OLLAMA_URL" "$MODEL_NAME")
+  echo ""
+  log_info "Selected model: ${CYAN}$MODEL_NAME${NC}"
+  echo ""
+fi
+
+# Ensure model is available
+ensure_ollama_model "$OLLAMA_URL" "$MODEL_NAME" || exit 1
+echo ""
+
+# Get current branch name
+CURRENT_BRANCH=$(get_current_branch)
+log_info "Current branch: ${CYAN}$CURRENT_BRANCH${NC}"
+
+# Get content based on review mode
+if [ "$FULL_REVIEW" = true ]; then
+  # Full review: get important files from current branch
+  log_info "Getting tracked files for full review..."
+
+  # Get list of key files (exclude tests, vendor, node_modules, etc.)
+  CHANGED_FILES=$(git ls-files | grep -E '\.(rb|yml|yaml|js|css|erb|rake)$' | \
+    grep -v -E '^(test/|spec/|vendor/|node_modules/|tmp/|log/)' | \
+    head -50)
+  CHANGED_FILES_COUNT=$(echo "$CHANGED_FILES" | wc -l | tr -d ' ')
+
+  log_success "Found $CHANGED_FILES_COUNT files to review"
+
+  # Build a summary diff from key files (first 100 lines of each)
+  log_info "Building file summaries..."
+  DIFF=""
+  for file in $CHANGED_FILES; do
+    if [ -f "$file" ]; then
+      DIFF+="
+=== $file ===
+$(head -100 "$file")
+"
+    fi
+  done
+
+  DIFF_LENGTH=${#DIFF}
+  DIFF=$(truncate_diff "$DIFF" 15000)
+  log_success "File summaries generated (${DIFF_LENGTH} chars)"
+
+  COMMIT_MESSAGES="Full review mode - reviewing current state of $CURRENT_BRANCH"
+  echo ""
+else
+  # Comparison mode: get diff
+  # Get the diff
+  DIFF=$(get_branch_diff "$BASE_BRANCH")
+
+  # Filter out ignored sections
+  DIFF=$(filter_ignored_sections "$DIFF")
+
+  # Truncate if too large
+  DIFF_LENGTH=${#DIFF}
+  DIFF=$(truncate_diff "$DIFF" 15000)
+
+  log_success "Diff generated (${DIFF_LENGTH} chars)"
+  echo ""
+
+  # Get changed files list
+  log_info "Getting changed files..."
+  CHANGED_FILES=$(get_changed_files "$BASE_BRANCH")
+  CHANGED_FILES_COUNT=$(echo "$CHANGED_FILES" | wc -l | tr -d ' ')
+  log_success "Found $CHANGED_FILES_COUNT changed files"
+  echo ""
+
+  # Get commit messages
+  log_info "Getting commit messages..."
+  COMMIT_MESSAGES=$(get_commit_messages "$BASE_BRANCH")
+  echo ""
+fi
+
+# Build system prompt using shared script
+log_info "Building system prompt..."
+SYSTEM_PROMPT=$(build_system_prompt)
+log_success "System prompt built"
+echo ""
+
+# Extract CLAUDE.md sections
+log_info "Extracting project guidelines..."
+CLAUDE_MD=$(extract_claude_sections)
+if [ -n "$CLAUDE_MD" ]; then
+  log_success "Guidelines extracted"
+else
+  log_warning "Guidelines not found, proceeding without them"
+fi
+echo ""
+
+# Build user prompt
+log_info "Building review prompt..."
+if [ "$FULL_REVIEW" = true ]; then
+  cat > "$TEMP_DIR/user.txt" <<USEREND
+**Branch:** $CURRENT_BRANCH (full review - all key files)
+
+**Review Mode:** Full codebase review
+
+**Files Reviewed:**
+$CHANGED_FILES
+
+**File Contents (first 100 lines each):**
+\`\`\`
+$DIFF
+\`\`\`
+
+**Focus Areas:**
+- Overall code quality and patterns
+- Security best practices
+- Rails conventions
+- Potential improvements
+USEREND
+else
+  cat > "$TEMP_DIR/user.txt" <<USEREND
+**Branch:** $CURRENT_BRANCH (comparing against $BASE_BRANCH)
+
+**Commits:**
+$COMMIT_MESSAGES
+
+**Changed Files:**
+$CHANGED_FILES
+
+**Diff:**
+\`\`\`diff
+$DIFF
+\`\`\`
+USEREND
+fi
+
+if [ -n "$CLAUDE_MD" ]; then
+  cat >> "$TEMP_DIR/user.txt" <<USEREND
+
+**Project Guidelines (CLAUDE.md):**
+$CLAUDE_MD
+USEREND
+fi
+
+USER_PROMPT=$(cat "$TEMP_DIR/user.txt")
+log_success "Review prompt built"
+echo ""
+
+# Call Ollama API
+log_info "Calling Ollama API for review..."
+echo -e "${CYAN}This may take a minute or two...${NC}"
+echo ""
+
+REVIEW=$(call_ollama_chat "$OLLAMA_URL" "$MODEL_NAME" "$SYSTEM_PROMPT" "$USER_PROMPT")
+
+# Smart deduplication
+REVIEW=$(deduplicate_review "$REVIEW")
+
+# Truncate if too long
+REVIEW=$(truncate_review "$REVIEW" 100)
+
+# Display the review
+if [ "$FULL_REVIEW" = true ]; then
+  ADDITIONAL_INFO="${CYAN}Branch:${NC} $CURRENT_BRANCH (full review)
+${CYAN}Files reviewed:${NC} $CHANGED_FILES_COUNT
+${CYAN}Content size:${NC} $DIFF_LENGTH chars"
+else
+  ADDITIONAL_INFO="${CYAN}Branch:${NC} $CURRENT_BRANCH (compared to $BASE_BRANCH)
+${CYAN}Changed files:${NC} $CHANGED_FILES_COUNT
+${CYAN}Diff size:${NC} $DIFF_LENGTH chars"
+fi
+
+display_review "$REVIEW" "AI Code Review - Local" "$MODEL_NAME" "$ADDITIONAL_INFO"
+
+# Save review to file
+REVIEW_FILE=$(save_review "$REVIEW" "docs/ai-reviews" "ai-review")
+log_success "Review saved to: ${CYAN}$REVIEW_FILE${NC}"
+echo ""
+
+# Check for critical issues
+if check_critical_issues "$REVIEW"; then
+  exit 0
+else
+  exit 1
+fi