import os
from dotenv import load_dotenv
import mysql.connector

load_dotenv()

DB_HOST = os.getenv("DB_HOST", "officialsite.kr")
DB_PORT = int(os.getenv("DB_PORT", "23306"))
DB_USER = os.getenv("DB_USER", "kssoft_ad_mgr")
DB_PASSWORD = os.getenv("DB_PASSWORD", "")
DB_NAME = os.getenv("DB_NAME", "kssoft_v2")

SEED_PLATFORMS = [
    ("기본", None),
]

SEED_MODULES = [
    ("naver-search", "네이버 검색기", 1),
    ("uploader", "업데이터", 1),
    ("inventory", "창고 상태", 1),
    ("fight-file", "파이팅 파일", 1),
    ("market", "상품 파이드", 1),
    ("admin", "업장 파이토", 1),
]

SEED_VERSIONS = [
    ("naver-search", "1.0.0"),
    ("uploader", "0.9.2"),
    ("inventory", "1.4.1"),
    ("fight-file", "2.0.0"),
    ("market", "1.2.0"),
    ("admin", "1.1.0"),
]


def main():
    conn = mysql.connector.connect(
        host=DB_HOST,
        port=DB_PORT,
        user=DB_USER,
        password=DB_PASSWORD,
        database=DB_NAME,
    )
    cursor = conn.cursor()

    # platforms
    for name, icon in SEED_PLATFORMS:
        cursor.execute(
            "INSERT INTO ad_master_platform (platform_kor, icon) VALUES (%s, %s)",
            (name, icon),
        )

    # modules
    for module_id, display_name, platform_id in SEED_MODULES:
        cursor.execute(
            "INSERT INTO ad_master (platform_id, c, name, icon) VALUES (%s, %s, %s, %s)",
            (platform_id, module_id, display_name, None),
        )

    # versions
    for programname, ver in SEED_VERSIONS:
        cursor.execute(
            "INSERT INTO ftp (programname, ver) VALUES (%s, %s)",
            (programname, ver),
        )

    conn.commit()
    cursor.close()
    conn.close()

    print("[OK] Seed data inserted")


if __name__ == "__main__":
    main()