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

// Tiny Telnet - A simple telnet client to test CTelnetSession API
#include "TTELNET.H"
//#include <debug.h>


// Device driver names
#if defined (__WINS__)
#define PDD_NAME _L("ECDRV")
#define LDD_NAME _L("ECOMM")
#else
#define PDD_NAME _L("EUART1")
#define LDD_NAME _L("ECOMM")
#endif

static CConsoleBase* console;



LOCAL_C void CommInitL()
//
// Initialisation code - loads the serial LDD and PDD
// starts the comm subsystem (for EPOC32 builds)
// On a full EPOC implementation, this code would not
// be required because higher level components (EIKON) 
// automatically start the services.
//
	{
	// Load the physical device driver
	// The OS will automatically append .PDD and 
	// search /System/Libs on all drives.
	TInt r=User::LoadPhysicalDevice(PDD_NAME);
	if (r != KErrNone && r!= KErrAlreadyExists)
		User::Leave(r);
	//test(r==KErrNone || r==KErrAlreadyExists);

	// Similarly for the Logical device driver
	r=User::LoadLogicalDevice(LDD_NAME);
	if (r != KErrNone && r != KErrAlreadyExists)
		User::Leave(r);
	//test(r==KErrNone|| r==KErrAlreadyExists);

	}



void DriveEngineL();
void SetupConsoleL();

GLDEF_C TInt E32Main() // main function called by E32
    {
	__UHEAP_MARK;
	CTrapCleanup* cleanup=CTrapCleanup::New(); // get clean-up stack
	TRAPD(error,SetupConsoleL()); // more initialization, then do example
	__ASSERT_ALWAYS(!error,User::Panic(_L("TTiny Telnet"),error));
	delete cleanup; // destroy clean-up stack
	
	__UHEAP_MARKEND;
	return 0; // and return
    }

void SetupConsoleL() // initialize and call example code under cleanup stack
    {
	console=Console::NewL(_L("Tiny Telnet"),
		TSize(KDefaultConsWidth,KDefaultConsHeight));
	CleanupStack::PushL(console);
	TRAPD(error,DriveEngineL()); // perform example function
	if (error) console->Printf(_L("failed: leave code=%d"), error);
	else console->Printf(_L("ok"));
	console->Printf(_L(" [press any key]"));
	console->Getch(); // get and ignore character
	CleanupStack::PopAndDestroy(); // close console
    }

/*
	TTelnet Test code starts here
*/

void DriveEngineL()
	{

	//initialization for COMM port 
	CommInitL();
	
	// Setup Active Scheduler
	CActiveScheduler* myActiveScheduler = new(ELeave) CActiveScheduler;
	CleanupStack::PushL(myActiveScheduler);
	CActiveScheduler::Install(myActiveScheduler);

	// Create Tiny Telnet
	CTinyTelnet* aTinyTelnet = CTinyTelnet::NewLC(console);
	//.. and post request to object
	aTinyTelnet->RequestCharacter();

	myActiveScheduler->Start();

	CleanupStack::PopAndDestroy(2); //aTinyTelnet, myActiveScheduler	
	}

CTinyTelnet::CTinyTelnet() : CActiveConsole (EPriorityStandard)
	{
	}

CTinyTelnet::~CTinyTelnet() 
	{
	delete iTelnetSession;
	delete iCommandLine;
	}

CTinyTelnet* CTinyTelnet::NewLC(CConsoleBase* aConsole)
	{
	CTinyTelnet* self = new (ELeave) CTinyTelnet();
    CleanupStack::PushL(self);
	self->ConstructL(aConsole);
	return self;
	}

void CTinyTelnet::ConstructL(CConsoleBase* aConsole)
	{
	CActiveConsole::ConstructL(aConsole);

	iState = EComplete;
	iLocalEcho = FALSE;
	iBinaryOnOff = FALSE;

	iCommandLine=CLineEdit::NewL(iConsole,10);
	// TODO 
	// Modify iTelnetConfig members here for different speed window size etc
//	iConfig.iTermType = _L8("vt100");
	iConfig.iServerEcho = TRUE;
	iTelnetSession = CTelnetSession::NewL(iConfig,this);
	iConsole->Printf(_L("'<ESC>' for command prompt\n"));
	};

void CTinyTelnet::Error(TInt aError)
	{
	iConsole->Printf(_L("Error(%d)\n"),aError);
	}

void CTinyTelnet::ReadComplete(const TDesC8& aBuffer)
	{
	// I know the driver won't chuck more than 128 bytes at us
	// The real code should use variable descriptors on the heap
	TBuf<128> tmpBuffer;
	// For unicode, use the overloaded copy
	tmpBuffer.Copy(aBuffer);
	iConsole->Printf(_L("%S"),&tmpBuffer);
	Read();
	}

void CTinyTelnet::WriteComplete()
	{
	// Request another character
	RequestCharacter();
	return;
	}

void CTinyTelnet::ConnectionClosed()
	{
	iState = EDisconnected;
	iConsole->Printf(_L("ConnectionClosed()\n"));
	}

