mirror of
http://88.130.71.182:3000/BlitTech/contexta_be.git
synced 2026-06-13 10:14:58 +00:00
134 lines
4.0 KiB
Python
134 lines
4.0 KiB
Python
from fastapi import APIRouter, HTTPException, status, Depends
|
|
from app.models import UserSignup, UserLogin, UserResponse, TokenResponse
|
|
from app.database import get_supabase
|
|
from app.dependencies import get_current_user
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
router = APIRouter(prefix="/auth", tags=["Authentication"])
|
|
|
|
|
|
@router.post("/signup", response_model=TokenResponse)
|
|
async def signup(data: UserSignup):
|
|
supabase = get_supabase()
|
|
try:
|
|
# Create auth user
|
|
auth_resp = supabase.auth.sign_up(
|
|
{"email": data.email, "password": data.password}
|
|
)
|
|
if not auth_resp.user:
|
|
raise HTTPException(status_code=400, detail="Failed to create account")
|
|
|
|
user = auth_resp.user
|
|
|
|
# Create company record
|
|
supabase.table("companies").insert(
|
|
{
|
|
"owner_id": user.id,
|
|
"name": data.company_name,
|
|
}
|
|
).execute()
|
|
|
|
# Create free subscription
|
|
supabase.table("subscriptions").insert(
|
|
{
|
|
"user_id": user.id,
|
|
"plan": "free",
|
|
"status": "active",
|
|
}
|
|
).execute()
|
|
|
|
token = auth_resp.session.access_token if auth_resp.session else ""
|
|
return TokenResponse(
|
|
access_token=token,
|
|
user=UserResponse(
|
|
id=user.id,
|
|
email=user.email,
|
|
company_name=data.company_name,
|
|
plan="free",
|
|
),
|
|
)
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
logger.error(f"Signup error: {e}")
|
|
if "already registered" in str(e).lower() or "already exists" in str(e).lower():
|
|
raise HTTPException(status_code=400, detail="Email already registered")
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
@router.post("/login", response_model=TokenResponse)
|
|
async def login(data: UserLogin):
|
|
supabase = get_supabase()
|
|
try:
|
|
auth_resp = supabase.auth.sign_in_with_password(
|
|
{"email": data.email, "password": data.password}
|
|
)
|
|
if not auth_resp.user or not auth_resp.session:
|
|
raise HTTPException(status_code=401, detail="Invalid credentials")
|
|
|
|
user = auth_resp.user
|
|
|
|
# Get company info
|
|
company = supabase.table("companies").select("name").eq("owner_id", user.id).execute()
|
|
company_name = company.data[0]["name"] if company.data else ""
|
|
|
|
# Get subscription
|
|
sub = (
|
|
supabase.table("subscriptions")
|
|
.select("plan")
|
|
.eq("user_id", user.id)
|
|
.eq("status", "active")
|
|
.execute()
|
|
)
|
|
plan = sub.data[0]["plan"] if sub.data else "free"
|
|
|
|
return TokenResponse(
|
|
access_token=auth_resp.session.access_token,
|
|
user=UserResponse(
|
|
id=user.id,
|
|
email=user.email,
|
|
company_name=company_name,
|
|
plan=plan,
|
|
),
|
|
)
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
logger.error(f"Login error: {e}")
|
|
raise HTTPException(status_code=401, detail="Invalid credentials")
|
|
|
|
|
|
@router.post("/logout")
|
|
async def logout(user=Depends(get_current_user)):
|
|
supabase = get_supabase()
|
|
try:
|
|
supabase.auth.sign_out()
|
|
except Exception:
|
|
pass
|
|
return {"message": "Logged out successfully"}
|
|
|
|
|
|
@router.get("/me", response_model=UserResponse)
|
|
async def get_me(user=Depends(get_current_user)):
|
|
supabase = get_supabase()
|
|
|
|
company = supabase.table("companies").select("name").eq("owner_id", user.id).execute()
|
|
company_name = company.data[0]["name"] if company.data else ""
|
|
|
|
sub = (
|
|
supabase.table("subscriptions")
|
|
.select("plan")
|
|
.eq("user_id", user.id)
|
|
.eq("status", "active")
|
|
.execute()
|
|
)
|
|
plan = sub.data[0]["plan"] if sub.data else "free"
|
|
|
|
return UserResponse(
|
|
id=user.id,
|
|
email=user.email,
|
|
company_name=company_name,
|
|
plan=plan,
|
|
)
|