/*

* Copyright (c) 2007-2010 Sebastian Brannstrom, Lars Persson, EmbedDev AB

*

* 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:

* EmbedDev AB - initial contribution.

*

* Contributors:

*

* Description:

*

*/

#include "ShowEngine.h"

#include "FeedEngine.h"

#include "FeedInfo.h"

#include <bautils.h>

#include <s32file.h>

#include "SettingsEngine.h"

#include <e32hashtab.h>

#include <httperr.h>

#include "debug.h"

#include "PodcastUtils.h"

#include <mpxcollectionhelperfactory.h>

const TUint KMaxDownloadErrors = 3;

const TInt KMimeBufLength = 100;

CShowEngine::CShowEngine(CPodcastModel& aPodcastModel) :

	iPodcastModel(aPodcastModel),

	iDB(*aPodcastModel.DB())

	{

	}

EXPORT_C CShowEngine::~CShowEngine()

	{	

	delete iShowClient;

	iObservers.Close();

	delete iShowDownloading;

	delete iMetaDataReader;

	iApaSession.Close();

	if (iCollectionHelper)

		delete iCollectionHelper;

	}

EXPORT_C CShowEngine* CShowEngine::NewL(CPodcastModel& aPodcastModel)

	{

	CShowEngine* self = new (ELeave) CShowEngine(aPodcastModel);

	CleanupStack::PushL(self);

	self->ConstructL();

	CleanupStack::Pop(self);

	return self;

	}

EXPORT_C void CShowEngine::GetMimeType(const TDesC& aFileName, TDes& aMimeType)

	{

	aMimeType.Zero();

	RFile file;

	if (file.Open(iPodcastModel.FsSession(), aFileName, 0) == KErrNone)

		{

		if (file.Read(iRecogBuffer) == KErrNone)

			{

			TDataRecognitionResult result;

			if (iApaSession.RecognizeData(aFileName, iRecogBuffer, result)

					== KErrNone)

				{

				aMimeType.Copy(result.iDataType.Des());

				}

			}

		}

	file.Close();

	}

void CShowEngine::ConstructL()

	{	

	iShowClient = CHttpClient::NewL(iPodcastModel, *this);

	iShowClient->SetResumeEnabled(ETrue);

	iMetaDataReader = new (ELeave) CMetaDataReader(*this, iPodcastModel.FsSession());

	iMetaDataReader->ConstructL();

	User::LeaveIfError(iApaSession.Connect());

	}

EXPORT_C void CShowEngine::SuspendDownloads()

	{

	iPodcastModel.SettingsEngine().SetDownloadSuspended(ETrue);

	iShowClient->Stop();

	}

EXPORT_C void CShowEngine::ResumeDownloadsL()

	{

	DP("CShowEngine::ResumeDownloadsL BEGIN");

	if (iPodcastModel.SettingsEngine().DownloadSuspended())

		{

		iPodcastModel.SettingsEngine().SetDownloadSuspended(EFalse);

		iDownloadErrors = 0;

		DownloadNextShowL();

		}

	DP("CShowEngine::ResumeDownloadsL END");

	}

EXPORT_C void CShowEngine::RemoveAllDownloadsL()

	{

	if (!iPodcastModel.SettingsEngine().DownloadSuspended())

		{

		SuspendDownloads();

		}

	DBRemoveAllDownloadsL();

	NotifyDownloadQueueUpdatedL();

	}

EXPORT_C void CShowEngine::RemoveDownloadL(TUint aUid)

	{

	DP("CShowEngine::RemoveDownloadL BEGIN");

	TBool resumeAfterRemove = EFalse;

	// if trying to remove the present download, we first stop it

	if (!iPodcastModel.SettingsEngine().DownloadSuspended() && iShowDownloading != NULL

			&& iShowDownloading->Uid() == aUid)

		{

		DP("CShowEngine::RemoveDownload\t This is the active download, we suspend downloading");

		SuspendDownloads();

		// partial downloads should be removed

		BaflUtils::DeleteFile(iPodcastModel.FsSession(), iShowDownloading->FileName());

		resumeAfterRemove = ETrue;

		}

	CShowInfo *info = DBGetShowByUidL(aUid);

	if (info != NULL)

		{

		info->SetDownloadState(ENotDownloaded);

		DBUpdateShowL(*info);

		delete info;

		}

	DBRemoveDownloadL(aUid);

	if (resumeAfterRemove)

		{

		ResumeDownloadsL();

		}

	else

		{

		NotifyShowDownloadUpdatedL(-1, -1);

		NotifyDownloadQueueUpdatedL();

		}

	DownloadNextShowL();

	DP("CShowEngine::RemoveDownloadL END");

	}

