#include "savepresetpopup.h"

// Tnz6 includes
#include "menubarcommandids.h"
#include "filebrowsermodel.h"
#include "tapp.h"

// TnzQt includes
#include "toonzqt/menubarcommand.h"

// TnzLib includes
#include "toonz/tfxhandle.h"
#include "toonz/tcolumnfx.h"
#include "toonz/toonzfolders.h"

// TnzBase includes
#include "tfx.h"
#include "tmacrofx.h"

// TnzCore includes
#include "tsystem.h"
#include "tstream.h"
#include "tfxattributes.h"

// Qt includes
#include <QPushButton>
#include <QMainWindow>

//=============================================================================
// SavePresetPopup
//-----------------------------------------------------------------------------

SavePresetPopup::SavePresetPopup()
	: Dialog(TApp::instance()->getMainWindow(), true, true, "SavePreset")
{
	setWindowTitle(tr("Save Preset"));
	addWidget(tr("Preset Name:"), m_nameFld = new DVGui::LineEdit(this));
	m_nameFld->setMinimumWidth(170);

	QPushButton *okBtn = new QPushButton(tr("Save"), this);
	okBtn->setDefault(true);
	QPushButton *cancelBtn = new QPushButton(tr("Cancel"), this);
	connect(okBtn, SIGNAL(clicked()), this, SLOT(onOkBtn()));
	connect(cancelBtn, SIGNAL(clicked()), this, SLOT(reject()));

	addButtonBarWidget(okBtn, cancelBtn);
}

//-----------------------------------------------------------------------------

SavePresetPopup::~SavePresetPopup()
{
}

//-----------------------------------------------------------------------------

bool SavePresetPopup::apply()
{
	TFxHandle *fxHandle = TApp::instance()->getCurrentFx();
	TFx *fx = fxHandle->getFx();
	if (TZeraryColumnFx *zfx = dynamic_cast<TZeraryColumnFx *>(fx))
		fx = zfx->getZeraryFx();
	if (!fx)
		return 0;

	TMacroFx *macroFx = dynamic_cast<TMacroFx *>(fx);
	bool isMacro = macroFx != 0;
	wstring name = m_nameFld->text().toStdWString();
	if (name.empty())
		return 0;
	TFilePath fp = ToonzFolder::getFxPresetFolder() + "presets" + fx->getFxType() + (name + (isMacro ? L".macrofx" : L".fx"));
	if (!TFileStatus(fp.getParentDir()).doesExist()) {
		try {
			TFilePath parent = fp.getParentDir();
			TSystem::mkDir(parent);
			DvDirModel::instance()->refreshFolder(parent.getParentDir());
		} catch (...) {
			DVGui::error(tr("It is not possible to create the preset folder %1.").arg(QString::fromStdString(fp.getParentDir().getName())));
			return 0;
		}
	}
	if (TFileStatus(fp).doesExist()) {
		if (DVGui::MsgBox(tr("Do you want to overwrite?"), tr("Yes"), tr("No")) == 2)
			return 0;
	}
	if (isMacro) {
		TOStream os(fp);
		TFx *fx2 = fx->clone(false);
		fx2->getAttributes()->passiveCacheDataIdx() = -1;
		os << fx2;
		delete fx2;
	} else {
		TOStream os(fp);
		fx->savePreset(os);
	}
	fxHandle->notifyFxPresetSaved();
	return true;
}

//-----------------------------------------------------------------------------

void SavePresetPopup::onOkBtn()
{
	if (apply())
		accept();
}

//-----------------------------------------------------------------------------

OpenPopupCommandHandler<SavePresetPopup> openSavePresetPopup(MI_SavePreset);