from pathlib import Path
import sys

from sqlalchemy import text

ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT))

from app.database import engine  # noqa: E402


def index_exists(conn, table_name: str, index_name: str) -> bool:
    return bool(
        conn.execute(
            text(
                """
                SELECT COUNT(1)
                FROM information_schema.statistics
                WHERE table_schema = DATABASE()
                  AND table_name = :table_name
                  AND index_name = :index_name
                """
            ),
            {"table_name": table_name, "index_name": index_name},
        ).scalar()
    )


def create_index(conn, table_name: str, index_name: str, columns: str) -> None:
    if not index_exists(conn, table_name, index_name):
        conn.execute(text(f"CREATE INDEX {index_name} ON {table_name} ({columns})"))


def main() -> None:
    with engine.begin() as conn:
        conn.execute(
            text(
                """
                CREATE TABLE IF NOT EXISTS production_report_allocations (
                    id BIGINT NOT NULL AUTO_INCREMENT,
                    report_id BIGINT NOT NULL,
                    report_item_id BIGINT NULL,
                    attendance_point_name VARCHAR(128) NOT NULL DEFAULT '',
                    employee_phone VARCHAR(20) NOT NULL DEFAULT '',
                    allocation_date DATE NOT NULL,
                    day_minutes DECIMAL(10, 2) NOT NULL DEFAULT 0,
                    overtime_minutes DECIMAL(10, 2) NOT NULL DEFAULT 0,
                    night_minutes DECIMAL(10, 2) NOT NULL DEFAULT 0,
                    effective_minutes DECIMAL(10, 2) NOT NULL DEFAULT 0,
                    good_qty DECIMAL(12, 2) NOT NULL DEFAULT 0,
                    defect_qty DECIMAL(12, 2) NOT NULL DEFAULT 0,
                    scrap_qty DECIMAL(12, 2) NOT NULL DEFAULT 0,
                    changeover_count DECIMAL(12, 2) NOT NULL DEFAULT 0,
                    reference_wage DECIMAL(12, 2) NOT NULL DEFAULT 0,
                    created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
                    updated_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
                    PRIMARY KEY (id),
                    CONSTRAINT fk_report_allocations_report
                        FOREIGN KEY (report_id) REFERENCES production_reports (id)
                        ON DELETE CASCADE,
                    CONSTRAINT fk_report_allocations_item
                        FOREIGN KEY (report_item_id) REFERENCES production_report_items (id)
                        ON DELETE CASCADE
                ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
                """
            )
        )

        create_index(conn, "production_report_allocations", "idx_report_allocations_report", "report_id")
        create_index(conn, "production_report_allocations", "idx_report_allocations_item", "report_item_id")
        create_index(conn, "production_report_allocations", "idx_report_allocations_date", "allocation_date")
        create_index(
            conn,
            "production_report_allocations",
            "idx_report_allocations_point_date",
            "attendance_point_name, allocation_date",
        )
        create_index(
            conn,
            "production_report_allocations",
            "idx_report_allocations_employee_date",
            "employee_phone, allocation_date",
        )

    print("report allocations schema migrated")


if __name__ == "__main__":
    main()