/*
 * Copyright 2014 Freeman Lou <freemanlou2430@yahoo.com>
 * All rights reserved. Distributed under the terms of the MIT license.
 */
#include "VirtualKeyboardWindow.h"

#include <Directory.h>
#include <Entry.h>
#include <FindDirectory.h>
#include <GroupLayoutBuilder.h>
#include <ListView.h>
#include <Locale.h>
#include <Menu.h>
#include <MenuItem.h>
#include <Path.h>
#include <Screen.h>

#include "KeyboardLayoutView.h"
#include "KeymapListItem.h"

static const uint32 kMsgMenuFontChange = 'mMFC';

static int
compare_key_list_items(const void* a, const void* b)
{
        KeymapListItem* item1 = *(KeymapListItem**)a;
        KeymapListItem* item2 = *(KeymapListItem**)b;

        return BLocale::Default()->StringCompare(item1->Text(), item2->Text());
}


VirtualKeyboardWindow::VirtualKeyboardWindow(BInputServerDevice* dev)
        :
        BWindow(BRect(0, 0, 0, 0), "Virtual Keyboard", B_NO_BORDER_WINDOW_LOOK,
                B_FLOATING_ALL_WINDOW_FEEL, B_WILL_ACCEPT_FIRST_CLICK | B_AVOID_FOCUS),
        fDevice(dev)
{
        BScreen screen;
        BRect screenRect(screen.Frame());
        ScreenChanged(screenRect, screen.ColorSpace());

        SetLayout(new BGroupLayout(B_VERTICAL));

        // Add to an options window later, use as list for now
        fMapListView = new BListView("Maps");
        fFontMenu = new BMenu("Font");
        fLayoutMenu = new BMenu("Layout");

        _LoadMaps();
        _LoadLayouts(fLayoutMenu);
        _LoadFonts();

        fCurrentKeymap.SetToCurrent();

        fKeyboardView = new KeyboardLayoutView("Keyboard", fDevice);
        fKeyboardView->GetKeyboardLayout()->SetDefault();
        fKeyboardView->SetKeymap(&fCurrentKeymap);

        AddChild(BGroupLayoutBuilder(B_VERTICAL).Add(fKeyboardView));
}


void
VirtualKeyboardWindow::_LoadLayouts(BMenu* menu)
{
        directory_which dataDirectories[] = {
                B_USER_NONPACKAGED_DATA_DIRECTORY,
                B_USER_DATA_DIRECTORY,
                B_SYSTEM_NONPACKAGED_DIRECTORY,
                B_SYSTEM_DATA_DIRECTORY,
        };

        for (uint i = 0; i < sizeof(dataDirectories) / sizeof(directory_which); i++) {
                BPath path;
                if (find_directory(dataDirectories[i], &path) != B_OK)
                        continue;

                path.Append("KeyboardLayouts");

                BDirectory directory;
                if (directory.SetTo(path.Path()) == B_OK)
                        _LoadLayoutMenu(menu, directory);
        }
}


void
VirtualKeyboardWindow::_LoadLayoutMenu(BMenu* menu, BDirectory directory)
{
        entry_ref ref;

        while (directory.GetNextRef(&ref) == B_OK) {
                if (menu->FindItem(ref.name) != NULL)
                        continue;

                BDirectory subdirectory;
                subdirectory.SetTo(&ref);
                if (subdirectory.InitCheck() == B_OK) {
                        BMenu* submenu = new BMenu(ref.name);
                        _LoadLayoutMenu(submenu, subdirectory);
                        menu->AddItem(submenu);
                } else {
                        // BMessage* message = new BMessage(kChangeKeyboardLayout);
                        // message->AddRed("ref",&ref);
                        menu->AddItem(new BMenuItem(ref.name, NULL));
                }
        }
}


void
VirtualKeyboardWindow::_LoadMaps()
{
        BPath path;
        if (find_directory(B_SYSTEM_DATA_DIRECTORY, &path) != B_OK)
                return;

        path.Append("Keymaps");
        BDirectory directory;
        entry_ref ref;

        if (directory.SetTo(path.Path()) == B_OK) {
                while (directory.GetNextRef(&ref) == B_OK) {
                        fMapListView->AddItem(new KeymapListItem(ref));
                }
        }
        fMapListView->SortItems(&compare_key_list_items);
}


void
VirtualKeyboardWindow::_LoadFonts()
{
        int32 numFamilies = count_font_families();
        font_family family, currentFamily;
        font_style currentStyle;
        uint32 flags;

        be_plain_font->GetFamilyAndStyle(&currentFamily, &currentStyle);

        for (int32 i = 0; i < numFamilies; i++) {
                if (get_font_family(i, &family, &flags) == B_OK) {
                        BMenuItem* item = new BMenuItem(family, NULL);  // new BMessage(kMsgMenuFontChanged));
                        fFontMenu->AddItem(item);
                        if (!strcmp(family, currentFamily))
                                item->SetMarked(true);
                }
        }
}


void
VirtualKeyboardWindow::MessageReceived(BMessage* message)
{
        if (message->what == kKeymapChange) {
                fCurrentKeymap.SetToCurrent();
                fKeyboardView->SetKeymap(&fCurrentKeymap);
        }
        BWindow::MessageReceived(message);
}


void
VirtualKeyboardWindow::ScreenChanged(BRect screenRect, color_space depth)
{
        ResizeTo(screenRect.Width(), screenRect.Height() / 3);
        MoveTo(0, screenRect.Height() - screenRect.Height() / 3);
}