from sqlalchemy.orm import Session

from api.database import SessionLocal
from api.hasher import argon2_hasher
import api.models as models


def seed():
    db: Session = SessionLocal()

    if db.query(models.User).count() == 0:
        admin = models.User(name="admin", password_hash=argon2_hasher.hash("admin"))
        db.add(admin)
        db.commit()

    if db.query(models.Service).count() == 0:
        lb = models.Loadbalancer()
        lb.servers.append(
            models.LoadbalancerServers(address="http://192.168.2.134:8000/")
        )
        lb.servers.append(
            models.LoadbalancerServers(address="http://192.168.2.132:80/")
        )

        test_service = models.Service(name="service")
        test_service.loadbalancer = lb

        db.add(test_service)
        db.commit()