'''
Created on 2017年8月15日

@author: CreatBot-SW
'''

import os
import sys

from PyQt5.Qt import pyqtSignal
from PyQt5.QtCore import QSize, QDir, QTimer
from PyQt5.QtGui import QIcon
from PyQt5.QtSerialPort import QSerialPortInfo
from PyQt5.QtWidgets import QApplication, QVBoxLayout, QGroupBox, \
	QRadioButton, QGridLayout, QWidget, QProgressBar, QStatusBar, QComboBox, QLabel, \
	QHBoxLayout, QLineEdit, QPushButton, QFileDialog, QCheckBox

from avr_isp.intelHex import formatError
from avr_isp.ispBase import IspError
from avr_isp.stk500v2 import stk500v2Thread, portError


if getattr(sys, 'frozen', False):
	bundle_dir = sys._MEIPASS
else:
	bundle_dir = os.path.dirname(os.path.abspath(__file__))


def portListAll():
	return [port.portName() for port in QSerialPortInfo.availablePorts()]


def portList():
	return [port.portName() for port in QSerialPortInfo.availablePorts() if port.description() == "Arduino Mega 2560"]


class countLabel(QLabel):

	def __init__(self, text):
		super(countLabel, self).__init__(text)

	def mouseDoubleClickEvent(self, e):
		self.setText("0")
		return super(countLabel, self).mouseDoubleClickEvent(e)


class portCombox(QComboBox):
	
	showPopupSignal = pyqtSignal(bool)
	
	def __init__(self):
		super(portCombox, self).__init__()
		
	def showPopup(self):
		self.showPopupSignal.emit(True)
		self.setSizeAdjustPolicy(QComboBox.AdjustToContents)
		return super(portCombox, self).showPopup()

class mainWindow(QWidget):

	def __init__(self):
		super(mainWindow, self).__init__()
		self.setWindowTitle(self.tr("Firmware Installer"))
		self.setWindowIcon(QIcon(os.path.join(bundle_dir, "ico.ico")))
		self.setFixedSize(QSize(480, 240))
		self.setAcceptDrops(True)

		self.portUpdateTimer = QTimer()
		self.portUpdateTimer.timeout.connect(self.portUpdate)
		self.portUpdateTimer.start(100)
		self.autoTimer = QTimer()
		self.autoTimer.setSingleShot(True)
		self.autoTimer.timeout.connect(self.installFile)
		self.task = None

		self.initUI()
		self.configUI()
		self.resize()

	def initUI(self):
		mainLayout = QVBoxLayout()

		self.portBox = QGroupBox(self.tr("Port Selection"))
		self.fileBox = QGroupBox(self.tr("File Selection"))
		self.ctrlBox = QGroupBox(self.tr("Control"))

		# PortBox widget
		self.autoRadio = QRadioButton(self.tr("Auto"))
		self.manualRadio = QRadioButton(self.tr("Manual"))

		self.manualBox = QGroupBox()
		manualLayout = QHBoxLayout()
		self.manualBox.setLayout(manualLayout)

		self.portCombo = portCombox()
		self.baudCombo = QComboBox()
		manualLayout.addWidget(QLabel(self.tr("Port:")))
		manualLayout.addWidget(self.portCombo)
		manualLayout.addStretch()
		manualLayout.addWidget(QLabel(self.tr("Baudrate:")))
		manualLayout.addWidget(self.baudCombo)
		manualLayout.addStretch()

		portLayout = QGridLayout()
		self.portBox.setLayout(portLayout)
		portLayout.addWidget(self.autoRadio, 0, 0)
		portLayout.addWidget(self.manualRadio, 0, 1)
		portLayout.addWidget(self.manualBox, 1, 0, 1, 2)

		# FileBox widget
		self.file = QLineEdit()
		self.file.setPlaceholderText("Please select a hex file.")
		self.file.setReadOnly(True)
		self.file.setFrame(False)
		self.file.setDragEnabled(True)
