//------------------------------------------------------------------------------
// <copyright file="MD5-2.h" company="LiquidRobotics">
//
// Copyright (c) 2014 Liquid Robotics, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
//
// This software is associated with a release.  Liquid Robotics makes no
// representation that the design, interfaces or component distribution will
// remain unchanged in future releases.
//
// </copyright>
//
// <summary>
// </summary>
//
// <owner></owner>
//------------------------------------------------------------------------------

#ifndef __MD5_2_H__
#define __MD5_2_H__

#include <iostream>
#include <stdexcept>
#include <vector>
#include <boost/utility.hpp>
#include <boost/serialization/nvp.hpp>
#include "LqrSMACByteArray.h"
#include "LqrSMACSharedLib.h"
#include "LqrSMACDeclareException.h"

namespace LiquidRobotics {
namespace PayloadInterface {

class SHARED_LIB_PUBLIC MD5 : boost::noncopyable
{
public:

	typedef SMACFramework::ByteArray::byte byte;
	typedef SMACFramework::ByteArray ArrayType;
	class Entity;

	DECLARE_EXCEPTION(FileNotFound, std::runtime_error);
	DECLARE_EXCEPTION(FileIOError, std::runtime_error);

	MD5();
	virtual ~MD5();

	template<typename IteratorType>
	void AppendIter(IteratorType i, int sequenceLength, const int bufferSize = 100)
	{
		ArrayType buffer(bufferSize);

		while(sequenceLength)
		{
			int bufferCount = 0;

			while(sequenceLength && (bufferCount < bufferSize))
			{
				buffer[bufferCount++] = *i;
				++i;
				--sequenceLength;
			}

			Append(&buffer.at(0), bufferCount);
		}
	}

	template<typename IteratorType>
	void Hash(IteratorType i, int sequenceLength, Entity& result, const int bufferSize = 100)
	{
			MD5 md5;
			md5.Append(i, sequenceLength);

			result = md5.Final();
	}

	void Append(const void* pData, int sequenceLength);
	void AppendFileContent(const std::string& fileName);
	Entity Final(bool resetContext = true);
	void Reset();

	class SHARED_LIB_PUBLIC Entity
	{
	public:

		Entity();
		Entity(const ArrayType& data);
		Entity(const byte* pData);
		Entity(const Entity& right);

		virtual ~Entity();

		Entity& operator=(const Entity& right);

		const ArrayType& GetBytes() const;
		void SetBytes(const ArrayType& newValue);

		bool operator==(const Entity& right) const;
		bool operator!=(const Entity& right) const;

	private:

		friend class boost::serialization::access;

		template<class Archive>
		void serialize(Archive& ar, const unsigned int version)
		{
			ar & BOOST_SERIALIZATION_NVP(bytes);
		}

		static void CheckArrayParam(const char* paramName, const char* caller, const ArrayType& data);

		static const unsigned MD5_Size;
		ArrayType bytes;
	};

private:

	void AllocateContext();

	void* pContext;
};

SHARED_LIB_PUBLIC std::ostream& operator<<(std::ostream& out, const MD5::Entity& right);

}
}

#endif
