#include "AddPrinterDlg.h"

#include <GroupLayoutBuilder.h>

#include "PrinterCap.h"
#include "PrinterData.h"


enum MSGS {
        kMsgCancel = 1,
        kMsgOK,
        kMsgProtocolClassChanged,
};


ProtocolClassItem::ProtocolClassItem(const ProtocolClassCap* cap)
        : BStringItem(cap->fLabel.c_str())
        , fProtocolClassCap(cap)
{
}


int
ProtocolClassItem::GetProtocolClass() const
{
        return fProtocolClassCap->fProtocolClass;
}

const char *
ProtocolClassItem::GetDescription() const
{
        return fProtocolClassCap->fDescription.c_str();
}


AddPrinterView::AddPrinterView(PrinterData* printerData,
        const PrinterCap* printerCap)
        :
        BView("addPrinter", B_WILL_DRAW),
        fPrinterData(printerData),
        fPrinterCap(printerCap)
{
}


AddPrinterView::~AddPrinterView()
{
}


void
AddPrinterView::AttachedToWindow()
{
        // protocol class box
        BBox* protocolClassBox = new BBox("protocolClass");
        protocolClassBox->SetLabel("Protocol classes:");

        // protocol class
        fProtocolClassList = new BListView("protocolClassList");
        fProtocolClassList->SetExplicitMinSize(BSize(500, 200));
        BScrollView* protocolClassScroller = new BScrollView(
                "protocolClassListScroller",
                fProtocolClassList,
                0,
                false,
                true,
                B_NO_BORDER);
        fProtocolClassList->SetSelectionMessage(
                new BMessage(kMsgProtocolClassChanged));
        fProtocolClassList->SetTarget(this);

        protocolClassBox->AddChild(protocolClassScroller);

        int count = fPrinterCap->CountCap(PrinterCap::kProtocolClass);
        ProtocolClassCap **protocolClasses =
                (ProtocolClassCap **)fPrinterCap->GetCaps(PrinterCap::kProtocolClass);
        while (count--) {
                const ProtocolClassCap *protocolClass = *protocolClasses;

                BStringItem* item = new ProtocolClassItem(protocolClass);
                fProtocolClassList->AddItem(item);
                if (protocolClass->fIsDefault) {
                        int index = fProtocolClassList->IndexOf(item);
                        fProtocolClassList->Select(index);
                }
                protocolClasses ++;
        }

        // description of protocol class box
        BBox* descriptionBox = new BBox("descriptionBox");
        descriptionBox->SetLabel("Description:");

        // description of protocol class
        fDescription = new BTextView("description");
        fDescription->SetExplicitMinSize(BSize(200, 200));
        fDescription->SetViewColor(descriptionBox->ViewColor());
        BScrollView* descriptionScroller = new BScrollView("descriptionScroller",
                        fDescription,
                        0,
                        false,
                        true,
                        B_NO_BORDER);
        fDescription->MakeEditable(false);

        descriptionBox->AddChild(descriptionScroller);

        // separator line
        BBox *separator = new BBox("separator");
        separator->SetExplicitMaxSize(BSize(B_SIZE_UNLIMITED, 1));

        // buttons
        BButton* cancel = new BButton("cancel", "Cancel",
                new BMessage(kMsgCancel));
        BButton* ok = new BButton("ok", "OK", new BMessage(kMsgOK));
        ok->MakeDefault(true);

        SetLayout(new BGroupLayout(B_VERTICAL));
        AddChild(BGroupLayoutBuilder(B_VERTICAL, 0)
                .Add(protocolClassBox)
                .Add(descriptionBox)
                .AddGlue()
                .Add(separator)
                .AddGroup(B_HORIZONTAL, 10, 1.0f)
                        .AddGlue()
                        .Add(cancel)
                        .Add(ok)
                .End()
                .SetInsets(0, 0, 0, 0)
        );


        // update description
        BMessage updateDescription(kMsgProtocolClassChanged);
        MessageReceived(&updateDescription);
}


ProtocolClassItem*
AddPrinterView::CurrentSelection()
{
        int selected = fProtocolClassList->CurrentSelection();
        if (selected >= 0) {
                return (ProtocolClassItem*)fProtocolClassList->ItemAt(selected);
        }
        return NULL;
}


void
AddPrinterView::MessageReceived(BMessage* msg)
{
        if (msg->what == kMsgProtocolClassChanged) {
                ProtocolClassItem *item = CurrentSelection();
                if (item != NULL) {
                        fDescription->SetText(item->GetDescription());
                }
        } else {
                BView::MessageReceived(msg);
        }
}


void
AddPrinterView::Save()
{
        ProtocolClassItem* item = CurrentSelection();
        if (item != NULL) {
                fPrinterData->SetProtocolClass(item->GetProtocolClass());
                fPrinterData->Save();
        }
}


AddPrinterDlg::AddPrinterDlg(PrinterData* printerData,
        const PrinterCap *printerCap)
        :
        DialogWindow(BRect(100, 100, 120, 120),
                "Add printer", B_TITLED_WINDOW_LOOK, B_MODAL_APP_WINDOW_FEEL,
                B_NOT_MINIMIZABLE | B_NOT_ZOOMABLE | B_ASYNCHRONOUS_CONTROLS
                        | B_AUTO_UPDATE_SIZE_LIMITS)
{
        SetResult(B_ERROR);

        fAddPrinterView = new AddPrinterView(printerData, printerCap);
        SetLayout(new BGroupLayout(B_VERTICAL));
        AddChild(BGroupLayoutBuilder(B_VERTICAL, 0)
                .Add(fAddPrinterView)
                .SetInsets(10, 10, 10, 10)
        );
}


void
AddPrinterDlg::MessageReceived(BMessage* msg)
{
        switch (msg->what) {
                case kMsgOK:
                        fAddPrinterView->Save();
                        SetResult(B_NO_ERROR);
                        PostMessage(B_QUIT_REQUESTED);
                        break;

                case kMsgCancel:
                        PostMessage(B_QUIT_REQUESTED);
                        break;

                default:
                        DialogWindow::MessageReceived(msg);
                        break;
        }
}