void CShowEngine::Connected(CHttpClient* /*aClient*/)

	{

	}

void CShowEngine::Progress(CHttpClient* /*aHttpClient */, TInt aBytes,

		TInt aTotalBytes)

	{	

	iShowDownloading->SetShowSize(aTotalBytes);

	TRAP_IGNORE(NotifyShowDownloadUpdatedL(aBytes, aTotalBytes));

	}

void CShowEngine::Disconnected(CHttpClient* /*aClient */)

	{

	}

void CShowEngine::DownloadInfo(CHttpClient* aHttpClient, TInt aTotalBytes)

	{

	DP1("About to download %d bytes", aTotalBytes);

	if (aHttpClient == iShowClient && iShowDownloading != NULL

			&& aTotalBytes != -1)

		{

		iShowDownloading->SetShowSize(aTotalBytes);

		}

	}

void CShowEngine::GetShowL(CShowInfo *info)

	{

	CFeedInfo *feedInfo = iPodcastModel.FeedEngine().GetFeedInfoByUid(

			info->FeedUid());

	TFileName filePath;

	filePath.Copy(iPodcastModel.SettingsEngine().BaseDir());

	TFileName relPath;

	relPath.Copy(feedInfo->Title());

	relPath.Append('\\');

	TFileName fileName;

	PodcastUtils::FileNameFromUrl(info->Url(), fileName);

	relPath.Append(fileName);

	PodcastUtils::EnsureProperPathName(relPath);

	// complete file path is base dir + rel path

	filePath.Append(relPath);

	info->SetFileNameL(filePath);

	User::LeaveIfError(iShowClient->GetL(info->Url(), filePath));

	}

EXPORT_C void CShowEngine::AddShowL(const CShowInfo& aItem)

	{

	DP1("CShowEngine::AddShowL, title=%S", &aItem.Title());

	CShowInfo *showInfo = DBGetShowByUidL(aItem.Uid());

	if (showInfo == NULL)

		{

		DBAddShowL(aItem);

		}

	else

		{

		delete showInfo;	

		User::Leave(KErrAlreadyExists);

		}	

	}

EXPORT_C void CShowEngine::AddObserver(MShowEngineObserver *observer)

	{

	iObservers.Append(observer);

	}

EXPORT_C void CShowEngine::RemoveObserver(MShowEngineObserver *observer)

	{

	TInt index = iObservers.Find(observer);

	if (index > KErrNotFound)

		{

		iObservers.Remove(index);

		}

	}

void CShowEngine::AddShowToMpxCollection(CShowInfo &aShowInfo)

	{

	if (!iCollectionHelper)

		iCollectionHelper = CMPXCollectionHelperFactory::NewCollectionHelperL();

	// if this leaves, not much we can do anyway

	TRAP_IGNORE(iCollectionHelper->AddL(aShowInfo.FileName(), this));

	}

