#!/bin/bash
# Script to update version number across the project

set -e

if [ -z "$1" ]; then
    echo "Usage: $0 <backend_version> [frontend_version]"
    echo "Example: $0 0.1.31              # Sets both to 0.1.31"
    echo "Example: $0 0.1.31 0.2.0        # Sets different versions"
    exit 1
fi

BACKEND_VERSION="$1"
FRONTEND_VERSION="${2:-$1}"  # Use backend version if frontend not specified

if [ "$BACKEND_VERSION" = "$FRONTEND_VERSION" ]; then
    echo "Updating both backend and frontend to version $BACKEND_VERSION..."
else
    echo "Updating backend to $BACKEND_VERSION and frontend to $FRONTEND_VERSION..."
fi

# Update backend version
BACKEND_VERSION_FILE="backend/config/version.go"
if [ -f "$BACKEND_VERSION_FILE" ]; then
    sed -i "s/const Version = \"[^\"]*\"/const Version = \"$BACKEND_VERSION\"/" "$BACKEND_VERSION_FILE"
    echo "✓ Updated $BACKEND_VERSION_FILE to $BACKEND_VERSION"
else
    echo "⚠ Warning: $BACKEND_VERSION_FILE not found"
fi

# Update frontend version
FRONTEND_VERSION_FILE="frontend/src/version.js"
if [ -f "$FRONTEND_VERSION_FILE" ]; then
    sed -i "s/export const VERSION = '[^']*'/export const VERSION = '$FRONTEND_VERSION'/" "$FRONTEND_VERSION_FILE"
    echo "✓ Updated $FRONTEND_VERSION_FILE to $FRONTEND_VERSION"
else
    echo "⚠ Warning: $FRONTEND_VERSION_FILE not found"
fi

# Update frontend package.json version
FRONTEND_PACKAGE="frontend/package.json"
if [ -f "$FRONTEND_PACKAGE" ]; then
    sed -i "s/\"version\": \"[^\"]*\"/\"version\": \"$FRONTEND_VERSION\"/" "$FRONTEND_PACKAGE"
    echo "✓ Updated $FRONTEND_PACKAGE to $FRONTEND_VERSION"
else
    echo "⚠ Warning: $FRONTEND_PACKAGE not found"
fi

# Update VERSION.md files
BACKEND_VERSION_MD="backend/VERSION.md"
if [ -f "$BACKEND_VERSION_MD" ]; then
    sed -i "s/## Current Version: .*/## Current Version: $BACKEND_VERSION/" "$BACKEND_VERSION_MD"
    echo "✓ Updated $BACKEND_VERSION_MD to $BACKEND_VERSION"
fi

FRONTEND_VERSION_MD="frontend/VERSION.md"
if [ -f "$FRONTEND_VERSION_MD" ]; then
    sed -i "s/## Current Version: .*/## Current Version: $FRONTEND_VERSION/" "$FRONTEND_VERSION_MD"
    echo "✓ Updated $FRONTEND_VERSION_MD to $FRONTEND_VERSION"
fi

echo ""
echo "✅ Version update complete!"
echo "   Backend:  $BACKEND_VERSION"
echo "   Frontend: $FRONTEND_VERSION"
echo ""
echo "Next steps:"
echo "1. Review changes: git diff"
if [ "$BACKEND_VERSION" = "$FRONTEND_VERSION" ]; then
    echo "2. Commit changes: git commit -am 'Bump version to $BACKEND_VERSION'"
    echo "3. Tag release: git tag v$BACKEND_VERSION"
else
    echo "2. Commit changes: git commit -am 'Bump backend to $BACKEND_VERSION, frontend to $FRONTEND_VERSION'"
    echo "3. Tag releases: git tag backend-v$BACKEND_VERSION && git tag frontend-v$FRONTEND_VERSION"
fi
echo "4. Push: git push && git push --tags"