haiku-beta6/src/kits/package/FetchFileJob.cpp

/*
 * Copyright 2011-2015, Haiku, Inc. All Rights Reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Axel Dörfler <[email protected]>
 *		Rene Gollent <[email protected]>
 *		Oliver Tappe <[email protected]>
 */


#include "FetchFileJob.h"

#include <stdio.h>
#include <sys/wait.h>

#include <Path.h>

#ifdef HAIKU_TARGET_PLATFORM_HAIKU
#	include <HttpRequest.h>
#	include <UrlRequest.h>
#	include <UrlProtocolRoster.h>
#endif


namespace BPackageKit {

namespace BPrivate {


#ifdef HAIKU_TARGET_PLATFORM_HAIKU

FetchFileJob::FetchFileJob(const BContext& context, const BString& title,
	const BString& fileURL, const BEntry& targetEntry)
	:
	inherited(context, title),
	fFileURL(fileURL),
	fTargetEntry(targetEntry),
	fTargetFile(&targetEntry, B_CREATE_FILE | B_ERASE_FILE | B_WRITE_ONLY),
	fError(B_ERROR),
	fDownloadProgress(0.0)
{
}


FetchFileJob::~FetchFileJob()
{
}


float
FetchFileJob::DownloadProgress() const
{
	return fDownloadProgress;
}


const char*
FetchFileJob::DownloadURL() const
{
	return fFileURL.String();
}


const char*
FetchFileJob::DownloadFileName() const
{
	return fTargetEntry.Name();
}


off_t
FetchFileJob::DownloadBytes() const
{
	return fBytes;
}


off_t
FetchFileJob::DownloadTotalBytes() const
{
	return fTotalBytes;
}


status_t
FetchFileJob::Execute()
{
	status_t result = fTargetFile.InitCheck();
	if (result != B_OK)
		return result;

	BUrlRequest* request = BUrlProtocolRoster::MakeRequest(fFileURL.String(),
		this);
	if (request == NULL)
		return B_BAD_VALUE;

	thread_id thread = request->Run();
	wait_for_thread(thread, NULL);

	return fError;
}


void
FetchFileJob::DataReceived(BUrlRequest*, const char* data, off_t position,
	ssize_t size)
{
	fTargetFile.WriteAt(position, data, size);
}


void
FetchFileJob::DownloadProgress(BUrlRequest*, ssize_t bytesReceived,
	ssize_t bytesTotal)
{
	if (bytesTotal != 0) {
		fBytes = bytesReceived;
		fTotalBytes = bytesTotal;
		fDownloadProgress = (float)bytesReceived/bytesTotal;
		NotifyStateListeners();
	}
}


void
FetchFileJob::RequestCompleted(BUrlRequest* request, bool success)
{
	fError = request->Status();

	if (success) {
		const BHttpResult& outResult = dynamic_cast<const BHttpResult&>
			(request->Result());
		uint16 code = outResult.StatusCode();
		uint16 codeClass = BHttpRequest::StatusCodeClass(code);

		switch (codeClass) {
			case B_HTTP_STATUS_CLASS_CLIENT_ERROR:
			case B_HTTP_STATUS_CLASS_SERVER_ERROR:
				fError = B_IO_ERROR;
				break;
			default:
				fError = B_OK;
				break;
		}
		switch (code) {
			case B_HTTP_STATUS_OK:
				fError = B_OK;
				break;
			case B_HTTP_STATUS_PARTIAL_CONTENT:
				fError = B_PARTIAL_READ;
				break;
			case B_HTTP_STATUS_REQUEST_TIMEOUT:
			case B_HTTP_STATUS_GATEWAY_TIMEOUT:
				fError = B_DEV_TIMEOUT;
				break;
			case B_HTTP_STATUS_NOT_IMPLEMENTED:
			case B_HTTP_STATUS_UNSUPPORTED_MEDIA_TYPE:
				fError = B_NOT_SUPPORTED;
				break;
			case B_HTTP_STATUS_UNAUTHORIZED:
				fError = B_PERMISSION_DENIED;
				break;
			case B_HTTP_STATUS_FORBIDDEN:
			case B_HTTP_STATUS_METHOD_NOT_ALLOWED:
			case B_HTTP_STATUS_NOT_ACCEPTABLE:
				fError = B_NOT_ALLOWED;
				break;
			case B_HTTP_STATUS_NOT_FOUND:
				fError = B_NAME_NOT_FOUND;
				break;
			case B_HTTP_STATUS_BAD_GATEWAY:
				fError = B_BAD_DATA;
				break;
			default:
				break;
		}
	}
}


void
FetchFileJob::Cleanup(status_t jobResult)
{
	if (jobResult != B_OK)
		fTargetEntry.Remove();
}


#else // HAIKU_TARGET_PLATFORM_HAIKU


FetchFileJob::FetchFileJob(const BContext& context, const BString& title,
	const BString& fileURL, const BEntry& targetEntry)
	:
	inherited(context, title),
	fFileURL(fileURL),
	fTargetEntry(targetEntry),
	fTargetFile(&targetEntry, B_CREATE_FILE | B_ERASE_FILE | B_WRITE_ONLY),
	fDownloadProgress(0.0)
{
}


FetchFileJob::~FetchFileJob()
{
}


float
FetchFileJob::DownloadProgress() const
{
	return fDownloadProgress;
}


const char*
FetchFileJob::DownloadURL() const
{
	return fFileURL.String();
}


const char*
FetchFileJob::DownloadFileName() const
{
	return fTargetEntry.Name();
}


off_t
FetchFileJob::DownloadBytes() const
{
	return fBytes;
}


off_t
FetchFileJob::DownloadTotalBytes() const
{
	return fTotalBytes;
}


status_t
FetchFileJob::Execute()
{
	return B_UNSUPPORTED;
}


void
FetchFileJob::Cleanup(status_t jobResult)
{
	if (jobResult != B_OK)
		fTargetEntry.Remove();
}


#endif // HAIKU_TARGET_PLATFORM_HAIKU

}	// namespace BPrivate

}	// namespace BPackageKit