/*
 * Copyright (c) 2011 Tobias Markmann
 * Licensed under the simplified BSD license.
 * See Documentation/Licenses/BSD-simplified.txt for more information.
 */

#include <Swiften/FileTransfer/IncrementalBytestreamHashCalculator.h>

#include <Swiften/StringCodecs/Hexify.h>
#include <Swiften/StringCodecs/MD5.h>
#include <Swiften/StringCodecs/SHA1.h>

namespace Swift {

IncrementalBytestreamHashCalculator::IncrementalBytestreamHashCalculator(bool doMD5, bool doSHA1) {
	md5Hasher = doMD5 ? new MD5() : NULL;
	sha1Hasher = doSHA1 ? new SHA1() : NULL;
}

IncrementalBytestreamHashCalculator::~IncrementalBytestreamHashCalculator() {
	delete md5Hasher;
	delete sha1Hasher;
}

void IncrementalBytestreamHashCalculator::feedData(const ByteArray& data) {
	if (md5Hasher) {
		md5Hasher->update(data);
	}
	if (sha1Hasher) {
		sha1Hasher->update(data);
	}
}
/*
void IncrementalBytestreamHashCalculator::feedData(const SafeByteArray& data) {
	if (md5Hasher) {
		md5Hasher->update(createByteArray(data.data(), data.size()));
	}
	if (sha1Hasher) {
		sha1Hasher->update(createByteArray(data.data(), data.size()));
	}
}*/

std::string IncrementalBytestreamHashCalculator::getSHA1String() {
	if (sha1Hasher) {
		ByteArray result = sha1Hasher->getHash();
		return Hexify::hexify(result);
	} else {
		return std::string();
	}
}

std::string IncrementalBytestreamHashCalculator::getMD5String() {
	if (md5Hasher) {
		ByteArray result = md5Hasher->getHash();
		return Hexify::hexify(result);
	} else {
		return std::string();
	}
}

}