/*
 *  The Mana Client
 *  Copyright (C) 2006-2009  The Mana World Development Team
 *  Copyright (C) 2009-2012  The Mana Developers
 *
 *  This file is part of The Mana Client.
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "gui/widgets/dropdown.h"

#include "graphics.h"

#include "gui/gui.h"
#include "gui/sdlinput.h"

#include "gui/widgets/listbox.h"
#include "gui/widgets/scrollarea.h"

#include "resources/theme.h"

#include <guichan/font.hpp>

DropDown::DropDown(gcn::ListModel *listModel):
    gcn::DropDown::DropDown(listModel,
                            new ScrollArea,
                            new ListBox(listModel))
{
    auto &skin = gui->getTheme()->getSkin(SkinType::DropDownFrame);
    setFrameSize(skin.frameSize);
    mPadding = skin.padding;

    setHeight(getFont()->getHeight() + 2 * mPadding);
}

DropDown::~DropDown()
{
    delete mScrollArea;
}

void DropDown::draw(gcn::Graphics* graphics)
{
    const int h = mDroppedDown ? mFoldedUpHeight : getHeight();

    const int alpha = gui->getTheme()->getGuiAlpha();
    gcn::Color faceColor = getBaseColor();
    faceColor.a = alpha;
    auto highlightColor = Theme::getThemeColor(Theme::HIGHLIGHT);
    highlightColor.a = alpha;
    gcn::Color shadowColor = faceColor - 0x303030;
    shadowColor.a = alpha;

    if (mListBox->getListModel() && mListBox->getSelected() >= 0)
    {
        graphics->setFont(getFont());
        graphics->setColor(Theme::getThemeColor(Theme::TEXT));
        graphics->drawText(mListBox->getListModel()->getElementAt(mListBox->getSelected()),
                           mPadding,
                           mPadding);
    }

    if (isFocused())
    {
        graphics->setColor(highlightColor);
        graphics->drawRectangle(
            gcn::Rectangle(mPadding, mPadding, getWidth() - h - mPadding * 2, h - 2 * mPadding));
    }

    drawButton(graphics);

    if (mDroppedDown)
    {
        drawChildren(graphics);

        // Draw two lines separating the ListBox with selected
        // element view.
        graphics->setColor(highlightColor);
        graphics->drawLine(0, h, getWidth(), h);
        graphics->setColor(shadowColor);
        graphics->drawLine(0, h + 1, getWidth(), h + 1);
    }
}

void DropDown::drawFrame(gcn::Graphics *graphics)
{
    const int bs = getFrameSize();

    WidgetState state(this);
    state.width += bs * 2;
    state.height += bs * 2;

    gui->getTheme()->drawSkin(static_cast<Graphics *>(graphics), SkinType::DropDownFrame, state);
}

// Overridden so that we can take mPadding into account
void DropDown::adjustHeight()
{
    const int listBoxHeight = mListBox->getHeight();
    int height = getFont()->getHeight() + 2 * mPadding;

    // The addition/subtraction of 2 compensates for the seperation lines
    // seperating the selected element view and the scroll area.

    if (mDroppedDown && getParent())
    {
        int availableHeight = getParent()->getChildrenArea().height - getY();

        if (listBoxHeight > availableHeight - height - 2)
        {
            mScrollArea->setHeight(availableHeight - height - 2);
            height = availableHeight;
        }
        else
        {
            height += listBoxHeight + 2;
            mScrollArea->setHeight(listBoxHeight);
        }
    }

    setHeight(height);

    mScrollArea->setWidth(getWidth());
    // Resize the ListBox to exactly fit the ScrollArea.
    mListBox->setWidth(mScrollArea->getChildrenArea().width);
    mScrollArea->setPosition(0, 0);
}

void DropDown::drawButton(gcn::Graphics *graphics)
{
    WidgetState state(this);
    if (mDroppedDown)
    {
        state.height = mFoldedUpHeight;
        state.flags |= STATE_SELECTED;
    }
    if (mPushed)
        state.flags |= STATE_HOVERED;

    auto &skin = gui->getTheme()->getSkin(SkinType::DropDownButton);

    // FIXME: Needs support for setting alignment in the theme.
    state.x = state.width - skin.getMinWidth();

    skin.draw(static_cast<Graphics *>(graphics), state);
}

// -- KeyListener notifications
void DropDown::keyPressed(gcn::KeyEvent& keyEvent)
{
    if (keyEvent.isConsumed())
        return;

    const gcn::Key key = keyEvent.getKey();
    const int selectedIndex = getSelected();

    if (key.getValue() == Key::ENTER || key.getValue() == Key::SPACE)
        dropDown();
    else if (key.getValue() == Key::UP)
        setSelected(getSelected() - 1);
    else if (key.getValue() == Key::DOWN)
        setSelected(getSelected() + 1);
    else if (key.getValue() == Key::HOME)
        setSelected(0);
    else if (key.getValue() == Key::END)
        setSelected(mListBox->getListModel()->getNumberOfElements() - 1);
    else
        return;

    keyEvent.consume();

    if (getSelected() != selectedIndex)
        distributeActionEvent();
}

void DropDown::focusLost(const gcn::Event& event)
{
    gcn::DropDown::focusLost(event);
    releaseModalMouseInputFocus();
}

void DropDown::mousePressed(gcn::MouseEvent& mouseEvent)
{
    gcn::DropDown::mousePressed(mouseEvent);

    if (0 <= mouseEvent.getY() && mouseEvent.getY() < getHeight() &&
        mouseEvent.getX() >= 0 && mouseEvent.getX() < getWidth() &&
        mouseEvent.getButton() == gcn::MouseEvent::LEFT && mDroppedDown &&
        mouseEvent.getSource() == mListBox)
    {
        mPushed = false;
        foldUp();
        releaseModalMouseInputFocus();
        distributeActionEvent();
    }
}

void DropDown::mouseWheelMovedUp(gcn::MouseEvent& mouseEvent)
{
    setSelected(getSelected() - 1);
    mouseEvent.consume();
    distributeActionEvent();
}

void DropDown::mouseWheelMovedDown(gcn::MouseEvent& mouseEvent)
{
    setSelected(getSelected() + 1);
    mouseEvent.consume();
    distributeActionEvent();
}

// Overridden to call our version of adjustHeight
void DropDown::dropDown()
{
    if (!mDroppedDown)
    {
        mDroppedDown = true;
        mFoldedUpHeight = getHeight();
        adjustHeight();

        if (getParent())
        {
            getParent()->moveToTop(this);
        }
    }

    mListBox->requestFocus();
}

// Overridden to call our version of adjustHeight
void DropDown::foldUp()
{
    if (mDroppedDown)
    {
        mDroppedDown = false;
        adjustHeight();
        mInternalFocusHandler.focusNone();
    }
}