# 		self.file.setDisabled(True)
		self.fileBtn = QPushButton(self.tr("&Open"))

		fileLayout = QHBoxLayout()
		self.fileBox.setLayout(fileLayout)
		fileLayout.addWidget(QLabel(self.tr("Hex file:")))
		fileLayout.addWidget(self.file)
		fileLayout.addWidget(self.fileBtn)

		# CtrlBox widget
		self.installBtn = QPushButton(self.tr("&Install"))
		self.stopBtn = QPushButton(self.tr("&Stop"))
		self.autoCheck = QCheckBox(self.tr("Auto Install"))
		self.autoTimeLabel = QLabel(self.tr("Idle Time:"))
		self.autoTimeLabel2 = QLabel(self.tr("s"))
		self.autoTime = QLineEdit("2")
		self.autoTime.setInputMask("00")
		self.autoTime.setMaximumWidth(20)

		layout1 = QVBoxLayout()
		layout1.addWidget(self.autoCheck)
		layout2 = QHBoxLayout()
		layout2.addWidget(self.autoTimeLabel)
		layout2.addWidget(self.autoTime)
		layout2.addWidget(self.autoTimeLabel2)
		layout1.addLayout(layout2)
		self.autoInstallWidget = QWidget()
		self.autoInstallWidget.setLayout(layout1)

		ctrlLayout = QHBoxLayout()
		self.ctrlBox.setLayout(ctrlLayout)

		ctrlLayout.addWidget(self.autoInstallWidget)
		ctrlLayout.addStretch()
		ctrlLayout.addWidget(self.installBtn)
		ctrlLayout.addStretch()
		ctrlLayout.addWidget(self.stopBtn)
		ctrlLayout.addStretch()

		# 进度条和状态栏
		self.progress = QProgressBar()
		self.progress.setTextVisible(False)
		self.progress.setRange(0, 100)
		self.statusBar = QStatusBar()
		self.statusBar.setSizeGripEnabled(False)

		self.tryAgainLabel = QLabel(self.tr("Try again..."))
		self.tryAgain = QLabel("0")
		self.statusBar.addPermanentWidget(self.tryAgainLabel)
		self.statusBar.addPermanentWidget(self.tryAgain)

		# 计数栏
		self.countBar = QStatusBar()
		self.countBar.setSizeGripEnabled(False)
		self.countBar.setStyleSheet("QStatusBar::item { border: 0 } QLabel {border:0; font-size: 14px; font-weight: bold}")
		countSuccessLabel = QLabel("Success: ")
		countFailureLabel = QLabel("Failure: ")
		self.countSuccess = countLabel("0")
		self.countFailure = countLabel("0")
		countSuccessLabel.setStyleSheet("QLabel {color: green}")
		self.countSuccess.setStyleSheet("QLabel {color: green}")
		countFailureLabel.setStyleSheet("QLabel {color: red}")
		self.countFailure.setStyleSheet("QLabel {color: red}")

# 		print(self.countFailure.mouseDoubleClickEvent())
		self.url = QLabel("<a href = www.creatbot.com>www.CreatBot.com</a>")
		self.url.setOpenExternalLinks(True)

		self.countBar.addWidget(QLabel(""), 1)
		self.countBar.addWidget(countSuccessLabel)
		self.countBar.addWidget(self.countSuccess, 1)
		self.countBar.addWidget(countFailureLabel)
		self.countBar.addWidget(self.countFailure, 1)
		self.countBar.addWidget(self.url)
		self.countBar.addWidget(QLabel(""), 1)

		# MainLayout
		mainLayout.addWidget(self.portBox)
		mainLayout.addWidget(self.fileBox)
		mainLayout.addWidget(self.ctrlBox)
		mainLayout.addWidget(self.progress)
		mainLayout.addWidget(self.statusBar)
		mainLayout.addWidget(self.countBar)
		self.setLayout(mainLayout)

	def configUI(self):
		self.baudCombo.addItems([str(baud) for baud in QSerialPortInfo.standardBaudRates()])

		self.progress.hide()
		self.statusBar.hide()
		self.stopBtn.setDisabled(True)
		self.portCombo.show()
		self.autoRadio.toggled.connect(self.manualBox.setDisabled)
		self.autoRadio.toggled.connect(self.autoInstallWidget.setVisible)
		self.autoRadio.toggled.connect(self.resize)
		self.manualRadio.clicked.connect(self.disableAutoInstall)
		self.manualRadio.clicked.connect(self.resize)
		self.autoCheck.toggled.connect(self.autoTimeLabel.setEnabled)
		self.autoCheck.toggled.connect(self.autoTime.setEnabled)
		self.autoCheck.toggled.connect(self.autoTimeLabel2.setEnabled)
		self.autoCheck.stateChanged.connect(self.autoStateChangeAction)
		self.portCombo.showPopupSignal.connect(self.portUpdate)
		self.autoTime.returnPressed.connect(self.autoTimeChangeAction)
		self.statusBar.messageChanged.connect(self.stateClearAction)

		self.autoCheck.click()
		self.autoCheck.click()
		self.autoRadio.click()

		self.fileBtn.clicked.connect(self.selectFile)
		self.installBtn.clicked.connect(self.installFile)
		self.installBtn.setFocus()
		self.stopBtn.clicked.connect(self.stopInstall)

		self.file.__class__.dragEnterEvent = self.dragEnterEvent

	def portUpdate(self, forceUpdate = False):
		if self.autoRadio.isChecked():
			self.baudCombo.setCurrentText("115200")
