#!/bin/bash

set -e

MODE="${1:-check}"
PROJECT_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"

PYTHON_MODULES=(
    "base_platform"
    "data_center/backend"
    "saas_portal/backend"
    "training_center/backend"
    "industry_accelerator/tax/backend"
    "llm_service/servers"
    "shared/common_logging"
    "shared/common_metrics"
)

echo "Python Format/Lint - Mode: $MODE"
echo "================================"

for module in "${PYTHON_MODULES[@]}"; do
    module_path="$PROJECT_ROOT/$module"

    if [ ! -d "$module_path" ]; then
        echo "⚠️  Skipping $module (directory not found)"
        continue
    fi

    echo ""
    echo "Processing: $module"

    if [ "$MODE" = "check" ]; then
        black --check "$module_path" || true
        ruff check "$module_path" || true
    else
        black "$module_path"
        ruff check --fix "$module_path"
    fi
done

echo ""
echo "Python format/lint complete!"