/**
 * @file llkeyboard.cpp
 * @brief Handler for assignable key bindings
 *
 * $LicenseInfo:firstyear=2001&license=viewerlgpl$
 * Second Life Viewer Source Code
 * Copyright (C) 2010, Linden Research, Inc.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation;
 * version 2.1 of the License only.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 *
 * Linden Research, Inc., 945 Battery Street, San Francisco, CA  94111  USA
 * $/LicenseInfo$
 */

#include "linden_common.h"
#include "indra_constants.h"
#include "llkeyboard.h"

#include "llwindowcallbacks.h"
#include "llgamecontrol.h"

//
// Globals
//

LLKeyboard *gKeyboard = NULL;

//static
std::map<KEY,std::string> LLKeyboard::sKeysToNames;
std::map<std::string,KEY> LLKeyboard::sNamesToKeys;
LLKeyStringTranslatorFunc*  LLKeyboard::mStringTranslator = NULL;   // Used for l10n + PC/Mac/Linux accelerator labeling

//
// Class Implementation
//

LLKeyboard::LLKeyboard() : mCallbacks(NULL)
{
    S32 i;

    // Constructor for LLTimer inits each timer. We want them to
    // be constructed without being initialized, so we shut them down here.
    for (i = 0; i < KEY_COUNT; i++)
    {
        mKeyLevelFrameCount[i] = 0;
        mKeyLevel[i] = false;
        mKeyUp[i]    = false;
        mKeyDown[i]  = false;
        mKeyRepeated[i] = false;
    }

    mInsertMode = LL_KIM_INSERT;
    mCurTranslatedKey = KEY_NONE;
    mCurScanKey = KEY_NONE;

    addKeyName(' ', "Space" );
    addKeyName(KEY_RETURN, "Enter" );
    addKeyName(KEY_LEFT, "Left" );
    addKeyName(KEY_RIGHT, "Right" );
    addKeyName(KEY_UP, "Up" );
    addKeyName(KEY_DOWN, "Down" );
    addKeyName(KEY_ESCAPE, "Esc" );
    addKeyName(KEY_HOME, "Home" );
    addKeyName(KEY_END, "End" );
    addKeyName(KEY_PAGE_UP, "PgUp" );
    addKeyName(KEY_PAGE_DOWN, "PgDn" );
    addKeyName(KEY_F1, "F1" );
    addKeyName(KEY_F2, "F2" );
    addKeyName(KEY_F3, "F3" );
    addKeyName(KEY_F4, "F4" );
    addKeyName(KEY_F5, "F5" );
    addKeyName(KEY_F6, "F6" );
    addKeyName(KEY_F7, "F7" );
    addKeyName(KEY_F8, "F8" );
    addKeyName(KEY_F9, "F9" );
    addKeyName(KEY_F10, "F10" );
    addKeyName(KEY_F11, "F11" );
    addKeyName(KEY_F12, "F12" );
    addKeyName(KEY_TAB, "Tab" );
    addKeyName(KEY_ADD, "Add" );
    addKeyName(KEY_SUBTRACT, "Subtract" );
    addKeyName(KEY_MULTIPLY, "Multiply" );
    addKeyName(KEY_DIVIDE, "Divide" );
    addKeyName(KEY_PAD_DIVIDE, "PAD_DIVIDE" );
    addKeyName(KEY_PAD_LEFT, "PAD_LEFT" );
    addKeyName(KEY_PAD_RIGHT, "PAD_RIGHT" );
    addKeyName(KEY_PAD_DOWN, "PAD_DOWN" );
    addKeyName(KEY_PAD_UP, "PAD_UP" );
    addKeyName(KEY_PAD_HOME, "PAD_HOME" );
    addKeyName(KEY_PAD_END, "PAD_END" );
    addKeyName(KEY_PAD_PGUP, "PAD_PGUP" );
    addKeyName(KEY_PAD_PGDN, "PAD_PGDN" );
    addKeyName(KEY_PAD_CENTER, "PAD_CENTER" );
    addKeyName(KEY_PAD_INS, "PAD_INS" );
    addKeyName(KEY_PAD_DEL, "PAD_DEL" );
    addKeyName(KEY_PAD_RETURN, "PAD_Enter" );
    addKeyName(KEY_BUTTON0, "PAD_BUTTON0" );
    addKeyName(KEY_BUTTON1, "PAD_BUTTON1" );
    addKeyName(KEY_BUTTON2, "PAD_BUTTON2" );
    addKeyName(KEY_BUTTON3, "PAD_BUTTON3" );
    addKeyName(KEY_BUTTON4, "PAD_BUTTON4" );
    addKeyName(KEY_BUTTON5, "PAD_BUTTON5" );
    addKeyName(KEY_BUTTON6, "PAD_BUTTON6" );
    addKeyName(KEY_BUTTON7, "PAD_BUTTON7" );
    addKeyName(KEY_BUTTON8, "PAD_BUTTON8" );
    addKeyName(KEY_BUTTON9, "PAD_BUTTON9" );
    addKeyName(KEY_BUTTON10, "PAD_BUTTON10" );
    addKeyName(KEY_BUTTON11, "PAD_BUTTON11" );
    addKeyName(KEY_BUTTON12, "PAD_BUTTON12" );
    addKeyName(KEY_BUTTON13, "PAD_BUTTON13" );
    addKeyName(KEY_BUTTON14, "PAD_BUTTON14" );
    addKeyName(KEY_BUTTON15, "PAD_BUTTON15" );

    addKeyName(KEY_BACKSPACE, "Backsp" );
    addKeyName(KEY_DELETE, "Del" );
    addKeyName(KEY_SHIFT, "Shift" );
    addKeyName(KEY_CONTROL, "Ctrl" );
    addKeyName(KEY_ALT, "Alt" );
    addKeyName(KEY_HYPHEN, "-" );
    addKeyName(KEY_EQUALS, "=" );
    addKeyName(KEY_INSERT, "Ins" );
    addKeyName(KEY_CAPSLOCK, "CapsLock" );
}


