Yajbir Singh
142 lines
4.0 KiB
C++
142 lines
4.0 KiB
C++
#include "IWork.h"
|
|
#include "../DesktopEditor/common/File.h"
|
|
#include "../DesktopEditor/common/Directory.h"
|
|
|
|
#include <libetonyek/libetonyek.h>
|
|
#include <libodfgen/OdtGenerator.hxx>
|
|
#include <libodfgen/OdsGenerator.hxx>
|
|
#include <libodfgen/OdpGenerator.hxx>
|
|
#include <libodfgen/test/StringDocumentHandler.hxx>
|
|
|
|
#include <memory>
|
|
#include <fstream>
|
|
|
|
class CIWorkFile_Private
|
|
{
|
|
public:
|
|
std::wstring m_sTempDirectory;
|
|
|
|
public:
|
|
CIWorkFile_Private()
|
|
{
|
|
}
|
|
~CIWorkFile_Private()
|
|
{
|
|
}
|
|
};
|
|
|
|
CIWorkFile::CIWorkFile()
|
|
{
|
|
m_internal = new CIWorkFile_Private();
|
|
}
|
|
|
|
CIWorkFile::~CIWorkFile()
|
|
{
|
|
delete m_internal;
|
|
}
|
|
|
|
#if !defined(_WIN32) && !defined(_WIN64)
|
|
#define DATA_TYPE_INPUTFILE std::string
|
|
#else
|
|
#define DATA_TYPE_INPUTFILE std::wstring
|
|
#endif
|
|
|
|
bool GetRVNGInputStream(const DATA_TYPE_INPUTFILE& sFile, std::shared_ptr<librevenge::RVNGInputStream>& oRVNGInputStream, libetonyek::EtonyekDocument::Type& oDocumentType)
|
|
{
|
|
oRVNGInputStream.reset(new librevenge::RVNGFileStream(sFile.c_str()));
|
|
|
|
oDocumentType = libetonyek::EtonyekDocument::TYPE_UNKNOWN;
|
|
const libetonyek::EtonyekDocument::Confidence confidence = libetonyek::EtonyekDocument::isSupported(oRVNGInputStream.get(), &oDocumentType);
|
|
|
|
return libetonyek::EtonyekDocument::CONFIDENCE_NONE != confidence;
|
|
}
|
|
|
|
IWorkFileType CIWorkFile::GetType(const std::wstring& sFile) const
|
|
{
|
|
//TODO:: так как на данный момент мы работает только напрямую с файлом, то работа с директорией нам пока не нужна
|
|
if (NSDirectory::PathIsDirectory(sFile))
|
|
return IWorkFileType::None;
|
|
|
|
std::shared_ptr<librevenge::RVNGInputStream> input;
|
|
libetonyek::EtonyekDocument::Type oDocumentType;
|
|
|
|
#if !defined(_WIN32) && !defined(_WIN64)
|
|
std::string sFileA = U_TO_UTF8(sFile);
|
|
if (!GetRVNGInputStream(sFileA, input, oDocumentType))
|
|
return IWorkFileType::None;
|
|
#else
|
|
if (!GetRVNGInputStream(sFile, input, oDocumentType))
|
|
return IWorkFileType::None;
|
|
#endif
|
|
|
|
switch (oDocumentType)
|
|
{
|
|
case libetonyek::EtonyekDocument::TYPE_PAGES:
|
|
return IWorkFileType::Pages;
|
|
case libetonyek::EtonyekDocument::TYPE_NUMBERS:
|
|
return IWorkFileType::Numbers;
|
|
case libetonyek::EtonyekDocument::TYPE_KEYNOTE:
|
|
return IWorkFileType::Keynote;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return IWorkFileType::None;
|
|
}
|
|
|
|
template<class Generator>
|
|
int Convert(const std::wstring& wsOutputFile, std::shared_ptr<librevenge::RVNGInputStream>& ptrInput, const std::wstring& wsPassword = L"", const std::wstring& wsTempDirectory = L"")
|
|
{
|
|
StringDocumentHandler content;
|
|
Generator generator;
|
|
generator.addDocumentHandler(&content, ODF_FLAT_XML);
|
|
|
|
bool bRes = libetonyek::EtonyekDocument::parse(ptrInput.get(), &generator);
|
|
if (!bRes)
|
|
return 1;
|
|
|
|
const std::string sOutputFileA = U_TO_UTF8(wsOutputFile);
|
|
std::ofstream output(sOutputFileA.c_str());
|
|
output << content.cstr();
|
|
|
|
if (output.bad())
|
|
return -1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int CIWorkFile::Convert2Odf(const std::wstring& sFile, const std::wstring& sOutputFile) const
|
|
{
|
|
//TODO:: так как на данный момент мы работает только напрямую с файлом, то работа с директорией нам пока не нужна
|
|
if (NSDirectory::PathIsDirectory(sFile))
|
|
return -1;
|
|
|
|
std::shared_ptr<librevenge::RVNGInputStream> input;
|
|
libetonyek::EtonyekDocument::Type oDocumentType;
|
|
|
|
#if !defined(_WIN32) && !defined(_WIN64)
|
|
std::string sFileA = U_TO_UTF8(sFile);
|
|
if (!GetRVNGInputStream(sFileA, input, oDocumentType))
|
|
return -1;
|
|
#else
|
|
if (!GetRVNGInputStream(sFile, input, oDocumentType))
|
|
return -1;
|
|
#endif
|
|
|
|
switch (oDocumentType)
|
|
{
|
|
case libetonyek::EtonyekDocument::TYPE_PAGES: return Convert<OdtGenerator>(sOutputFile, input);
|
|
case libetonyek::EtonyekDocument::TYPE_NUMBERS: return Convert<OdsGenerator>(sOutputFile, input);
|
|
case libetonyek::EtonyekDocument::TYPE_KEYNOTE: return Convert<OdpGenerator>(sOutputFile, input);
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
void CIWorkFile::SetTmpDirectory(const std::wstring& sFolder)
|
|
{
|
|
m_internal->m_sTempDirectory = sFolder;
|
|
}
|