/****
 * 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.
 *
 * Message.cpp
 *
 ****/

#include "include/Network/Mdns/Message.h"
#include "Packet.h"
#include <Data/HexString.h>
#include <debug_progmem.h>

namespace mDNS
{
bool Message::parse()
{
	// TODO: If it's truncated we can expect more data soon so we should wait for additional records before deciding whether to respond.
	// if(isTruncated())
	// {
	// }

	// Non zero Response code implies error.
	if(getResponseCode() != 0) {
		debug_w("Got errored MDNS message");
		return false;
	}

	Packet pkt{data, 4};

	// Number of incoming queries.
	auto questionsCount = pkt.read16();

	// Number of incoming answers.
	auto answersCount = pkt.read16();

	// Number of incoming Name Server resource records.
	auto nsCount = pkt.read16();

	// Number of incoming Additional resource records.
	auto additionalCount = pkt.read16();

	// List of questions
	for(uint16_t i = 0; i < questionsCount; i++) {
		auto question = new Question(*this);
		if(!question->parse(pkt)) {
			delete question;
			return false;
		}
		questions.add(question);
	}

	// List of answers, namespaces and additional records
	auto parseRecords = [&](Answer::Kind kind, uint16_t count) -> bool {
		for(uint16_t i = 0; i < count; i++) {
			auto answer = new Answer(*this, kind);
			if(!answer->parse(pkt)) {
				delete answer;
				return false;
			}
			answers.add(answer);
		}
		return true;
	};

	bool ok = parseRecords(Answer::Kind::answer, answersCount);
	if(ok) {
		ok = parseRecords(Answer::Kind::name, nsCount);
		if(ok) {
			ok = parseRecords(Answer::Kind::additional, additionalCount);
		}
	}

	return true;
}

Answer* Message::operator[](ResourceType type)
{
	for(auto& ans : answers) {
		if(ans.getType() == type) {
			return &ans;
		}
	}
	return nullptr;
}

} // namespace mDNS