/*
 * Copyright 2010-2017 Haiku Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Christophe Huriaux, c.huriaux@gmail.com
 *		Adrien Destugues, pulkomandy@pulkomandy.tk
 */

#include <HttpResult.h>

#include <errno.h>
#include <Debug.h>


using std::ostream;

#ifndef LIBNETAPI_DEPRECATED
using namespace BPrivate::Network;
#endif


BHttpResult::BHttpResult(const BUrl& url)
	:
	fUrl(url),
	fHeaders(),
	fStatusCode(0)
{
}


BHttpResult::BHttpResult(BMessage* archive)
	:
	BUrlResult(archive),
	fUrl(archive->FindString("http:url")),
	fHeaders(),
	fStatusCode(archive->FindInt32("http:statusCode"))
{
	fStatusString = archive->FindString("http:statusString");

	BMessage headers;
	archive->FindMessage("http:headers", &headers);
	fHeaders.PopulateFromArchive(&headers);
}


BHttpResult::BHttpResult(const BHttpResult& other)
	:
	fUrl(other.fUrl),
	fHeaders(other.fHeaders),
	fStatusCode(other.fStatusCode),
	fStatusString(other.fStatusString)
{
}


BHttpResult::~BHttpResult()
{
}


// #pragma mark Result parameters modifications


void
BHttpResult::SetUrl(const BUrl& url)
{
	fUrl = url;
}


// #pragma mark Result parameters access


const BUrl&
BHttpResult::Url() const
{
	return fUrl;
}


BString
BHttpResult::ContentType() const
{
	return Headers()["Content-Type"];
}


size_t
BHttpResult::Length() const
{
	const char* length = Headers()["Content-Length"];
	if (length == NULL)
		return 0;

	/* NOTE: Not RFC7230 compliant:
	 * - If Content-Length is a list, all values must be checked and verified
	 *   to be duplicates of each other, but this is currently not supported.
	 */
	size_t result = 0;
	/* strtoul() will ignore a prefixed sign, so we verify that there aren't
	 * any before continuing (RFC7230 only permits digits).
	 *
	 * We can check length[0] directly because header values are trimmed by
	 * HttpHeader beforehand. */
	if (length[0] != '-' && length[0] != '+') {
		errno = 0;
		char *endptr = NULL;
		result = strtoul(length, &endptr, 10);
		/* ERANGE will be signalled if the result is too large (which can
		 * happen), in that case, return 0. */
		if (errno != 0 || *endptr != '\0')
			result = 0;
	}
	return result;
}


const BHttpHeaders&
BHttpResult::Headers() const
{
	return fHeaders;
}


int32
BHttpResult::StatusCode() const
{
	return fStatusCode;
}


const BString&
BHttpResult::StatusText() const
{
	return fStatusString;
}


// #pragma mark Result tests


bool
BHttpResult::HasHeaders() const
{
	return fHeaders.CountHeaders() > 0;
}


// #pragma mark Overloaded members


BHttpResult&
BHttpResult::operator=(const BHttpResult& other)
{
	if (this == &other)
		return *this;

	fUrl = other.fUrl;
	fHeaders = other.fHeaders;
	fStatusCode = other.fStatusCode;
	fStatusString = other.fStatusString;

	return *this;
}


status_t
BHttpResult::Archive(BMessage* target, bool deep) const
{
	status_t result = BUrlResult::Archive(target, deep);
	if (result != B_OK)
		return result;

	target->AddString("http:url", fUrl);
	target->AddInt32("http:statusCode", fStatusCode);
	target->AddString("http:statusString", fStatusString);

	BMessage headers;
	fHeaders.Archive(&headers);
	target->AddMessage("http:headers", &headers);

	return B_OK;
}


/*static*/ BArchivable*
BHttpResult::Instantiate(BMessage* archive)
{
	if (!validate_instantiation(archive, "BHttpResult"))
		return NULL;

	return new BHttpResult(archive);
}