/*
* Copyright (c) 2008-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 <fstream>
#include "utils.h"
#include "logger.h"

#ifndef __LINUX__
#include <windows.h>
#endif

typedef unsigned char TUint8;
typedef unsigned int TUint;

#ifdef __LINUX__
void OpenUtf8FStreamForRead(std::fstream &aFStream, const char *aUtf8FileName)
{
	aFStream.open(aUtf8FileName, std::ios_base::binary | std::ios_base::in);
}

void OpenUtf8FStreamForWrite(std::fstream &aFStream, const char *aUtf8FileName)
{
	aFStream.open(aUtf8FileName, std::ios_base::binary | std::ios_base::trunc | std::ios_base::out);
}

#else

WCHAR *WindowsUtf8ToUtf16(const char *aUtf8)
{
	int utf8Len = strlen(aUtf8) + 1; // We want to copy the terminator to the new string
	int utf16Space = MultiByteToWideChar(CP_UTF8, 0, aUtf8, utf8Len, 0, 0);
	if(utf16Space <= 0)
		{
		dbg << Log::Indent() << "Failed to convert UTF-8 '" << aUtf8 << "' to UTF16" << Log::Endl();
		FatalError();
		return 0;
		}
	// prog << Log::Indent() << "utf16Space " << utf16Space << Log::Endl();
	WCHAR *utf16 = new WCHAR[utf16Space];
	(void) MultiByteToWideChar(CP_UTF8, 0, aUtf8, utf8Len, utf16, utf16Space);
#if 0
	prog << Log::Indent() << "Raw UTF16 name is (chars " << utf16Space << ")" << Log::Endl();
	for(int i=0; i<utf16Space*2; ++i)
		{
		prog << TUint(((TUint8 *)utf16)[i]) << Log::Endl();
		}
#endif
	return utf16;
}

CHAR *WindowsUtf16ToUtf8(const WCHAR *aUtf16)
{
	int utf16Len = wcslen(aUtf16) + 1; // We want to copy the terminator to the new string
	int utf8Space = WideCharToMultiByte(CP_UTF8, 0, aUtf16, utf16Len, 0, 0, 0, 0);
	if(utf8Space <= 0)
		{
		dbg << Log::Indent() << "Failed to convert UTF16 string to UTF8" << Log::Endl();
		FatalError();
		return 0;
		}
	//prog << Log::Indent() << "utf8Space " << utf8Space << Log::Endl();
	CHAR *utf8 = new CHAR[utf8Space];
	(void) WideCharToMultiByte(CP_UTF8, 0, aUtf16, utf16Len, utf8, utf8Space, 0, 0);

#if 0
	prog << Log::Indent() << "Raw UTF8 name is (chars " << utf8Space << ")" << Log::Endl();
	for(int i=0; i<utf8Space; ++i)
		{
		prog << TUint(((TUint8 *)utf8)[i]) << Log::Endl();
		}
#endif
	return utf8;
}

WCHAR *WindowsRetrieveDosName(const WCHAR *aLongName)
{
	// Convert to a short DOS file name. File MUST already exist!
	// len will be space required for string and terminator
	DWORD len = GetShortPathNameW((WCHAR *)aLongName, 0, 0);
	if( len <= 0 )
		{
		prog << Log::Indent() << "Failed to find existing file and retrieve short name" << Log::Endl();
		return 0;
		}
	
	//prog << Log::Indent() << "space required for short name is " << len << Log::Endl();
	WCHAR *shortName = new WCHAR[len];
	(void) GetShortPathNameW(aLongName, shortName, len);
	return shortName;
}

void OpenUtf8FStreamForRead(std::fstream &aFStream, const char *aUtf8FileName)
{
	// Unfortunately the windows C++ STL library will not open files specified via a UTF-8 filename.
	// The workaround is to convert the UTF-8 name to the short DOS compatible name and open that instead

	// First convert the UTF-8 name to UTF-16 for use with the windows APIs
	WCHAR *utf16Name = WindowsUtf8ToUtf16(aUtf8FileName);

	// Obtain short DOS compatible name for file
	WCHAR *utf16NameDos = WindowsRetrieveDosName(utf16Name);
	if(utf16NameDos == 0)
		{
		aFStream.setstate(std::ios_base::badbit);
		delete [] utf16Name;
		return;
		}

	char *utf8NameDos = WindowsUtf16ToUtf8(utf16NameDos);

	aFStream.open(utf8NameDos, std::ios_base::binary | std::ios_base::in);

	delete [] utf8NameDos;
	delete [] utf16NameDos;
	delete [] utf16Name;
	return;
}

void OpenUtf8FStreamForWrite(std::fstream &aFStream, const char *aUtf8FileName)
{
	// Unfortunately the windows C++ STL library will not open files specified via a UTF-8 filename.
	// The workaround is to make sure the file already exists and then convert the UTF-8 name to the short DOS compatible name 
	// and open that instead

	// First convert the UTF-8 name to UTF-16 for use with the windows APIs
	WCHAR *utf16Name = WindowsUtf8ToUtf16(aUtf8FileName);

	// Make sure the file exists
	HANDLE fh = CreateFileW((WCHAR *)utf16Name, GENERIC_WRITE, FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE, 0,  CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0 );
	if(fh == INVALID_HANDLE_VALUE)
		{
		dbg << Log::Indent() << "Failed to create file '" << aUtf8FileName << "'" << Log::Endl();
		aFStream.setstate(std::ios_base::badbit);
		delete [] utf16Name;
		return;
		}
	// Close handle
	CloseHandle(fh);


	// Obtain short DOS compatible name for file
	WCHAR *utf16NameDos = WindowsRetrieveDosName(utf16Name);
	if(utf16NameDos == 0)
		{
		aFStream.setstate(std::ios_base::badbit);
		delete [] utf16Name;
		return;
		}

	char *utf8NameDos = WindowsUtf16ToUtf8(utf16NameDos);

	aFStream.open(utf8NameDos, std::ios_base::binary | std::ios_base::trunc | std::ios_base::out);

	delete [] utf8NameDos;
	delete [] utf16NameDos;
	delete [] utf16Name;
	return;
}
#endif

// End of file