from flask_sqlalchemy import SQLAlchemy
from flask_login import UserMixin
from werkzeug.security import generate_password_hash, check_password_hash

db = SQLAlchemy()


class User(UserMixin, db.Model):
    __tablename__ = "users"

    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(80), unique=True, nullable=False, index=True)
    password_hash = db.Column(db.String(256), nullable=False)

    def set_password(self, password: str) -> None:
        self.password_hash = generate_password_hash(password)

    def check_password(self, password: str) -> bool:
        return check_password_hash(self.password_hash, password)


class ServiceGroup(db.Model):
    __tablename__ = "service_groups"

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(120), nullable=False)
    sort_order = db.Column(db.Integer, nullable=False, default=0)

    services = db.relationship(
        "Service",
        backref="group",
        lazy="dynamic",
        cascade="all, delete-orphan",
    )


class Service(db.Model):
    __tablename__ = "services"

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(120), nullable=False)
    host = db.Column(db.String(255), nullable=False)
    port = db.Column(db.Integer, nullable=False)
    path = db.Column(db.String(512), nullable=False, default="/")
    sort_order = db.Column(db.Integer, nullable=False, default=0)
    group_id = db.Column(
        db.Integer, db.ForeignKey("service_groups.id"), nullable=False, index=True
    )

    def build_url(self) -> str:
        p = (self.path or "/").strip()
        if not p.startswith("/"):
            p = "/" + p
        return f"http://{self.host}:{self.port}{p}"