"""
QGIS 图层映射配置。

所有模板→目标库的表名/图层映射集中管理，避免散落在业务代码中。
"""
from pathlib import Path
import os

# ============================================================
# 静态底图映射：模板中 PostgreSQL 图层 → 本地 GPKG 文件
# key: 模板图层名（用于匹配 maplayer/layername）
# table: 原始数据源表名（schema.table 格式，用于匹配 datasource 替换）
# gpkg: 本地 GPKG 文件名（相对于 GPKG 目录）
# ============================================================
STATIC_LAYERS = {
    "水库":               {"table": "base.rivers",                     "gpkg": "rivers.gpkg"},
    "市州驻地":           {"table": "base.sx_capital",                 "gpkg": "sx_capital.gpkg"},
    "河流":               {"table": "base.river",                      "gpkg": "river.gpkg"},
    "active_fault":       {"table": "base.active_fault",               "gpkg": "active_fault.gpkg"},
    "陕西省":             {"table": "base.sx",                         "gpkg": "sx.gpkg"},
    "乡镇驻地":           {"table": "base.sx_street",                  "gpkg": "sx_street.gpkg"},
    "区县驻地":           {"table": "base.sx_xa_county",               "gpkg": "sx_xa_county.gpkg"},
    "县界":               {"table": "base.sx_xa_county_boundary",      "gpkg": "sx_xa_county_boundary.gpkg"},
    "周边区县":           {"table": "base.sx_zb_county_boundary",      "gpkg": "sx_zb_county_boundary.gpkg"},
    "周边市州":           {"table": "base.sx_zb_city",                 "gpkg": "sx_zb_city.gpkg"},
    "周边县区":           {"table": "base.sx_zb_county",               "gpkg": "sx_zb_county.gpkg"},
    "traffic_expressway": {"table": "base.traffic_expressway",         "gpkg": "traffic_expressway.gpkg"},
    "traffic_provincial": {"table": "base.traffic_provincial",         "gpkg": "traffic_provincial.gpkg"},
    "traffic_railway":    {"table": "base.traffic_railway",            "gpkg": "traffic_railway.gpkg"},
    "traffic_township":   {"table": "base.traffic_township",           "gpkg": "traffic_township.gpkg"},
    "traffic_trunk_line": {"table": "base.traffic_trunk_line",         "gpkg": "traffic_trunk_line.gpkg"},
    # ── 新增静态层 ──
    "积水点":             {"table": "base.hazard_hydrops",             "gpkg": "hazard_hydrops.gpkg"},
    "排水口":             {"table": "base.lifeline_outfall",           "gpkg": "lifeline_outfall.gpkg"},
    "供水管网":           {"table": "base.lifeline_pipe",              "gpkg": "lifeline_pipe.gpkg"},
    "risk_census_population": {"table": "base.risk_census_population", "gpkg": "risk_census_population.gpkg"},
    "西安乡镇":           {"table": "base.sx_xa_towns",                "gpkg": "sx_xa_towns.gpkg"},
}

# ============================================================
# OGR 本地图层 → PostgreSQL 转换
# 设计师用本地 JSON/shp 文件制作模板，数据已迁移到 qgis schema
# key: layername（模板 datasource 中 |layername=xxx 的值）
# value: qgis schema 中的目标表名
# ============================================================
OGR_TO_POSTGRES = {
    "公园":   "shelter_park",
    "学校":   "shelter_school",
    "文化馆": "shelter_cultural",
    "人防设施": "shelter_defence",
    "体育馆": "shelter_gymnasium",
    "广场":   "shelter_square",
    "住宿":   "shelter_stay",
}

# ============================================================
# 表名映射：模板中引用的表名 → 目标库中实际表名
# （模板由多个源库的表拼合，迁移后表名可能不同）
# ============================================================
TABLE_RENAMES = {
    "hazard_waterlogging": "hazard_hydrops",  # 积水点
}

# ============================================================
# Schema 替换：模板源 schema → 目标 schema
# ============================================================
SCHEMA_REPLACEMENTS = ["base", "kspg", "dzxx"]

# ============================================================
# 图层过滤配置（layer_filter.py 使用）
# ============================================================
EVENT_LAYERS = ["eqcenter", "震中"]  # 按 event 字段过滤
QUEUE_LAYERS = [
    "intensity", "intensity_mian",
    "dz_ryss", "dz_jjss", "dz_rysw", "dz_jzph", "dz_xzjl",
]  # 按 eqqueue_id 字段过滤

# ============================================================
# GPKG 目录路径（项目根目录相对路径）
# ============================================================
GPKG_SUBDIR = "app/data/gpkg"


def get_gpkg_dir(project_root: str = None) -> str:
    """获取 GPKG 目录绝对路径（从配置读取，避免硬编码）"""
    try:
        from config import settings
        gpkg_subdir = getattr(settings, "QGIS_GPKG_DIR", None) or GPKG_SUBDIR
    except Exception:
        gpkg_subdir = GPKG_SUBDIR

    if project_root is None:
        project_root = os.path.dirname(
            os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
        )
    gpkg_dir = os.path.join(project_root, gpkg_subdir)
    return os.path.normpath(gpkg_dir).replace("\\", "/")


def build_static_layers_config(gpkg_dir: str = None) -> dict:
    """
    构建 template_modifier / qgis_runner 使用的静态底图配置。

    返回格式与原有 template_override 的 static_layers 段兼容。
    """
    if gpkg_dir is None:
        gpkg_dir = get_gpkg_dir()

    layers = {}
    for name, info in STATIC_LAYERS.items():
        layers[name] = {
            "file": info["gpkg"],
            "table": info["table"],
        }

    return {
        "enabled": True,
        "gpkg_dir": gpkg_dir,
        "layers": layers,
    }