/*
 * pawscontrolwindow.cpp - Author: Andrew Craig
 *
 * Copyright (C) 2003 Atomic Blue (info@planeshift.it, http://www.atomicblue.org)
 *
 *
 * 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 (version 2 of the License)
 * 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, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 *
 */
//////////////////////////////////////////////////////////////////////

#include <psconfig.h>

// CS INCLUDES
#include <csgeom/vector3.h>
#include <csutil/xmltiny.h>
#include <iutil/vfs.h>
#include <iutil/objreg.h>

// COMMON INCLUDES

// CLIENT INCLUDES
#include "globals.h"
#include "util/psxmlparser.h"
#include <iutil/cfgmgr.h>
#include <iutil/csinput.h>

// PAWS INCLUDES
#include "pawscontrolwindow.h"
#include "pawsconfigwindow.h"
#include "psmainwidget.h"
#include "paws/pawstextbox.h"
#include "paws/pawsbutton.h"
#include "paws/pawsmanager.h"
#include "paws/pawsyesnobox.h"

#include "net/cmdhandler.h"
#include "net/msghandler.h"

//////////////////////////////////////////////////////////////////////
// GUI BUTTONS NOTE: ONLY THOSE WHO NEEDS SPECIAL TREATMENT SHOULD BE
// PLACED HERE!
//////////////////////////////////////////////////////////////////////
#define CONTROL_MINIUP    10
#define CONTROL_MINIDOWN  20
#define CONTROL_QUIT      100
#define CONTROL_GUILD     500
#define CONTROL_QUEST     600

#define IMAGE_SIZE 52
#define ICONS 14
//////////////////////////////////////////////////////////////////////


//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

pawsControlWindow::pawsControlWindow()
{
    hidden = false;
    style = 0;
    alwaysResize = true;
}

pawsControlWindow::~pawsControlWindow()
{
    visible = true; // Force visibility on next load
    if (hidden) Toggle(); // Unhide so we get the right sizes saved

    csRef<iConfigFile> file = psengine->GetConfig()->LookupDomain(PawsManager::GetSingleton().GetConfigFile());
    file->SetInt("PlaneShift.GUI.ControlWindow.CurrentStyle", style);
    file->Save();
}

bool pawsControlWindow::PostSetup()
{
    SetAlwaysOnTop(true);

    AddWindow( "InventoryWindow" ,   "InventoryButton" );
    AddWindow( "ConfigWindow" ,      "OptionsButton" );
    AddWindow( "SpellBookWindow" ,   "SpellBookButton" );
    AddWindow( "InfoWindow" ,        "InfoButton" );
    AddWindow( "HelpWindow" ,        "HelpButton" );
    AddWindow( "ShortcutWindow" ,    "ShortcutButton" );        
    AddWindow( "BuddyWindow" ,       "BuddyButton" );
    AddWindow( "GroupWindow" ,       "GroupButton" );
    AddWindow( "PetitionWindow" ,    "PetitionButton" );
    AddWindow( "ChatWindow" ,        "ChatButton" );
    AddWindow( "SkillWindow" ,       "SkillsButton" );
    AddWindow( "QuestNotebook" ,     "QuestButton" );
    AddWindow( "GuildWindow" ,       "GuildButton" );

    keyboard = CS_QUERY_REGISTRY(PawsManager::GetSingleton().GetObjectRegistry(),iKeyboardDriver);

    //The quit button is a bit special
    //We need to manualy register it
    Icon* icon = new Icon;
    icon->window = NULL;
    icon->theirButton = (pawsButton*)FindWidget("QuitButton");
    icon->orgRes = icon->theirButton->GetBackground();
    icon->IsActive = false;
    icon->IsOver = false;
    buttons.Push(icon);

    csRef<iConfigFile> file = psengine->GetConfig()->LookupDomain(PawsManager::GetSingleton().GetConfigFile());
    int loadStyle = file->GetInt("PlaneShift.GUI.ControlWindow.CurrentStyle", 1);
    for (int i=0; i < loadStyle; i++)
        NextStyle(); // Switch to saved style

    buttonUp = FindWidget("ShowButtonUp");
    buttonDown = FindWidget("ShowButtonDown");
    return true;
}

