"""
地图导出模块。布局文本更新、比例尺调整、图片导出。
"""
from qgis.core import QgsLayoutExporter, QgsScaleBarSettings

from app.config.paths import get_logger

logger = get_logger("qgis.exporter")


class MapExporter:
    def __init__(self, config: dict, layout):
        self.config = config
        self.layout = layout

    def update_texts(self, model: dict) -> None:
        """更新布局中的文本标签"""
        for key in ["mapTitle", "mapTime", "mapUint", "info"]:
            label = self.layout.itemById(key)
            if label is not None:
                label.setText(model[key])

    def update_scale_bar(self) -> None:
        """调整比例尺为自适应宽度模式"""
        scale_bar = self.layout.itemById("ScaleBar")
        if scale_bar is None:
            logger.warning("比例尺控件不存在")
            return

        scale_bar.setSegmentSizeMode(
            QgsScaleBarSettings.SegmentSizeMode.SegmentSizeFitWidth
        )
        scale_bar.setMaximumBarWidth(70)
        scale_bar.setMinimumBarWidth(40)

    def export(self, path: str) -> None:
        """导出布局为图片"""
        dpi = self.config["qgis"]["exportDpi"]

        settings = QgsLayoutExporter.ImageExportSettings()
        settings.dpi = dpi

        exporter = QgsLayoutExporter(self.layout)
        result = exporter.exportToImage(path, settings)

        if result != QgsLayoutExporter.Success:
            raise RuntimeError(f"图片导出失败: {path}")

        logger.info(f"图片已导出: {path}")