LLKeyboard::~LLKeyboard()
{
    // nothing
}

void LLKeyboard::addKeyName(KEY key, const std::string& name)
{
    sKeysToNames[key] = name;
    std::string nameuc = name;
    LLStringUtil::toUpper(nameuc);
    sNamesToKeys[nameuc] = key;
}

void LLKeyboard::resetKeyDownAndHandle()
{
    MASK mask = currentMask(false);
    for (S32 i = 0; i < KEY_COUNT; i++)
    {
        if (mKeyLevel[i])
        {
            mKeyDown[i] = false;
            mKeyLevel[i] = false;
            mKeyUp[i] = true;
            mCurTranslatedKey = (KEY)i;
            mCallbacks->handleTranslatedKeyUp(i, mask);
        }
    }
    LLGameControl::clearAllKeys();
}

// BUG this has to be called when an OS dialog is shown, otherwise modifier key state
// is wrong because the keyup event is never received by the main window. JC
void LLKeyboard::resetKeys()
{
    S32 i;

    for (i = 0; i < KEY_COUNT; i++)
    {
        if( mKeyLevel[i] )
        {
            mKeyLevel[i] = false;
        }
    }

    for (i = 0; i < KEY_COUNT; i++)
    {
        mKeyUp[i] = false;
    }

    for (i = 0; i < KEY_COUNT; i++)
    {
        mKeyDown[i] = false;
    }

    for (i = 0; i < KEY_COUNT; i++)
    {
        mKeyRepeated[i] = false;
    }
}


bool LLKeyboard::translateKey(const NATIVE_KEY_TYPE os_key, KEY *out_key)
{

    // Only translate keys in the map, ignore all other keys for now
    auto iter = mTranslateKeyMap.find(os_key);
    if (iter == mTranslateKeyMap.end())
    {
        //LL_WARNS() << "Unknown virtual key " << os_key << LL_ENDL;
        *out_key = 0;
        return false;
    }
    else
    {
        *out_key = iter->second;
        return true;
    }
}

LLKeyboard::NATIVE_KEY_TYPE LLKeyboard::inverseTranslateKey(const KEY translated_key)
{
    auto iter = mInvTranslateKeyMap.find(translated_key);
    if (iter == mInvTranslateKeyMap.end())
    {
        return 0;
    }
    else
    {
        return iter->second;
    }
}


