from app.core.extensions import db
from app.core.models import Subnet, Server, App, Port
from app.core.auth import User  # Import User from auth module
import json


def seed_database():
    """Add sample data to the database"""
    # Create a default subnet if none exists
    if Subnet.query.count() == 0:
        subnet = Subnet(
            cidr="192.168.1.0/24",
            location="Office",
            auto_scan=True,
            active_hosts=json.dumps([]),
        )
        db.session.add(subnet)

        # Create a sample server
        server = Server(
            hostname="server1",
            ip_address="192.168.1.10",
            subnet=subnet,
            documentation="# Server 1\n\nThis is a sample server.",
        )
        db.session.add(server)

        # Create a sample app
        app = App(
            name="Web App",
            server=server,
            documentation="# Welcome to Web App\n\nThis is a sample application.",
        )
        db.session.add(app)

        # Add some ports
        ports = [
            Port(app=app, port_number=80, protocol="TCP", description="HTTP"),
            Port(app=app, port_number=443, protocol="TCP", description="HTTPS"),
        ]
        db.session.add_all(ports)

    # Create a default user if none exists
    if User.query.count() == 0:
        admin = User(username="admin", email="admin@example.com", is_admin=True)
        admin.set_password("admin")
        db.session.add(admin)

    try:
        db.session.commit()
        print("Database seeded successfully")
    except Exception as e:
        db.session.rollback()
        print(f"Error seeding database: {e}")