/*
 * (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 "RtfLex.h"
#include "RtfProperty.h"
#include "RtfDocument.h"

class RtfConvertationManager;

class RtfReader
{
public: 
	class ReaderState;
	typedef boost::shared_ptr<ReaderState> ReaderStatePtr;

    RtfConvertationManager *m_convertationManager;
	
	class ReaderState
	{
		public: 
            int                     m_nUD; // количество символов игнорируемых за юникодом
            RtfCharProperty         m_oCharProp;
            RtfParagraphProperty    m_oParagraphProp;
            RtfRowProperty          m_oRowProperty;
            RtfCellProperty         m_oCellProperty;
            RtfOldList              m_oCurOldList;
		//	RtfSectionProperty		m_oSectionProp;
			ReaderStatePtr			m_pSaveState;
            std::string             m_sCurText;
			
			ReaderState()
			{
				m_nUD = 1;
				m_oCharProp.SetDefaultRtf();
				m_oParagraphProp.SetDefaultRtf();
				m_oRowProperty.SetDefaultRtf();
				m_oCellProperty.SetDefaultRtf();
				m_oCurOldList.SetDefault();
			}
	};

//---------------------------------TextParser----------------------------------
    ReaderStatePtr      m_oState;
    RtfSectionProperty  m_oCurSectionProp;
    RtfLex              m_oLex;
    int                 m_nFootnote; //толко для симовола chftn. основано на том что вложенных footnote быть не может
    int                 m_nDefFont;
    std::wstring             m_sTempFolder;

	RtfReader(RtfDocument& oDocument, std::wstring sFilename );
	~RtfReader()
	{
	}
	void PushState();
	void PopState();
	bool Load();
	long GetProgress()
	{
		return (long)( g_cdMaxPercent * m_oLex.GetProgress());
	}
	void Stop()
	{
		m_oLex.CloseSource();
	}
private:
	RtfDocument& m_oDocument;
	std::wstring m_sFilename;
};

class RtfAbstractReader
{
public:
	NFileWriter::CBufferedFileWriter* m_oFileWriter;

	RtfAbstractReader()
	{
		m_bCanStartNewReader = false;
		m_bSkip = false;
		m_nSkipChars = 0;
		m_nCurGroups = 1;
		m_oFileWriter = NULL;
		m_bStopReader = false;

		m_bUseGlobalCodepage = false;
	}
	bool Parse(RtfDocument& oDocument, RtfReader& oReader )
	{
		NFileWriter::CBufferedFileWriter* poOldWriter = oReader.m_oLex.m_oFileWriter;
		oReader.m_oLex.m_oFileWriter = m_oFileWriter;

		int res = 0;
		m_oTok = oReader.m_oLex.NextCurToken();
		if( m_oTok.Type == m_oTok.None )
			m_oTok = oReader.m_oLex.NextToken();
	
		while (m_oTok.Type != RtfToken::Eof && false == m_bStopReader)
		{
			switch (m_oTok.Type)
			{
				case RtfToken::GroupStart:
						ExecuteTextInternal2(oDocument, oReader, m_oTok.Key, m_nSkipChars);
						PushState(oReader);
						break;
				case RtfToken::GroupEnd:
						ExecuteTextInternal2(oDocument, oReader, m_oTok.Key, m_nSkipChars);
						PopState(oDocument, oReader);
						break;
				case RtfToken::Keyword:
						ExecuteTextInternal2(oDocument, oReader, m_oTok.Key, m_nSkipChars);
                        if( m_oTok.Key == "u")
						{
							ExecuteText( oDocument, oReader, ExecuteTextInternal( oDocument, oReader, m_oTok.Key, m_oTok.HasParameter, m_oTok.Parameter, m_nSkipChars) );
							break;
						}
						else
						{
							if( true == m_bSkip )
							{
                                if( false ==  ExecuteCommand( oDocument, oReader, m_oTok.Key, m_oTok.HasParameter, m_oTok.Parameter ) )
									Skip( oDocument, oReader );
								m_bSkip = false;
							}
							else
                                ExecuteCommand( oDocument, oReader, m_oTok.Key, m_oTok.HasParameter, m_oTok.Parameter );
						}
						if( true == m_bCanStartNewReader )
							m_bCanStartNewReader = false;
						break;
				case RtfToken::Control:
                        if( m_oTok.Key == "42" )
							m_bSkip = true;
                        if( m_oTok.Key == "39" && true == m_oTok.HasParameter )
                            oReader.m_oState->m_sCurText += m_oTok.Parameter ;
						break;
				case RtfToken::Text:
                        oReader.m_oState->m_sCurText += m_oTok.Key;
						break;
			}
			if( false == m_bStopReader)
				m_oTok = oReader.m_oLex.NextToken();
		}

		oReader.m_oLex.m_oFileWriter = poOldWriter;

		return true;
	}
	virtual void PushState(RtfReader& oReader)
	{
		oReader.PushState();
		m_nCurGroups++;
		m_bCanStartNewReader = true;
	}
	virtual void PopState(RtfDocument& oDocument, RtfReader& oReader)
	{
		if( m_nCurGroups > 0 )
			m_nCurGroups--;
		else
			;//ATLASSERT(false);
		if( m_nCurGroups == 0 )
		{
			m_bStopReader = true;
			ExitReader( oDocument, oReader );
		}
		oReader.PopState();
		if( m_nCurGroups == 0 )
			ExitReader2( oDocument, oReader );
	}
	bool StartSubReader( RtfAbstractReader& poNewReader, RtfDocument& oDocument, RtfReader& oReader  )
	{
		if( true == m_bCanStartNewReader )
		{
			m_bCanStartNewReader = false;
			m_nCurGroups--;
			//poNewReader.PushState( oReader );
			poNewReader.m_bSkip = m_bSkip;
			return poNewReader.Parse(oDocument, oReader);
		}
		return false;
	}
	void Skip( RtfDocument& oDocument, RtfReader& oReader )
	{
		int cGroup = 1;
		while( cGroup >= 1 )
		{
			m_oTok = oReader.m_oLex.NextToken();
			if(m_oTok.Type == RtfToken::GroupStart)
				cGroup++;
			else if(m_oTok.Type == RtfToken::GroupEnd)
				cGroup--;
			else if(m_oTok.Type == RtfToken::Eof)
				break;
		}
		PopState( oDocument, oReader );
	}
    virtual bool ExecuteCommand( RtfDocument& oDocument, RtfReader& oReader, std::string sKey, bool bHasPar, int nPar )
	{
		return true;
	}
    virtual void ExecuteText( RtfDocument& oDocument, RtfReader& oReader, std::wstring oText )
	{
	}
	virtual void ExitReader( RtfDocument& oDocument, RtfReader& oReader )
	{
	}
	virtual void ExitReader2( RtfDocument& oDocument, RtfReader& oReader )
	{
	}
    /*static */std::wstring ExecuteTextInternal( RtfDocument& oDocument, RtfReader& oReader, std::string & sKey, bool bHasPar, int nPar, int& nSkipChars)
	{
        std::wstring sResult;

        if( "u" == sKey )
		{
			if( true == bHasPar )
                sResult += wchar_t( nPar );
		}
		else
		{
            std::string sCharString;
            if( "39" == sKey )
			{
				if( true == bHasPar )
                    sCharString += (char)nPar ;
			}
			else
				sCharString = sKey;

			sResult = ExecuteTextInternalCodePage(sCharString, oDocument, oReader);
		}
		ExecuteTextInternalSkipChars(sResult, oReader, sKey, nSkipChars);
		return sResult;
	}

    virtual void ExecuteTextInternal2( RtfDocument& oDocument, RtfReader& oReader, std::string & sKey, int& nSkipChars)
	{
        if(oReader.m_oState->m_sCurText.length() > 0)
		{
            std::wstring sResult = ExecuteTextInternalCodePage(oReader.m_oState->m_sCurText, oDocument, oReader);
            oReader.m_oState->m_sCurText.erase();
            if(sResult.length() > 0)
			{
                std::string str;
                ExecuteTextInternalSkipChars (sResult, oReader, str, nSkipChars);
				ExecuteText					 ( oDocument, oReader, sResult);
			}
		}
	}
    /*static */void ExecuteTextInternalSkipChars(std::wstring & sResult, RtfReader& oReader, std::string & sKey, int& nSkipChars)
	{
		//удаляем символы вслед за юникодом
		if( nSkipChars > 0 )
		{
            int nLength = (int)sResult.length();
			if( nSkipChars >= nLength )
			{
				nSkipChars -= nLength;
                sResult.clear();
			}
			else
			{
                sResult = sResult.substr( nSkipChars );
				nSkipChars = 0;
			}
		}
        if( "u" == sKey )
		{
			//надо правильно установить m_nSkipChars по значению \ucN
			nSkipChars = oReader.m_oState->m_nUD;
		}
	}
    /*static */std::wstring ExecuteTextInternalCodePage( std::string & sCharString, RtfDocument & oDocument, RtfReader & oReader);
	
	bool		m_bUseGlobalCodepage;

private:
	RtfToken	m_oTok;
	bool		m_bCanStartNewReader;

	int			m_nSkipChars;
	bool		m_bSkip;
	bool		m_bStopReader;

protected: 
	int			m_nCurGroups;

};