/*

* Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of the License "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description: 

*

*/

#include "UTF16String.h"

#include <iostream>

#include <fstream>

#include <string>

using namespace std ;

#ifdef __LINUX__

#define stricmp strcasecmp

#endif

void PrintHelp(){

	cout <<  "Syntax: TestUTF16Str  <-[mbtou|utomb] > -i inputfilename -o outputfilename "<<endl; 

	cout << "	mbtou is by default."<<endl;

}

int main(int argc, char* argv[]){

	const char* input = 0 ;

	const char* output = 0 ;

	if(argc < 5){

		PrintHelp();

		return 1;

	}

	bool mbtou = true ;

	int i = 1;

	while(i < argc){

		if('-' == *argv[i] || '/' == *argv[i]){

			if(!stricmp(&(argv[i][1]),"utomb"))

				mbtou = false ;

			else if((argv[i][1] | 0x20) == 'i'){

				i++ ;

				if(i < argc)

					input = argv[i];

			}

			else if((argv[i][1] | 0x20) == 'o'){

				i++ ;

				if(i < argc)

					output = argv[i];

			}

			else if(stricmp(&(argv[i][1]),"mbtou")){

				cerr << "Unrecognized option "<< argv[i] << endl ;

			}				

		}

		else {

			cerr << "Unrecognized option "<< argv[i] << endl ;

		}

		i++ ;			

	}

	if(!input || !output){

		PrintHelp();

		return 2;

	}

	fstream ifs(input, ios_base::in + ios_base::binary);	

	if(!ifs.is_open()){

		cerr << "Cannot open \""<< input << "\" for reading."<<endl ;

		return 3;

	}

	fstream ofs(output, ios_base::out + ios_base::binary + ios_base::trunc);

	if(!ofs.is_open()){

		cerr << "Cannot open \""<< output << "\" for writing."<<endl ;

		ifs.close();

		return 4;

	}

	ifs.seekg(0,ios_base::end);

	size_t length = ifs.tellg();

	ifs.seekg(0,ios_base::beg);

	char* buffer = new char[length + 2];

	ifs.read(buffer,length);

	buffer[length] = 0 ;

	buffer[length + 1] = 0 ;

	ifs.close();

	static unsigned char const utf16FileHdr[2] = {0xFF,0xFE};

	static unsigned char const utf8FileHdr[3] = {0xEF,0xBB,0xBF};

	if(mbtou){

		char* mbstr = buffer ;

		if(length > 3){

			if(memcmp(buffer,utf8FileHdr,sizeof(utf8FileHdr)) == 0){

				mbstr += 3;

				length -= 3 ;

			}

		}

		UTF16String theStr(mbstr , length);

		if(length > 0 && theStr.IsEmpty()){

			cerr << "Convert Error[From UTF8 To UTF16]."<<endl ;

		}

		else{

			length = theStr.length() << 1;			

			ofs.write(reinterpret_cast<const char*>(utf16FileHdr),sizeof(utf16FileHdr));

			ofs.write(reinterpret_cast<const char*>(theStr.c_str()),length);

			cout << "Done."<<endl ;

		}		

	}

	else{

		TUint16* unistr = reinterpret_cast<TUint16*>(buffer);

		length >>= 1;

		if(*unistr == 0xFEFF){

			unistr ++ ;

			length -- ;		

		}

		UTF16String theStr(unistr , length);

		string mbstr ;

		if(!theStr.ToUTF8(mbstr)){

			cerr << "Convert Error[From UTF16 To UTF8]."<<endl ;

		}else{

			//ofs.write(reinterpret_cast<const char*>(utf8FileHdr),sizeof(utf8FileHdr));

			ofs.write(mbstr.c_str(),mbstr.length());

			cout << "Done."<<endl ;

		}

	}

	ofs.close();	

	delete []buffer ;	

	return 0;

}