//show/hide buttons
void pawsControlWindow::Toggle()
{
    
    static int oldW = 0, oldH = 0;
    static int oldMinW = 0, oldMinH = 0;

    for (size_t z = 0; z < children.Length(); z++ )
    {
        if (hidden) children[z]->Show();
        else children[z]->Hide();
    }
    
    if (hidden) 
    {
        buttonUp->Show();
        buttonDown->Hide();  
        this->SetForceSize(oldW,oldH);
        this->SetMinSize(oldMinW,oldMinH);
    }
    else
    {
        oldW = this->ScreenFrame().Width();
        oldH = this->ScreenFrame().Height();
        this->GetMinSize(oldMinW,oldMinH);

        int w = buttonUp->ScreenFrame().Width();
        int h = buttonUp->ScreenFrame().Height();

        this->SetForceSize(w,h);
        this->SetMinSize(w,h);
        buttonUp->Hide();
        buttonDown->Show();
        buttonDown->SetRelativeFramePos(0,0);
    }

    hidden = !hidden;
}

bool pawsControlWindow::OnMouseEnter()
{
    if(keyboard->GetKeyState(CSKEY_SHIFT) && !alwaysResize)
    {
        SetResizeShow(true);
        pawsWidget::OnMouseEnter();
        return true;
    }
    else
        return pawsWidget::OnMouseEnter();
}

bool pawsControlWindow::OnMouseExit()
{
    if(!alwaysResize)
        SetResizeShow(false);

    pawsWidget::OnMouseExit();
    return true;
}

bool pawsControlWindow::OnButtonPressed( int mouseButton, int keyModifier, pawsWidget* reporter )
{
    if(reporter->GetID() == CONTROL_MINIDOWN || reporter->GetID() == CONTROL_MINIUP)
    {
        if (mouseButton == csmbRight)
        { //If the user clicked on the right mouse button, switch to another style
            NextStyle();
            return true;
        }
        else if (mouseButton == csmbMiddle)
        {
            printf("Resetting toolbar to orginal size (%d,%d)\n",orgw,orgh);
            SetRelativeFrameSize(orgw,orgh);
            return true;
        }
    }

    switch ( reporter->GetID() )
    {
        case CONTROL_MINIUP:
        {
            if ( !hidden )
                Toggle();        
            return true; 
        }

        case CONTROL_MINIDOWN:
        {
            if ( hidden )
                Toggle();
            return true; 
        }

        case CONTROL_QUIT:
        {
            HandleQuit();
            return true;
        }
        ////////////////////////////////////////////////////////////////////
        //Special cases that need more than the standard to open correctly
        ////////////////////////////////////////////////////////////////////

        case CONTROL_QUEST:
        {
            if (HandleWindow("QuestNotebook"))
            {
                psUserCmdMessage msg("/quests");
                psengine->GetMsgHandler()->SendMessage(msg.msg);
            }
            return true;
        }
        
        ////////////////////////////////////////////////////////////////////
        // These two are from the confirm window buttons for quiting.
        ////////////////////////////////////////////////////////////////////
        case CONFIRM_YES:
        {
            psengine->QuitClient();            
            return true;
        }
        case CONFIRM_NO:
        {
            PawsManager::GetSingleton().SetModalWidget( 0 );
            reporter->GetParent()->Hide();
            return true;
        }
        ////////////////////////////////////////////////////////////////////

        ////////////////////////////////////////////////////////////////////
        //Default handle for windows
        ////////////////////////////////////////////////////////////////////
        default:
        {
            pawsWidget* wdg = FindWidget(reporter->GetID());
            if (!wdg)
                return false;

            pawsControlledWindow* wnd = FindWindowFromButton(wdg->GetName());
            if (!wnd)
                return false;

            HandleWindow(wnd->GetName());
            return true;
        }
    }
    return false;
}

