#include <QTimer>
#include "inputeventselectiondialog.h"
#include "ui_inputeventselectiondialog.h"

#define COUNTDOWN_SECS 3

InputEventSelectionDialog::InputEventSelectionDialog(QWidget* parent)
    : QDialog(parent)
    , ui(new Ui::InputEventSelectionDialog)
{
	ui->setupUi(this);
	ui->countdownlabel->setText(m_countingText.arg(COUNTDOWN_SECS));

	m_countdownTimer = new QTimer(this);
	connect(m_countdownTimer, SIGNAL(timeout()), this, SLOT(updateCountdown()));

	setWindowFlags(Qt::Window | Qt::WindowCloseButtonHint);
	setFixedSize(size());

	// workaround to avoid direct thread ui access
	connect(this, SIGNAL(startCountdown(QString)), this, SLOT(handleStartCountdown(QString)));
	connect(this, SIGNAL(cancelCountdown()), this, SLOT(handleCancelCountdown()));
}

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

void InputEventSelectionDialog::Setup(const char* text, CInputBindingManager* inputManager, CInputProviderQtKey* qtKeyInputProvider, CInputProviderQtMouse* qtMouseInputProvider, uint32 padIndex, PS2::CControllerInfo::BUTTON button)
{
	m_inputManager = inputManager;
	m_padIndex = padIndex;
	m_button = button;
	m_buttonName = QString::fromUtf8(text);
	m_qtKeyInputProvider = qtKeyInputProvider;
	m_qtMouseInputProvider = qtMouseInputProvider;
	ui->bindinglabel->setText(m_bindingText.arg(m_buttonName));

	m_providersOverrideConnection = m_inputManager->OverrideInputEventHandler([this](auto target, auto value) { this->onInputEvent(target, value); });
	connect(this, SIGNAL(countdownComplete()), this, SLOT(confirmBinding()));
}

static bool IsAxisIdle(uint32 value)
{
	uint32 triggerRange = (255 * 20) / 100;
	uint32 triggerVal1 = 255 - triggerRange;
	uint32 triggerVal2 = 0 + triggerRange;
	return (value < triggerVal1) && (value > triggerVal2);
}

void InputEventSelectionDialog::onInputEvent(const BINDINGTARGET& target, uint32 value)
{
	auto setSelection =
	    [this](CInputBindingManager::BINDINGTYPE bindingType, const auto& target) {
		    m_selectedTarget = target;
		    m_bindingType = bindingType;
		    m_state = STATE::SELECTED;
		    auto targetDescription = m_inputManager->GetTargetDescription(target);
		    startCountdown(QString::fromStdString(targetDescription));
	    };

	auto resetSelection =
	    [this]() {
		    m_selectedTarget = BINDINGTARGET();
		    m_state = STATE::WAITING;
		    m_bindingType = CInputBindingManager::BINDING_UNBOUND;
		    cancelCountdown();
	    };

	auto setSelectionSimulatedAxis =
	    [this](const auto& target) {
		    m_selectedTargetSimulatedAxis = target;
		    m_state = STATE::SIMULATEDAXIS_SELECTED;
		    auto targetDescription = m_inputManager->GetTargetDescription(target);
		    startCountdown(QString::fromStdString(targetDescription));
	    };

	auto resetSelectionSimulatedAxis =
	    [this]() {
		    m_selectedTargetSimulatedAxis = BINDINGTARGET();
		    m_state = STATE::SIMULATEDAXIS_WAITING;
		    cancelCountdown();
	    };

	switch(m_state)
	{
	case STATE::WAITING:
		//Check if we've pressed something
		if(PS2::CControllerInfo::IsAxis(m_button))
		{
			switch(target.keyType)
			{
			case BINDINGTARGET::KEYTYPE::BUTTON:
				if(value != 0)
				{
					setSelection(CInputBindingManager::BINDING_SIMULATEDAXIS, target);
				}
				break;
			case BINDINGTARGET::KEYTYPE::AXIS:
				if(!IsAxisIdle(value))
				{
					setSelection(CInputBindingManager::BINDING_SIMPLE, target);
				}
				break;
			}
		}
		else
		{
			switch(target.keyType)
			{
			case BINDINGTARGET::KEYTYPE::BUTTON:
				if(value != 0)
				{
					setSelection(CInputBindingManager::BINDING_SIMPLE, target);
				}
				break;
			case BINDINGTARGET::KEYTYPE::AXIS:
				if(!IsAxisIdle(value))
				{
					setSelection(CInputBindingManager::BINDING_SIMPLE, target);
				}
				break;
			case BINDINGTARGET::KEYTYPE::POVHAT:
				if(value < BINDINGTARGET::POVHAT_MAX)
				{
					m_bindingValue = value;
					setSelection(CInputBindingManager::BINDING_POVHAT, target);
				}
			}
		}
		break;
	case STATE::SELECTED:
		if(target != m_selectedTarget) break;
		switch(target.keyType)
		{
		case BINDINGTARGET::KEYTYPE::BUTTON:
			if(value == 0)
			{
				resetSelection();
			}
			break;
		case BINDINGTARGET::KEYTYPE::AXIS:
			if(IsAxisIdle(value))
			{
				resetSelection();
			}
			break;
		case BINDINGTARGET::KEYTYPE::POVHAT:
			if(m_bindingValue != value)
			{
				resetSelection();
			}
		}
		break;
	case STATE::SIMULATEDAXIS_WAITING:
		if((target.keyType == BINDINGTARGET::KEYTYPE::BUTTON) && (value != 0))
		{
			setSelectionSimulatedAxis(target);
		}
		break;
	case STATE::SIMULATEDAXIS_SELECTED:
		if(target != m_selectedTargetSimulatedAxis) break;
		if(value == 0)
		{
			resetSelectionSimulatedAxis();
		}
		break;
	}
}

