/*
 * (c) Copyright Ascensio System SIA 2010-2017
 *
 * This program is a free software product. You can redistribute it and/or
 * modify it under the terms of the GNU Affero General Public License (AGPL)
 * version 3 as published by the Free Software Foundation. In accordance with
 * Section 7(a) of the GNU AGPL its Section 15 shall be amended to the effect
 * that Ascensio System SIA expressly excludes the warranty of non-infringement
 * of any third-party rights.
 *
 * This program is distributed WITHOUT ANY WARRANTY; without even the implied
 * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR  PURPOSE. For
 * details, see the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
 *
 * You can contact Ascensio System SIA at Lubanas st. 125a-25, Riga, Latvia,
 * EU, LV-1021.
 *
 * The  interactive user interfaces in modified source and object code versions
 * of the Program must display Appropriate Legal Notices, as required under
 * Section 5 of the GNU AGPL version 3.
 *
 * Pursuant to Section 7(b) of the License you must retain the original Product
 * logo when distributing the program. Pursuant to Section 7(e) we decline to
 * grant you any rights under trademark law for use of our trademarks.
 *
 * All the Product's GUI elements, including illustrations and icon sets, as
 * well as technical writing content are licensed under the terms of the
 * Creative Commons Attribution-ShareAlike 4.0 International. See the License
 * terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode
 *
 */
#pragma once

#include "../../Common/3dParty/pole/pole.h"

namespace DocFileFormat
{
	class StructuredStorageReader
	{
	public:
		StructuredStorageReader () : m_pStorage(NULL)
		{

		}
		~StructuredStorageReader()
		{
			if(m_pStorage)
			{
				delete m_pStorage;
				m_pStorage = NULL;
			}
		}

		bool SetFile (const wchar_t* filename)
		{
			m_pStorage = new POLE::Storage(filename);
			
			if (m_pStorage)//return true;
			{	
				return m_pStorage->open();
			}
			return false;
		}
		bool isDirectory( const std::string& name )
		{
			if (!m_pStorage) return false;
			
			return m_pStorage->isDirectory(name);
		}

		bool GetStream (const char *path, POLE::Stream** ppStream)
		{
			if (( m_pStorage != NULL ) && ( path != NULL ))
			{
				*ppStream = new POLE::Stream(m_pStorage, path);
			}
			if ((*ppStream)  && ((*ppStream)->size() > 0)) 
				return true;

			return false;
		}

		inline POLE::Storage* GetStorage()
		{
			return m_pStorage;
		}
	
		void copy( int indent, std::string path, POLE::Storage * storageOut, bool withRoot = true)
		{
			std::list<std::string> entries, entries_sort;
			entries = m_pStorage->entries( path );
			
			for( std::list<std::string>::iterator it = entries.begin(); it != entries.end(); it++ )
			{
				std::string name = *it;
				std::string fullname = path + name;
		       
				if( m_pStorage->isDirectory( fullname ) )
				{
					entries_sort.push_back(name);
				}
				else
				{
					entries_sort.push_front(name);
				}
			}
			//for( std::list<std::string>::iterator it = entries.begin(); it != entries.end(); it++ )
			for( std::list<std::string>::iterator it = entries_sort.begin(); it != entries_sort.end(); it++ )
			{
				std::string name = *it;
				std::string fullname = path + name;
		       
				if( m_pStorage->isDirectory( fullname ) )
				{
					copy( indent + 1, fullname + "/", storageOut, withRoot );
				}
				else
				{
					copy_stream(fullname, storageOut, withRoot);
				}
			}
		}

	private:
		void copy_stream(std::string streamName, POLE::Storage * storageOut, bool withRoot = true)
		{
			POLE::Stream *stream = new POLE::Stream(m_pStorage, streamName);
			if (!stream) return;

			stream->seek(0);
			int size_stream = stream->size();

			if (withRoot == false)
			{
				int pos = streamName.find("/");
				if (pos >= 0)
					streamName = streamName.substr(pos + 1);
			}
			
			POLE::Stream *streamNew = new POLE::Stream(storageOut, streamName, true, size_stream);
			if (!streamNew) return;

			unsigned char* data_stream = new unsigned char[size_stream];
			if (data_stream)
			{
				stream->read(data_stream, size_stream);

				streamNew->write(data_stream, size_stream);

				delete []data_stream;
				data_stream = NULL;
			}

			streamNew->flush();
					
			delete streamNew;
			delete stream;
		}

		POLE::Storage* m_pStorage;
	};
}