Spaces:
Paused
Paused
File size: 2,912 Bytes
3e6b063 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 |
import os
import time
from flask import Blueprint, render_template, request, jsonify, redirect, url_for, session, current_app
from flask_login import LoginManager, login_user, logout_user, login_required, current_user
import requests
from models.user import User
auth_bp = Blueprint('auth', __name__)
# Google OAuth configuration (these would normally come from environment variables)
GOOGLE_CLIENT_ID = os.environ.get('GOOGLE_CLIENT_ID', 'your-google-client-id')
GOOGLE_CLIENT_SECRET = os.environ.get('GOOGLE_CLIENT_SECRET', 'your-google-client-secret')
GOOGLE_REDIRECT_URI = os.environ.get('GOOGLE_REDIRECT_URI', 'http://localhost:5000/auth/google/callback')
# Initialize login manager
login_manager = LoginManager()
def init_app(app):
"""Initialize the login manager with the app"""
login_manager.init_app(app)
# Set login view - using setattr to avoid type checking issues
setattr(login_manager, 'login_view', 'auth.login')
return login_manager
@login_manager.user_loader
def load_user(user_id):
return User.get(user_id)
@auth_bp.route("/login")
def login():
"""Display the login page"""
return render_template('login.html')
@auth_bp.route("/google/login")
def google_login():
"""Redirect to Google OAuth login"""
# In a real implementation, you would redirect to Google's OAuth endpoint
# For now, we'll create a dummy user for testing
user = User.create_or_update(
id="google_user_123",
email="user@gmail.com",
name="Google User",
picture=None
)
if user:
login_user(user)
return redirect(url_for('index'))
else:
return "Failed to create user", 500
@auth_bp.route("/guest/login", methods=['POST'])
def guest_login():
"""Login as a guest user"""
name = request.form.get('name')
if not name:
return "Name is required", 400
# Create a guest user
user = User.create_or_update(
id=f"guest_{name.lower().replace(' ', '_')}_{int(time.time())}",
email=f"{name.lower().replace(' ', '.')}@guest.texlab.com",
name=name,
picture=None
)
if user:
login_user(user)
return redirect(url_for('index'))
else:
return "Failed to create user", 500
@auth_bp.route("/logout")
@login_required
def logout():
logout_user()
return redirect(url_for('auth.login'))
@auth_bp.route("/user")
def get_current_user():
if current_user.is_authenticated:
return jsonify({
'authenticated': True,
'user': {
'id': current_user.id,
'email': current_user.email,
'name': current_user.name,
'picture': current_user.picture
}
})
else:
return jsonify({'authenticated': False}) |