# 			self.portCombo.addItems(portList())
			self.portCombo.clear()
			for port in QSerialPortInfo.availablePorts():
				if port.description() not in ["Arduino Mega 2560", "USB-SERIAL CH340"]:  # 过滤2560和CH340
						continue
				self.portCombo.addItem(port.portName() + " (" + port.description() + ")" , port.portName())
		else:
			currentPortData = self.portCombo.currentData()
			if forceUpdate or (currentPortData and currentPortData not in [port.portName() for port in QSerialPortInfo.availablePorts()]):
				self.portCombo.clear()
				for port in QSerialPortInfo.availablePorts():
					self.portCombo.addItem(port.portName() + " (" + port.description() + ")" , port.portName())
				self.portCombo.setCurrentIndex(self.portCombo.findData(currentPortData))
					
		self.portCombo.setSizeAdjustPolicy(QComboBox.AdjustToContents)
		self.baudCombo.setSizeAdjustPolicy(QComboBox.AdjustToContents)

	def disableAutoInstall(self):
		self.autoCheck.setChecked(False)

	def autoStateChangeAction(self, check):
		if not check and self.autoTimer.remainingTime() > 0:
			self.stopInstall()

	def autoTimeChangeAction(self):
		self.autoTime.clearFocus()
		if self.autoCheck.isChecked() and self.autoTimer.remainingTime() > 0:
			self.autoTimer.stop()
			self.installFile()

	def stateClearAction(self, msg):
		if msg is "" and self.statusBar.isVisible():
			self.statusBar.hide()
		if msg is not "" and self.statusBar.isHidden():
			self.statusBar.show()
		self.resize()

	def selectFile(self):
		hexFileDialog = QFileDialog()
		hexFileDialog.setWindowTitle(self.tr("Select hex file"))
		hexFileDialog.setNameFilter(self.tr("Hex files(*.hex)"))
		hexFileDialog.setFileMode(QFileDialog.ExistingFile)
		if(self.file.text() is ""):
			hexFileDialog.setDirectory(QDir.home())  # 设置为Home目录
		else:
			fileDir = QDir(self.file.text())
			fileDir.cdUp()
			if(fileDir.exists()):
				hexFileDialog.setDirectory(fileDir)  # 设置为当前文件所在目录
			else:
				hexFileDialog.setDirectory(QDir.home())
		if(hexFileDialog.exec()):
			self.file.setText(hexFileDialog.selectedFiles()[0])

	def installFile(self, notFromButton = True):
		if(self.file.text() is ""):
			if(not notFromButton):
				self.selectFile()
				self.installFile(True)
		else:
			if self.autoTimer.remainingTime() is not -1:
				self.autoTimer.stop()

			self.portBox.setDisabled(True)
			self.fileBox.setDisabled(True)
			self.installBtn.setDisabled(True)
			self.stopBtn.setEnabled(True)
			self.progress.show()
			self.resize()

			self.task = stk500v2Thread(self, self.portCombo.currentData(), int(self.baudCombo.currentText()), self.file.text(), self.progressUpdate)
			self.task.stateCallback[str].connect(self.stateUpdate)
			self.task.stateCallback[Exception].connect(self.stateUpdate)
			self.task.finished.connect(self.autoAction)
			self.task.start()
			self.statusBar.showMessage(" ")

	def stopInstall(self, succeed = False, autoInstall = False):
		if autoInstall:
			self.progress.reset()
		else:
			self.portBox.setEnabled(True)
			self.fileBox.setEnabled(True)
			self.installBtn.setEnabled(True)
			self.stopBtn.setDisabled(True)
			self.progress.reset()
			self.progress.hide()
			self.tryAgain.setText("0")
			self.resize()

		if succeed:
			self.countSuccess.setText(str(int(self.countSuccess.text()) + 1))
			self.task = None
		else:
			if self.autoTimer.remainingTime() is not -1:
				self.autoTimer.stop()
			if self.task is not None and self.task.isRunning():
					self.task.finished.disconnect()
					if self.task.isReady():
						self.countFailure.setText(str(int(self.countFailure.text()) + 1))

					if(self.task.isInterruptionRequested()):
						pass
					else:
