/**************************************************************
 *
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 *
 *************************************************************/



// MARKER(update_precomp.py): autogen include statement, do not remove
#include "precompiled_shell.hxx"
#include "internal/global.hxx"
#include "internal/infotips.hxx"
#include "internal/shlxthdl.hxx"
#include "internal/metainforeader.hxx"
#include "internal/contentreader.hxx"
#include "internal/utilities.hxx"
#include "internal/registry.hxx"
#include "internal/fileextensions.hxx"
#include "internal/iso8601_converter.hxx"
#include "internal/config.hxx"

#include "internal/resource.h"
#include <stdio.h>
#include <utility>
#include <stdlib.h>


#define MAX_STRING 80
#define KB 1024.0
const std::wstring WSPACE = std::wstring(SPACE);

//-----------------------------
//
//-----------------------------

CInfoTip::CInfoTip(long RefCnt) :
	m_RefCnt(RefCnt)
{
	ZeroMemory(m_szFileName, sizeof(m_szFileName));
	InterlockedIncrement(&g_DllRefCnt);
}

//-----------------------------
//
//-----------------------------

CInfoTip::~CInfoTip()
{
	InterlockedDecrement(&g_DllRefCnt);
}

//-----------------------------
// IUnknown methods
//-----------------------------

HRESULT STDMETHODCALLTYPE CInfoTip::QueryInterface(REFIID riid, void __RPC_FAR *__RPC_FAR *ppvObject)
{
	*ppvObject = 0;

	IUnknown* pUnk = 0;

	if (IID_IUnknown == riid || IID_IQueryInfo == riid)
	{
		pUnk = static_cast<IQueryInfo*>(this);
		pUnk->AddRef();
		*ppvObject = pUnk;
		return S_OK;
	}
	else if (IID_IPersistFile == riid)
	{
		pUnk = static_cast<IPersistFile*>(this);
		pUnk->AddRef();
		*ppvObject = pUnk;
		return S_OK;
	}

	return E_NOINTERFACE;
}

//----------------------------
//
//----------------------------

ULONG STDMETHODCALLTYPE CInfoTip::AddRef(void)
{
	return InterlockedIncrement(&m_RefCnt);
}

//----------------------------
//
//----------------------------

ULONG STDMETHODCALLTYPE CInfoTip::Release( void)
{
	long refcnt = InterlockedDecrement(&m_RefCnt);

	if (0 == m_RefCnt)
		delete this;

	return refcnt;
}

//********************helper functions for GetInfoTip functions**********************

/** get file type information from registry.
*/
std::wstring getFileTypeInfo(const std::string& file_extension)
{
	char extKeyValue[MAX_STRING];
	char typeKeyValue[MAX_STRING];
	::std::string sDot(".");
	if (QueryRegistryKey(HKEY_CLASSES_ROOT, (sDot.append(file_extension)).c_str(), "", extKeyValue, MAX_STRING))
		if (QueryRegistryKey( HKEY_CLASSES_ROOT, extKeyValue, "",typeKeyValue, MAX_STRING))
			return StringToWString(typeKeyValue);

	return EMPTY_STRING;
}

/** get file size.
*/
DWORD getSizeOfFile( char* FileName )
{
	HANDLE hFile = CreateFile(StringToWString(FileName).c_str(),			// open file
						GENERIC_READ,										// open for reading
						FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE,	// share for all operations
						NULL,												// no security
						OPEN_EXISTING,										// existing file only
						FILE_ATTRIBUTE_NORMAL,								// normal file
						NULL);												// no attr. template

	if (hFile != INVALID_HANDLE_VALUE)
	{
		DWORD dwSize = GetFileSize( HANDLE(hFile), NULL );
		CloseHandle( HANDLE(hFile) );
		return dwSize;
	}

	return INVALID_FILE_SIZE;
}

