/*
 * Copyright 2015-2016, Rene Gollent, rene@gollent.com.
 * Distributed under the terms of the MIT License.
 */
#include "SignalDispositionEditWindow.h"

#include <signal.h>

#include <Button.h>
#include <LayoutBuilder.h>
#include <MenuField.h>

#include <AutoDeleter.h>
#include <AutoLocker.h>

#include "AppMessageCodes.h"
#include "SignalDispositionMenu.h"
#include "SignalDispositionTypes.h"
#include "Team.h"
#include "UiUtils.h"
#include "UserInterface.h"


enum {
	MSG_SELECTED_SIGNAL_CHANGED 		= 'ssic',
	MSG_SELECTED_DISPOSITION_CHANGED 	= 'sdic',
	MSG_SAVE_SIGNAL_DISPOSITION 		= 'ssid'
};


SignalDispositionEditWindow::SignalDispositionEditWindow(::Team* team,
	int32 signal, UserInterfaceListener* listener, BHandler* target)
	:
	BWindow(BRect(), "Edit signal disposition", B_FLOATING_WINDOW,
		B_AUTO_UPDATE_SIZE_LIMITS | B_CLOSE_ON_ESCAPE),
	fTeam(team),
	fListener(listener),
	fEditMode(signal > 0),
	fCurrentSignal(signal),
	fSaveButton(NULL),
	fCancelButton(NULL),
	fSignalSelectionField(NULL),
	fDispositionSelectionField(NULL),
	fTarget(target)
{
}


SignalDispositionEditWindow::~SignalDispositionEditWindow()
{
	BMessenger(fTarget).SendMessage(MSG_SIGNAL_DISPOSITION_EDIT_WINDOW_CLOSED);
}


SignalDispositionEditWindow*
SignalDispositionEditWindow::Create(::Team* team, int32 signal,
	UserInterfaceListener* listener, BHandler* target)
{
	SignalDispositionEditWindow* self = new SignalDispositionEditWindow(
		team, signal, listener, target);

	try {
		self->_Init();
	} catch (...) {
		delete self;
		throw;
	}

	return self;

}

void
SignalDispositionEditWindow::MessageReceived(BMessage* message)
{
	switch (message->what) {
		case MSG_SELECTED_SIGNAL_CHANGED:
		{
			int32 signal;
			if (message->FindInt32("signal", &signal) == B_OK)
				fCurrentSignal = signal;
			break;
		}
		case MSG_SELECTED_DISPOSITION_CHANGED:
		{
			int32 disposition;
			if (message->FindInt32("disposition", &disposition) == B_OK)
				fCurrentDisposition = disposition;
			break;
		}
		case MSG_SAVE_SIGNAL_DISPOSITION:
		{
			fListener->SetCustomSignalDispositionRequested(fCurrentSignal,
				fCurrentDisposition);
			// fall through
		}
		case B_CANCEL:
			Quit();
			break;

		default:
			BWindow::MessageReceived(message);
			break;
	}

}


void
SignalDispositionEditWindow::Show()
{
	CenterOnScreen();
	BWindow::Show();
}


void
SignalDispositionEditWindow::_Init()
{
	SignalDispositionMenu* menu = new SignalDispositionMenu("dispositionMenu",
		new BMessage(MSG_SELECTED_DISPOSITION_CHANGED));

	BLayoutBuilder::Group<>(this, B_VERTICAL)
		.SetInsets(B_USE_DEFAULT_SPACING)
		.AddGroup(B_HORIZONTAL)
			.Add((fSignalSelectionField = new BMenuField("Signal:",
				_BuildSignalSelectionMenu())))
			.Add((fDispositionSelectionField = new BMenuField("Disposition:",
				menu)))
		.End()
		.AddGroup(B_HORIZONTAL)
			.AddGlue()
			.Add((fSaveButton = new BButton("Save",
				new BMessage(MSG_SAVE_SIGNAL_DISPOSITION))))
			.Add((fCancelButton = new BButton("Cancel",
				new BMessage(B_CANCEL))))
		.End()
	.End();

	fSignalSelectionField->Menu()->SetLabelFromMarked(true);
	fSignalSelectionField->Menu()->SetTargetForItems(this);
	menu->SetLabelFromMarked(true);
	menu->SetTargetForItems(this);

	AutoLocker< ::Team> teamLocker(fTeam);
	_UpdateState();

	// if we're editing an existing row, don't allow changing the signal
	// selection
	if (fEditMode)
		fSignalSelectionField->SetEnabled(false);


}


