from sqlalchemy import create_engine, Column, String, Text, Integer, Boolean, DateTime
from sqlalchemy.orm import declarative_base, sessionmaker, Session
from datetime import datetime
import json
import os

DATA_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "data")
os.makedirs(DATA_DIR, exist_ok=True)
DATABASE_URL = f"sqlite:///{DATA_DIR}/wochenplaner.db"

engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
Base = declarative_base()

class User(Base):
    __tablename__ = "users"
    id = Column(String, primary_key=True)
    username = Column(String, unique=True, index=True, nullable=False)
    password_hash = Column(String, nullable=False)
    role = Column(String, default="user")
    created_at = Column(DateTime, default=datetime.utcnow)

class AppData(Base):
    __tablename__ = "app_data"
    id = Column(Integer, primary_key=True, autoincrement=True)
    key = Column(String, unique=True, default="main")
    resources = Column(Text, default="[]")
    tasks = Column(Text, default="{}")
    calendars = Column(Text, default="[]")
    general_config = Column(Text, default="{}")
    views = Column(Text, default="[]")
    print_options = Column(Text, default="{}")
    global_row_height = Column(Integer, default=36)
    base_font_size = Column(Integer, default=14)
    header_options = Column(Text, default="{}")
    hidden_resource_ids = Column(Text, default="[]")
    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
    updated_by = Column(String, nullable=True)

class LoginAttempt(Base):
    __tablename__ = "login_attempts"
    id = Column(Integer, primary_key=True, autoincrement=True)
    username = Column(String, index=True)
    ip_address = Column(String, nullable=True)
    attempted_at = Column(DateTime, default=datetime.utcnow)
    success = Column(Boolean, default=False)

def init_db():
    Base.metadata.create_all(bind=engine)