/** format file size in to be more readable.
*/
std::wstring formatSizeOfFile( DWORD dwSize )
{
	if ( dwSize < 1000 )
	{
		char buffer[3];
		int dFileSize = dwSize;

		_itoa( dFileSize, buffer, 10 );
		return StringToWString( buffer ).append(StringToWString(" B"));
	}

	char *buffer=NULL;
	int decimal, sign;
	double dFileSize = (double)dwSize/(double)KB;

	buffer = _fcvt( dFileSize, 1, &decimal, &sign );

	::std::wstring wsTemp = StringToWString( buffer );
	int pos=decimal % 3;
	::std::wstring wsBuffer = wsTemp.substr( 0,pos);

	if ( decimal )
		for (;decimal - pos > 2;pos += 3)
		{
			if (pos)
				wsBuffer.append(StringToWString(","));
			wsBuffer.append( wsTemp.substr( pos, 3) );
		}
	else
		wsBuffer.append(StringToWString("0"));

	wsBuffer.append(StringToWString("."));
	wsBuffer.append(wsTemp.substr( decimal, wsTemp.size()-decimal ));
	wsBuffer.append(StringToWString(" KB"));

	return wsBuffer;
}


/** get file size information.
*/
std::wstring getFileSizeInfo(char* FileName)
{
	DWORD dwSize=getSizeOfFile(FileName);
	if (dwSize != INVALID_FILE_SIZE)
		return formatSizeOfFile( dwSize );

	return EMPTY_STRING;
}

//----------------------------
// IQueryInfo methods
//----------------------------

HRESULT STDMETHODCALLTYPE CInfoTip::GetInfoTip(DWORD /*dwFlags*/, wchar_t** ppwszTip)
{
	std::wstring msg;
	const std::wstring CONST_SPACE(SPACE);

	// display File Type, no matter other info is loaded successfully or not.
	std::wstring tmpTypeStr = getFileTypeInfo( get_file_name_extension(m_szFileName) );
	if ( tmpTypeStr != EMPTY_STRING )
	{
		msg += GetResString(IDS_TYPE_COLON) + CONST_SPACE;
		msg += tmpTypeStr;
	}

	try
	{
		CMetaInfoReader meta_info_accessor(m_szFileName);

		// display document title;
		if ( meta_info_accessor.getTagData( META_INFO_TITLE ).length() > 0)
		{
			if ( msg != EMPTY_STRING )
				 msg += L"\n";
			msg += GetResString(IDS_TITLE_COLON) + CONST_SPACE;
			msg += meta_info_accessor.getTagData( META_INFO_TITLE );
		}
		else
		{
			if ( msg != EMPTY_STRING )
				 msg += L"\n";
			msg += GetResString(IDS_TITLE_COLON) + CONST_SPACE;
			msg += m_FileNameOnly;
		}

		// display document author;
		if ( meta_info_accessor.getTagData( META_INFO_AUTHOR ).length() > 0)
		{
			if ( msg != EMPTY_STRING )
				 msg += L"\n";
			msg += GetResString( IDS_AUTHOR_COLON ) + CONST_SPACE;
			msg += meta_info_accessor.getTagData( META_INFO_AUTHOR );
		}

		// display document subject;
		if ( meta_info_accessor.getTagData( META_INFO_SUBJECT ).length() > 0)
		{
			if ( msg != EMPTY_STRING )
				 msg += L"\n";
			msg += GetResString(IDS_SUBJECT_COLON) + CONST_SPACE;
			msg += meta_info_accessor.getTagData( META_INFO_SUBJECT );
		}

		// display document description;
		if ( meta_info_accessor.getTagData( META_INFO_DESCRIPTION ).length() > 0)
		{
			if ( msg != EMPTY_STRING )
				 msg += L"\n";
			msg += GetResString( IDS_COMMENTS_COLON ) + CONST_SPACE;
			msg += meta_info_accessor.getTagData( META_INFO_DESCRIPTION );
		}

		// display modified time formatted into locale representation.
		if ( iso8601_date_to_local_date(meta_info_accessor.getTagData(META_INFO_MODIFIED )).length() > 0)
		{
			if ( msg != EMPTY_STRING )
				 msg += L"\n";
			msg += GetResString( IDS_MODIFIED_COLON ) + CONST_SPACE;
			msg += iso8601_date_to_local_date(meta_info_accessor.getTagData(META_INFO_MODIFIED ));
		}
	}

	catch (const std::exception&)
	{
		//return E_FAIL;
	}

	// display file size, no matter other information is loaded successfully or not.
	std::wstring tmpSizeStr = getFileSizeInfo( m_szFileName );
	if ( tmpSizeStr != EMPTY_STRING )
	{
		msg += L"\n";
		msg += GetResString( IDS_SIZE_COLON ) + CONST_SPACE;
		msg += tmpSizeStr;
	}


	// finalize and assign the string.
	LPMALLOC lpMalloc;
	HRESULT hr = SHGetMalloc(&lpMalloc);

	if (SUCCEEDED(hr))
	{
		size_t len = sizeof(wchar_t) * msg.length() + sizeof(wchar_t);
		wchar_t* pMem = reinterpret_cast<wchar_t*>(lpMalloc->Alloc(len));

		ZeroMemory(pMem, len);

		msg.copy(pMem,msg.length() + 1);

		*ppwszTip = pMem;
		lpMalloc->Release();

		return S_OK;
	}

	return E_FAIL;
}

