# Flet imports
import flet as ft

# other imports
import os
from math import pi
from typing import Optional
from loguru import logger
import logging
#logging.basicConfig(level=logging.DEBUG)

# utils imports
from scripts import flet_utils
from scripts.flet_settings_window import settings_window
from scripts.flet_gallery_window import gallery_window
from scripts.flet_file_manager import file_picker, uploads, imports
from scripts.flet_titlebar import titlebar
from scripts.flet_tool_manager import tool_manager
from scripts.flet_asset_manager import asset_manager
from scripts.flet_canvas import canvas
from scripts.flet_messages import messages
from scripts.flet_property_manager import property_manager

# for debugging
from pprint import pprint

os.environ["FLET_WS_MAX_MESSAGE_SIZE"] = "8000000"


#	main ###############################################################
@logger.catch(reraise=True)
def main(page: ft.Page):

#	init ###############################################################
	# messages
	page.messages = messages
	page.message = messages.message
	page.max_message_history = 50

	# ui
	page.current_layout = 'Default'
	page.titlebar_height = 50
	page.bottom_panel_height = page.height * 0.2
	page.toolbox_height = 250
	page.tool_manager_width = 50
	page.tool_manager_button_size = 40
	page.left_panel_width = 200
	page.right_panel_width = 250

	page.background_color = None
	page.primary_color = None
	page.secondary_color = 'white_10'
	page.tertiary_color = 'blue'

	page.text_color = None
	page.text_size = 14
	page.icon_size = 20
	page.padding = 0
	page.margin = 0
	page.container_padding = 0
	page.container_margin = 0
	page.tab_color = 'white_10'
	page.tab_padding = ft.padding.only(left = 2, top = 12, right = 2, bottom = 8)
	page.tab_margin = 0
	page.divider_height = 10
	page.vertical_divider_width = 10

	# titlebar
	page.titlebar = titlebar

	def change_theme_mode(e):
		page.theme_mode = "dark" if page.theme_mode == "light" else "light"

		if "(Light theme)" in titlebar.theme_switcher.tooltip:
			titlebar.theme_switcher.tooltip = titlebar.theme_switcher.tooltip.replace("(Light theme)", '')

		if "(Dark theme)" in titlebar.theme_switcher.tooltip:
			titlebar.theme_switcher.tooltip = titlebar.theme_switcher.tooltip.replace("(Dark theme)", '')

		titlebar.theme_switcher.tooltip += "(Light theme)" if page.theme_mode == "light" else "(Dark theme)"
		page.update()

	page.change_theme_mode = change_theme_mode

	# layouts
	def set_layout(e):
		page.current_layout = e.control.value
		page.update()

	page.set_layout = set_layout

	# tools
	page.tool_manager = tool_manager
	page.current_tool = 'pan'

	# asset manager
	page.asset_manager = asset_manager
	page.active_layer = None
	page.layer_height = 50

	# canvas
	page.canvas = canvas
	page.canvas_background = flet_utils.get_canvas_background('webui/flet/assets/images/default_grid_texture.png')
	page.canvas_size = [512,512]

	def get_viewport_size():
		viewport_width = page.width - (page.tool_manager_width + (page.vertical_divider_width * 3) + page.left_panel_width + page.right_panel_width)
		viewport_height = page.height - (page.titlebar_height * 3) - page.bottom_panel_height
		return viewport_width, viewport_height

	page.get_viewport_size = get_viewport_size

	# property manager
	page.property_manager = property_manager

	# settings
	def load_settings():
		settings = flet_utils.get_user_settings_from_config()
		page.session.set('settings',settings)
		page.update()

	def save_settings_to_config():
		settings = page.session.get('settings')
		flet_utils.save_user_settings_to_config(settings)

	def reset_settings_from_config():
		settings = flet_utils.get_default_settings_from_config()
		page.session.remove('settings')
		page.session.set('settings',settings)
		save_settings_to_config()

	if not page.session.contains_key('settings'):
		load_settings()
		settings = page.session.get('settings')
		if 'webui_page' in settings:
			if 'default_theme' in settings['webui_page']:
				page.theme_mode = settings['webui_page']['default_theme']['value']
			if 'max_message_history' in settings['webui_page']:
				MAX_MESSAGE_HISTORY = settings['webui_page']['max_message_history']['value']

		page.session.set('layout','default')


