#!/bin/bash
set -euo pipefail

echo "=== Fixing unwrap_used violations in services (Agent W17) ==="

# Count before
BEFORE=$(grep -r "\.unwrap()" services/api_gateway/src services/trading_service/src services/backtesting_service/src services/ml_training_service/src 2>/dev/null | wc -l)
echo "Found $BEFORE .unwrap() calls in services"

# Pattern 1: std::env::current_dir().unwrap()
# Fix: Replace with .expect()
echo "Fixing Pattern 1: current_dir().unwrap() → current_dir().expect()"
find services/ -name "*.rs" -type f -exec sed -i \
  's/std::env::current_dir()\.unwrap()/std::env::current_dir().expect("INVARIANT: Current directory should be accessible")/g' {} +

# Pattern 2: Duration operations
# Fix: duration_since().unwrap() → duration_since().expect()
echo "Fixing Pattern 2: duration_since().unwrap() → duration_since().expect()"
find services/ -name "*.rs" -type f -exec sed -i \
  's/\.duration_since(\([^)]*\))\.unwrap()/\.duration_since(\1).expect("INVARIANT: System clock should not go backwards")/g' {} +

# Pattern 3: serde_json operations in tests (be conservative)
echo "Fixing Pattern 3: serde_json operations → expect()"
find services/ -name "*.rs" -type f -exec sed -i \
  's/serde_json::to_string(\([^)]*\))\.unwrap()/serde_json::to_string(\1).expect("INVARIANT: Serialization should succeed for valid types")/g' {} +

find services/ -name "*.rs" -type f -exec sed -i \
  's/serde_json::from_str(\([^)]*\))\.unwrap()/serde_json::from_str(\1).expect("INVARIANT: Deserialization should succeed for valid JSON")/g' {} +

find services/ -name "*.rs" -type f -exec sed -i \
  's/serde_json::from_slice(\([^)]*\))\.unwrap()/serde_json::from_slice(\1).expect("INVARIANT: Deserialization should succeed for valid JSON")/g' {} +

# Pattern 4: Request::builder().unwrap() (common in tests)
echo "Fixing Pattern 4: Request builder operations → expect()"
find services/ -name "*.rs" -type f -exec sed -i \
  's/\.body(Body::empty())\.unwrap()/\.body(Body::empty()).expect("INVARIANT: Empty body should always be valid")/g' {} +

# Pattern 5: chrono date/time operations
echo "Fixing Pattern 5: Chrono operations → expect()"
find services/ -name "*.rs" -type f -exec sed -i \
  's/\.with_ymd_and_hms(\([^)]*\))\.unwrap()/\.with_ymd_and_hms(\1).expect("INVARIANT: Valid date\/time parameters")/g' {} +

find services/ -name "*.rs" -type f -exec sed -i \
  's/\.and_hms_opt(\([^)]*\))\.unwrap()/\.and_hms_opt(\1).expect("INVARIANT: Valid time parameters")/g' {} +

# Pattern 6: Duration::from_std().unwrap()
echo "Fixing Pattern 6: Duration::from_std().unwrap() → expect()"
find services/ -name "*.rs" -type f -exec sed -i \
  's/Duration::from_std(\([^)]*\))\.unwrap()/Duration::from_std(\1).expect("INVARIANT: Duration should fit in chrono::Duration")/g' {} +

# Pattern 7: .join().unwrap() for thread handles
echo "Fixing Pattern 7: handle.join().unwrap() → expect()"
find services/ -name "*.rs" -type f -exec sed -i \
  's/\.join()\.unwrap()/\.join().expect("INVARIANT: Thread should complete successfully")/g' {} +

# Count after
AFTER=$(grep -r "\.unwrap()" services/api_gateway/src services/trading_service/src services/backtesting_service/src services/ml_training_service/src 2>/dev/null | wc -l)
FIXED=$((BEFORE - AFTER))

echo ""
echo "Fixed $FIXED violations"
echo "Remaining: $AFTER .unwrap() calls"
echo ""
echo "✓ Fix complete"
echo "Review changes with: git diff"