deploy_cloud_run.sh

#!/bin/bash
# Deployment script for Cloud Run service
# Deploys a single Cloud Run service that handles both weekday and weekly predictions

set -e

# Get the directory where this script is located
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
cd "$SCRIPT_DIR"

# Load PROJECT_ID from .env file (without sourcing to avoid issues with special characters)
if [ -f .env ]; then
  PROJECT_ID=$(cat .env | grep -i "FIREBASE_PROJECT_ID" | cut -d'=' -f2 | tr -d ' ')
  if [ -z "$PROJECT_ID" ]; then
    echo "Error: FIREBASE_PROJECT_ID not found in .env file!"
    exit 1
  fi
else
  echo "Error: .env file not found!"
  exit 1
fi
REGION="us-central1"
SERVICE_NAME="trading-predictions"
IMAGE_NAME="gcr.io/${PROJECT_ID}/${SERVICE_NAME}"

echo "🚀 Deploying to Cloud Run..."
echo "Project: ${PROJECT_ID}"
echo "Region: ${REGION}"
echo "Service: ${SERVICE_NAME}"
echo "Working directory: $(pwd)"
echo ""

# Verify Dockerfile exists
if [ ! -f Dockerfile ]; then
  echo "❌ Error: Dockerfile not found in $(pwd)"
  exit 1
fi

# Build the Docker image
echo "📦 Building Docker image..."
gcloud builds submit . --tag ${IMAGE_NAME} --project=${PROJECT_ID}

echo "✅ Image built successfully"
echo ""

# Deploy to Cloud Run
echo "🚀 Deploying to Cloud Run..."
gcloud run deploy ${SERVICE_NAME} \
  --image=${IMAGE_NAME} \
  --platform=managed \
  --region=${REGION} \
  --allow-unauthenticated \
  --memory=2Gi \
  --cpu=2 \
  --timeout=3600 \
  --max-instances=1 \
  --min-instances=0 \
  --port=8080 \
  --project=${PROJECT_ID}

echo ""
echo "✅ Cloud Run service deployed successfully!"
echo ""

# Get the service URL
SERVICE_URL=$(gcloud run services describe ${SERVICE_NAME} --region=${REGION} --format="value(status.url)")
echo "Service URL: ${SERVICE_URL}"
echo ""

# Set up Cloud Scheduler jobs
echo "⏰ Setting up Cloud Scheduler jobs..."
echo ""

# 1. LSTM Strategy - Every 15 minutes on weekdays (8:30 AM - 3:00 PM CT)
# Schedule: 30,45 8 * * 1-5 and 0,15,30,45 9-14 * * 1-5 and 0 15 * * 1-5
# Since Cloud Scheduler only supports one cron per job, we use: 30,45 8 * * 1-5 and 0,15,30,45 9-15 * * 1-5
# This gives us 8:30, 8:45, then 9:00-15:45 every 15 min. We'll handle 15:00 separately or accept 15:45 as close enough.
# Actually, we can't combine two schedules easily. Let's use: 30,45 8-14 * * 1-5 and 0,15,30,45 9-15 * * 1-5
# But that's still two expressions. Simplest: Use 30,45 8 * * 1-5 and 0,15,30,45 9-15 * * 1-5 as two jobs, OR
# Use 0,15,30,45 8-15 * * 1-5 and filter in code to only process 8:30 onwards and stop at 3:00 PM
# For now, let's use 30,45 8 * * 1-5 and 0,15,30,45 9-15 * * 1-5 as two separate jobs

# Use existing job name to preserve configuration
SCHEDULER_JOB_LSTM="weekday-predictions-job"
SCHEDULER_JOB_LSTM_830="weekday-predictions-830-job"
SERVICE_ACCOUNT_EMAIL="${PROJECT_ID}@appspot.gserviceaccount.com"

echo "📅 Creating/updating LSTM scheduler jobs (every 15 min, Mon-Fri, 8:30 AM - 3:00 PM CT)..."

# Job 1: 8:30, 8:45 AM
if gcloud scheduler jobs describe ${SCHEDULER_JOB_LSTM_830} --location=${REGION} --project=${PROJECT_ID} &>/dev/null; then
  echo "   Updating LSTM 8:30 job..."
  gcloud scheduler jobs update http ${SCHEDULER_JOB_LSTM_830} \
    --location=${REGION} \
    --schedule="30,45 8 * * 1-5" \
    --uri="${SERVICE_URL}/weekday" \
    --http-method=POST \
    --oidc-service-account-email=${SERVICE_ACCOUNT_EMAIL} \
    --time-zone="America/Chicago" \
    --project=${PROJECT_ID} \
    --attempt-deadline=600s