//----------------------------
//
//----------------------------

HRESULT STDMETHODCALLTYPE CInfoTip::GetInfoFlags(DWORD * /*pdwFlags*/ )
{
	return E_NOTIMPL;
}

//----------------------------
// IPersist methods
//----------------------------

HRESULT STDMETHODCALLTYPE CInfoTip::GetClassID(CLSID* pClassID)
{
	pClassID = const_cast<CLSID*>(&CLSID_INFOTIP_HANDLER);
	return S_OK;
}

//----------------------------
// IPersistFile methods
//----------------------------

HRESULT STDMETHODCALLTYPE CInfoTip::Load(LPCOLESTR pszFileName, DWORD /*dwMode*/)
{
	std::wstring fname = pszFileName;

	// there must be a '\' and there must even be an
	// extension, else we would not have been called
	std::wstring::iterator begin = fname.begin() + fname.find_last_of(L"\\") + 1;
	std::wstring::iterator end   = fname.end();

	m_FileNameOnly = std::wstring(begin, end);

	fname = getShortPathName( fname );

	std::string fnameA = WStringToString(fname);

	// #115531#
	// ZeroMemory because strncpy doesn't '\0'-terminate the destination
	// string; reserve the last place in the buffer for the final '\0'
	// that's why '(sizeof(m_szFileName) - 1)'
	ZeroMemory(m_szFileName, sizeof(m_szFileName));
	strncpy(m_szFileName, fnameA.c_str(), (sizeof(m_szFileName) - 1));

	return S_OK;
}

//----------------------------
//
//----------------------------

HRESULT STDMETHODCALLTYPE CInfoTip::IsDirty(void)
{
	return E_NOTIMPL;
}

//----------------------------
//
//----------------------------

HRESULT STDMETHODCALLTYPE CInfoTip::Save(LPCOLESTR /*pszFileName*/, BOOL /*fRemember*/)
{
	return E_NOTIMPL;
}

//----------------------------
//
//----------------------------

HRESULT STDMETHODCALLTYPE CInfoTip::SaveCompleted(LPCOLESTR /*pszFileName*/)
{
	return E_NOTIMPL;
}

//----------------------------
//
//----------------------------

HRESULT STDMETHODCALLTYPE CInfoTip::GetCurFile(LPOLESTR __RPC_FAR * /*ppszFileName*/)
{
	return E_NOTIMPL;
}

/* vim: set noet sw=4 ts=4: */