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


#include "SelectPPDDlg.h"

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

#include <Button.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 "PPDParser.h"

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

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


class PPDStringItem : public BStringItem {
public:
                        PPDStringItem(const BString& text, const BString& path)
                        :
                        BStringItem(text.String()),
                        fPPDPath(path)
                        {
                        }

        BString fPPDPath;
};


SelectPPDDlg::SelectPPDDlg(PSData* data)
        : DialogWindow(BRect(10, 10, 400, 400),
                "Select PPD", B_TITLED_WINDOW_LOOK, B_MODAL_APP_WINDOW_FEEL,
                B_NOT_MINIMIZABLE | B_NOT_ZOOMABLE | B_ASYNCHRONOUS_CONTROLS
                        | B_CLOSE_ON_ESCAPE),
                fPSData(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(kMsgManuSelected));
        PopulateManufacturers(B_SYSTEM_DATA_DIRECTORY);

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

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


void
SelectPPDDlg::PopulateManufacturers(directory_which data_dir)
{
        char name[1024];
        BDirectory dir;
        BEntry entry;
        BPath path;

        if (find_directory(data_dir, &path) == B_OK
                && path.Append("ppd") == B_OK
                && dir.SetTo(path.Path()) == B_OK) {
                // Got the directory, now scan it
                while (dir.GetNextEntry(&entry) == B_OK)
                        if (entry.IsDirectory() 
                                && entry.GetName(name) == B_OK)
                                fManufacturersListView->AddItem(new BStringItem(name));
        }
}


void
SelectPPDDlg::PopulatePrinters(directory_which data_dir)
{
        int32 idx = fManufacturersListView->CurrentSelection();
        char name[1024];
        BDirectory dir;
        BString manu;
        BEntry entry;
        BPath path;

        // Bail out if no manufacturer is selected
        if (idx < 0)
                return;

        manu = ((BStringItem*)fManufacturersListView->ItemAt(idx))->Text();

        if (find_directory(data_dir, &path) == B_OK
                && path.Append("ppd") == B_OK
                && path.Append(manu) == B_OK
                && dir.SetTo(path.Path()) == B_OK) {
                // Found manufacturer PPD directory, now fill our printer list
                while (dir.GetNextEntry(&entry) == B_OK)
                        if (entry.GetName(name) == B_OK) {
                                PPDParser parser(dir, name);
                                if (parser.InitCheck() == B_OK) {
                                        BString modelName = parser.GetParameter("ModelName");
                                        BPath ppdPath = path;
                                        ppdPath.Append(name);
                                        fPrintersListView->AddItem(new PPDStringItem(modelName,
                                                ppdPath.Path()));
                                }
                        }
        }
}


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


void
SelectPPDDlg::Save()
{
        BString ppdPath;
        int32 idx;

        idx = fPrintersListView->CurrentSelection();
        if (idx >= 0)
                ppdPath = dynamic_cast<PPDStringItem*>
                        (fPrintersListView->ItemAt(idx))->fPPDPath;

        fPSData->fPPD = ppdPath;
        fPSData->Save();
}


void 
SelectPPDDlg::MessageReceived(BMessage* msg)
{
        switch (msg->what) {
                case kMsgManuSelected:
                        fPrintersListView->MakeEmpty();
                        PopulatePrinters(B_SYSTEM_DATA_DIRECTORY);
                        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;
        }
}