Create streak_routes.py

api/streak_routes.py

@@ -0,0 +1,186 @@
+from fastapi import APIRouter, HTTPException, Depends
+from datetime import datetime, timedelta
+from app import verify_token, db_service
+
+router = APIRouter()
+
+@router.post("/check-in")
+async def check_in(
+    user_id: str = Depends(verify_token)
+):
+    """
+    User check-in for streak
+    """
+    try:
+        today = datetime.utcnow().date()
+        
+        # Get current streak
+        streak = db_service.get_streak(user_id)
+        
+        if not streak:
+            # First check-in
+            streak_data = {
+                "current_streak": 1,
+                "longest_streak": 1,
+                "total_days": 1,
+                "last_check_in": today.isoformat(),
+                "badges": []
+            }
+            db_service.update_streak(user_id, streak_data)
+            return {
+                "streak": 1,
+                "rewards": [],
+                "message": "First check-in! Welcome to your streak journey!"
+            }
+        
+        last_check = datetime.fromisoformat(streak["last_check_in"]).date()
+        
+        # Check if already checked in today
+        if last_check == today:
+            return {
+                "streak": streak["current_streak"],
+                "rewards": [],
+                "message": "Already checked in today!"
+            }
+        
+        # Check if streak is consecutive
+        if (today - last_check).days == 1:
+            # Consecutive day
+            new_streak = streak["current_streak"] + 1
+        else:
+            # Streak broken
+            new_streak = 1
+        
+        # Update streak
+        streak_data = {
+            "current_streak": new_streak,
+            "longest_streak": max(new_streak, streak["longest_streak"]),
+            "total_days": streak["total_days"] + 1,
+            "last_check_in": today.isoformat(),
+            "badges": streak.get("badges", [])
+        }
+        
+        # Check for rewards
+        rewards = []
+        if new_streak == 7:
+            rewards.append({
+                "type": "badge",
+                "name": "7-Day Dedication",
+                "icon": "🌟"
+            })
+            streak_data["badges"].append({
+                "id": "7-day",
+                "earned_at": datetime.utcnow().isoformat()
+            })
+            
+        elif new_streak == 30:
+            rewards.append({
+                "type": "discount",
+                "value": "15%",
+                "code": f"STREAK30_{user_id[:6]}",
+                "message": "15% off your next purchase!"
+            })
+            
+        elif new_streak == 50:
+            rewards.append({
+                "type": "raffle",
+                "chance": "5%",
+                "message": "You've been entered into our monthly raffle!"
+            })
+            
+        elif new_streak == 100:
+            rewards.append({
+                "type": "badge",
+                "name": "Skincare Master",
+                "icon": "👑"
+            })
+            streak_data["badges"].append({
+                "id": "100-day",
+                "earned_at": datetime.utcnow().isoformat()
+            })
+        
+        db_service.update_streak(user_id, streak_data)
+        
+        messages = {
+            1: "Great start! Day 1 of your streak!",
+            7: "You've earned your first badge! Keep it up!",
+            30: "One month streak! You're on fire!",
+            50: "50 days! You're a skincare pro!",
+            100: "100 DAYS! You're a legend! 🏆"
+        }
+        
+        message = messages.get(new_streak, f"{new_streak} day streak!")
+        
+        return {
+            "streak": new_streak,
+            "rewards": rewards,
+            "message": message
+        }
+        
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@router.get("/current")
+async def get_streak(
+    user_id: str = Depends(verify_token)
+):
+    """
+    Get user's current streak
+    """
+    try:
+        streak = db_service.get_streak(user_id)
+        if not streak:
+            return {
+                "current_streak": 0,
+                "longest_streak": 0,
+                "total_days": 0,
+                "badges": []
+            }
+        return streak
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@router.get("/leaderboard")
+async def get_leaderboard(
+    timeframe: str = "weekly",
+    user_id: str = Depends(verify_token)
+):
+    """
+    Get leaderboard rankings
+    """
+    try:
+        leaderboard = db_service.get_leaderboard(timeframe)
+        return {"leaderboard": leaderboard}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@router.get("/badges")
+async def get_badges(
+    user_id: str = Depends(verify_token)
+):
+    """
+    Get user's earned badges
+    """
+    try:
+        badges = db_service.get_user_badges(user_id)
+        return {"badges": badges}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@router.post("/share")
+async def share_streak(
+    user_id: str = Depends(verify_token)
+):
+    """
+    Share streak achievement
+    """
+    try:
+        streak = db_service.get_streak(user_id)
+        share_data = {
+            "streak": streak["current_streak"],
+            "message": f"I've maintained a {streak['current_streak']} day skincare streak on Glow AI! 🔥",
+            "share_url": f"https://glowai.app/share/streak/{user_id}"
+        }
+        return share_data
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))