models.py

from flask_sqlalchemy import SQLAlchemy
from datetime import datetime
import uuid
import bcrypt

db = SQLAlchemy()

class User(db.Model):
    """User model with OAuth support"""
    __tablename__ = 'users'
    
    id = db.Column(db.String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    email = db.Column(db.String(120), unique=True, nullable=False, index=True)
    name = db.Column(db.String(120), nullable=False)
    password_hash = db.Column(db.String(255), nullable=True)
    
    # OAuth
    oauth_provider = db.Column(db.String(50), nullable=True)  # google, apple, microsoft
    oauth_provider_user_id = db.Column(db.String(255), nullable=True, index=True)
    
    # Verification
    email_verified = db.Column(db.Boolean, default=False)
    email_verify_token = db.Column(db.String(255), nullable=True)
    email_verify_token_expires = db.Column(db.DateTime, nullable=True)
    
    # Role & permissions
    role = db.Column(db.String(20), default='user')  # user, admin, moderator
    
    # Metadata
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    last_login = db.Column(db.DateTime, nullable=True)
    active = db.Column(db.Boolean, default=True)
    
    # Relations
    sessions = db.relationship('Session', backref='user', lazy='dynamic', cascade='all, delete-orphan')
    runs = db.relationship('Run', backref='user', lazy='dynamic', cascade='all, delete-orphan')
    hardware_scans = db.relationship('HardwareScan', backref='user', lazy='dynamic', cascade='all, delete-orphan')
    audit_logs = db.relationship('AuditLog', backref='user', lazy='dynamic', cascade='all, delete-orphan')
    
    def set_password(self, password):
        self.password_hash = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')
    
    def check_password(self, password):
        if not self.password_hash:
            return False
        return bcrypt.checkpw(password.encode('utf-8'), self.password_hash.encode('utf-8'))
    
    def to_dict(self):
        return {
            'id': self.id,
            'email': self.email,
            'name': self.name,
            'role': self.role,
            'email_verified': self.email_verified,
            'created_at': self.created_at.isoformat(),
            'last_login': self.last_login.isoformat() if self.last_login else None,
            'active': self.active
        }

class Session(db.Model):
    """User session/connection tracking"""
    __tablename__ = 'sessions'
    
    id = db.Column(db.String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    user_id = db.Column(db.String(36), db.ForeignKey('users.id'), nullable=False, index=True)
    session_token = db.Column(db.String(255), unique=True, nullable=False, index=True)
    refresh_token = db.Column(db.String(255), nullable=True, unique=True)
    
    ip_address = db.Column(db.String(45), nullable=True)
    user_agent = db.Column(db.String(500), nullable=True)
    device_name = db.Column(db.String(120), nullable=True)
    
    persistent = db.Column(db.Boolean, default=False)  # "remember me"
    
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    expires_at = db.Column(db.DateTime, nullable=False)
    last_activity = db.Column(db.DateTime, default=datetime.utcnow)
    
    active = db.Column(db.Boolean, default=True)

class Connection(db.Model):
    """Live user connection/location tracking"""
    __tablename__ = 'connections'
    
    id = db.Column(db.String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    user_id = db.Column(db.String(36), db.ForeignKey('users.id'), nullable=False, index=True)
    session_id = db.Column(db.String(36), db.ForeignKey('sessions.id'), nullable=True)
    
    socket_id = db.Column(db.String(100), nullable=False, unique=True, index=True)
    
    # Location
    latitude = db.Column(db.Float, nullable=True)
    longitude = db.Column(db.Float, nullable=True)
    accuracy = db.Column(db.Float, nullable=True)
    
    # Timing
    connected_at = db.Column(db.DateTime, default=datetime.utcnow)
    last_seen = db.Column(db.DateTime, default=datetime.utcnow)
    
    # Status
    online = db.Column(db.Boolean, default=True)
    
    def to_dict(self):
        return {
            'id': self.id,
            'user_id': self.user_id,
            'socket_id': self.socket_id,
            'latitude': self.latitude,
            'longitude': self.longitude,
            'accuracy': self.accuracy,
            'last_seen': self.last_seen.isoformat() if self.last_seen else None,
            'online': self.online
        }

class HardwareScan(db.Model):
    """Hardware/device info snapshots"""
    __tablename__ = 'hardware_scans'
    
    id = db.Column(db.String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    user_id = db.Column(db.String(36), db.ForeignKey('users.id'), nullable=False, index=True)
    
    timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True)
    
    # Browser/Device info
    user_agent = db.Column(db.String(500), nullable=True)
    platform = db.Column(db.String(100), nullable=True)
    
    # Hardware (approximate)
    cpu_threads = db.Column(db.Integer, nullable=True)
    memory_gb = db.Column(db.Float, nullable=True)
    gpu_renderer = db.Column(db.String(255), nullable=True)
    
    # Screen
    screen_width = db.Column(db.Integer, nullable=True)
    screen_height = db.Column(db.Integer, nullable=True)
    
    # Network
    connection_type = db.Column(db.String(50), nullable=True)  # 4g, wifi, etc
    
    # Media devices
    cameras_count = db.Column(db.Integer, default=0)
    microphones_count = db.Column(db.Integer, default=0)
    
    # Battery (if available)
    battery_percent = db.Column(db.Float, nullable=True)
    
    # Extra JSON data
    extra_data = db.Column(db.JSON, nullable=True, default={})

class AuditLog(db.Model):
    """Admin action audit log"""
    __tablename__ = 'audit_logs'
    
    id = db.Column(db.String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    admin_id = db.Column(db.String(36), db.ForeignKey('users.id'), nullable=True, index=True)
    
    action = db.Column(db.String(100), nullable=False)  # force_logout, change_role, etc
    target_user_id = db.Column(db.String(36), nullable=True, index=True)
    
    timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True)
    ip_address = db.Column(db.String(45), nullable=True)
    extra_data = db.Column(db.JSON, nullable=True, default={})

class Run(db.Model):
    """Athlete runs (migrated from localStorage)"""
    __tablename__ = 'runs'
    
    id = db.Column(db.String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    user_id = db.Column(db.String(36), db.ForeignKey('users.id'), nullable=False, index=True)
    
    start_time_iso = db.Column(db.DateTime, nullable=False, index=True)
    end_time_iso = db.Column(db.DateTime, nullable=False)
    
    elapsed_ms = db.Column(db.Integer, nullable=False)
    distance_m = db.Column(db.Integer, nullable=False)
    avg_kmh = db.Column(db.Float, nullable=False)
    max_kmh = db.Column(db.Float, nullable=False)
    pace_min_km = db.Column(db.String(50), nullable=True)
    
    # Fuel
    fuel_liters_used = db.Column(db.Float, nullable=True)
    fuel_cost_cad = db.Column(db.Float, nullable=True)
    
    # Trace (GeoJSON-like)
    trace = db.Column(db.JSON, nullable=True, default=[])
    
    created_at = db.Column(db.DateTime, default=datetime.utcnow, index=True)