#include "oMainWindow.h"
#include "ui_oMainWindow.h"
#include "oEditor.h"
#include "oTextEdit.h"
#include "oXmlResolver.h"
#include "oSyntaxHighlighter.h"

oMainWindow::oMainWindow(QWidget *parent)
: QMainWindow(parent)
, _closingTabIndex(-1)
, ui(new Ui::oMainWindow)
{
 ui->setupUi(this);
 QObject::connect(ui->action_Open,SIGNAL(triggered()),this,SLOT(openFileEvent()));
	QObject::connect(ui->actionSave,SIGNAL(triggered()),this,SLOT(saveCurrentTab()));
	QObject::connect(ui->actionSave_All,SIGNAL(triggered()),this,SLOT(saveAllTabs()));
	QObject::connect(ui->actionClose,SIGNAL(triggered()),this,SLOT(saveAndCloseTab()));
	QObject::connect(ui->action_New,SIGNAL(triggered()),this,SLOT(newFileEvent()));
	QObject::connect(ui->actionFont,SIGNAL(triggered()),this,SLOT(fontSettingEvent()));
 QObject::connect(ui->actionColorPicker,SIGNAL(triggered()),this,SLOT(colorPickerEvent()));
 QObject::connect(ui->tabWidget,SIGNAL(tabCloseRequested(int)),this,SLOT(tabCloseEvent(int)));

	QSettings settings(QSettings::IniFormat,QSettings::UserScope, "Dorothy", "CodeEditor");

	_defaultFont = settings.value("Font",_defaultFont).value<QFont>();

	QString pathStr = settings.value("LastOpened").toString();
	if (!pathStr.isEmpty())
	{
 QStringList paths = pathStr.split(";",Qt::SkipEmptyParts);
		foreach (QString path, paths)
		{
			if (QFileInfo(path).exists())
			{
				this->addTab(path);
			}
		}
	}
}

oMainWindow::~oMainWindow()
{
 delete ui;
}

void oMainWindow::addTab(const QString& filePath)
{
	oEditor* editor = new oEditor(ui->tabWidget, filePath);
	editor->setFont(_defaultFont);
	int index = ui->tabWidget->addTab(editor, QFileInfo(filePath).fileName());
	QObject::connect(editor,SIGNAL(editorChanged(oEditor*,bool)),this,SLOT(tabChanged(oEditor*,bool)));
	ui->tabWidget->setCurrentIndex(index);
	ui->tabWidget->setTabToolTip(index,editor->getFilePath());
}

void oMainWindow::colorPickerEvent()
{
 _colorPickerOptions = QColorDialog::ShowAlphaChannel;
 oEditor* editor = static_cast<oEditor*>(ui->tabWidget->currentWidget());
 if (editor)
 {
 oTextEdit* textEdit = editor->view();
 QString text = textEdit->toPlainText();
 QTextCursor cursor = textEdit->textCursor();
 oXmlResolver* resolver = textEdit->resolver();
 resolver->resolve(text.mid(0,cursor.position()));
 if (resolver->isCurrentInTag() && !resolver->currentAttribute().isEmpty())
 {
 if (resolver->currentAttribute() == "Color3")
 {
 _colorPickerOptions = 0;
 }
 }
 }
 QColorDialog* colorDialog = new QColorDialog();
 colorDialog->setOptions(QColorDialog::ColorDialogOptions(_colorPickerOptions));
 colorDialog->show();
 QObject::connect(colorDialog,SIGNAL(colorSelected(QColor)),this,SLOT(colorSelected(QColor)));
}

void oMainWindow::colorSelected(const QColor& color)
{
 QString result = _colorPickerOptions ?
 QString("0x%1%2%3%4")
 .arg(color.alpha(),2,16,QChar('0'))
 .arg(color.red(),2,16,QChar('0'))
 .arg(color.green(),2,16,QChar('0'))
 .arg(color.blue(),2,16,QChar('0')) :
 QString("0x%1%2%3")
 .arg(color.red(),2,16,QChar('0'))
 .arg(color.green(),2,16,QChar('0'))
 .arg(color.blue(),2,16,QChar('0'));
 oEditor* editor = static_cast<oEditor*>(ui->tabWidget->currentWidget());
 if (editor)
 {
 oTextEdit* textEdit = editor->view();
 QTextCursor cursor = textEdit->textCursor();
 cursor.insertText(result);
 textEdit->highlighter()->rehighlightBlock(cursor.block());
 textEdit->setTextCursor(cursor);
 }
}

