seed.py

from database import get_db_connection, init_db
from utils.auth import hash_password

# Creates initial test data with three RBAC roles for immediate testing
# Production: Replace with proper user onboarding system

def seed_data():
    init_db() # Ensure tables exist
    conn = get_db_connection()
    c = conn.cursor()

    # TENDERS TABLE - Projects for contractors to bid on
    c.execute('''
        CREATE TABLE IF NOT EXISTS tenders (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            title TEXT NOT NULL,
            description TEXT NOT NULL,
            status TEXT DEFAULT 'OPEN',
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        )
    ''')
    
    # Seed initial tenders (idempotent - safe to run multiple times)
    check_tenders = c.execute("SELECT count(*) FROM tenders").fetchone()[0]
    if check_tenders == 0:
        c.execute("INSERT INTO tenders (title, description) VALUES ('City Road Repaving', 'Resurfacing of Main St and 1st Ave.')")
        c.execute("INSERT INTO tenders (title, description) VALUES ('District School Renovation', 'Structural repairs for District 4 School.')")
        c.execute("INSERT INTO tenders (title, description) VALUES ('Smart Traffic Lights', 'Installation of AI traffic system in downtown.')")
        print("✅ Seeded initial tenders.")

    # Three roles for RBAC demonstration
    users = [
        ("contractor", "contractor@example.com", "pass123", "contractor"),
        ("official", "official@gov.in", "admin123", "official"),
        ("auditor", "auditor@agency.com", "audit123", "auditor")
    ]

    print("Seeding Users...")
    for username, email, pwd, role in users:
        exists = c.execute("SELECT 1 FROM users WHERE username=?", (username,)).fetchone()
        if not exists:
            hashed, salt = hash_password(pwd)
            c.execute(
                "INSERT INTO users (username, email, password_hash, salt, role) VALUES (?, ?, ?, ?, ?)",
                (username, email, hashed, salt, role)
            )
            print(f" -> Created User: {username} (Role: {role})")
        else:
            print(f" -> User {username} already exists.")

    conn.commit()
    conn.close()
    print("Database seeding complete.")

if __name__ == "__main__":
    seed_data()