/*
 * (c) Copyright Ascensio System SIA 2010-2019
 *
 * 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 20A-12 Ernesta Birznieka-Upisha
 * street, Riga, Latvia, EU, LV-1050.
 *
 * 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 "PPTFileDefines.h"
#include "../Enums/RecordType.h"
#include "../Reader/ReadStructures.h"
#include "../../Common/Vml/PPTShape/Enums.h"
#include "../../XlsFile/Format/Binary/CFStream.h"
#include "../../../Common/3dParty/pole/pole.h"
#include "../../../OfficeCryptReader/source/CryptTransform.h"
#include "../../../OOXML/Base/Nullable.h"


#include <boost/smart_ptr/shared_array.hpp>

#include <iostream>
#include <iomanip>
#include <fstream>


//#include <execinfo.h>

//using namespace PPT_FORMAT;
using namespace XLS;

using NSCommon::nullable;

std::string GetRecordName(PPT_FORMAT::RecordType dwType);

class SRecordHeader 
{ 
public:
    unsigned char           RecVersion;
    unsigned short          RecInstance;
    PPT_FORMAT::RecordType	RecType;
    _UINT32                 RecLen;

    bool bBadHeader;

    void Clear()
    {
        RecVersion = 0;
        RecInstance = 0;
        RecType = RT_NONE;
        RecLen = 0;

        bBadHeader = false;
    }
    SRecordHeader()
    {
        Clear();
    }
    bool ReadFromStream(const CFStreamPtr &pStream)
    {
        Clear();

        if (pStream->isEOF()) return FALSE;
        POLE::uint64 nRd = 0;

        unsigned short rec =0;
        pStream->read((unsigned char*)&(rec), 2);

        RecInstance = rec >> 4;
        RecVersion	= rec - (RecInstance << 4);

        *pStream >> RecType >> RecLen;

        unsigned long sz = pStream->getStreamSize() - pStream->getStreamPointer();

        if (RecLen > sz)
        {
            RecLen = (UINT)sz;
            bBadHeader = true; // GZoabli_PhD.ppt ... RecLen & 0xffff ????
        }

        return true;
    }

    bool ReadFromStream(POLE::Stream * pStream)
    {
        Clear();
        if (!pStream) return false;

        POLE::uint64 nRd = 0;

        unsigned short rec =0;
        nRd = pStream->read((unsigned char*)&(rec), 2);

        if (nRd != 2) return false;

        RecInstance = rec >> 4;
        RecVersion	= rec - (RecInstance<<4);

        nRd = pStream->read((unsigned char*)&(RecType), 2);

        nRd = pStream->read((unsigned char*)&(RecLen), 4);

        POLE::uint64 sz = pStream->size() - pStream->tell();

        if (RecLen > sz)
        {
            RecLen = (UINT)sz;
            bBadHeader = true; // GZoabli_PhD.ppt ... RecLen & 0xffff ????
        }

//        void** backTraceData = (void**)(new char*[40]);
//        int backTraceSize = backtrace(backTraceData, 40);

//        std::ofstream file("/home/ivaz28/pp/dia/ppt/pptRecords.txt", std::ios::out | std::ios::app);

//        file << std::string(backTraceSize - 11, ' ')
//             << "0x" << std::setw(4) << std::setfill('0') << std::hex << (int)RecType
//             << " " << std::setw(40) << std::setfill(' ') << std::left << GetRecordName(RecType)
//             << " " << std::setw(5) << std::dec << RecLen
//             << " " << backTraceSize << std::endl;

//        delete [] backTraceData;
//        file.close();

        return true;
    }

    bool IsContainer()
    {
        /*if ((RecVersion == PSFLAG_CONTAINER) || ((RecVersion & 0x0F) == 0x0F))
        {
            return TRUE;
        }*/
        if (1064 == RecType)
            return false;

        if (RecVersion == 0x0F)
        {
            return true;
        }
        return false;
    }

    SRecordHeader& operator =(const SRecordHeader& oSrc)
    {
        RecVersion	= oSrc.RecVersion;
        RecInstance = oSrc.RecInstance;
        RecType		= oSrc.RecType;
        RecLen		= oSrc.RecLen;
        return (*this);
    }

};

class IRecord
{
public:
    SRecordHeader m_oHeader;

    virtual ~IRecord(){}
    virtual void ReadFromStream(SRecordHeader & oHeader, const CFStreamPtr &pStream) = 0;
    virtual void ReadFromStream(SRecordHeader & oHeader, POLE::Stream* pStream) = 0;
};

class CUnknownRecord : public IRecord
{
    // этот класс - просто для того, чтобы нигде не проверять,
    // реализована ли у нас такая запись

public:
    CUnknownRecord()
    {
    }

    ~CUnknownRecord()
    {
    }
    virtual void ReadFromStream(SRecordHeader & oHeader, const CFStreamPtr &pStream)
    {
        m_oHeader = oHeader;

        pStream->seekFromCurForward(m_oHeader.RecLen);
    }
    virtual void ReadFromStream(SRecordHeader & oHeader, POLE::Stream* pStream)
    {
        m_oHeader = oHeader;

        StreamUtils::StreamSkip((long)m_oHeader.RecLen, pStream);
    }

    std::wstring ReadStringW(const CFStreamPtr &pStream, int size);
    std::string	 ReadStringA(const CFStreamPtr &pStream, int size);
};