void CShowEngine::CompleteL(CHttpClient* /*aHttpClient*/, TInt aError)

	{

	if (iShowDownloading != NULL)

		{

		DP2("CShowEngine::CompleteL file=%S, aError=%d", &iShowDownloading->FileName(), aError);		

		if(aError != KErrCouldNotConnect)

			{

			if(aError == KErrDisconnected && iPodcastModel.SettingsEngine().DownloadSuspended())

				{

				// no error if disconnect happened because of suspended downloading

				}

			else

				{

				iShowDownloading->SetLastError(aError);

				}

			if (aError == KErrNone)

				{

				TBuf<KMimeBufLength> mimeType;

				GetMimeType(iShowDownloading->FileName(), mimeType);

				_LIT(KMimeAudio,"audio");

				_LIT(KMimeVideo,"video");

				if (mimeType.Left(5) == KMimeAudio)

					{

					iShowDownloading->SetShowType(EAudioPodcast);

					}

				else if (mimeType.Left(5) == KMimeVideo)

					{

					iShowDownloading->SetShowType(EVideoPodcast);

					}

				iShowDownloading->SetDownloadState(EDownloaded);

				DBUpdateShowL(*iShowDownloading);

				DBRemoveDownloadL(iShowDownloading->Uid());

				AddShowToMpxCollection(*iShowDownloading);				

				NotifyShowFinishedL(aError);

				iDownloadErrors = 0;

				delete iShowDownloading;

				iShowDownloading = NULL;

				}

			else

				{

				// 400 and 500 series errors are serious errors on which probably another download will fail

				if(aError >= HTTPStatus::EBadRequest && aError <= HTTPStatus::EBadRequest+200)

					{

					iShowDownloading->SetDownloadState(EFailedDownload);

					DBUpdateShowL(*iShowDownloading);

					DBRemoveDownloadL(iShowDownloading->Uid());

					NotifyShowFinishedL(aError);

					delete iShowDownloading;

					iShowDownloading = NULL;

					}

				else if (aError == KErrDiskFull)

					{

					// stop downloading immediately if disk is full

					iShowDownloading->SetDownloadState(EQueued);

					DBUpdateShowL(*iShowDownloading);

					iDownloadErrors = KMaxDownloadErrors;

					}

				else // other kind of error, missing network etc, reque this show

					{

					iShowDownloading->SetDownloadState(EQueued);

					DBUpdateShowL(*iShowDownloading);

					}

				NotifyDownloadQueueUpdatedL();

				iDownloadErrors++;

				if (iDownloadErrors >= KMaxDownloadErrors)

					{

					DP("Too many downloading errors, suspending downloads");

					iPodcastModel.SettingsEngine().SetDownloadSuspended(ETrue);

					NotifyShowFinishedL(aError);

					}

				}

			DownloadNextShowL();

			}		

		else

			{

			// Connection error

			if(iShowDownloading)

				{

				iShowDownloading->SetDownloadState(EQueued);

				DBUpdateShowL(*iShowDownloading);

				}

			iPodcastModel.SettingsEngine().SetDownloadSuspended(ETrue);

			NotifyShowFinishedL(aError);

			}

		}

	}

EXPORT_C CShowInfo* CShowEngine::ShowDownloading()

	{

	return iShowDownloading;

	}

EXPORT_C CShowInfo* CShowEngine::GetShowByUidL(TUint aShowUid)

	{

	return DBGetShowByUidL(aShowUid);

	}

CShowInfo* CShowEngine::DBGetShowByUidL(TUint aUid)

	{

	DP("CShowEngine::DBGetShowByUid");

	CShowInfo *showInfo = NULL;

	_LIT(KSqlStatement, "select url, title, description, filename, position, playtime, playstate, downloadstate, feeduid, uid, showsize, trackno, pubdate, showtype from shows where uid=%u");

	iSqlBuffer.Format(KSqlStatement, aUid);

	sqlite3_stmt *st;

	//DP1("SQL: %S", &iSqlBuffer.Left(KSqlDPLen));

	int rc = sqlite3_prepare16_v2(&iDB, (const void*) iSqlBuffer.PtrZ(), -1,

			&st, (const void**) NULL);

	if (rc == SQLITE_OK)

		{

		rc = sqlite3_step(st);

		Cleanup_sqlite3_finalize_PushL(st);

		if (rc == SQLITE_ROW)

			{

			showInfo = CShowInfo::NewLC();

			DBFillShowInfoFromStmtL(st, showInfo);

			CleanupStack::Pop(showInfo);

			}

		CleanupStack::PopAndDestroy();//st

		}

	return showInfo;

	}

EXPORT_C CShowInfo* CShowEngine::DBGetShowByFileNameL(TFileName aFileName)

	{

	DP("CShowEngine::DBGetShowByUid");

	CShowInfo *showInfo = NULL;

	_LIT(KSqlStatement, "select url, title, description, filename, position, playtime, playstate, downloadstate, feeduid, uid, showsize, trackno, pubdate, showtype from shows where filename=\"%S\"");

	iSqlBuffer.Format(KSqlStatement, &aFileName);

	sqlite3_stmt *st;

	int rc = sqlite3_prepare16_v2(&iDB, (const void*) iSqlBuffer.PtrZ(), -1,

			&st, (const void**) NULL);

	if (rc == SQLITE_OK)

		{

		rc = sqlite3_step(st);

		Cleanup_sqlite3_finalize_PushL(st);

		if (rc == SQLITE_ROW)

			{

			showInfo = CShowInfo::NewLC();

			DBFillShowInfoFromStmtL(st, showInfo);

			CleanupStack::Pop(showInfo);

			}

		CleanupStack::PopAndDestroy();//st

		}

	return showInfo;

	}

