#pragma once
/******************************************************************************
*
*   Copyright (c) 2019 AT&T Intellectual Property.
*   Copyright (c) 2018-2019 Nokia.
*
*   Licensed under the Apache License, Version 2.0 (the "License");
*   you may not use this file except in compliance with the License.
*   You may obtain a copy of the License at
*
*       http://www.apache.org/licenses/LICENSE-2.0
*
*   Unless required by applicable law or agreed to in writing, software
*   distributed under the License is distributed on an "AS IS" BASIS,
*   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*   See the License for the specific language governing permissions and
*   limitations under the License.
*
******************************************************************************/

// Standard Includes: ANSI C/C++, MSA, and Third-Party Libraries
#include <climits>

// Local Includes: Application specific classes, functions, and libraries
#include "asn/utility.hpp"
#include "asn/buffer.hpp"
#include "asn/error_context.hpp"

namespace asn {
namespace ber {

/********************************************************************************
EncoderCtx
*********************************************************************************/
class EncoderCtx : boost::noncopyable
{
public:
	typedef buffer<u8*> buf_type;

	EncoderCtx(u8* data, size_t size)
		: m_buffer{ m_errCtx }
	{
		Reset(data, size);
	}

	template <size_t SIZE>
	explicit EncoderCtx(u8(&buff)[SIZE])
		: EncoderCtx(buff, SIZE)
	{
	}

	explicit EncoderCtx()
		: EncoderCtx(nullptr, 0)
	{
	}

	buf_type& refBuffer() { return m_buffer; }
	error_context& refErrorCtx() { return m_errCtx; }
	explicit operator bool() const { return static_cast<bool>(m_errCtx); }

	void Reset(void* data = nullptr, size_t size = 0)
	{
		m_buffer.reset(static_cast<u8*>(data), size);
		m_errCtx.reset();
	}

	//Name of the being processed IE
	void ie_name(const char* name) { m_errCtx.ie_name(name); }
	char const* ie_name() const { return m_errCtx.ie_name(); }

private:
	template <class IE, int IE_TYPE>
	friend struct Encode;

	error_context			m_errCtx;
	buf_type				m_buffer;
};


/********************************************************************************
DecoderCtx
*********************************************************************************/
class DecoderCtx : boost::noncopyable
{
public:
	typedef buffer<u8 const*>	buf_type;
	typedef allocator			alloc_type;

	DecoderCtx(const void* data = nullptr, size_t size = 0, void* alloc_buffer = nullptr, size_t alloc_buffer_size = 0)
		: m_buffer{ m_errCtx }
	{
		reset(data, size, alloc_buffer, alloc_buffer_size);
	}

	buf_type& refBuffer() { return m_buffer; }
	alloc_type& refAllocator() { return m_allocator; }
	error_context& refErrorCtx() { return m_errCtx; }
	explicit operator bool() const { return static_cast<bool>(m_errCtx); }

	void reset(const void* data = nullptr, size_t size = 0, void* alloc_buffer = nullptr, size_t alloc_buffer_size = 0)
	{
		m_allocator.reset(alloc_buffer, alloc_buffer_size);
		m_buffer.reset(static_cast<u8 const*>(data), size);
		m_errCtx.reset();
	}

	//Name of the being processed IE
	void ie_name(const char* name) { m_errCtx.ie_name(name); }
	char const* ie_name() const { return m_errCtx.ie_name(); }

private:
	error_context			m_errCtx;
	buf_type				m_buffer;
	alloc_type				m_allocator;
};

} //namespace ber
} //namespace asn