IRecord* CreateByType(SRecordHeader oHeader);

class CRecordsContainer : public CUnknownRecord
{
public:
    std::vector<IRecord*> m_arRecords;

    CRecordsContainer() : m_arRecords()
    {
    }

    virtual ~CRecordsContainer()
    {
        Clear();
    }

    void Clear()
    {
        size_t nCount = m_arRecords.size();
        while(0 != nCount)
        {
            if (NULL != m_arRecords[nCount-1])
            {
                delete m_arRecords[nCount-1];
                m_arRecords[nCount-1] = NULL;
            }
            m_arRecords.pop_back();
            --nCount;
        }
    }

    virtual void ReadFromStream(SRecordHeader & oHeader, const CFStreamPtr &pStream);
    virtual void ReadFromStream(SRecordHeader & oHeader, POLE::Stream* pStream);

    template <typename T>
    void GetRecordsByType(std::vector<T>* pArray, bool bIsChild, bool bOnlyFirst = false) const
    {
        if (NULL == pArray)
            return;

        // возвращаем указатели, их не удалять наверху!!!!
        for (size_t nIndex = 0; nIndex < m_arRecords.size(); ++nIndex)
        {
            T pRec = dynamic_cast<T>(m_arRecords[nIndex]);
            if (NULL != pRec)
            {
                pArray->push_back(pRec);
                if (bOnlyFirst)
                {
                    return;
                }
            }
            else if ((bIsChild) && (m_arRecords[nIndex]->m_oHeader.IsContainer()))
            {
                CRecordsContainer* pContainer = dynamic_cast<CRecordsContainer*>(m_arRecords[nIndex]);
                if (NULL != pContainer)
                {
                    pContainer->GetRecordsByType(pArray, bIsChild, bOnlyFirst);
                }
            }
        }
    }
};

/**************************************************************
    контейнеры... (по идее они не нужны, потом можно убрать)
**************************************************************/

/*
class CRecordDocument					: public CRecordsContainer {};
class CRecordExAviMovie					: public CRecordsContainer {};
class CRecordExCDAudio					: public CRecordsContainer {};
class CRecordExControl					: public CRecordsContainer {};
class CRecordExEmbed					: public CRecordsContainer {};
class CRecordExHyperlink				: public CRecordsContainer {};
class CRecordExHyperlink9				: public CRecordsContainer {};
class CRecordExLink						: public CRecordsContainer {};
class CRecordExMCIMovie					: public CRecordsContainer {};
class CRecordExMIDIAudio				: public CRecordsContainer {};
class CRecordExObjList					: public CRecordsContainer {};
class CRecordExOleObjStg				: public CRecordsContainer {};
class CRecordExQuickTimeMovie			: public CRecordsContainer {};
class CRecordExVideo					: public CRecordsContainer {};
class CRecordExWAVAudioEmbedded			: public CRecordsContainer {};
class CRecordExWAVAudioLink				: public CRecordsContainer {};
class CRecordFontCollection10			: public CRecordsContainer {};
class CRecordHandout					: public CRecordsContainer {};
class CRecordHeadersFooters				: public CRecordsContainer {};
class CRecordHTMLPublishInfo			: public CRecordsContainer {};
class CRecordInteractiveInfo			: public CRecordsContainer {};
class CRecordList						: public CRecordsContainer {};
class CRecordMainMaster					: public CRecordsContainer {};
class CRecordMsofbtClientData			: public CRecordsContainer {};
class CRecordNamedShow					: public CRecordsContainer {};
class CRecordNamedShows					: public CRecordsContainer {};
class CRecordNotes						: public CRecordsContainer {};
class CRecordNotesTextViewInfo			: public CRecordsContainer {};
class CRecordOutlineTextProps9			: public CRecordsContainer {};
class CRecordOutlineTextProps10			: public CRecordsContainer {};
class CRecordOutlineTextProps11			: public CRecordsContainer {};
class CRecordOutlineViewInfo			: public CRecordsContainer {};
class CRecordParaBuild					: public CRecordsContainer {};
class CRecordProgBinaryTag				: public CRecordsContainer {};
class CRecordProgStringTag				: public CRecordsContainer {};
class CRecordProgTags					: public CRecordsContainer {};
class CRecordSlide						: public CRecordsContainer {};
class CRecordSlideListTable10			: public CRecordsContainer {};
class CRecordSlideListWithText			: public CRecordsContainer {};
class CRecordSlideSyncInfo12			: public CRecordsContainer {};
class CRecordSlideViewInfo				: public CRecordsContainer {};
class CRecordSorterViewInfo				: public CRecordsContainer {};
class CRecordSound						: public CRecordsContainer {};
class CRecordSoundCollection			: public CRecordsContainer {};
class CRecordSrKinsoku					: public CRecordsContainer {};
class CRecordSummary					: public CRecordsContainer {};
class CRecordVBAInfo					: public CRecordsContainer {};
*/

/********************************************************************************
        создаем по типу (остальные будем добавлять по мере необходимости)
********************************************************************************/

//-------------------------------------------------------------------------------
#define CREATE_BY_TYPE(RECORD_TYPE, CLASS_RECORD_NAME)							\
    case RECORD_TYPE: { pRecord = new CLASS_RECORD_NAME(); break; }				\
    //-------------------------------------------------------------------------------