else
  echo "   Creating LSTM 8:30 job..."
  gcloud scheduler jobs create http ${SCHEDULER_JOB_LSTM_830} \
    --location=${REGION} \
    --schedule="30,45 8 * * 1-5" \
    --uri="${SERVICE_URL}/weekday" \
    --http-method=POST \
    --oidc-service-account-email=${SERVICE_ACCOUNT_EMAIL} \
    --time-zone="America/Chicago" \
    --project=${PROJECT_ID} \
    --attempt-deadline=600s \
    --description="Weekday LSTM predictions - 8:30, 8:45 AM CT (Mon-Fri)"
fi

# Job 2: 9:00 AM - 2:45 PM (every 15 min for hours 9-14)
if gcloud scheduler jobs describe ${SCHEDULER_JOB_LSTM} --location=${REGION} --project=${PROJECT_ID} &>/dev/null; then
  echo "   Updating LSTM main job (9:00 AM - 2:45 PM)..."
  gcloud scheduler jobs update http ${SCHEDULER_JOB_LSTM} \
    --location=${REGION} \
    --schedule="0,15,30,45 9-14 * * 1-5" \
    --uri="${SERVICE_URL}/weekday" \
    --http-method=POST \
    --oidc-service-account-email=${SERVICE_ACCOUNT_EMAIL} \
    --time-zone="America/Chicago" \
    --project=${PROJECT_ID} \
    --attempt-deadline=600s
else
  echo "   Creating LSTM main job (9:00 AM - 2:45 PM)..."
  gcloud scheduler jobs create http ${SCHEDULER_JOB_LSTM} \
    --location=${REGION} \
    --schedule="0,15,30,45 9-14 * * 1-5" \
    --uri="${SERVICE_URL}/weekday" \
    --http-method=POST \
    --oidc-service-account-email=${SERVICE_ACCOUNT_EMAIL} \
    --time-zone="America/Chicago" \
    --project=${PROJECT_ID} \
    --attempt-deadline=600s \
    --description="Weekday LSTM predictions - every 15 minutes from 9:00 AM to 2:45 PM CT (Mon-Fri)"
fi

# Job 3: 3:00 PM exactly (15:00)
SCHEDULER_JOB_LSTM_3PM="weekday-predictions-3pm-job"
if gcloud scheduler jobs describe ${SCHEDULER_JOB_LSTM_3PM} --location=${REGION} --project=${PROJECT_ID} &>/dev/null; then
  echo "   Updating LSTM 3PM job (3:00 PM only)..."
  gcloud scheduler jobs update http ${SCHEDULER_JOB_LSTM_3PM} \
    --location=${REGION} \
    --schedule="0 15 * * 1-5" \
    --uri="${SERVICE_URL}/weekday" \
    --http-method=POST \
    --oidc-service-account-email=${SERVICE_ACCOUNT_EMAIL} \
    --time-zone="America/Chicago" \
    --project=${PROJECT_ID} \
    --attempt-deadline=600s
else
  echo "   Creating LSTM 3PM job (3:00 PM only)..."
  gcloud scheduler jobs create http ${SCHEDULER_JOB_LSTM_3PM} \
    --location=${REGION} \
    --schedule="0 15 * * 1-5" \
    --uri="${SERVICE_URL}/weekday" \
    --http-method=POST \
    --oidc-service-account-email=${SERVICE_ACCOUNT_EMAIL} \
    --time-zone="America/Chicago" \
    --project=${PROJECT_ID} \
    --attempt-deadline=600s \
    --description="Weekday LSTM predictions - 3:00 PM CT only (Mon-Fri)"
fi

echo "   ✅ LSTM scheduler job configured"
echo ""

# 2. MLP Strategy - Every 1 minute on weekdays (8:30 AM - 3:00 PM CT)
# Schedule: 30-59 8 * * 1-5 and * 9-14 * * 1-5 and 0-0 15 * * 1-5
# Since Cloud Scheduler only supports one cron per job, we need two jobs:
# - Job 1: 30-59 8 * * 1-5 (8:30-8:59 AM)
# - Job 2: * 9-15 * * 1-5 (9:00 AM - 3:59 PM, but we want to stop at 3:00 PM)
# For Job 2, we'll use * 9-14 * * 1-5 and 0 15 * * 1-5, but that's still two expressions
# Simplest: Use two jobs: 30-59 8 * * 1-5 and * 9-15 * * 1-5, then filter to stop at 15:00 in code
# Or: Use * 8-15 * * 1-5 and filter in code to only process 8:30 onwards and stop at 3:00 PM

SCHEDULER_JOB_MLP_830="mlp-minute-predictions-830"
SCHEDULER_JOB_MLP="mlp-minute-predictions"
echo "📅 Creating/updating MLP scheduler jobs (every 1 minute, Mon-Fri, 8:30 AM - 3:00 PM CT)..."

# Job 1: 8:30-8:59 AM
if gcloud scheduler jobs describe ${SCHEDULER_JOB_MLP_830} --location=${REGION} --project=${PROJECT_ID} &>/dev/null; then
  echo "   Updating MLP 8:30 job..."
  gcloud scheduler jobs update http ${SCHEDULER_JOB_MLP_830} \
    --location=${REGION} \
    --schedule="30-59 8 * * 1-5" \
    --uri="${SERVICE_URL}/mlp_minute" \
    --http-method=POST \
    --oidc-service-account-email=${SERVICE_ACCOUNT_EMAIL} \
    --time-zone="America/Chicago" \
    --project=${PROJECT_ID} \
    --attempt-deadline=60s
