# -*- encoding: utf-8 -*-

import os
import sys
import re
import time
import yaml
import webbrowser

from PyQt6.QtCore import Qt
from PyQt6.QtWidgets import (
	QMainWindow, QFileDialog,
	QMessageBox,
)
from PyQt6.QtGui import (
	QIcon, QAction, QFont,
)

import color
import config
from widget_display import widget_display
from widget_tree import widget_tree
from widget_interact import widget_interact
from widget_term import widget_term
import formatter
import sheet

font = QFont("consolas", 10)
font.setFamilies(["consolas", "黑体"])

class message_about(QMessageBox):
	def __init__(self, window, *arg):
		super(message_about, self).__init__(*arg)
		self.window = window
		self.setIcon(QMessageBox.Icon.Information)
		self.setText("关于")
		self.setInformativeText("by 李隆坤\n nugnikoll@qq.com")
		self.setWindowTitle("关于")
		# self.setDetailedText("The details are as follows:")
		self.setStandardButtons(QMessageBox.StandardButton.Ok)

class main_window(QMainWindow):
	def __init__(self, ipshell, *arg):
		super(main_window, self).__init__(*arg)
		self.ipshell = ipshell
		self.stdout_save = sys.stdout
		self.stderr_save = sys.stderr

		sheet.window = self
		self.ipshell.user_ns["formatter"] = formatter
		self.ipshell.user_ns["sheet"] = sheet.sheet
		self.ipshell.user_ns["cell"] = sheet.cell
		self.ipshell.user_ns["graph"] = sheet.graph
		self.ipshell.user_ns["plot"] = sheet.plot
		self.ipshell.user_ns["param"] = sheet.param

		self.setFont(font)
		self.setWindowTitle("数据处理")
		self.statusBar().showMessage("就绪")

		self.create_menubar()

		self.widget_display = widget_display(self)
		self.widget_display.setObjectName("widget display")
		self.setCentralWidget(self.widget_display)
		self.widget_tree = widget_tree(self)
		self.widget_tree.setObjectName("widget tree")
		self.addDockWidget(Qt.DockWidgetArea.LeftDockWidgetArea, self.widget_tree)
		self.widget_interact = widget_interact(self)
		self.widget_interact.setObjectName("widget interact")
		self.addDockWidget(Qt.DockWidgetArea.RightDockWidgetArea, self.widget_interact)
		self.widget_term = widget_term(self)
		self.widget_term.setObjectName("widget term")
		self.addDockWidget(Qt.DockWidgetArea.BottomDockWidgetArea, self.widget_term)

		self.widget_display.initial_table()

		if "main_window" in config.config:
			if "geometry" in config.config["main_window"]:
				self.restoreGeometry(config.config["main_window"]["geometry"])
			if "state" in config.config["main_window"]:
				self.restoreState(config.config["main_window"]["state"])

	#redirect iostream
	def redirect(self):
		self.stdout_save = sys.stdout
		self.stderr_save = sys.stderr

		sys.stdout = self.widget_term.text_term
		sys.stderr = self.widget_term.text_term

	def restore_redirect(self):
		sys.stdout = self.stdout_save
		sys.stderr = self.stderr_save

	def create_menubar(self):
		menu_file = self.menuBar().addMenu("文件(&F)")
		# act_new_game = QAction(
		# 	"新文件(&N)", self, shortcut = "Ctrl+N",
		# 	statusTip = "创建一个新文件", triggered = self.new_file
		# )
		# menu_file.addAction(act_new_game)
		act_import_data = QAction(
			"导入数据(&I)", self, shortcut = "Ctrl+I",
			statusTip = "从文件导入数据到表格", triggered = self.on_import_data
		)
		menu_file.addAction(act_import_data)
		act_export_data = QAction(
			"导出数据(&E)", self, shortcut = "Ctrl+T",
			statusTip = "将表格数据导出到文件", triggered = self.on_export_data
		)
		menu_file.addAction(act_export_data)
		act_reset_data = QAction(
			"重置数据(&R)", self, #shortcut = "Ctrl+R",
			statusTip = "将表格数据重置为初始数据", triggered = self.on_reset_data
		)
		menu_file.addAction(act_reset_data)
		act_load_script = QAction(
			"运行脚本(&L)", self, shortcut = "Ctrl+L",
			statusTip = "加载并运行一个Python脚本", triggered = self.on_load_script
		)
		menu_file.addAction(act_load_script)
		act_quit = QAction(
			"退出(&Q)", self, shortcut = "Alt+F4",
			statusTip = "退出程序", triggered = self.close
		)
		menu_file.addAction(act_quit)

		# menu_edit = self.menuBar().addMenu("编辑(&E)")
		# act_undo = QAction(
		# 	"撤销(&U)", self, shortcut = "Ctrl+Z",
		# 	statusTip = "撤销最新修改", triggered = self.undo
		# )
		# menu_edit.addAction(act_undo)
		# act_redo = QAction(
		# 	"重做(&R)", self, shortcut = "Ctrl+Y",
		# 	statusTip = "恢复最新修改", triggered = self.redo
		# )
		# menu_edit.addAction(act_redo)

		menu_window = self.menuBar().addMenu("窗口(&W)")
		act_widget_tree = QAction(
			"项目窗口(&P)", self,
			statusTip = "显示/隐藏 项目窗口", triggered = self.toggle_widget_tree
		)
		menu_window.addAction(act_widget_tree)
		act_widget_interact = QAction(
			"交互窗口(&I)", self,
			statusTip = "显示/隐藏 交互窗口", triggered = self.toggle_widget_interact
		)
		menu_window.addAction(act_widget_interact)
		act_widget_term = QAction(
			"终端窗口(&T)", self,
			statusTip = "显示/隐藏 终端窗口", triggered = self.toggle_widget_term
		)
		menu_window.addAction(act_widget_term)

		menu_help = self.menuBar().addMenu("帮助(&H)")
		act_doc_develop = QAction(
			"开发指南(&D)", self, shortcut = "F1",
			statusTip = "在浏览器中阅读开发指南", triggered = self.show_doc_develop
		)
		menu_help.addAction(act_doc_develop)
		act_about = QAction(
			"关于(&A)", self,
			statusTip = "关于数据处理", triggered = self.show_about
		)
		menu_help.addAction(act_about)

	def on_import_data(self, filename = None):
		if self.widget_display.get_current_widget() is None:
			return
		if not filename:
			if hasattr(self.on_import_data, "filename"):
				path_hint = os.path.dirname(self.on_import_data.filename)
			else:
				path_hint = ""
			filename, _ = QFileDialog.getOpenFileName(self, "从文件导入数据", path_hint, "CSV文件(*.csv)")
		if not filename:
			return

		if len(filename) < 4 or filename[-4:] != ".csv":
			filename += ".csv"

		filename = filename.replace("\\", "\\\\")
		filename = filename.replace("\"", "\\\"")

		widget_id = self.widget_display.get_current_id()
		self.process(f"sheet(\"{widget_id}\").load_file(\"{filename}\")")

	def on_export_data(self, filename = None):
		if self.widget_display.get_current_widget() is None:
			return
		if not filename:
			if hasattr(self.on_export_data, "filename"):
				path_hint = os.path.dirname(self.on_export_data.filename)
			else:
				path_hint = ""
			filename, _ = QFileDialog.getSaveFileName(self, "导出数据到文件", path_hint, "CSV文件(*.csv)")
		if not filename:
			return

		if len(filename) < 4 or filename[-4:] != ".csv":
			filename += ".csv"

		filename = filename.replace("\\", "\\\\")
		filename = filename.replace("\"", "\\\"")

		widget_id = self.widget_display.get_current_id()
		self.process(f"sheet(\"{widget_id}\").dump_file(\"{filename}\")")

	def on_reset_data(self):
		widget_id = self.widget_display.get_current_id()
		self.process(f"sheet(\"{widget_id}\").reset_content()")

	def get_config(self):
		data = {
			"main_window": {
				"geometry": self.saveGeometry().data(),
				"state": self.saveState().data(),
				#"widget_interact": self.widget_interact.get_config(),
			},
		}
		return data

	def closeEvent(self, event):
		self.restore_redirect()

		data = self.get_config()
		with open(config.config_file, "w", encoding = "utf-8") as fobj:
			yaml.safe_dump(data, fobj, allow_unicode = True)

		super().closeEvent(event)

	#process command
	def process(self, s):
		s = s.replace("\u2029", "\n")

		print(f"{color.light_green}>>{color.nc}" + s)

		if not s:
			return
		if s[0] == "%":
			result = self.ipshell.magic(s)
			if result:
				print(result)
		else:
			query = re.search(r"[_0-9a-zA-Z.]+\?", s)
			if query:
				result = self.ipshell.magic("pinfo " + query.group()[:-1])
			else:
				query = re.search(r"[_0-9a-zA-Z.]+\?\?", s)
				if query:
					result = self.ipshell.magic("pinfo2 " + query.group()[:-1])
				self.ipshell.ex(s)

	#load and execute a script
	def load_script(self, filename):
		with open(filename, "r", encoding = "utf-8") as fobj:
			content = fobj.read()
		self.ipshell.ex(content)

	#load and execute a script
	def on_load_script(self, filename = None):
		if not filename:
			filename, _ = QFileDialog.getOpenFileName(self, "加载python脚本", "", "Python脚本(*.py)")
		if not filename:
			return
		filename = filename.replace("\\", "\\\\")
		filename = filename.replace("\"", "\\\"")
		self.process(f"window.load_script(\"{filename}\")")

	def undo(self):
		pass

	def redo(self):
		pass

	def show_doc_develop(self, event):
		doc_file = os.path.join(config.doc_path, "开发指南.html")
		webbrowser.open_new_tab(doc_file)

	def show_about(self, event):
		if not hasattr(self, "msg_about"):
			self.msg_about = message_about(self)
		self.msg_about.exec()

	def toggle_widget_interact(self, event):
		if self.widget_interact.isVisible():
			self.widget_interact.hide()
		else:
			self.widget_interact.show()

	def toggle_widget_tree(self, event):
		if self.widget_tree.isVisible():
			self.widget_tree.hide()
		else:
			self.widget_tree.show()

	def toggle_widget_term(self, event):
		if self.widget_term.isVisible():
			self.widget_term.hide()
		else:
			self.widget_term.show()