#	settings window ####################################################

	def close_settings_window(e):
		settings_window.open = False
		page.update()

	page.close_settings = close_settings_window

	def open_settings_window(e):
		page.dialog = settings_window
		settings_window.open = True
		page.update()

	page.open_settings = open_settings_window

	page.settings_window = settings_window
	settings_window.content.width = page.width * 0.50
	settings_window.content.bgcolor = page.primary_color
	settings_window.content.padding = page.container_padding
	settings_window.content.margin = page.container_margin


#	gallery window #####################################################

	def close_gallery_window(e):
		gallery_window.open = False
		page.update()

	page.close_gallery = close_gallery_window

	def open_gallery_window(e):
		page.dialog = gallery_window
		gallery_window.open = True
		page.update()

	page.open_gallery = open_gallery_window

	page.gallery_window = gallery_window
	page.refresh_gallery = gallery_window.refresh_gallery
	gallery_window.content.width = page.width * 0.5
	gallery_window.content.bgcolor = page.primary_color
	gallery_window.content.padding = page.container_padding
	gallery_window.content.margin = page.container_margin

	gallery_window.outputs_gallery.height = page.height * 0.75
	gallery_window.outputs_gallery.bgcolor = page.primary_color
	gallery_window.outputs_gallery.padding = page.container_padding
	gallery_window.outputs_gallery.margin = page.container_margin

	gallery_window.uploads_gallery.height = page.height * 0.75
	gallery_window.uploads_gallery.bgcolor = page.primary_color
	gallery_window.uploads_gallery.padding = page.container_padding
	gallery_window.uploads_gallery.margin = page.container_margin


#	file manager #######################################################

	def close_upload_window(e):
		uploads.open = False
		page.update()

	page.close_uploads = close_upload_window

	def open_upload_window(e):
		page.dialog = uploads
		uploads.open = True
		page.update()

	page.open_uploads = open_upload_window

	def close_import_window(e):
		imports.open = False
		page.update()

	page.close_imports = close_import_window

	def open_import_window(e):
		page.dialog = imports
		imports.open = True
		page.update()

	page.open_imports = open_import_window

	page.uploads = uploads
	page.imports = imports
	page.file_picker = file_picker
	page.overlay.append(file_picker)


#	center panel #############################################################

	text_editor = ft.Container(
			content = ft.Text('Under Construction.'),
			bgcolor = page.secondary_color,
			expand = True,
	)

	viewport = ft.Container(
			bgcolor = page.primary_color,
			padding = page.container_padding,
			margin = page.container_margin,
			content = ft.Tabs(
					selected_index = 0,
					animation_duration = 300,
					tabs = [
							ft.Tab(
									content = canvas,
									tab_content = ft.Text(
											value = 'Canvas',
											size = page.text_size,
									),
							),
							ft.Tab(
									text = 'Text Editor',
									content = text_editor,
									tab_content = ft.Text(
											value = 'Text Editor',
											size = page.text_size,
									),
							),
					],
			),
			expand = True,
	)

	center_panel = ft.Container(
			content = ft.Column(
					controls = [
							viewport,
							messages,
					],
			),
			bgcolor = page.primary_color,
			padding = page.container_padding,
			margin = page.container_margin,
			expand = True,
	)


#	layouts ############################################################

	default_layout = ft.Row(
			controls = [
				tool_manager,
				asset_manager,
				center_panel,
				property_manager,
			],
			expand=True,
	)

	current_layout = default_layout


#	workspace ##########################################################

	workspace = ft.Container(
			bgcolor = page.background_color,
			padding = 0,
			margin = 0,
			expand = True,
			content = ft.Column(
					controls = [
						titlebar,
						current_layout,
					],
			),
			height = page.height,
			width = page.width,
	)

	page.title = "Stable Diffusion Playground"
	page.add(workspace)
	page.settings_window.setup(page.session.get('settings'))
	page.gallery_window.setup()
	page.titlebar.setup()
	page.tool_manager.setup()
	page.asset_manager.setup()
	page.canvas.setup()
	page.messages.setup()
	page.property_manager.setup()
	page.update()


ft.app(target=main, route_url_strategy="path", port= 8505, assets_dir="assets", upload_dir="assets/uploads")