from qgis.PyQt.QtWidgets import QToolBar

from .AddMap import AddMapBtn
from .FitZoom import FitZoomAction
from .Search import SearchAction
from .Setting import SettingAction
from .icons import icons
from ..utils import PluginConfig


class TiandituToolbar(QToolBar):
    def __init__(self, iface, parent=None) -> None:
        super().__init__("文件服务器", parent)
        self.iface = iface
        self.icons = icons
        self.conf = PluginConfig()
        self.setToolTip("文件服务器")
        self.add_button = None
        self.actions = []

        self.init_config()
        self.setup_action()

    def setup_action(self):
        # self.add_button = AddMapBtn(self)
        # self.addWidget(self.add_button)
        # 添加 Action
        action_setting = SettingAction(self)
        action_search = SearchAction(iface=self.iface, parent=self)
        action_fitzoom = FitZoomAction(iface=self.iface, parent=self)

        # self.actions.extend([action_setting, action_search, action_fitzoom])
        self.actions.extend([action_search])
        self.addActions(self.actions)

    def init_config(self):
        self.conf.init_config()

    def remove_dock(self):
        print(self.actions)
        if self.actions is not None and len(self.actions) > 1:
            dock = self.actions[1]
            dock.unload()