void oMainWindow::fontSettingEvent()
{
	bool ok;
	_defaultFont = QFontDialog::getFont(&ok, _defaultFont, this);
	if (ok)
	{
		QTabBar* tabBar = ui->tabWidget->tabBar();
		for (int i = 0;i < tabBar->count();i++)
		{
 oEditor* editor = static_cast<oEditor*>(ui->tabWidget->widget(i));
			editor->setFont(_defaultFont);
		}
		QSettings settings(QSettings::IniFormat,QSettings::UserScope, "Dorothy", "CodeEditor");
		settings.setValue("Font", _defaultFont);
	}
}

void oMainWindow::newFileEvent()
{
	QString filePath = QFileDialog::getSaveFileName(this,"","","*.xml");
	if (!filePath.isEmpty())
	{
 if (QFileInfo(filePath).suffix().toLower() != "xml")
 {
 filePath += ".xml";
 }
		this->addTab(filePath);
	}
}

void oMainWindow::openFileEvent()
{
 QStringList filePaths = QFileDialog::getOpenFileNames(this,"","","*.xml");
 if (!filePaths.isEmpty())
 {
 foreach(QString filePath, filePaths)
 {
 QTabBar* tabBar = ui->tabWidget->tabBar();
 for (int i = 0;i < tabBar->count();i++)
 {
 if (filePath == tabBar->tabToolTip(i))
 {
 ui->tabWidget->setCurrentIndex(i);
 return;
 }
 }
			this->addTab(filePath);
 }
 }
}

void oMainWindow::tabChanged(oEditor* editor, bool modified)
{
 ui->tabWidget->setTabText(ui->tabWidget->indexOf(editor), (modified ? "*" : "")+editor->getFileName());
}

void oMainWindow::tabCloseEvent(int index)
{
 oEditor* editor = static_cast<oEditor*>(ui->tabWidget->widget(index));
 if (editor->isModified())
 {
 _closingTabIndex = index;
		QMessageBox* box = new QMessageBox(QMessageBox::Warning,"Closing Tab","Close the modified tab?",QMessageBox::Ok|QMessageBox::Save|QMessageBox::Cancel);
 QObject::connect(box,SIGNAL(buttonClicked(QAbstractButton*)),this,SLOT(closeModifiedTabEvent(QAbstractButton*)));
 box->show();
 }
 else ui->tabWidget->removeTab(index);
}

void oMainWindow::saveCurrentTab()
{
 oEditor* editor = static_cast<oEditor*>(ui->tabWidget->currentWidget());
 if (editor) editor->save();
}

void oMainWindow::closeModifiedTabEvent(QAbstractButton* button)
{
 if (button->text() == "OK")
 {
 ui->tabWidget->removeTab(_closingTabIndex);
 }
 else if (button->text() == "Save")
 {
 oEditor* editor = static_cast<oEditor*>(ui->tabWidget->widget(_closingTabIndex));
 editor->save();
 ui->tabWidget->removeTab(_closingTabIndex);
 }
 _closingTabIndex = -1;
}

void oMainWindow::saveAllTabs()
{
	QTabBar* tabBar = ui->tabWidget->tabBar();
	for (int i = 0;i < tabBar->count();i++)
	{
 oEditor* editor = static_cast<oEditor*>(ui->tabWidget->widget(i));
		editor->save();
	}
}

void oMainWindow::saveAndCloseTab()
{
 oEditor* editor = static_cast<oEditor*>(ui->tabWidget->currentWidget());
 if (editor)
 {
 editor->save();
 ui->tabWidget->removeTab(ui->tabWidget->indexOf(editor));
 }
}

void oMainWindow::closeEvent(QCloseEvent* event)
{
	bool needSave = false;
 QTabBar* tabBar = ui->tabWidget->tabBar();
	for (int i = 0;i < tabBar->count();i++)
	{
 oEditor* editor = static_cast<oEditor*>(ui->tabWidget->widget(i));
		if (editor->isModified())
		{
			needSave = true;
			break;
		}
	}
	if (needSave)
	{
		QMessageBox::StandardButton ret = QMessageBox::warning(this,
						tr("Application"),
						 tr("The document has been modified.\n"
							"Do you want to save your changes?"),
						 QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel);
		if (ret == QMessageBox::Save)
		{
			saveAllTabs();
			event->accept();
		}
		else if (ret == QMessageBox::Cancel)
		{
			event->ignore();
		}
		else event->accept();
	}
	QSettings settings(QSettings::IniFormat,QSettings::UserScope, "Dorothy", "CodeEditor");
	QString paths;
	for (int i = 0;i < tabBar->count();i++)
	{
		paths += (tabBar->tabToolTip(i) + ";");
	}
	settings.setValue("LastOpened", paths);
}