from __future__ import annotations

import os
from pathlib import Path

# Gói hằng số cấu hình riêng để file chính gọn hơn.
ROOT_DIR = Path("HĐ-2026")
SOURCE_CODE = "HD-2026"
SOURCE_NAME = "HĐ-2026"
SOURCE_YEAR = 2026
SOURCE_TYPE = "mixed_dossier_repository"
INGEST_BATCH = "hd2026_baseline_v2"
MANIFEST_PATH = Path("tmp/hd2026_project_coverage_manifest.json")

DATABASE_URL = os.getenv(
    "DATABASE_URL",
    "postgresql://dscons_user:dscons_password@127.0.0.1:5432/dscons_operations",
)

TOP_LEVEL_CATEGORY_RULES = {
    "Anh Quang-2026": "commercial_contract",
    "Anh Thư-2026": "service_transport",
    "Bình Minh PL-2026": "material_supply",
    "Hoàng Phúc-2026": "commercial_contract",
    "Kiến Hải-2026": "unknown_mixed",
    "Lâm Cường Thịnh-2026": "material_supply",
    "Mai Hoa-2026": "service_transport",
    "Mai Hoan-2026": "commercial_contract",
    "Phòng KT xã Kiến Minh-2026": "owner_project_dossier",
    "Phú Quý-2026": "service_transport",
    "Taibus-2026": "vendor_profile",
    "Thái An-2026": "service_transport",
    "Thiên Khang-2026": "commercial_contract",
    "X88-2026": "vendor_profile",
    "Đa Độ 2026": "owner_project_dossier",
}

EXPECTED_DOCUMENT_TYPES = (
    "quyet_dinh_phe_duyet_nhiem_vu",
    "quyet_dinh_phe_duyet_bcktkt",
    "quyet_dinh_phe_duyet_khlcnt",
    "quyet_dinh_phe_duyet_kqlcnt",
    "bien_ban_thuong_thao_hop_dong",
    "hop_dong_thi_cong",
    "phu_luc_hop_dong",
    "bao_lanh_thuc_hien_hop_dong",
    "bao_lanh_tam_ung",
    "ban_ve_thi_cong",
    "ban_ve_chi_tiet",
    "du_toan",
    "bang_khoi_luong",
    "bien_ban_ban_giao_mat_bang",
    "nhat_ky_thi_cong",
    "ho_so_quan_ly_chat_luong",
    "bien_ban_nghiem_thu_cong_viec",
    "bang_xac_nhan_khoi_luong",
    "ho_so_thanh_toan",
    "bang_gia_tri_thanh_toan",
    "ho_so_hoan_cong",
    "ban_ve_hoan_cong",
    "ho_so_quyet_toan_tong_hop",
    "bien_ban_thanh_ly_hop_dong",
)