package org.apache.poi.hpsf;

import java.io.IOException;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;

import org.apache.poi.util.Internal;
import org.apache.poi.util.LittleEndian;

@Internal
class CodePageString
{

    private static String codepageToEncoding( final int codepage )
            throws UnsupportedEncodingException
    {
        if ( codepage <= 0 )
            throw new UnsupportedEncodingException(
                    "Codepage number may not be " + codepage );
        switch ( codepage )
        {
        case Constants.CP_UTF16:
            return "UTF-16";
        case Constants.CP_UTF16_BE:
            return "UTF-16BE";
        case Constants.CP_UTF8:
            return "UTF-8";
        case Constants.CP_037:
            return "cp037";
        case Constants.CP_GBK:
            return "GBK";
        case Constants.CP_MS949:
            return "ms949";
        case Constants.CP_WINDOWS_1250:
            return "windows-1250";
        case Constants.CP_WINDOWS_1251:
            return "windows-1251";
        case Constants.CP_WINDOWS_1252:
            return "windows-1252";
        case Constants.CP_WINDOWS_1253:
            return "windows-1253";
        case Constants.CP_WINDOWS_1254:
            return "windows-1254";
        case Constants.CP_WINDOWS_1255:
            return "windows-1255";
        case Constants.CP_WINDOWS_1256:
            return "windows-1256";
        case Constants.CP_WINDOWS_1257:
            return "windows-1257";
        case Constants.CP_WINDOWS_1258:
            return "windows-1258";
        case Constants.CP_JOHAB:
            return "johab";
        case Constants.CP_MAC_ROMAN:
            return "MacRoman";
        case Constants.CP_MAC_JAPAN:
            return "SJIS";
        case Constants.CP_MAC_CHINESE_TRADITIONAL:
            return "Big5";
        case Constants.CP_MAC_KOREAN:
            return "EUC-KR";
        case Constants.CP_MAC_ARABIC:
            return "MacArabic";
        case Constants.CP_MAC_HEBREW:
            return "MacHebrew";
        case Constants.CP_MAC_GREEK:
            return "MacGreek";
        case Constants.CP_MAC_CYRILLIC:
            return "MacCyrillic";
        case Constants.CP_MAC_CHINESE_SIMPLE:
            return "EUC_CN";
        case Constants.CP_MAC_ROMANIA:
            return "MacRomania";
        case Constants.CP_MAC_UKRAINE:
            return "MacUkraine";
        case Constants.CP_MAC_THAI:
            return "MacThai";
        case Constants.CP_MAC_CENTRAL_EUROPE:
            return "MacCentralEurope";
        case Constants.CP_MAC_ICELAND:
            return "MacIceland";
        case Constants.CP_MAC_TURKISH:
            return "MacTurkish";
        case Constants.CP_MAC_CROATIAN:
            return "MacCroatian";
        case Constants.CP_US_ACSII:
        case Constants.CP_US_ASCII2:
            return "US-ASCII";
        case Constants.CP_KOI8_R:
            return "KOI8-R";
        case Constants.CP_ISO_8859_1:
            return "ISO-8859-1";
        case Constants.CP_ISO_8859_2:
            return "ISO-8859-2";
        case Constants.CP_ISO_8859_3:
            return "ISO-8859-3";
        case Constants.CP_ISO_8859_4:
            return "ISO-8859-4";
        case Constants.CP_ISO_8859_5:
            return "ISO-8859-5";
        case Constants.CP_ISO_8859_6:
            return "ISO-8859-6";
        case Constants.CP_ISO_8859_7:
            return "ISO-8859-7";
        case Constants.CP_ISO_8859_8:
            return "ISO-8859-8";
        case Constants.CP_ISO_8859_9:
            return "ISO-8859-9";
        case Constants.CP_ISO_2022_JP1:
        case Constants.CP_ISO_2022_JP2:
        case Constants.CP_ISO_2022_JP3:
            return "ISO-2022-JP";
        case Constants.CP_ISO_2022_KR:
            return "ISO-2022-KR";
        case Constants.CP_EUC_JP:
            return "EUC-JP";
        case Constants.CP_EUC_KR:
            return "EUC-KR";
        case Constants.CP_GB2312:
            return "GB2312";
        case Constants.CP_GB18030:
            return "GB18030";
        case Constants.CP_SJIS:
            return "SJIS";
        default:
            return "cp" + codepage;
        }
    }

    private byte[] _value;

    CodePageString( final byte[] data, final int startOffset )
    {
        int offset = startOffset;

        int size = LittleEndian.getInt( data, offset );
        offset += LittleEndian.INT_SIZE;

        _value = LittleEndian.getByteArray( data, offset, size );
        if ( _value[size - 1] != 0 )
            throw new IllegalPropertySetDataException(
                    "CodePageString started at offset #" + offset
                            + " is not NULL-terminated" );
    }

    CodePageString( String string, int codepage )
            throws UnsupportedEncodingException
    {
        setJavaValue( string, codepage );
    }

    String getJavaValue( int codepage ) throws UnsupportedEncodingException
    {
        String result;
        if ( codepage == -1 )
            result = new String( _value );
        else
            result = new String( _value, codepageToEncoding( codepage ) );
        return result.substring( 0, result.length() - 1 );
    }

    int getSize()
    {
        return LittleEndian.INT_SIZE + _value.length;
    }

    void setJavaValue( String string, int codepage )
            throws UnsupportedEncodingException
    {
        if ( codepage == -1 )
            _value = ( string + "\0" ).getBytes();
        else
            _value = ( string + "\0" )
                    .getBytes( codepageToEncoding( codepage ) );
    }

    int write( OutputStream out ) throws IOException
    {
        LittleEndian.putInt( _value.length, out );
        out.write( _value );
        return LittleEndian.INT_SIZE + _value.length;
    }
}