BMenu*
SignalDispositionEditWindow::_BuildSignalSelectionMenu()
{
	BMenu* menu = new BMenu("signals");
	BMenuItem* item;

	#undef ADD_SIGNAL_MENU_ITEM
	#define ADD_SIGNAL_MENU_ITEM(x)										\
		menu->AddItem((item = new BMenuItem(#x, new BMessage(			\
			MSG_SELECTED_SIGNAL_CHANGED))));							\
		item->Message()->AddInt32("signal", x);

	ADD_SIGNAL_MENU_ITEM(SIGHUP)
	ADD_SIGNAL_MENU_ITEM(SIGINT)
	ADD_SIGNAL_MENU_ITEM(SIGQUIT)
	ADD_SIGNAL_MENU_ITEM(SIGILL)
	ADD_SIGNAL_MENU_ITEM(SIGCHLD)
	ADD_SIGNAL_MENU_ITEM(SIGABRT)
	ADD_SIGNAL_MENU_ITEM(SIGPIPE)
	ADD_SIGNAL_MENU_ITEM(SIGFPE)
	ADD_SIGNAL_MENU_ITEM(SIGKILL)
	ADD_SIGNAL_MENU_ITEM(SIGSTOP)
	ADD_SIGNAL_MENU_ITEM(SIGSEGV)
	ADD_SIGNAL_MENU_ITEM(SIGCONT)
	ADD_SIGNAL_MENU_ITEM(SIGTSTP)
	ADD_SIGNAL_MENU_ITEM(SIGALRM)
	ADD_SIGNAL_MENU_ITEM(SIGTERM)
	ADD_SIGNAL_MENU_ITEM(SIGTTIN)
	ADD_SIGNAL_MENU_ITEM(SIGTTOU)
	ADD_SIGNAL_MENU_ITEM(SIGUSR1)
	ADD_SIGNAL_MENU_ITEM(SIGUSR2)
	ADD_SIGNAL_MENU_ITEM(SIGWINCH)
	ADD_SIGNAL_MENU_ITEM(SIGKILLTHR)
	ADD_SIGNAL_MENU_ITEM(SIGTRAP)
	ADD_SIGNAL_MENU_ITEM(SIGPOLL)
	ADD_SIGNAL_MENU_ITEM(SIGPROF)
	ADD_SIGNAL_MENU_ITEM(SIGSYS)
	ADD_SIGNAL_MENU_ITEM(SIGURG)
	ADD_SIGNAL_MENU_ITEM(SIGVTALRM)
	ADD_SIGNAL_MENU_ITEM(SIGXCPU)
	ADD_SIGNAL_MENU_ITEM(SIGXFSZ)
	ADD_SIGNAL_MENU_ITEM(SIGBUS)

	BString signalName;
	for (int32 i = SIGRTMIN; i <= SIGRTMAX; i++) {
		menu->AddItem((item = new BMenuItem(UiUtils::SignalNameToString(i,
					signalName), new BMessage(MSG_SELECTED_SIGNAL_CHANGED))));
		item->Message()->AddInt32("signal", i);
	}

	return menu;
}


void
SignalDispositionEditWindow::_UpdateState()
{
	if (fCurrentSignal <= 0)
		fCurrentSignal = SIGHUP;

	fSignalSelectionField->Menu()->ItemAt(fCurrentSignal - 1)->SetMarked(true);

	fCurrentDisposition = fTeam->SignalDispositionFor(fCurrentSignal);
	fDispositionSelectionField->Menu()->ItemAt(fCurrentDisposition)->SetMarked(
		true);
}