/**************************************************************************************
 * Copyright (c) 2016, Tomoaki Yamaguchi
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * and Eclipse Distribution License v1.0 which accompany this distribution.
 *
 * The Eclipse Public License is available at
 *    http://www.eclipse.org/legal/epl-v10.html
 * and the Eclipse Distribution License is available at
 *   http://www.eclipse.org/org/documents/edl-v10.php.
 *
 * Contributors:
 *    Tomoaki Yamaguchi - initial API and implementation and/or initial documentation
 **************************************************************************************/

#include <string.h>
#include <stdio.h>

#include "LGwProxy.h"
#include "LMqttsnClientApp.h"
#include "LMqttsnClient.h"
#include "LScreen.h"

using namespace std;
using namespace linuxAsyncClient;

extern TaskList      theTaskList[];
extern TestList      theTestList[];
extern OnPublishList theOnPublishList[];
extern MQTTSNCONF;
extern UDPCONF;
extern void setup(void);
/*=====================================
          LMqttsnClient
 ======================================*/
LMqttsnClient* theClient = new LMqttsnClient();
LScreen* theScreen = new LScreen();
bool theOTAflag = false;
bool theClientMode = true;

int run(void)
{
	char c = 0;
	setup();
	printf("\n%s", PAHO_COPYRIGHT4);
	printf("\n%s\n", PAHO_COPYRIGHT0);
	printf("%s\n", PAHO_COPYRIGHT1);
	printf("%s\n", PAHO_COPYRIGHT2);
	printf(" *\n%s\n", PAHO_COPYRIGHT3);
	printf("%s\n", TESTER_VERSION);
	printf("%s\n", PAHO_COPYRIGHT4);

#ifndef CLIENT_MODE
	{
		theClientMode = false;
		PROMPT(" Do you like Tomoaki ? ( y/n ) : ");
		while (true)
		{
			if (CHECKKEYIN(&c))
			{
				if ( toupper(c) == 'N' )
				{
					ASSERT("\033[0;31m\n**** Sorry ****\033[0;37m\n\n");
					PROMPT("");
					return 0;
				}
			}
			else if ( toupper(c) == 'Y' )
			{
				ASSERT("\033[0m\033[0;32mAttempting to Connect the Broker.....\033[0m\033[0;37m\n");
				PROMPT("");
				break;
			}
		}
	}
#endif

	theClient->addTask(theClientMode);
	theClient->initialize( theNetcon, theMqcon);
	do
	{
		theClient->run();
	}
	while (theClientMode);

	delete theScreen;
	delete theClient;
	return 0;
}

/*=====================================
        Class LMqttsnClient
 ======================================*/
LMqttsnClient::LMqttsnClient()
{

}

LMqttsnClient::~LMqttsnClient()
{

}

void LMqttsnClient::initialize(LUdpConfig netconf, LMqttsnConfig mqconf)
{
	_gwProxy.initialize(netconf, mqconf);
	setSleepDuration(mqconf.sleepDuration);
}

void LMqttsnClient::addTask(bool clientMode)
{
	if ( clientMode )
	{
		_taskMgr.add(theTaskList);
	}
	else
	{
		_taskMgr.add(theTestList);
	}
}


LGwProxy* LMqttsnClient::getGwProxy(void)
{
	return &_gwProxy;
}

LPublishManager* LMqttsnClient::getPublishManager(void)
{
	return &_pubMgr;
}
;

LSubscribeManager* LMqttsnClient::getSubscribeManager(void)
{
	return &_subMgr;
}
;

LRegisterManager* LMqttsnClient::getRegisterManager(void)
{
	return _gwProxy.getRegisterManager();
}

LTaskManager* LMqttsnClient::getTaskManager(void)
{
	return &_taskMgr;
}
;

LTopicTable* LMqttsnClient::getTopicTable(void)
{
	return _gwProxy.getTopicTable();
}

void LMqttsnClient::publish(const char* topicName, Payload* payload, uint8_t qos, bool retain)
{
	_pubMgr.publish(topicName, payload, qos, retain);
}

void LMqttsnClient::publish(const char* topicName, uint8_t* payload, uint16_t len, uint8_t qos, bool retain)
{
	_pubMgr.publish(topicName, payload, len, qos, retain);
}

void LMqttsnClient::publish(uint16_t topicId, Payload* payload, uint8_t qos, bool retain)
{
	_pubMgr.publish(topicId, payload, qos, retain);
}

void LMqttsnClient::publish(uint16_t topicId, uint8_t* payload, uint16_t len, uint8_t qos, bool retain)
{
	_pubMgr.publish(topicId, payload, len, qos, retain);
}

void LMqttsnClient::subscribe(const char* topicName, TopicCallback onPublish, uint8_t qos)
{
	_subMgr.subscribe(topicName, onPublish, qos);
}

void LMqttsnClient::subscribe(uint16_t topicId, TopicCallback onPublish, uint8_t qos, uint8_t topicType)
{
	_subMgr.subscribe(topicId, onPublish, qos, topicType);
}

void LMqttsnClient::unsubscribe(const char* topicName)
{
	_subMgr.unsubscribe(topicName);
}

void LMqttsnClient::disconnect(uint16_t sleepInSecs)
{
	_gwProxy.disconnect(sleepInSecs);
}

void LMqttsnClient::run()
{
	_gwProxy.connect();
	_taskMgr.run();
	sleep();
}

void LMqttsnClient::sleep(void)
{

}

void LMqttsnClient::setSleepDuration(uint32_t duration)
{
	_sleepDuration = duration;
}

void LMqttsnClient::onConnect(void)
{
	_subMgr.onConnect();
}

const char* LMqttsnClient::getClientId(void)
{
	return _gwProxy.getClientId();
}