void CTinyTelnet::OptionsChanged()
	{
	TOptionStatus status;
	if(iState != EConnected)
		{
		iConsole->Printf(_L("No Connection , status not valid\n"));
		}
	else
		iTelnetSession->OptionStatus(status);
	}

void CTinyTelnet::Connected()
	{
	iConsole->Printf(_L("Connected()"));
	Read();
	iState = EConnected;
	}

void CTinyTelnet::DisplayOptionStates()
// It's debatable whether a user would ever want to see this stuff
// It's just here as demo of the call
	{
	if(iState != EConnected)
		{
		iConsole->Printf(_L("No Connection , status not valid\n"));
		}
	else
		{
		TOptionStatus aStatus;
		iTelnetSession->OptionStatus(aStatus);

		iConsole->Printf(_L("Server Send   Binary         Option = %d\n"),aStatus.iServerBinary);
		iConsole->Printf(_L("Client Send   Binary         Option = %d\n"),aStatus.iClientBinary);
		iConsole->Printf(_L("Server Echo   Data           Option = %d\n"),aStatus.iEcho);
		iConsole->Printf(_L("Client Supply Window Size    Option = %d\n"),aStatus.iNAWS);
		iConsole->Printf(_L("Client Supply Terminal Speed Option = %d\n"),aStatus.iTerminalSpeed);
		iConsole->Printf(_L("Client Supply Terminal Type  Option = %d\n"),aStatus.iTerminalType);
		iConsole->Printf(_L("Server Supply Status         Option = %d\n"),aStatus.iServerStatus);
		iConsole->Printf(_L("Client Supply Status         Option = %d\n"),aStatus.iClientStatus);
		}
	}

void CTinyTelnet::ProcessKeyPress(TChar aChar)
	{
	if (aChar != EKeyEscape)
		{
		// Send input Character to echo host
		if(Write(aChar) != KErrNone)
			{
			RequestCharacter();
			}
		else
			{
			if(iLocalEcho)
				iConsole->Printf(_L("%C"), TUint(aChar));
			}
		}
	else
		{
		if(DoCommand())
			RequestCharacter();
		}
	}

TBool CTinyTelnet::DoCommand()
// Handle command strings from the user
// Demo only
	{
	iCommandLine->Edit(_L("\nTinyTelnet>"), &iCommand);

	// Seperate Command from Command Arguments.
	TInt commandEnd;
	TBuf<80> commandArgs;
	TBool ret = TRUE;

	iCommand.TrimAll();  // Remove leading/trailing/double white spaces              
	
	commandEnd = iCommand.Locate(' ');

	if ( (commandEnd != iCommand.Length()) && (commandEnd != KErrNotFound) )
		{
		commandArgs = iCommand.Mid(commandEnd+1);
		iCommand.SetLength(commandEnd);	
		}
	
	if(iCommand.Length()==0) 
		Cancel(); // do nothing
	else if ( iCommand.MatchF(_L("open")) == 0 )
		// Attempt to open a telnet session
		Open(commandArgs);
	else if ( iCommand.MatchF(_L("status")) == 0 )
		{
		// Display the state of the Telnet RFC options
		DisplayOptionStates();
		}
	else if ( iCommand.MatchF(_L("openip")) == 0 )
		// Other open
		OpenIP(commandArgs);
	else if ( iCommand.MatchF(_L("close")) == 0 )
		// Close a connection
		Close(commandArgs);
	else if ( iCommand.MatchF(_L("help")) == 0 )
		Help(commandArgs);
	else if ( iCommand.MatchF(_L("echo")) == 0 )
		{
		// Toggle local echo
		LocalEcho(commandArgs);
		(iConfig.iServerEcho == TRUE) ? (iConfig.iServerEcho = FALSE) : (iConfig.iServerEcho = TRUE);
		iTelnetSession->DoModifyConfig(iConfig);
		}
	else if ( iCommand.MatchF(_L("Logout")) == 0 )
		// Force Telnet level logout to the host ,host could refuse 
		iTelnetSession->DoForceLogout();
	else if ( iCommand.MatchF(_L("AllowLogout")) == 0 )
		// Demo only Toggle allow server to logout option
		{
		(iConfig.iAllowLogout == TRUE) ? (iConfig.iAllowLogout = FALSE) : (iConfig.iAllowLogout = TRUE);
		iTelnetSession->DoModifyConfig(iConfig);
		}
	else if ( iCommand.MatchF(_L("Window")) == 0)
		// Demo only, changes the window size
		{
		iConfig.iWindowSize.x = 60;
		iConfig.iWindowSize.y = 20;
		iTelnetSession->DoModifyConfig(iConfig);
		}
	else if ( iCommand.MatchF(_L("Speed")) == 0)
		// Demo only, sets the speed
		{
		iConfig.iTermSpeed = _L8("9600");
		iTelnetSession->DoModifyConfig(iConfig);
		}
	else if ( iCommand.MatchF(_L("Type")) == 0)
		// Demo only, changes the terminal type
		// See API document implementation of terminal type configuration
		{
		iConfig.iTermType = _L8("vt100");
		iTelnetSession->DoModifyConfig(iConfig);
		}

	else if(iCommand.MatchF(_L("exit")) == 0)
		{
		// Clean exit, all destructors called
		ret = FALSE;
		CActiveScheduler::Stop();
		}
	else
		{
		iConsole->Printf(_L("command = %S.\n"),&iCommand);
		iConsole->Printf(_L("unknown command\n"));
		}
	return ret;
	}

