// vtc_base.cpp
// 
// Copyright (c) 2009 - 2010 Accenture. All rights reserved.
// This component and the accompanying materials are made available
// under the terms of the "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:
// Accenture - Initial contribution
//

#include "vtc_base.h"
#include <fshell/common.mmh>

_LIT(KIniFileName, "\\system\\console\\vt100.ini");
_LIT(KIniDesciptionFile, "\\resource\\vt100.idf");
_LIT(KAttConsoleSizeDetect, "console_size_detect");

_LIT(KNewLine, "\r\n");

class TOverflowTruncate : public TDes16Overflow
	{
public:
	virtual void Overflow(TDes16&) {}
	};
	
EXPORT_C CVtcConsoleBase::CVtcConsoleBase()
	{
	}

EXPORT_C CVtcConsoleBase::~CVtcConsoleBase()
	{
	delete iIniFile;
	delete iInputController;
	delete iOutputController;
	delete iUnderlyingConsole; // In case of leave during construction, this might still be non-null
	}

EXPORT_C TInt CVtcConsoleBase::Create(const TDesC& aTitle, TSize /*aSize*/)
	{
	TRAPD(err, ConstructL(aTitle));
	if (err)
		{
		TBuf<512> message;
		message.Format(_L("Failed to create console (%d)."), err);
			
		if (iUnderlyingConsole && (LazyConsole::IsConstructed(iUnderlyingConsole) || !LazyConsole::IsLazy(iUnderlyingConsole)))
		// if we have an underlyconsole, which is either not lazy or is lazy but already constructed, then print the error to it.
			{
			iUnderlyingConsole->Write(message);
			iUnderlyingConsole->Write(KNewLine);
			}
		else
		// else display a dialog
			{
			RNotifier notifier;
			if (notifier.Connect() == KErrNone)
				{
				TInt buttonVal;
				TRequestStatus notifierStatus;
				notifier.Notify(_L("vt100"), message, _L("OK"), KNullDesC, buttonVal, notifierStatus);
				User::WaitForRequest(notifierStatus);
				notifier.Close();
				}
			}
		}
	Message(EDebug, _L("VT100 console create completed with err=%d"), err);
	return err;
	}

EXPORT_C void CVtcConsoleBase::ConstructL(const TDesC&)
	{
	iIniFile = LtkUtils::CIniFile::NewL(KIniFileName, KIniDesciptionFile);

	TSize screenSize(80, 24); // If sizeDetect is not specified, we default to (and only support) 80x24
	if (iIniFile->GetBool(KAttConsoleSizeDetect))
		{
		DetectScreenSizeL(screenSize);
		}
	iOutputController = CVtConsoleOutputController::NewL(*this, *iIniFile, screenSize);
	iInputController = CVtConsoleInputController::NewL(*this, *iIniFile);
	ClearScreen();
	delete iUnderlyingConsole;
	iUnderlyingConsole = NULL;
	}
	
EXPORT_C TInt CVtcConsoleBase::Extension_(TUint aExtensionId, TAny*& a0, TAny* a1)
	{
	if (aExtensionId == ConsoleMode::KSetConsoleModeExtension)
		{
		ConsoleMode::TMode mode = (ConsoleMode::TMode)(TInt)a1;
		iInputController->SetMode(mode);
		iOutputController->SetMode(mode);
		return KErrNone;
		}
	else if (aExtensionId == UnderlyingConsole::KSetUnderlyingConsoleExtension)
		{
		iUnderlyingConsole = (CConsoleBase*)a1;
		return KErrNone;
		}
	else if (aExtensionId == ConsoleAttributes::KSetConsoleAttributesExtension)
		{
		ConsoleAttributes::TAttributes* attributes = (ConsoleAttributes::TAttributes*)a1;
		return iOutputController->SetAttributes(attributes->iAttributes, attributes->iForegroundColor, attributes->iBackgroundColor);
		}
	else
		{
		return CConsoleBase::Extension_(aExtensionId, a0, a1);
		}
	}

EXPORT_C void CVtcConsoleBase::Message(TVerbosity aVerbosity, TRefByValue<const TDesC> aFmt, ...)
	{
	if (Debug() || (aVerbosity == EInformation) || (aVerbosity == EError))
		{
		TOverflowTruncate overflow;
		VA_LIST list;
		VA_START(list, aFmt);
		TBuf<0x100> buf;
		buf.AppendFormatList(aFmt, list, &overflow);
		
		if (iUnderlyingConsole)
			{
			iUnderlyingConsole->Write(buf);
			iUnderlyingConsole->Write(KNewLine);
			}
		else
			{
			// Cover all the bases
			User::InfoPrint(buf);
			RDebug::Print(buf);
			}
		}
	}
	
EXPORT_C TBool CVtcConsoleBase::Debug()
	{
	return iDebug;
	}
	
