// Copyright (c) 1998-2009 Nokia Corporation and/or its subsidiary(-ies).
// All rights reserved.
// This component and the accompanying materials are made available
// under the terms of "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 <jni.h>
#include "DbmsConnection.h"
#include "Utils.h"
#include "dbmsjni/com_symbian_dbms_jdbc_DbmsConnection.h"
#include <stdlib.h>
#include <stdio.h>

CTrapCleanup* gCleanup = NULL; // get clean-up stack
TInt gConnectionCount = 0;
jclass gExcCls = NULL;
TUint32 gThirdUid = 0;

extern TPanicHandler gPanicHandler;
extern TInt gBlockSize;
extern TInt gClusterSize;
extern TExtendedLocale gLocale;

// //
// DbmsConnection peer
// //

DbmsConnection::DbmsConnection(){
	iOpen = EFalse;
}

DbmsConnection::~DbmsConnection(){
}

TInt DbmsConnection::Open(const TDesC& aFileName){
	RFs fs;
	TInt res = iDatabase.Open(fs, aFileName);
	if ( res == KErrNotFound ) {
		if ( gThirdUid != 0 ) {
			TBuf<64> buf;
			buf.Append(_L("SECURE["));
			buf.AppendNum(gThirdUid, EHex);
			buf.Append(_L("]"));
			res = iDatabase.Create(fs, aFileName, buf);
		} else {
			res = iDatabase.Create(fs, aFileName);
		}
	}
	if ( res == KErrNone ) {
		iOpen = ETrue;
	}
	return res;
}

void DbmsConnection::Close(){
	if ( iOpen )
		{
		iDatabase.Compact();
		}
	iDatabase.Close();
	iOpen = EFalse;
}

// //
// JNI
// //

/*
 * Class:     com_symbian_dbms_jdbc_DbmsConnection
 * Method:    _globalInit
 * Signature: ()V
 */
JNIEXPORT void JNICALL Java_com_symbian_dbms_jdbc_DbmsConnection__1globalInit
  (JNIEnv *aEnv, jobject) {
	if ( gCleanup == NULL ) {
		// init globals
		gCleanup=CTrapCleanup::New(); // get clean-up stack
		gPanicHandler = &SosPanicHandler;
		jclass excCls = aEnv->FindClass("java/lang/RuntimeException");
		if ( excCls == NULL )
			{
			printf("Could not create exception class java/lang/RuntimeException in DbmsConnection::init ...\n");
			}
		else
			{
			gExcCls = (jclass)(aEnv->NewGlobalRef(excCls));
			if ( gExcCls == NULL )
				{
				printf("Could not create glref to exception class java/lang/RuntimeException in DbmsConnection::init ...\n");
				}
			}
	}
}


JNIEXPORT jint JNICALL Java_com_symbian_dbms_jdbc_DbmsConnection__1init
  (JNIEnv *aEnv, jobject aObject, jstring aString) {
	gConnectionCount++;
	DbmsConnection* connection = new DbmsConnection();
	RJString file(*aEnv, aString);
	TInt res = connection->Open(file);
	if ( res != KErrNone ){
		connection->Close();
		delete connection;
		return (jint) res;
	} else {
		return (jint) connection;
	}
}

/*
 * Class:     com_symbian_dbms_jdbc_DbmsConnection
 * Method:    _setSecureId
 * Signature: (I)V
 */
JNIEXPORT void JNICALL Java_com_symbian_dbms_jdbc_DbmsConnection__1setSecureId
  (JNIEnv *, jobject, jint aSecureId) {
	gThirdUid = aSecureId;
}

/*
 * Class:     com_symbian_dbms_jdbc_DbmsConnection
 * Method:    _close
 * Signature: (I)V
 */
JNIEXPORT void JNICALL Java_com_symbian_dbms_jdbc_DbmsConnection__1close
  (JNIEnv * aEnv, jobject aJavaObj, jint aPeer) {
	DbmsConnection* connection = (DbmsConnection*) aPeer;
	connection->Close();
	delete connection;
	gConnectionCount--;
	if ( gConnectionCount == 0 && gCleanup != NULL ) {
		delete gCleanup;
		gCleanup = NULL;
	}
}

/*
 * Class:     com_symbian_dbms_jdbc_DbmsConnection
 * Method:    _setLocaleDll
 * Signature: (Ljava/lang/String;)I
 */
JNIEXPORT jint JNICALL Java_com_symbian_dbms_jdbc_DbmsConnection__1setLocaleDll
  (JNIEnv *aEnv, jobject, jstring aDllFile){
	RJString rj(*aEnv, aDllFile);
	return gLocale.LoadLocale(rj);
}

/*
 * Class:     com_symbian_dbms_jdbc_DbmsConnection
 * Method:    _setBlockSize
 * Signature: (I)V
 */
JNIEXPORT void JNICALL Java_com_symbian_dbms_jdbc_DbmsConnection__1setBlockSize
  (JNIEnv *, jobject, jint aBlockSize){
	gBlockSize = aBlockSize;
}