void InputEventSelectionDialog::keyPressEvent(QKeyEvent* ev)
{
	if(ev->key() == Qt::Key_Escape)
	{
		reject();
		return;
	}
	if(ev->isAutoRepeat()) return;
	m_qtKeyInputProvider->OnKeyPress(ev->key());
}

void InputEventSelectionDialog::keyReleaseEvent(QKeyEvent* ev)
{
	if(ev->isAutoRepeat()) return;
	m_qtKeyInputProvider->OnKeyRelease(ev->key());
}

void InputEventSelectionDialog::mousePressEvent(QMouseEvent* ev)
{
	m_qtMouseInputProvider->OnMousePress(ev->button());
}

void InputEventSelectionDialog::mouseReleaseEvent(QMouseEvent* ev)
{
	m_qtMouseInputProvider->OnMouseRelease(ev->button());
}

void InputEventSelectionDialog::handleStartCountdown(QString bindingDesc)
{
	m_countdownRemain = COUNTDOWN_SECS - 1;
	static_assert(COUNTDOWN_SECS >= 1, "COUNTDOWN_SECS must be at least 1");
	ui->countdownlabel->setText(m_countingText.arg(m_countdownRemain));
	ui->selectedbuttonlabel->setText(m_selectedkeyText.arg(bindingDesc));
	m_countdownTimer->start(1000);
}

void InputEventSelectionDialog::handleCancelCountdown()
{
	m_countdownTimer->stop();
	ui->countdownlabel->setText(m_countingText.arg(COUNTDOWN_SECS));
	ui->selectedbuttonlabel->setText(m_selectedkeyText.arg(QString("None")));
}

void InputEventSelectionDialog::updateCountdown()
{
	if(m_countdownRemain == 0)
	{
		m_countdownTimer->stop();
		countdownComplete();
	}
	else
	{
		m_countdownRemain--;
		ui->countdownlabel->setText(m_countingText.arg(m_countdownRemain));
	}
}

void InputEventSelectionDialog::confirmBinding()
{
	switch(m_bindingType)
	{
	case CInputBindingManager::BINDING_SIMPLE:
		m_inputManager->SetSimpleBinding(m_padIndex, m_button, m_selectedTarget);
		accept();
		break;
	case CInputBindingManager::BINDING_POVHAT:
		m_inputManager->SetPovHatBinding(m_padIndex, m_button, m_selectedTarget, m_bindingValue);
		accept();
		break;
	case CInputBindingManager::BINDING_SIMULATEDAXIS:
		if(m_state == STATE::SELECTED)
		{
			m_state = STATE::SIMULATEDAXIS_WAITING;
			ui->countdownlabel->setText(m_countingText.arg(COUNTDOWN_SECS));
			ui->bindinglabel->setText(m_nextbindingText.arg(m_buttonName));
			ui->selectedbuttonlabel->setText(m_selectedkeyText.arg(QString("None")));
		}
		else if(m_state == STATE::SIMULATEDAXIS_SELECTED)
		{
			m_inputManager->SetSimulatedAxisBinding(m_padIndex, m_button, m_selectedTarget, m_selectedTargetSimulatedAxis);
			accept();
		}
		break;
	}
}