EXPORT_C void CVtcConsoleBase::SetDebug(TBool aDebug)
	{
	iDebug = aDebug;
	}
		
EXPORT_C void CVtcConsoleBase::Read(TRequestStatus& aStatus)
	{
	iInputController->GetKeyPress(iKeyPress, aStatus);
	}

EXPORT_C void CVtcConsoleBase::ReadCancel()
	{
	iInputController->CancelGetKeyPress();
	}

EXPORT_C void CVtcConsoleBase::Write(const TDesC& aDes)
	{
	iOutputController->Write(aDes);
	}
	
EXPORT_C TPoint CVtcConsoleBase::CursorPos() const
	{
	TPoint pos;
	iOutputController->GetCursorPos(pos);
	return pos;
	}

EXPORT_C void CVtcConsoleBase::SetCursorPosAbs(const TPoint& aPoint)
	{
	iOutputController->SetCursorPosAbs(aPoint);
	}

EXPORT_C void CVtcConsoleBase::SetCursorPosRel(const TPoint& aPoint)
	{
	iOutputController->SetCursorPosRel(aPoint);
	}

EXPORT_C void CVtcConsoleBase::SetCursorHeight(TInt aPercentage)
	{
	iOutputController->SetCursorHeight(aPercentage);
	}

EXPORT_C void CVtcConsoleBase::SetTitle(const TDesC& aDes)
	{
	iOutputController->SetTitle(aDes);
	}

EXPORT_C void CVtcConsoleBase::ClearScreen()
	{
	iOutputController->ClearScreen();
	}

EXPORT_C void CVtcConsoleBase::ClearToEndOfLine()
	{
	iOutputController->ClearToEndOfLine();
	}

EXPORT_C TSize CVtcConsoleBase::ScreenSize() const
	{
	TSize size;
	iOutputController->GetScreenSize(size);
	return size;
	}

EXPORT_C TKeyCode CVtcConsoleBase::KeyCode() const
	{
	return iKeyPress.iCode;
	}

EXPORT_C TUint CVtcConsoleBase::KeyModifiers() const
	{
	return iKeyPress.iModifiers;
	}

void CVtcConsoleBase::DetectScreenSizeL(TSize& aSize)
	{
	TInt err;
	do
		{
		// If we get a KErrCorrupt error during the detect, we restart it from the top
		// This is because KErrCorrupt is (hopefully) only ever returned as a result of the DSR in ReadCursorPos failing, in which case it is worth retrying.
		// If it's any other error, the console is actually dead so we should bail
		err = DoDetectScreenSize(aSize);
		Message(EDebug, _L("DoDetectScreenSize returned %d"), err);
		}
#ifdef FSHELL_PLATFORM_OPP // Temporary OPP specific change - the drivers for the mid-sized prototype currently complete requests with KErrAbort just before power management sends the device to sleep.
	while ((err == KErrCorrupt) || (err == KErrAbort));
#else
	while (err == KErrCorrupt);
#endif
	User::LeaveIfError(err);
	}

TInt CVtcConsoleBase::DoDetectScreenSize(TSize& aSize)
	{
	Message(EDebug, _L("Beginning VT100 console size detect"));
	_LIT8(KSpace, " ");
	_LIT8(KNewLine, "\r\n");
	_LIT8(KResetCursorPosAbs, "\x1b[1;1H");
	TInt err = Output(KResetCursorPosAbs);
	if (err) return err;

#ifdef FSHELL_VT100_WORK_AROUND_TERATERM_CURSOR_BUG
	// Note, at the point where the cursor has just wrap to the next line, TeraTerm reports
	// the same cursor position has just before is wrapped. When the cursor is moved to the
	// right again, TeraTerm corrects its reckoning of the cursor position. The following
	// code works around this bug by keeping track of the previous cursor position and noticing
	// if it doesn't increment. This is treated as though the cursor had wrapped.

	TInt previousCursorPosX = -1;
	for (TInt x = 0; ; ++x)
		{
		err = Output(KSpace);
		if (err) return err;
		TPoint pos;
		TInt err = ReadCursorPos(pos);
		if (err) return err;
		TInt cursorPosX = pos.iX;
		if ((cursorPosX == 0) || (cursorPosX == previousCursorPosX))
			{
			aSize.iWidth = x + 1;
			break;
			}
		previousCursorPosX = cursorPosX;
		}
#else
	for (TInt x = 0; ; ++x)
		{
		err = Output(KSpace);
		if (err) return err;
		TPoint pos;
		err = ReadCursorPos(pos);
		if (err) return err;
		if (pos.iX == 0)
			{
			aSize.iWidth = x + 1;
			break;
			}
		}
#endif
	err = Output(KResetCursorPosAbs);
	if (err) return err;
	TInt prevYPos = 0;
	for (TInt y = 0; ; ++y)
		{
		err = Output(KNewLine);
		if (err) return err;
		TPoint pos;
		TInt err = ReadCursorPos(pos);
		if (err) return err;
		if (pos.iY == prevYPos)
			{
			aSize.iHeight = y;
			break;
			}
		else
			{
			prevYPos = y;
			}
		}
	err = Output(KResetCursorPosAbs);
	Message(EDebug, _L("Completed VT100 console size detect = %dx%d"), aSize.iWidth, aSize.iHeight);
	return err;
	}