/*
 * Class:     com_symbian_dbms_jdbc_DbmsConnection
 * Method:    _setClusterSize
 * Signature: (I)V
 */
JNIEXPORT void JNICALL Java_com_symbian_dbms_jdbc_DbmsConnection__1setClusterSize
  (JNIEnv *, jobject, jint aClusterSize){
	gClusterSize = aClusterSize;
}


void PrintColTypeName(TDes8& aDes, TDbCol& col)
	{
	_LIT8(KNotNull, "not-null ");
	_LIT8(KAutoIncrement, "auto-increment");
	_LIT8(KColNameInt8, "EDbColInt8");
	_LIT8(KColNameInt16, "EDbColInt16");
	_LIT8(KColNameInt32, "EDbColInt32");
	_LIT8(KColNameInt64, "EDbColInt64");
	_LIT8(KColNameUInt8, "EDbColUint8");
	_LIT8(KColNameUInt16, "EDbColUint16");
	_LIT8(KColNameUInt32, "EDbColUint32");
	_LIT8(KColNameBit, "EDbColBit");
	_LIT8(KColNameReal32,"EDbColReal32");
	_LIT8(KColNameReal64,"EDbColReal64");
	_LIT8(KColNameDateTime,"EDbColDateTime");
	_LIT8(KColNameBinary,"EDbColBinary");
	_LIT8(KColNameText8,"EDbColText8");
	_LIT8(KColNameText16,"EDbColText16");
	_LIT8(KColNameLongBinary,"EDbColLongBinary");
	_LIT8(KColNameLongText8,"EDbColLongText8");
	_LIT8(KColNameLongText16,"EDbColLongText16");
	_LIT8(KFormattingPrefix, "| ");
	aDes.Append(KFormattingPrefix);
	switch ( col.iType )
		{
		case EDbColInt8:
			{
			aDes.Append(KColNameInt8);
			break;
			}
		case EDbColInt16:
			{
			aDes.Append(KColNameInt16);
			break;
			}
		case EDbColInt32:
			{
			aDes.Append(KColNameInt32);
			break;
			}
		case EDbColInt64:
			{
			aDes.Append(KColNameInt64);
			break;
			}
		case EDbColUint8:
			{
			aDes.Append(KColNameUInt8);
			break;
			}
		case EDbColUint16:
			{
			aDes.Append(KColNameUInt16);
			break;
			}
		case EDbColUint32:
			{
			aDes.Append(KColNameUInt32);
			break;
			}
		case EDbColBit:
			{
			aDes.Append(KColNameBit);
			break;
			}
		case EDbColReal32:
			{
			aDes.Append(KColNameReal32);
			break;
			}
		case EDbColReal64:
			{
			aDes.Append(KColNameReal64);
			break;
			}
		case EDbColDateTime:
			{
			aDes.Append(KColNameDateTime);
			break;
			}
		case EDbColBinary:
			{
			aDes.Append(KColNameBinary);
			break;
			}
		case EDbColText8:
			{
			aDes.Append(KColNameText8);
			aDes.Append('[');
			aDes.AppendNum(col.iMaxLength);
			aDes.Append(']');
			break;
			}
		case EDbColText16:
			{
			aDes.Append(KColNameText16);
			aDes.Append('[');
			aDes.AppendNum(col.iMaxLength);
			aDes.Append(']');
			break;
			}
		case EDbColLongBinary:
			{
			aDes.Append(KColNameLongBinary);
			break;
			}
		case EDbColLongText8:
			{
			aDes.Append(KColNameLongText8);
			break;
			}
		case EDbColLongText16:
			{
			aDes.Append(KColNameLongText16);
			break;
			}
		} // end switch
		TInt len = aDes.Length();
		for (TInt i = len; i < 46 ; i++ )
			aDes.Append(' ');

		aDes.Append(KFormattingPrefix);
		if ( col.iAttributes & TDbCol::ENotNull )
			{
			aDes.Append(KNotNull);
			}
		if ( col.iAttributes & TDbCol::EAutoIncrement )
			{
			aDes.Append(KAutoIncrement);
			}
	}


