/*
* Copyright 2010, Haiku. All rights reserved.
* Distributed under the terms of the MIT License.
*
* Authors:
*		Ithamar R. Adema <ithamar.adema@team-embedded.nl>
*		Michael Pfeiffer
*/


#include "SelectPrinterDialog.h"

#include <GroupLayout.h>
#include <GroupLayoutBuilder.h>

#include <Button.h>
#include <Debug.h>
#include <Directory.h>
#include <Entry.h>
#include <ListView.h>
#include <Message.h>
#include <Path.h>
#include <ScrollBar.h>
#include <ScrollView.h>
#include <String.h>
#include <StringItem.h>

#include "GPBinding.h"


enum {
	kMsgCancel = 'stop',
	kMsgOK = 'okok',

	kMsgManufacturerSelected = 'msel',
	kMsgPrinterSelected = 'psel',
};


class StringValueItem : public BStringItem {
public:
			StringValueItem(const BString& text, const BString& value)
				:
				BStringItem(text.String()),
				fValue(value)
				{
				}

	BString	fValue;
};


SelectPrinterDialog::SelectPrinterDialog(GPData* data)
	:
	DialogWindow(BRect(100, 100, 500, 500),
		"Select Printer", B_TITLED_WINDOW_LOOK, B_MODAL_APP_WINDOW_FEEL,
		B_NOT_MINIMIZABLE | B_NOT_ZOOMABLE | B_ASYNCHRONOUS_CONTROLS
			| B_CLOSE_ON_ESCAPE),
	fData(data)
{
	SetResult(B_ERROR);

	BButton* ok;
	BButton* cancel;

	ok = new BButton("btn:ok", "OK", new BMessage(kMsgOK));
	ok->MakeDefault(true);
	ok->SetEnabled(false);
	fOKButton = ok;

	cancel = new BButton("btn:cancel", "Cancel", new BMessage(kMsgCancel));

	BScrollView* manuScroller, *printerScroller;
	fManufacturersListView = new BListView("olv:manufacturers");
	manuScroller = new BScrollView("scr:manufacturers", fManufacturersListView,
		0, false, true);
	fPrintersListView = new BListView("olv:printers");
	printerScroller = new BScrollView("scr:printers", fPrintersListView, 0,
		false, true);

	fPrintersListView->SetSelectionMessage(new BMessage(kMsgPrinterSelected));
	fManufacturersListView->SetSelectionMessage(
		new BMessage(kMsgManufacturerSelected));
	PopulateManufacturers();

	// Build the layout
	SetLayout(new BGroupLayout(B_VERTICAL));

	AddChild(BGroupLayoutBuilder(B_VERTICAL, 10)
		.AddGroup(B_HORIZONTAL, 5, 1.0f)
			.Add(manuScroller)
			.Add(printerScroller)
		.End()
		.AddGroup(B_HORIZONTAL, 5, 2.0f)
			.AddGlue()
			.Add(cancel)
			.Add(ok)
		.End()
		.SetInsets(10, 10, 10, 10)
	);
}


void
SelectPrinterDialog::PopulateManufacturers()
{
	BMessage manufacturers;
	GPBinding binding;
	binding.GetPrinterManufacturers(manufacturers);

	BString id;
	BString displayName;
	for (int32 index = 0; binding.ExtractManufacturer(manufacturers, index, id,
		displayName); index ++) {
		fManufacturersListView->AddItem(new StringValueItem(id, displayName));
	}
}


void
SelectPrinterDialog::PopulatePrinters()
{
	BString manufacturer = GetSelectedItemValue(fManufacturersListView).String();
	if (manufacturer.Length() == 0)
		return;

	BMessage models;
	GPBinding binding;
	binding.GetPrinterModels(manufacturer, models);

	BString displayName;
	BString driver;
	for (int32 index = 0; binding.ExtractModel(models, index, displayName, driver);
		index ++) {
		StringValueItem* item = new StringValueItem(displayName, driver);
		fPrintersListView->AddItem(item);
	}
}


BString
SelectPrinterDialog::GetSelectedItemValue(BListView* listView)
{
	int32 index = listView->CurrentSelection();

	// Bail out if no manufacturer is selected
	if (index < 0)
		return "";
	BListItem* item = listView->ItemAt(index);
	StringValueItem* valueItem = dynamic_cast<StringValueItem*>(item);
	ASSERT(valueItem != NULL);

	return valueItem->fValue;
}

void
SelectPrinterDialog::PrinterSelected()
{
	int32 index = fPrintersListView->CurrentSelection();
	fOKButton->SetEnabled(index >= 0);
}


void
SelectPrinterDialog::Save()
{
	BString driverName = GetSelectedItemValue(fPrintersListView);
	fData->fGutenprintDriverName = driverName;
	fData->Save();
}


void 
SelectPrinterDialog::MessageReceived(BMessage* msg)
{
	switch (msg->what) {
		case kMsgManufacturerSelected:
			fPrintersListView->MakeEmpty();
			PopulatePrinters();
			PrinterSelected();
			break;
		case kMsgPrinterSelected:
			PrinterSelected();
			break;
		case kMsgOK:
			Save();
			SetResult(B_NO_ERROR);
			PostMessage(B_QUIT_REQUESTED);
			break;
		case kMsgCancel:
			PostMessage(B_QUIT_REQUESTED);
			break;
		default:
			DialogWindow::MessageReceived(msg);
			break;
	}
}