//Returns true if it made it visible, returns false if it made it invisible or error
bool pawsControlWindow::HandleWindow(csString widgetStr)
{
    pawsWidget* widget = PawsManager::GetSingleton().FindWidget(widgetStr.GetData());
    if (!widget)
    {
        printf("%s isn't loaded\n",widgetStr.GetData());
        return false;
    }

    if ( widget->IsVisible() )
    {
        widget->Hide();
        if ( this->Includes(PawsManager::GetSingleton().GetCurrentFocusedWidget()) )
            PawsManager::GetSingleton().SetCurrentFocusedWidget(NULL);  // Don't leave focus on us after hiding
        return false;
    }
    else
    {
        widget->Show();
        return true;
    }
}

bool pawsControlWindow::HandleWindowName(csString widgetStr)
{
    csString widget;
    widgetStr.Downcase();
    if(widgetStr == "options" )
        widget = "ConfigWindow";
    else if(widgetStr == "stats" || widgetStr =="skills")
        widget = "SkillWindow";  
    else if(widgetStr == "spell book" || widgetStr == "spells")
        widget = "SpellBookWindow";
    else if(widgetStr == "inventory" || widgetStr == "inv")
        widget = "InventoryWindow";
    else if(widgetStr == "help")
        widget = "HelpWindow";
    else if(widgetStr == "buddy")
        widget = "BuddyWindow";
    else if(widgetStr == "status window" || widgetStr == "info")
        widget = "InfoWindow";
    else if(widgetStr == "petition")
        widget = "PetitionWindow";
    else if(widgetStr == "quest")
        widget = "QuestNotebook";
    else if(widgetStr == "gm")
        widget = "GmGUI";
    else if(widgetStr == "shortcut")
        widget = "ShortcutWindow";
    else if(widgetStr == "group")
        widget = "GroupWindow";
    else if(widgetStr == "guild")
        widget = "GuildWindow";
    else if(widgetStr == "glyph")
        widget = "GlyphWindow";
    else if(widgetStr == "bag" || widgetStr == "smallinventory")
        widget = "SmallInventoryWindow";
    else if(widgetStr == "talk" || widgetStr == "chat" || widgetStr == "communications")
        widget = "ChatWindow";
    else if(widgetStr == "quit")
    {
        HandleQuit();
        return true;
    }
    else if(widgetStr == "buy")
    {
        psengine->GetCmdHandler()->Execute("/buy");
        return true;
    }
    if(widget)
        HandleWindow(widget);
    else return false;
    return true;
}

void pawsControlWindow::HandleQuit()
{
    PawsManager::GetSingleton().CreateYesNoBox( "\nDo you really wish to leave Yliakum?", this );
}

bool pawsControlWindow::OnChildMouseEnter(pawsWidget *child)
{
    pawsButton* btn = (pawsButton*)child;
    Icon* icon = GetIcon(child->GetName());

    if (!icon)
        return false;

    csString bg(icon->orgRes); //Get the icon and then get the orignal resource bg
    bg += "_over";

    btn->SetBackground(bg.GetData());
    return true;
}

bool pawsControlWindow::OnChildMouseExit(pawsWidget *child)
{
    pawsButton* btn = (pawsButton*)child;
    Icon* icon = GetIcon(child->GetName());
    if (!icon)
        return false;

    csString bg(icon->orgRes);
    btn->SetBackground(bg.GetData());

    if (GetIcon(child->GetName())->IsActive)
    {
        bg = GetIcon(child->GetName())->orgRes;
        bg += "_active";
        btn->SetBackground(bg.GetData());
    }

    return true;
}

