import firebase_admin
from firebase_admin import credentials, firestore
import json
import os

# Manual init for script
FIREBASE_CREDENTIALS_PATH = "serviceAccountKey.json"
STORAGE_BUCKET = "bussymath.firebasestorage.app"

if not firebase_admin._apps:
    with open(FIREBASE_CREDENTIALS_PATH, "r", encoding="utf-8") as f:
        cred_dict = json.load(f)
    cred = credentials.Certificate(cred_dict)
    firebase_admin.initialize_app(cred, {
        'storageBucket': STORAGE_BUCKET
    })

db = firestore.client()

def create_mock_user():
    uid = "dev-bypass-user"
    user_ref = db.collection('users').document(uid)
    
    today = "2026-03-12" # Static for bypass
    month = "2026-03"
    
    user_data = {
        'uid': uid,
        'name': 'Dev Bypass User',
        'status': 'approved',
        'tier': 'student_premium',
        'used_today': 0,
        'used_tokens_this_month': 0,
        'monthly_token_budget': 9999999,
        'last_usage_month': month,
        'quota': {
            'limit': 9999,
            'used_today': 0,
            'last_reset_date': today,
        },
        'isAdmin': True,
        'role': 'admin'
    }
    
    user_ref.set(user_data, merge=True)
    print(f"✅ Created mock user: {uid}")

if __name__ == "__main__":
    create_mock_user()