# 						self.task.requestInterruption()
						self.task.terminate()
						self.statusBar.clearMessage()
# 						self.task = None
			else:
				self.statusBar.clearMessage()

	def autoAction(self):
		self.stopInstall(True, self.autoCheck.isChecked())
		if self.autoCheck.isChecked():
			self.autoTimer.start(int(self.autoTime.text()) * 1000)

	def resize(self):
		self.setFixedHeight(self.sizeHint().height())

	def progressUpdate(self, cur, total):
		self.progress.setMaximum(total)
		self.progress.setValue(cur)

	def stateUpdate(self, stateOrError):
		self.tryAgainLabel.setHidden(True)
		self.tryAgain.setHidden(True)
		if self.task.isReady():
			self.tryAgain.setText("0")

		if type(stateOrError) == str:
# 		self.statusBar.setStyleSheet("QStatusBar::item { border: 0 } QLabel {color: red; font-weight: bold}")
			self.statusBar.setStyleSheet("QStatusBar {font-weight: bold; color: black}  QStatusBar::item { border: 0 } QLabel {font-weight: bold}")
			if self.task is not None and not self.task.isWork and not self.autoCheck.isChecked():
				self.statusBar.showMessage(stateOrError, 3000)
			else:
				self.statusBar.showMessage(stateOrError)
		else:
			self.task.requestInterruption()
			self.statusBar.setStyleSheet("QStatusBar {font-weight: bold; color: red} QStatusBar::item { border: 0 } QLabel {font-weight: bold; color: red}")

			if type(stateOrError) == portError:
				if (stateOrError.value in [portError.errorInvalid, portError.errorBusy]) and (int(self.tryAgain.text()) < 20):
					self.statusBar.showMessage("PortError: " + str(stateOrError))
					self.tryAgain.setText(str(int(self.tryAgain.text()) + 1))
					self.tryAgainLabel.setVisible(True)
					self.tryAgain.setVisible(True)
					self.stopInstall(autoInstall = True)
					self.autoTimer.start(1000)  # 1秒自动重试
				else:
					self.statusBar.showMessage("PortError: " + str(stateOrError), 5000)
					self.stopInstall()
			elif type(stateOrError) == IspError:
				if self.autoCheck.isChecked():
					self.statusBar.showMessage("IspError: " + str(stateOrError))
					self.tryAgainLabel.setVisible(True)
					self.tryAgain.setVisible(True)
					self.stopInstall(autoInstall = True)
					self.autoTimer.start(int(self.autoTime.text()) * 1000)
				else:
					self.statusBar.showMessage("IspError: " + str(stateOrError), 5000)
					self.stopInstall()
			elif type(stateOrError) == formatError:
				self.statusBar.showMessage("HexError: " + str(stateOrError), 5000)
				self.stopInstall()
			else:
				self.statusBar.showMessage("Error: " + str(stateOrError), 5000)
				self.stopInstall()

			self.task.isWork = False
			self.task.wait(100)
			self.task = None


if __name__ == '__main__':
	app = QApplication(sys.argv)

	win = mainWindow()
	win.show()

	if(len(sys.argv) > 1):  # 关联hex文件自动安装
		win.portUpdate()
		win.file.setText(sys.argv[1])
		win.installBtn.click()

	sys.exit(app.exec())