// When a new window registers make a new entry for them in our list.
void pawsControlWindow::Register( pawsControlledWindow* window )
{
    Icon * icon = new Icon;
    icon->window = window;
    icon->theirButton = FindButtonFromWindow( window->GetName() );
    icon->orgRes = icon->theirButton->GetBackground();
    icon->IsActive = false;
    icon->IsOver = false;

    buttons.Push( icon );
}

void pawsControlWindow::AddWindow(csString wndName, csString btnName)
{
    WBName newName;
    pawsButton* btn = (pawsButton*)FindWidget(btnName.GetData());

    if (!btn)
    {
        printf("Couldn't find button %s!\n",btnName.GetData());
        return;
    }

    newName.buttonName = btnName;
    newName.windowName = wndName;
    newName.id = btn->GetID();
    wbs.Push(newName);
}

pawsControlledWindow* pawsControlWindow::FindWindowFromButton(csString btnName)
{
    for (size_t x = 0; x < wbs.Length(); x++ )
    {
        if ( wbs[x].buttonName == btnName )
        {
            pawsWidget* wdg = PawsManager::GetSingleton().FindWidget(wbs[x].windowName.GetData());
            if (!wdg)
            {
                printf("Found %s but FindWidget returned null!\n",btnName.GetData());
                return NULL;
            }

            return (pawsControlledWindow*)wdg;
        }
    }
    return NULL;
}


pawsButton* pawsControlWindow::FindButtonFromWindow(csString wndName)
{
    for (size_t x = 0; x < wbs.Length(); x++ )
    {
        if ( wbs[x].windowName == wndName )
        {
            pawsWidget* wdg = FindWidget(wbs[x].buttonName.GetData());
            if (!wdg)
            {
                printf("Couldn't find window %s!\n",wndName.GetData());
                return NULL;
            }
            return (pawsButton*)wdg;
        }
    }
    return NULL;
}

Icon* pawsControlWindow::GetIcon(csString btnName)
{
    for (size_t x = 0; x < buttons.Length(); x++ )
    {
        if ( !strcmp(buttons[x]->theirButton->GetName(), btnName.GetData()) )
        {
            return buttons[x];
        }
    }
    return NULL;
}

//This is called everytime a window that has a button opens
void pawsControlWindow::WindowOpen(pawsWidget* wnd)
{
    pawsButton* btn = FindButtonFromWindow(wnd->GetName());
    Icon* icon = GetIcon(btn->GetName());

    icon->IsActive = true;

    csString bg(icon->orgRes);
    bg += "_active";
    
    btn->SetBackground(bg.GetData());
}

//This is called everytime a window that has a button closes
void pawsControlWindow::WindowClose(pawsWidget* wnd)
{
    pawsButton* btn = FindButtonFromWindow(wnd->GetName());
    Icon* icon = GetIcon(btn->GetName());
    icon->IsActive = false;   
    btn->SetBackground(icon->orgRes.GetData());
}