bool LLKeyboard::handleTranslatedKeyDown(KEY translated_key, U32 translated_mask)
{
    bool handled = false;
    bool repeated = false;

    // is this the first time the key went down?
    // if so, generate "character" message
    if( !mKeyLevel[translated_key] )
    {
        mKeyLevel[translated_key] = true;
        mKeyLevelTimer[translated_key].reset();
        mKeyLevelFrameCount[translated_key] = 0;
        mKeyRepeated[translated_key] = false;
    }
    else
    {
        // Level is already down, assume it's repeated.
        repeated = true;
        mKeyRepeated[translated_key] = true;
    }

    mKeyDown[translated_key] = true;
    mCurTranslatedKey = (KEY)translated_key;
    handled = mCallbacks->handleTranslatedKeyDown(translated_key, translated_mask, repeated);
    return handled;
}


bool LLKeyboard::handleTranslatedKeyUp(KEY translated_key, U32 translated_mask)
{
    bool handled = false;
    if( mKeyLevel[translated_key] )
    {
        mKeyLevel[translated_key] = false;

        // Only generate key up events if the key is thought to
        // be down.  This allows you to call resetKeys() in the
        // middle of a frame and ignore subsequent KEY_UP
        // messages in the same frame.  This was causing the
        // sequence W<return> in chat to move agents forward. JC
        mKeyUp[translated_key] = true;
        handled = mCallbacks->handleTranslatedKeyUp(translated_key, translated_mask);
    }

    LL_DEBUGS("UserInput") << "keyup -" << translated_key << "-" << LL_ENDL;

    return handled;
}

bool LLKeyboard::handleKeyDown(const U16 key, const U32 mask)
{
    U32 translated_mask = updateModifiers(mask);

    KEY     translated_key = 0;
    bool    handled = false;
    if(translateKey(key, &translated_key))
    {
        handled = handleTranslatedKeyDown(translated_key, translated_mask);
    }
    if (!handled)
    {
        LLGameControl::onKeyDown(translated_key, translated_mask);
    }

    return handled;
}


bool LLKeyboard::handleKeyUp(const U16 key, const U32 mask)
{
    U32 translated_mask = updateModifiers(mask);

    KEY     translated_key = 0;
    bool    handled = false;
    if(translateKey(key, &translated_key))
    {
        handled = handleTranslatedKeyUp(translated_key, translated_mask);
    }
    if (!handled)
    {
        LLGameControl::onKeyUp(translated_key, translated_mask);
    }

    return handled;
}


void LLKeyboard::toggleInsertMode()
{
    if (LL_KIM_INSERT == mInsertMode)
    {
        mInsertMode = LL_KIM_OVERWRITE;
    }
    else
    {
        mInsertMode = LL_KIM_INSERT;
    }
}


// Returns time in seconds since key was pressed.
F32 LLKeyboard::getKeyElapsedTime(KEY key)
{
    return mKeyLevelTimer[key].getElapsedTimeF32();
}

// Returns time in frames since key was pressed.
S32 LLKeyboard::getKeyElapsedFrameCount(KEY key)
{
    return mKeyLevelFrameCount[key];
}

// static
bool LLKeyboard::keyFromString(const std::string& str, KEY *key)
{
    std::string instring(str);
    size_t length = instring.size();

    if (length < 1)
    {
        return false;
    }
    if (length == 1)
    {
        char ch = toupper(instring[0]);
        if (('0' <= ch && ch <= '9') ||
            ('A' <= ch && ch <= 'Z') ||
            ('!' <= ch && ch <= '/') || // !"#$%&'()*+,-./
            (':' <= ch && ch <= '@') || // :;<=>?@
            ('[' <= ch && ch <= '`') || // [\]^_`
            ('{' <= ch && ch <= '~'))   // {|}~
        {
            *key = ch;
            return true;
        }
    }

    LLStringUtil::toUpper(instring);
    KEY res = get_if_there(sNamesToKeys, instring, (KEY)0);
    if (res != 0)
    {
        *key = res;
        return true;
    }
    LL_WARNS() << "keyFromString failed: " << str << LL_ENDL;
    return false;
}


// static
std::string LLKeyboard::stringFromKey(KEY key, bool translate)
{
    std::string res = get_if_there(sKeysToNames, key, std::string());
    if (res.empty())
    {
        char buffer[2];     /* Flawfinder: ignore */
        buffer[0] = key;
        buffer[1] = '\0';
        res = std::string(buffer);
    }

    if (translate)
    {
        LLKeyStringTranslatorFunc *trans = gKeyboard->mStringTranslator;
        if (trans != NULL)
        {
            res = trans(res);
        }
    }

    return res;
}

