xref: /haiku/src/apps/installer/CopyEngine.h (revision 7f78035b8b9c9c43bfa0aa34b48c5ff4bd23bc3b)

/*
 * Copyright 2008-2009, Stephan Aßmus <superstippi@gmx.de>
 *  All rights reserved. Distributed under the terms of the MIT License.
 */
#ifndef COPY_ENGINE_H
#define COPY_ENGINE_H


#include <stdlib.h>

#include <Entry.h>
#include <File.h>
#include <Messenger.h>

#include "BlockingQueue.h"

class BFile;
class ProgressReporter;


class CopyEngine {
public:
								CopyEngine(ProgressReporter* reporter);
	virtual						~CopyEngine();

			void				ResetTargets(const char* source);
			status_t			CollectTargets(const char* source,
									sem_id cancelSemaphore = -1);

			status_t			CopyFolder(const char* source,
									const char* destination,
									sem_id cancelSemaphore = -1);

			status_t			CopyFile(const BEntry& entry,
									const BEntry& destination,
									sem_id cancelSemaphore = -1);

private:
			status_t			_CollectCopyInfo(const char* source,
									int32& level, sem_id cancelSemaphore);
			status_t			_CopyFolder(const char* source,
									const char* destination,
									int32& level, sem_id cancelSemaphore);

			bool				_ShouldCopyEntry(const BEntry& entry,
									const char* name,
									const struct stat& statInfo,
									int32 level) const;

			bool				_ShouldClobberFolder(const char* name,
									const struct stat& statInfo,
									int32 level) const;

			status_t			_RemoveFolder(BEntry& entry);

			void				_UpdateProgress();

	static	int32				_WriteThreadEntry(void* cookie);
			void				_WriteThread();

private:
			enum {
				BUFFER_COUNT	= 16,
				BUFFER_SIZE		= 1024 * 1024
			};
			struct Buffer {
				Buffer(BFile* file)
					:
					file(file),
					buffer(malloc(BUFFER_SIZE)),
					size(BUFFER_SIZE),
					validBytes(0),
					deleteFile(false)
				{
				}
				~Buffer()
				{
					if (deleteFile)
						delete file;
					free(buffer);
				}
				BFile*			file;
				void*			buffer;
				size_t			size;
				size_t			validBytes;
				bool			deleteFile;
			};

	BlockingQueue<Buffer>		fBufferQueue;

			thread_id			fWriterThread;
	volatile bool				fQuitting;

			off_t				fBytesRead;
			off_t				fLastBytesRead;
			uint64				fItemsCopied;
			uint64				fLastItemsCopied;
			bigtime_t			fTimeRead;

			off_t				fBytesWritten;
			bigtime_t			fTimeWritten;

			off_t				fBytesToCopy;
			uint64				fItemsToCopy;

			const char*			fCurrentTargetFolder;
			const char*			fCurrentItem;

			ProgressReporter*	fProgressReporter;

	// TODO: Should be made into a list of BEntris to be ignored, perhaps.
	// settable by method...
			BEntry				fVarDirectory;
};


#endif // COPY_ENGINE_H