void CTinyTelnet::DoCancel()
	{

	}

void CTinyTelnet::Open(TDesC& aCommandArg)
	{
	TInt err;
	iCommandArg = aCommandArg;
	err = iTelnetSession->Connect(iCommandArg);
	if(err != KErrNone)
		iState = EConnecting;
	iConsole->Printf(_L("Connect returned - %d\n"), err);
	}

void CTinyTelnet::Help(TDesC& /*aCommandArg*/)
	{
	iConsole->Printf(_L("\nHelp\n"));
	iConsole->Printf(_L("----\n\n"));
	iConsole->Printf(_L(" <esc>				   - command prompt\n"));
	iConsole->Printf(_L(" open <server>		   - connect to server on telnet port 23\n"));
	iConsole->Printf(_L(" openip <aa.bb.cc.dd> - connect to server on telnet port 23\n"));
	iConsole->Printf(_L(" close                - disconnect connection\n"));
	iConsole->Printf(_L(" echo                 - toggle local echo on/off\n"));
	iConsole->Printf(_L(" status               - print option status\n\n"));
	}

void CTinyTelnet::LocalEcho(TDesC& /*aCommandArg*/)
	{
	iLocalEcho = (!iLocalEcho);
	}

void CTinyTelnet::Logout(TDesC& /*aCommandArg*/)
	{
	iTelnetSession->DoForceLogout();
	}

void CTinyTelnet::OpenIP(TDesC& aCommandArg)
	{
	TInt err;
	iAddress.Input(aCommandArg);	// Accepts a, a.b, a.b.c, a.b.c.d or 0xaabbccdd
	err = iTelnetSession->Connect(iAddress);
	iConsole->Printf(_L("Connect returned - %d\n"), err);
	}

void CTinyTelnet::Read()
	{
	iTelnetSession->Read();
	}


TInt CTinyTelnet::Write(TChar aChar)
// Write characters to the Server
// For Demo, we are trapping :-
// CTRL+A AO  (Abort Output)
// CTRL+B AYT (Are You There)
// CTRL+C IP  (Interrupt Process)
// CTRL+D EC  (Erase Character, NVT only)
// CTRL+E EL  (Erase Line, NVT only)
// CTRL+F BRK (Break)
	{

	TInt ret = KErrNone;
	TTelnetUserControl code;
	if (iState==EConnected)
		{
		TBuf8<10> buffer;
		buffer.SetLength(0);

		if(aChar == 1)
			{
			// ABORT OUTPUT
			code = EAo;
			ret = iTelnetSession->Write(code);
			}
		else if(aChar == 2)
			{
			// ARE YOU THERE
			code = EAyt;
			ret = iTelnetSession->Write(code);
			}
		else if(aChar == 3)
			{
			// INTERRUPT PROCESS
			code = EIp;
			ret = iTelnetSession->Write(code);
			}
		else if(aChar == 4)
			{
			// ERASE CHARACTER
			code = EEc;
			ret = iTelnetSession->Write(code);
			}
		else if(aChar == 5)
			{
			// ERASE LINE
			code = EEl;
			ret = iTelnetSession->Write(code);
			}
		else if(aChar == 6)
			{
			// BREAK
			code = EBrk;
			ret = iTelnetSession->Write(code);
			}
		else
			{
			buffer.Append(aChar);
			ret = iTelnetSession->Write(buffer);
			}
		
		
		//iConsole->Printf(_L("Write returned - (%d)\n"), err);
		}
	else
		{
		ret = KErrDisconnected;
		iConsole->Printf(_L("Not connected [write failed]\n"));
		}
	return ret;
	}

void CTinyTelnet::Close(TDesC& /*aCommandArg*/)
	{
	TInt err;
	err = iTelnetSession->Disconnect();
	iConsole->Printf(_L("Disconnect returned - %d\n"), err);
	}


//
// CActiveConsole Implementation
//
void CActiveConsole::ConstructL(CConsoleBase* aConsole)
	{
	iConsole = aConsole;
	CActiveScheduler::Add(this);
	}

CActiveConsole::~CActiveConsole()
	{
	Cancel();
	}

void CActiveConsole::RequestCharacter()
	{
	__ASSERT_DEBUG(!IsActive(), User::Panic(_L("CActiveConsole::RequestCharacter()"), KRequestPending));
	iConsole->Read(iStatus);
	SetActive();
	}

void CActiveConsole::RunL()
	{
	ProcessKeyPress(TChar(iConsole->KeyCode()));
	}

void CActiveConsole::DoCancel()
	{
	iConsole->ReadCancel();
	}