/****
 * Sming Framework Project - Open Source framework for high efficiency native ESP8266 development.
 * Created 2015 by Skurydin Alexey
 * http://github.com/SmingHub/Sming
 * All files of the Sming Core are provided under the LGPL v3 license.
 *
 * HttpUpdater.cpp
 *
 */

#include "include/Ota/Network/HttpUpgrader.h"
#include <Ota/Manager.h>

namespace Ota
{
namespace Network
{
void HttpUpgrader::start()
{
	for(unsigned i = 0; i < items.count(); i++) {
		auto& it = items[i];
		debug_d("Download file:\r\n"
				"    (%u) %s -> %s @ 0x%X",
				currentItem, it.url.c_str(), it.partition.name().c_str(), it.partition.address());

		HttpRequest* request;
		if(baseRequest != nullptr) {
			request = baseRequest->clone();
			request->setURL(it.url);
		} else {
			request = new HttpRequest(it.url);
		}

		request->setMethod(HTTP_GET);
		request->setResponseStream(it.getStream());

		if(i == items.count() - 1) {
			request->onRequestComplete(RequestCompletedDelegate(&HttpUpgrader::updateComplete, this));
		} else {
			request->onRequestComplete(RequestCompletedDelegate(&HttpUpgrader::itemComplete, this));
		}

		if(!send(request)) {
			debug_e("ERROR: Rejected sending new request.");
			break;
		}
	}
}

int HttpUpgrader::itemComplete(HttpConnection& client, bool success)
{
	if(!success) {
		updateFailed();
		return -1;
	}

	auto& it = items[currentItem];
	debug_d("Finished: URL: %s, Offset: 0x%X, Length: %u", it.url.c_str(), it.partition.address(),
			it.stream->available());

	it.size = it.stream->available();
	it.stream = nullptr; // the actual deletion will happen outside of this class
	currentItem++;

	return 0;
}

int HttpUpgrader::updateComplete(HttpConnection& client, bool success)
{
	int hasError = itemComplete(client, success);
	if(hasError != 0) {
		return hasError;
	}

	debug_d("\r\nFirmware download finished!");
	for(unsigned i = 0; i < items.count(); i++) {
		debug_d(" - item: %u, addr: 0x%X, url: %s", i, items[i].partition.address(), items[i].url.c_str());
	}

	if(!success) {
		updateFailed();
		return -1;
	}

	if(updateDelegate) {
		updateDelegate(*this, true);
	}

	applyUpdate();

	return 0;
}

void HttpUpgrader::updateFailed()
{
	debug_e("\r\nFirmware download failed..");
	if(updateDelegate) {
		updateDelegate(*this, false);
	}
	items.clear();
}

void HttpUpgrader::applyUpdate()
{
	items.clear();
	if(romSlot == NO_ROM_SWITCH) {
		debug_d("Firmware updated.");
		return;
	}

	// set to boot new rom and then reboot
	debug_d("Firmware updated, rebooting to rom %u...\r\n", romSlot);

	OtaManager.setBootPartition(items[romSlot].partition);
	System.restart();
}

} // namespace Network

} // namespace Ota