void CShowEngine::DBGetAllShowsL(RShowInfoArray& aShowArray)

	{

	DP("CShowEngine::DBGetAllShows");

	_LIT(KSqlStatement, "select url, title, description, filename, position, playtime, playstate, downloadstate, feeduid, uid, showsize, trackno, pubdate, showtype from shows");

	iSqlBuffer.Format(KSqlStatement);

	sqlite3_stmt *st;

	int rc = sqlite3_prepare16_v2(&iDB, (const void*) iSqlBuffer.PtrZ(), -1,

			&st, (const void**) NULL);

	if (rc == SQLITE_OK)

		{

		rc = sqlite3_step(st);

		Cleanup_sqlite3_finalize_PushL(st);

		while (rc == SQLITE_ROW)

			{

			CShowInfo* showInfo = CShowInfo::NewLC();

			DBFillShowInfoFromStmtL(st, showInfo);

			aShowArray.Append(showInfo);

			CleanupStack::Pop(showInfo);

			rc = sqlite3_step(st);

			}

		CleanupStack::PopAndDestroy();//st

		}

	}

void CShowEngine::DBGetAllDownloadsL(RShowInfoArray& aShowArray)

	{

	DP("CShowEngine::DBGetAllDownloads");

	_LIT(KSqlStatement, "select url, title, description, filename, position, playtime, playstate, downloadstate, feeduid, shows.uid, showsize, trackno, pubdate, showtype, lasterror from downloads, shows where downloads.uid=shows.uid");

	iSqlBuffer.Format(KSqlStatement);

#ifndef DONT_SORT_SQL

	_LIT(KSqlSort, " order by dl_index");

	iSqlBuffer.Append(KSqlSort);

#endif

	sqlite3_stmt *st;

	int rc = sqlite3_prepare16_v2(&iDB, (const void*) iSqlBuffer.PtrZ(), -1,

			&st, (const void**) NULL);

	if (rc == SQLITE_OK)

		{

		rc = sqlite3_step(st);

		Cleanup_sqlite3_finalize_PushL(st);

		while (rc == SQLITE_ROW)

			{

			CShowInfo* showInfo = CShowInfo::NewLC();

			DBFillShowInfoFromStmtL(st, showInfo);

			aShowArray.Append(showInfo);

			CleanupStack::Pop(showInfo);

			rc = sqlite3_step(st);

			}

		CleanupStack::PopAndDestroy();//st

		}

	else

		{

		User::Leave(KErrCorrupt);

		}

	// delete downloads that don't have a show

	_LIT(KSqlStatement2, "delete from downloads where uid not in (select downloads.uid from shows, downloads where shows.uid=downloads.uid)");

	iSqlBuffer.Format(KSqlStatement2);

	rc = sqlite3_prepare16_v2(&iDB, (const void*) iSqlBuffer.PtrZ(), -1,	&st, (const void**) NULL);

	if (rc == SQLITE_OK)

		{

		Cleanup_sqlite3_finalize_PushL(st);

		rc = sqlite3_step(st);

		CleanupStack::PopAndDestroy(); // st

		}

	else

		{

		User::Leave(KErrCorrupt);

		}

	}

CShowInfo* CShowEngine::DBGetNextDownloadL()

	{

	DP("CShowEngine::DBGetNextDownload");

	CShowInfo *showInfo = NULL;

	_LIT(KSqlStatement, "select url, title, description, filename, position, playtime, playstate, downloadstate, feeduid, shows.uid, showsize, trackno, pubdate, showtype, lasterror from downloads, shows where downloads.uid=shows.uid");

	iSqlBuffer.Format(KSqlStatement);

#ifdef DONT_SORT_SQL

	_LIT(KSqlSort, " limit 1");

	iSqlBuffer.Append(KSqlSort);

#else

	_LIT(KSqlNoSort, " order by dl_index limit 1");

	iSqlBuffer.Append(KSqlNoSort);

#endif

	sqlite3_stmt *st;

	int rc = sqlite3_prepare16_v2(&iDB, (const void*) iSqlBuffer.PtrZ(), -1,

			&st, (const void**) NULL);

	if (rc == SQLITE_OK)

		{

		rc = sqlite3_step(st);

		Cleanup_sqlite3_finalize_PushL(st);

		if (rc == SQLITE_ROW)

			{

			showInfo = CShowInfo::NewLC();

			DBFillShowInfoFromStmtL(st, showInfo);