import logging.config

import yaml
from dbengine.handlers import handlers
from fasthtml import serve
from fasthtml.components import Div

from myfasthtml.controls.CommandsDebugger import CommandsDebugger
from myfasthtml.controls.DataGridFormattingManager import DataGridFormattingManager
from myfasthtml.controls.DataGridsManager import DataGridsManager
from myfasthtml.controls.Dropdown import Dropdown
from myfasthtml.controls.FileUpload import FileUpload
from myfasthtml.controls.InstancesDebugger import InstancesDebugger
from myfasthtml.controls.Keyboard import Keyboard
from myfasthtml.controls.Layout import Layout
from myfasthtml.controls.TabsManager import TabsManager
from myfasthtml.controls.helpers import Ids, mk
from myfasthtml.core.dbengine_utils import DataFrameHandler
from myfasthtml.core.instances import UniqueInstance
from myfasthtml.icons.carbon import volume_object_storage
from myfasthtml.icons.fluent_p2 import key_command16_regular
from myfasthtml.icons.fluent_p3 import folder_open20_regular, text_edit_style20_regular
from myfasthtml.myfastapp import create_app

with open('logging.yaml', 'r') as f:
  config = yaml.safe_load(f)

# At the top of your script or module
logging.config.dictConfig(config)

app, rt = create_app(protect_routes=True,
                     mount_auth_app=True,
                     pico=False,
                     vis=True,
                     title="MyFastHtml",
                     live=True,
                     base_url="http://localhost:5003")


@rt("/")
def index(session):
  session_instance = UniqueInstance(session=session,
                                    _id=Ids.UserSession,
                                    on_init=lambda: handlers.register_handler(DataFrameHandler()))
  layout = Layout(session_instance, "Testing Layout")
  layout.footer_left.add("Goodbye World")
  
  tabs_manager = TabsManager(layout, _id=f"-tabs_manager")
  add_tab = tabs_manager.commands.add_tab
  btn_show_right_drawer = mk.button("show",
                                    command=layout.commands.toggle_drawer("right"),
                                    id="btn_show_right_drawer_id")
  
  instances_debugger = InstancesDebugger(layout)
  btn_show_instances_debugger = mk.label("Instances",
                                         icon=volume_object_storage,
                                         command=add_tab("Instances", instances_debugger),
                                         id=instances_debugger.get_id())
  
  commands_debugger = CommandsDebugger(layout)
  btn_show_commands_debugger = mk.label("Commands",
                                        icon=key_command16_regular,
                                        command=add_tab("Commands", commands_debugger),
                                        id=commands_debugger.get_id())
  
  btn_file_upload = mk.label("Upload",
                             icon=folder_open20_regular,
                             command=add_tab("File Open", FileUpload(layout, _id="-file_upload")),
                             id="file_upload_id")
  
  btn_popup = mk.label("Popup",
                       command=add_tab("Popup", Dropdown(layout, "Content", button="button", _id="-dropdown")))
  
  layout.header_left.add(tabs_manager.add_tab_btn())
  layout.header_right.add(btn_show_right_drawer)
  layout.left_drawer.add(btn_show_instances_debugger, "Debugger")
  layout.left_drawer.add(btn_show_commands_debugger, "Debugger")
  
  # Parameters
  formatting_manager = DataGridFormattingManager(layout)
  btn_show_formatting_manager = mk.label("Formatting",
                                         icon=text_edit_style20_regular,
                                         command=add_tab("Formatting", formatting_manager),
                                         id=formatting_manager.get_id())
  layout.left_drawer.add(btn_show_formatting_manager, "Parameters")
  
  layout.left_drawer.add(btn_file_upload, "Test")
  layout.left_drawer.add(btn_popup, "Test")
  
  # data grids
  dgs_manager = DataGridsManager(session_instance, save_state=True)
  layout.left_drawer.add_group("Documents", Div("Documents",
                                                dgs_manager.mk_main_icons(),
                                                cls="mf-layout-group flex gap-3"))
  layout.left_drawer.add(dgs_manager, "Documents")
  
  layout.set_main(tabs_manager)
  
  # keyboard shortcuts
  keyboard = Keyboard(layout, _id="-keyboard").add("ctrl+o",
                                                   add_tab("File Open", FileUpload(layout, _id="-file_upload")))
  keyboard.add("ctrl+n", add_tab("File Open", FileUpload(layout, _id="-file_upload")))
  return layout, keyboard


if __name__ == "__main__":
  # debug_routes(app)
  serve(port=5003)