jstring DumpSchemaL(JNIEnv *aEnv, jobject, jint aPeer){
	_LIT8(KSchemaTableLabel, "Table: ");
	_LIT8(KSchemaIndexLabel, "| INDEX: ");
	_LIT8(KLine, "----------------------------------------------------------------------------\n");
	DbmsConnection* connection = (DbmsConnection*) aPeer;
	RDbNamedDatabase& db = connection->iDatabase;
	TInt KMaxSchemaSize = 1024*1024; // 1Mb

	CDbTableNames* tableNames = db.TableNamesL();
	RBuf8 resultBuf;
	RBuf retBuf;
	TBuf8<512> printBuf;
	TInt err = resultBuf.Create(KMaxSchemaSize);
	if ( err != KErrNone ) {
		ThrowExc(aEnv, err );
		return NULL;
	}
	err = retBuf.Create(KMaxSchemaSize);
	if ( err != KErrNone ) {
		resultBuf.Close();
		ThrowExc(aEnv, err );
		return NULL;
	}
	for (int i = 0; i < tableNames->Count(); i++ )
		{
		TBuf<64> tableName;
		tableName.Copy((*tableNames)[i]);
		printBuf.Copy(KSchemaTableLabel);
		printBuf.Append(tableName);
		resultBuf.Append(	KLine);
		resultBuf.Append(_L8("| "));
		resultBuf.Append(printBuf);
		for( TInt j = printBuf.Length() ; j < 74; j++)
			resultBuf.Append(_L8(" "));
		resultBuf.Append(_L8("|\n"));
		resultBuf.Append(KLine);

		CDbColSet* colSet = db.ColSetL(tableName);
		CleanupStack::PushL(colSet);
		printBuf.Zero();

		resultBuf.Append(_L8("| Column name               "));
		resultBuf.Append(_L8("| Type              "));
		resultBuf.Append(_L8("| Flags                    |\n"));
		resultBuf.Append(KLine);
		for ( TInt ii = 0 ; ii < colSet->Count() ; ii++ )
			{
			TDbCol col = (*colSet)[ii+1];
			TBuf8<128> cn;
			cn.Copy(col.iName);
			printBuf.Copy(cn);
			printBuf.Append(' ');
			for(TInt j = cn.Length() ; j < 25; j++ )
				printBuf.Append(' ');
			PrintColTypeName(printBuf, col);
			printBuf.Append(' ');
			resultBuf.Append(_L8("| "));
			resultBuf.Append(printBuf);
			for( TInt j = printBuf.Length() ; j < 74; j++)
				resultBuf.Append(_L8(" "));
			resultBuf.Append(_L8("|\n"));
			}

		CleanupStack::PopAndDestroy();

		// indices
		_LIT8(KPrimary, " PRIMARY KEY ");
		_LIT8(KUnique, " UNIQUE ");
		_LIT8(KFolded, " Comparison: FOLDED ");
		_LIT8(KCollated, " Comparison: COLLATED ");
		_LIT8(KNormal, " Comparison: NORMAL ");

		CDbIndexNames* indexNames = db.IndexNamesL(tableName);

		if ( indexNames->Count() > 0 )
			resultBuf.Append(KLine);

		for (int j = 0; j < indexNames->Count(); j++ )
			{
			TBuf<64> indexName;
			indexName.Copy((*indexNames)[j]);
			printBuf.Copy(KSchemaIndexLabel);
			printBuf.Append(indexName);
			printBuf.Append(' ');
			printBuf.Append('(');
			printBuf.Append(' ');
			CDbKey* key = db.KeyL(indexName, tableName);
			CleanupStack::PushL(key);
			// indexed columns
			for ( int k = 0 ; k < key->Count(); k ++ )
				{
				if ( k != 0 )
					{
					printBuf.Append(',');
					printBuf.Append(' ');
					}
				TDbKeyCol col = (*key)[k];
				printBuf.Append(col.iName);
				}
			printBuf.Append(' ');
			printBuf.Append(')');
			printBuf.Append(' ');
			if ( key->IsPrimary() )
				{
				printBuf.Append(KPrimary);
				}
			if ( key->IsUnique() )
				{
				printBuf.Append(KUnique);
				}
			switch ( key->Comparison() )
				{
				case EDbCompareNormal:
					{
					printBuf.Append(KNormal);
					break;
					}
				case EDbCompareFolded:
					{
					printBuf.Append(KFolded);
					break;
					}
				case EDbCompareCollated:
					{
					printBuf.Append(KCollated);
					break;
					}
				}
			CleanupStack::PopAndDestroy();

			resultBuf.Append(printBuf);
			for( TInt j = printBuf.Length() ; j < 76; j++)
				resultBuf.Append(_L8(" "));
			resultBuf.Append(_L8("|\n"));
			}

		resultBuf.Append(KLine);
		resultBuf.Append(_L8("\n"));
		}
	delete tableNames;
	// create java string - must copy into 16 bit desc
	retBuf.Copy(resultBuf);
	resultBuf.Close();
	jstring jret = CreateJavaString( aEnv, retBuf );
	retBuf.Close();
	return jret;
}

/*
 * Class:     com_symbian_dbms_jdbc_DbmsConnection
 * Method:    _schema
 * Signature: (I)Ljava/lang/String;
 */
JNIEXPORT jstring JNICALL Java_com_symbian_dbms_jdbc_DbmsConnection__1schema
  (JNIEnv *aEnv, jobject aObj, jint aPeer) {
	jstring ret = NULL;
	TRAPD(err, ret = DumpSchemaL(aEnv, aObj, aPeer));
	if ( err != KErrNone ) {
		ThrowExc(aEnv, err );
		return NULL;
	}
	return ret;
}