from sqlalchemy import Column, Integer, String, ForeignKey, Table
from sqlalchemy.orm import relationship, backref
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from utils.config.config import ServerConfig

config = ServerConfig()
engine = create_engine(f"sqlite:///{config.database_file}")
Base = declarative_base(bind=engine)
Base.metadata.create_all()

client_image_table = Table(
    "client_image",
    Base.metadata,
    Column("client_mac", String, ForeignKey("clients.mac_address")),
    Column("image_id", Integer, ForeignKey("vm_images.image_id"))
)


class Client(Base):
    __tablename__ = "clients"
    mac_address = Column(String, primary_key=True)
    ip_address = Column(String(16), nullable=False)
    hostname = Column(String(100), nullable=False)
    client_version = Column(String(100), nullable=False)
    vm_list_on_machine = relationship(
        "VMImage",
        secondary=client_image_table,
    )

    def has_vm_installed(self, vm_object):
        for vm in self.vm_list_on_machine:
            if vm.image_hash == vm_object.image_hash:
                return True
        return False


class VMImage(Base):
    __tablename__ = "vm_images"
    image_id = Column(Integer, primary_key=True)
    image_name = Column(String(100), unique=True, nullable=False)
    image_file = Column(String(500), unique=False, nullable=False)
    image_version = Column(String(100), nullable=False)
    image_hash = Column(String(500), nullable=False)
    clients = relationship(
        "Client",
        secondary=client_image_table
    )


class User(Base):
    __tablename__ = "users"
    user_id = Column(Integer, primary_key=True, autoincrement=True)
    username = Column(String)
    password_hash = Column(String)