void pawsControlWindow::NextStyle()
{
    // We load it each time to make the user able to change the style without the need to
    // restart the client
    //Increase style
    style++;
    csString filename = "/this/data/gui/control_styles.xml";
    if (!psengine->GetVFS()->Exists(filename.GetData()))
    {
        Error2( "Could not find XML: %s",filename.GetData());
        return;
    }

    csRef<iDocument> doc = ParseFile(psengine->GetObjectRegistry(),filename.GetData());
    if (!doc)
    {
        Error2("Error parsing file %s", filename.GetData());
        return;
    }

    csString topNodestr("style");
    topNodestr += style;

    csRef<iDocumentNode> root = doc->GetRoot();
    if (!root)
    {
        Error2("No XML root in %s", filename.GetData());
        return;
    }
    csRef<iDocumentNode> topNode = root->GetNode(topNodestr.GetData());

    //If "style"+style doesn't exist, jump to start
    if (!topNode)
    {
        if (style == 1) //No styles at all :(
            return;

        style=0;
        NextStyle();
        return;
    }

    csRef<iDocumentNodeIterator> iter = topNode->GetNodes();

    //Loop through the XML
    int bwidth=52, bheight=52, mwidth=16, mheight=16;
    float xscale=1.0f, yscale=1.0f;

    // get the current width and height of the buttons (should be the same for all)
    pawsButton* btn = (pawsButton*)FindWidget("QuitButton");
    int boldwidth = btn->ClipRect().Width();
    int boldheight = btn->ClipRect().Height();

    while ( iter->HasNext() )
    {
        csRef<iDocumentNode> node = iter->Next();
        if (!strcmp(node->GetValue(),"button"))
        {
            int x = GetActualWidth(node->GetAttributeValueAsInt("x"));
            int y = GetActualHeight(node->GetAttributeValueAsInt("y"));
            csString name = node->GetAttributeValue("name");

            pawsWidget* wdg = FindWidget(name.GetData());
            if (!wdg)
            {
                Error2("No such widget as %s!",name.GetData());
                continue;
            }
            wdg->SetSize(bwidth, bheight);
            wdg->SetRelativeFramePos(x, y);
        }
        else if (!strcmp(node->GetValue(),"buttons"))
        {
            bwidth = GetActualWidth(node->GetAttributeValueAsInt("width"));
            bheight = GetActualHeight(node->GetAttributeValueAsInt("height"));
            mwidth = GetActualWidth(node->GetAttributeValueAsInt("hide_width"));
            mheight = GetActualHeight(node->GetAttributeValueAsInt("hide_height"));

            // get the multiplier for the button and the widget size
            if (boldwidth != 0 && boldheight != 0)
            {
                xscale = float(boldwidth) / float(bwidth);
                yscale = float(boldheight) / float(bheight);
            }
        }
        else if (!strcmp(node->GetValue(), "bar"))
        {
            orgw = GetActualWidth(node->GetAttributeValueAsInt("width"));
            orgh = GetActualWidth(node->GetAttributeValueAsInt("height"));
            SetRelativeFrameSize(orgw, orgh);
            min_width = GetActualWidth(node->GetAttributeValueAsInt("min_w"));
            min_height = GetActualWidth(node->GetAttributeValueAsInt("min_h"));

            alwaysResize = node->GetAttributeValueAsBool("always_resize", true);
            SetResizeShow(alwaysResize);
        }
    }

    /* Resize the widget according to the proportions of the previous style.
     *
     * When the new styles dimensions exceeds its max it will be downsized to fit.
     * See: pawsWidget::Resize(int deltaX, int deltaY, int flags)
     */
    Resize( int(xscale*orgw-orgw), int(yscale*orgh-orgh), RESIZE_RIGHT | RESIZE_BOTTOM );

    // Set the minimize buttons size
    pawsButton* minUp = (pawsButton*)FindWidget("ShowButtonUp");
    minUp->SetSize(mwidth,mheight);

    pawsButton* minDown = (pawsButton*)FindWidget("ShowButtonDown");
    minDown->SetSize(mwidth,mheight);
}

void pawsControlWindow::Hide()
{
    if (!hidden)
        Toggle();
}

void pawsControlWindow::Show()
{
    if (hidden)
        Toggle();

    pawsWidget::Show();
}

bool pawsControlWindow::Contains( int x, int y )
{
    if(!pawsWidget::Contains(x, y))
        return false;

    // Special case of a resize border
    if(keyboard->GetKeyState(CSKEY_SHIFT) || (ResizeFlags(x,y) && showResize) || alwaysResize)
        return true;

    // Typical case, this window is transparent
    for (size_t z = 0; z < children.Length(); z++ )
    {
        if (children[z]->IsVisible() )
        { 
            if ( children[z]->Contains(x,y) )
                return true;
        }
    }
    return false;
}