TInt CVtcConsoleBase::ReadCursorPos(TPoint& aPosition)
	{
	// Ideally this functionality should be moved into the input and output controllers, but currently
	// this is problematic because:
	//
	// 1) TeraTerm contains bugs in the way it reports cursor position that need to be worked around.
	//    See FSHELL_VT100_WORK_AROUND_TERATERM_CURSOR_BUG.
	//
	// 2) The CVtConsoleInputController state machine would need some fairly heavy changes to be able
	//    to cope with ANSI Device Status Report escape sequences.
	//
	// 3) The fact that CVtConsoleInputController and CVtConsoleOutputController run in the same thread
	//    means that implementing CVtcConsoleBase::CursorPos (which is synchronous) would be tricky.
	//    Possible solutions are a) using CActiveSchedulerWait (which is almost always a bad idea) and
	//    b) putting CVtConsoleInputController in a separate thread (which would be a fair bit of work,
	//       and we've been there before...).
	//
	// For the time being then, vt100.dll has its own console size detection code (a duplication of
	// the version in iosrv) and uses this function only during construction to initialize the size of
	// the output controller's cursor tracker. From that point on, the cursor position is always retrieved
	// via the cursor tracker (the horrible thing that it is).

	// Note, this function can only safely be used before the input and output controllers are created
	// because in uses the MConsoleInput and MConsoleOutput interfaces directly.
	ASSERT((iOutputController == NULL) && (iInputController == NULL));

	_LIT8(KDeviceStatusReport, "\x1b[6n");
	//Message(EDebug, _L("Sending VT100 DSR"));
	TInt err = Output(KDeviceStatusReport);
	if (err < 0)
		{
		Message(EDebug, _L("Error %d sending DSR"), err);
		return err;
		}
	
	TPoint pos;
	TBuf8<32> buf;
	TPtr8 ptr(const_cast<TUint8*>(buf.Ptr()), 0, buf.MaxLength());
	FOREVER
		{
		TRequestStatus status;
		//Message(EDebug, _L("Waiting for input"));
		Input(ptr, status);
		User::WaitForRequest(status);
		err = status.Int();
		if (err < 0)
			{
			Message(EDebug, _L("Error getting input %d"), err);
			return err;
			}

		buf.SetLength(buf.Length() + ptr.Length());
		ptr.Set(const_cast<TUint8*>(buf.Ptr()) + buf.Length(), 0, buf.MaxLength() - buf.Length());
		if (ptr.MaxLength() == 0) return KErrOverflow; // Not sure how this could happen, maybe if the terminal returned really really long but otherwise valid numbers for the x and y pos

		// The sequence we're looking for is \x1b[nnn;nnnR
		TLex8 lex(buf);
		if (lex.Eos()) continue;
		if (lex.Get() != 0x1b) return KErrCorrupt;
		if (lex.Eos()) continue;
		if (lex.Get() != '[') return KErrCorrupt;
		if (lex.Eos()) continue;
		err = lex.Val(pos.iY);
		if (err) return err;
		if (lex.Eos()) continue;
		if (lex.Get() != ';') return KErrCorrupt;
		if (lex.Eos()) continue;
		err = lex.Val(pos.iX);
		if (err) return err;
		if (lex.Eos()) continue;
		if (lex.Get() != 'R') return KErrCorrupt;
		// If we reach here, we've successfully read the whole sequence

		// I assume we subtract one here because the VT100 indexes are 1-based and we use zero-based? -TomS
		pos.iX--;
		pos.iY--;
		aPosition = pos;
		return KErrNone;
		}
	}

EXPORT_C TInt CVtcConsoleBase::ReadKeywordValuePair(TLex& aLex, TPtrC& aKeyword, TPtrC& aValue)
	{
	TLexMark mark;
	aLex.SkipSpaceAndMark(mark);
	while (!aLex.Eos() && !aLex.Peek().IsSpace() && (aLex.Peek() != '='))
		{
		aLex.Get();
		}
	aKeyword.Set(aLex.MarkedToken(mark));
	aLex.SkipSpace();
	if (aLex.Get() != '=')
		{
		return KErrArgument;
		}
	aLex.SkipSpaceAndMark(mark);
	while (!aLex.Eos() && (aLex.Peek() != ','))
		{
		aLex.Get();
		}
	aValue.Set(aLex.MarkedToken(mark));
	if (aLex.Peek() == ',')
		{
		aLex.Get();
		}
	return KErrNone;
	}