//static
std::string LLKeyboard::stringFromMouse(EMouseClickType click, bool translate)
{
    std::string res;
    switch (click)
    {
        case CLICK_LEFT:
            res = "LMB";
            break;
        case CLICK_MIDDLE:
            res = "MMB";
            break;
        case CLICK_RIGHT:
            res = "RMB";
            break;
        case CLICK_BUTTON4:
            res = "MB4";
            break;
        case CLICK_BUTTON5:
            res = "MB5";
            break;
        case CLICK_DOUBLELEFT:
            res = "Double LMB";
            break;
        default:
            break;
    }

    if (translate && !res.empty())
    {
        LLKeyStringTranslatorFunc* trans = gKeyboard->mStringTranslator;
        if (trans != NULL)
        {
            res = trans(res);
        }
    }
    return res;
}

//static
std::string LLKeyboard::stringFromAccelerator(MASK accel_mask)
{
    std::string res;

    LLKeyStringTranslatorFunc *trans = gKeyboard->mStringTranslator;

    if (trans == NULL)
    {
        LL_ERRS() << "No mKeyStringTranslator" << LL_ENDL;
        return res;
    }

    // Append any masks
#ifdef LL_DARWIN
    // Standard Mac names for modifier keys in menu equivalents
    // We could use the symbol characters, but they only exist in certain fonts.
    if (accel_mask & MASK_CONTROL)
    {
        if (accel_mask & MASK_MAC_CONTROL)
        {
            res.append(trans("accel-mac-control"));
        }
        else
        {
            res.append(trans("accel-mac-command"));     // Symbol would be "\xE2\x8C\x98"
        }
    }
    if (accel_mask & MASK_ALT)
        res.append(trans("accel-mac-option"));      // Symbol would be "\xE2\x8C\xA5"
    if (accel_mask & MASK_SHIFT)
        res.append(trans("accel-mac-shift"));       // Symbol would be "\xE2\x8C\xA7"
#else
    if (accel_mask & MASK_CONTROL)
        res.append(trans("accel-win-control"));
    if (accel_mask & MASK_ALT)
        res.append(trans("accel-win-alt"));
    if (accel_mask & MASK_SHIFT)
        res.append(trans("accel-win-shift"));
#endif
    return res;
}
//static
std::string LLKeyboard::stringFromAccelerator( MASK accel_mask, KEY key )
{
    std::string res;

    // break early if this is a silly thing to do.
    if( KEY_NONE == key )
    {
        return res;
    }

    res.append(stringFromAccelerator(accel_mask));
    std::string key_string = LLKeyboard::stringFromKey(key);
    if ((accel_mask & MASK_NORMALKEYS) &&
        (key_string[0] == '-' || key_string[0] == '=' || key_string[0] == '+'))
    {
        res.append( " " );
    }

    std::string keystr = stringFromKey( key );
    res.append( keystr );

    return res;
}

//static
std::string LLKeyboard::stringFromAccelerator(MASK accel_mask, EMouseClickType click)
{
    std::string res;
    if (CLICK_NONE == click)
    {
        return res;
    }
    res.append(stringFromAccelerator(accel_mask));
    res.append(stringFromMouse(click));
    return res;
}

//static
bool LLKeyboard::maskFromString(const std::string& str, MASK *mask)
{
    std::string instring(str);
    if (instring == "NONE")
    {
        *mask = MASK_NONE;
        return true;
    }
    else if (instring == "SHIFT")
    {
        *mask = MASK_SHIFT;
        return true;
    }
    else if (instring == "CTL")
    {
        *mask = MASK_CONTROL;
        return true;
    }
    else if (instring == "ALT")
    {
        *mask = MASK_ALT;
        return true;
    }
    else if (instring == "CTL_SHIFT")
    {
        *mask = MASK_CONTROL | MASK_SHIFT;
        return true;
    }
    else if (instring == "ALT_SHIFT")
    {
        *mask = MASK_ALT | MASK_SHIFT;
        return true;
    }
    else if (instring == "CTL_ALT")
    {
        *mask = MASK_CONTROL | MASK_ALT;
        return true;
    }
    else if (instring == "CTL_ALT_SHIFT")
    {
        *mask = MASK_CONTROL | MASK_ALT | MASK_SHIFT;
        return true;
    }
    else
    {
        return false;
    }
}


//static
void LLKeyboard::setStringTranslatorFunc( LLKeyStringTranslatorFunc *trans_func )
{
    mStringTranslator = trans_func;
}