#ifndef GAMEPADUI_STRING_H
#define GAMEPADUI_STRING_H
#ifdef _WIN32
#pragma once
#endif

#include "vgui/ILocalize.h"
#include "tier1/utlvector.h"

extern vgui::ILocalize *g_pVGuiLocalize;

class GamepadUIString
{
public:
    GamepadUIString()
    {
    }

    GamepadUIString( const char *pszText )
    {
        SetText( pszText );
    }

    GamepadUIString( const wchar_t *pszText )
    {
        SetText( pszText );
    }

    GamepadUIString( const wchar_t *pszText, int nLength )
    {
        SetText( pszText, nLength );
    }

    const wchar_t*String() const
    {
        if ( m_ManagedText.Count() )
            return m_ManagedText.Base();

        return L"";
    }

    int Length() const
    {
        if ( m_ManagedText.Count() )
            return m_ManagedText.Count()-1;

        return 0;
    }

    bool IsEmpty() const
    {
        return Length() == 0;
    }

    void SetText( const char *pszText )
    {
        m_ManagedText.Purge();

        if ( !pszText || !*pszText )
            return;

        const wchar_t *pszFoundText = g_pVGuiLocalize->Find( pszText );
        if ( !pszFoundText )
        {
            SetRawUTF8( pszText );
        }
        else
        {
            int nChars = V_wcslen( pszFoundText );
            SetText( pszFoundText, nChars );
        }
    }

    void SetText( const wchar_t *pszText, int nLength )
    {
        m_ManagedText.Purge();

        if ( !pszText || !nLength )
            return;

        m_ManagedText.EnsureCapacity( nLength + 1 );
        for ( int i = 0; i < nLength; i++ )
            m_ManagedText.AddToTail( pszText[ i ] );
        m_ManagedText.AddToTail( L'\0' );
    }

    void SetText( const wchar_t *pszText )
    {
        if ( !pszText )
            SetText( NULL, 0 );
        else
            SetText( pszText, V_wcslen( pszText ) );
    }

    void SetRawUTF8( const char* pszText )
    {
        m_ManagedText.Purge();

        if ( !pszText || !*pszText )
            return;

        wchar_t szUnicode[ 4096 ];
		memset( szUnicode, 0, sizeof( wchar_t ) * 4096 );
        
		V_UTF8ToUnicode( pszText, szUnicode, sizeof( szUnicode ) );

        int nChars = V_strlen(pszText);
        if ( nChars > 1 )
            SetText( szUnicode, nChars );
    }
private:
    CCopyableUtlVector< wchar_t > m_ManagedText;
};

#endif // GAMEPADUI_STRING_H