else
  echo "   Creating MLP 8:30 job..."
  gcloud scheduler jobs create http ${SCHEDULER_JOB_MLP_830} \
    --location=${REGION} \
    --schedule="30-59 8 * * 1-5" \
    --uri="${SERVICE_URL}/mlp_minute" \
    --http-method=POST \
    --oidc-service-account-email=${SERVICE_ACCOUNT_EMAIL} \
    --time-zone="America/Chicago" \
    --project=${PROJECT_ID} \
    --attempt-deadline=60s \
    --description="MLP trading predictions - every 1 minute from 8:30-8:59 AM CT (Mon-Fri)"
fi

# Job 2: 9:00 AM - 2:59 PM (all minutes for hours 9-14)
if gcloud scheduler jobs describe ${SCHEDULER_JOB_MLP} --location=${REGION} --project=${PROJECT_ID} &>/dev/null; then
  echo "   Updating MLP main job (9:00 AM - 2:59 PM)..."
  gcloud scheduler jobs update http ${SCHEDULER_JOB_MLP} \
    --location=${REGION} \
    --schedule="* 9-14 * * 1-5" \
    --uri="${SERVICE_URL}/mlp_minute" \
    --http-method=POST \
    --oidc-service-account-email=${SERVICE_ACCOUNT_EMAIL} \
    --time-zone="America/Chicago" \
    --project=${PROJECT_ID} \
    --attempt-deadline=60s
else
  echo "   Creating MLP main job (9:00 AM - 2:59 PM)..."
  gcloud scheduler jobs create http ${SCHEDULER_JOB_MLP} \
    --location=${REGION} \
    --schedule="* 9-14 * * 1-5" \
    --uri="${SERVICE_URL}/mlp_minute" \
    --http-method=POST \
    --oidc-service-account-email=${SERVICE_ACCOUNT_EMAIL} \
    --time-zone="America/Chicago" \
    --project=${PROJECT_ID} \
    --attempt-deadline=60s \
    --description="MLP trading predictions - every 1 minute from 9:00 AM to 2:59 PM CT (Mon-Fri)"
fi

# Job 3: 3:00 PM exactly (15:00)
SCHEDULER_JOB_MLP_3PM="mlp-minute-predictions-3pm"
if gcloud scheduler jobs describe ${SCHEDULER_JOB_MLP_3PM} --location=${REGION} --project=${PROJECT_ID} &>/dev/null; then
  echo "   Updating MLP 3PM job (3:00 PM only)..."
  gcloud scheduler jobs update http ${SCHEDULER_JOB_MLP_3PM} \
    --location=${REGION} \
    --schedule="0 15 * * 1-5" \
    --uri="${SERVICE_URL}/mlp_minute" \
    --http-method=POST \
    --oidc-service-account-email=${SERVICE_ACCOUNT_EMAIL} \
    --time-zone="America/Chicago" \
    --project=${PROJECT_ID} \
    --attempt-deadline=60s
else
  echo "   Creating MLP 3PM job (3:00 PM only)..."
  gcloud scheduler jobs create http ${SCHEDULER_JOB_MLP_3PM} \
    --location=${REGION} \
    --schedule="0 15 * * 1-5" \
    --uri="${SERVICE_URL}/mlp_minute" \
    --http-method=POST \
    --oidc-service-account-email=${SERVICE_ACCOUNT_EMAIL} \
    --time-zone="America/Chicago" \
    --project=${PROJECT_ID} \
    --attempt-deadline=60s \
    --description="MLP trading predictions - 3:00 PM CT only (Mon-Fri)"
fi

echo "   ✅ MLP scheduler job configured"
echo ""

echo "✅ Deployment complete!"
echo ""
echo "Scheduler Jobs:"
echo "  - LSTM 8:30 (8:30, 8:45 AM): ${SCHEDULER_JOB_LSTM_830}"
echo "  - LSTM main (15 min, 9:00 AM - 2:45 PM CT): ${SCHEDULER_JOB_LSTM}"
echo "  - LSTM 3PM (3:00 PM only): ${SCHEDULER_JOB_LSTM_3PM}"
echo "  - MLP 8:30 (8:30-8:59 AM): ${SCHEDULER_JOB_MLP_830}"
echo "  - MLP main (1 min, 9:00 AM - 2:59 PM CT): ${SCHEDULER_JOB_MLP}"
echo "  - MLP 3PM (3:00 PM only): ${SCHEDULER_JOB_MLP_3PM}"
echo ""
echo "To view scheduler jobs:"
echo "  gcloud scheduler jobs list --location=${REGION} --project=${PROJECT_ID}"