You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
427 lines
15 KiB
427 lines
15 KiB
/****************************************************************************
|
|
** Copyright (c) 2013 Debao Zhang <hello@debao.me>
|
|
** All right reserved.
|
|
**
|
|
** Permission is hereby granted, free of charge, to any person obtaining
|
|
** a copy of this software and associated documentation files (the
|
|
** "Software"), to deal in the Software without restriction, including
|
|
** without limitation the rights to use, copy, modify, merge, publish,
|
|
** distribute, sublicense, and/or sell copies of the Software, and to
|
|
** permit persons to whom the Software is furnished to do so, subject to
|
|
** the following conditions:
|
|
**
|
|
** The above copyright notice and this permission notice shall be
|
|
** included in all copies or substantial portions of the Software.
|
|
**
|
|
** THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
** EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
** MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
** NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
|
** LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
|
** OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
|
** WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
**
|
|
****************************************************************************/
|
|
#include "xlsxpackage_p.h"
|
|
#include "xlsxworkbook.h"
|
|
#include "xlsxworksheet.h"
|
|
#include "xlsxworkbook_p.h"
|
|
#include "xlsxutility_p.h"
|
|
#include "xlsxcontenttypes_p.h"
|
|
#include "xlsxsharedstrings_p.h"
|
|
#include "xlsxdocpropscore_p.h"
|
|
#include "xlsxdocpropsapp_p.h"
|
|
#include "xlsxtheme_p.h"
|
|
#include "xlsxstyles_p.h"
|
|
#include "xlsxrelationships_p.h"
|
|
#include "xlsxzipwriter_p.h"
|
|
#include "xlsxdrawing_p.h"
|
|
#include "xlsxzipreader_p.h"
|
|
#include "xlsxdocument.h"
|
|
#include <QBuffer>
|
|
#include <QDebug>
|
|
#include <QDir>
|
|
#include <QFileInfo>
|
|
|
|
namespace QXlsx {
|
|
|
|
/*
|
|
From Wikipedia: The Open Packaging Conventions (OPC) is a
|
|
container-file technology initially created by Microsoft to store
|
|
a combination of XML and non-XML files that together form a single
|
|
entity such as an Open XML Paper Specification (OpenXPS)
|
|
document. http://en.wikipedia.org/wiki/Open_Packaging_Conventions.
|
|
|
|
At its simplest an Excel XLSX file contains the following elements:
|
|
|
|
____ [Content_Types].xml
|
|
|
|
|
|____ docProps
|
|
| |____ app.xml
|
|
| |____ core.xml
|
|
|
|
|
|____ xl
|
|
| |____ workbook.xml
|
|
| |____ worksheets
|
|
| | |____ sheet1.xml
|
|
| |
|
|
| |____ styles.xml
|
|
| |
|
|
| |____ theme
|
|
| | |____ theme1.xml
|
|
| |
|
|
| |_____rels
|
|
| |____ workbook.xml.rels
|
|
|
|
|
|_____rels
|
|
|____ .rels
|
|
|
|
The Packager class coordinates the classes that represent the
|
|
elements of the package and writes them into the XLSX file.
|
|
*/
|
|
typedef QSharedPointer<Worksheet> WorksheetPtrType;
|
|
|
|
Package::Package(Document *document) :
|
|
m_document(document)
|
|
{
|
|
m_workbook = m_document->workbook();
|
|
m_worksheet_count = 0;
|
|
m_chartsheet_count = 0;
|
|
foreach (WorksheetPtrType sheet, m_workbook->worksheets()) {
|
|
if (sheet->isChartsheet())
|
|
m_chartsheet_count += 1;
|
|
else
|
|
m_worksheet_count += 1;
|
|
}
|
|
}
|
|
|
|
bool Package::parsePackage(QIODevice *packageDevice)
|
|
{
|
|
ZipReader zipReader(packageDevice);
|
|
QStringList filePaths = zipReader.filePaths();
|
|
|
|
if (!filePaths.contains(QLatin1String("_rels/.rels")))
|
|
return false;
|
|
|
|
Relationships rootRels = Relationships::loadFromXmlData(zipReader.fileData(QStringLiteral("_rels/.rels")));
|
|
|
|
//load core property
|
|
QList<XlsxRelationship> rels_core = rootRels.packageRelationships(QStringLiteral("/metadata/core-properties"));
|
|
if (!rels_core.isEmpty()) {
|
|
//Get the core property file name if it exists.
|
|
//In normal case, this should be "docProps/core.xml"
|
|
QString docPropsCore_Name = rels_core[0].target;
|
|
|
|
DocPropsCore props = DocPropsCore::loadFromXmlData(zipReader.fileData(docPropsCore_Name));
|
|
foreach (QString name, props.propertyNames())
|
|
m_document->setDocumentProperty(name, props.property(name));
|
|
}
|
|
|
|
//load app property
|
|
QList<XlsxRelationship> rels_app = rootRels.documentRelationships(QStringLiteral("/extended-properties"));
|
|
if (!rels_app.isEmpty()) {
|
|
//Get the app property file name if it exists.
|
|
//In normal case, this should be "docProps/app.xml"
|
|
QString docPropsApp_Name = rels_app[0].target;
|
|
|
|
DocPropsApp props = DocPropsApp::loadFromXmlData(zipReader.fileData(docPropsApp_Name));
|
|
foreach (QString name, props.propertyNames())
|
|
m_document->setDocumentProperty(name, props.property(name));
|
|
}
|
|
|
|
//load workbook now, Get the workbook file path from the root rels file
|
|
//In normal case, this should be "xl/workbook.xml"
|
|
QList<XlsxRelationship> rels_xl = rootRels.documentRelationships(QStringLiteral("/officeDocument"));
|
|
if (rels_xl.isEmpty())
|
|
return false;
|
|
QString xlworkbook_Path = rels_xl[0].target;
|
|
QStringList xlworkbook_PathList = splitPath(xlworkbook_Path);
|
|
QString xlworkbook_Dir = xlworkbook_PathList[0];
|
|
QString xlworkbook_Name = xlworkbook_PathList[1];
|
|
QSharedPointer<Workbook> book = Workbook::loadFromXmlData(zipReader.fileData(xlworkbook_Path));
|
|
QList<QPair<QString, QString> > sheetNameIdPairList = book->d_func()->sheetNameIdPairList;
|
|
Relationships xlworkbook_Rels = Relationships::loadFromXmlData(
|
|
zipReader.fileData(xlworkbook_Dir+QStringLiteral("/_rels/")+xlworkbook_Name+QStringLiteral(".rels")));
|
|
|
|
//load styles
|
|
QList<XlsxRelationship> rels_styles = xlworkbook_Rels.documentRelationships(QStringLiteral("/styles"));
|
|
if (!rels_styles.isEmpty()) {
|
|
//In normal case this should be styles.xml which in xl
|
|
QString name = rels_styles[0].target;
|
|
QString path = xlworkbook_Dir + QLatin1String("/") + name;
|
|
QSharedPointer<Styles> styles (new Styles(true));
|
|
styles->loadFromXmlData(zipReader.fileData(path));
|
|
m_document->workbook()->d_ptr->styles = styles;
|
|
}
|
|
|
|
//load sharedStrings
|
|
QList<XlsxRelationship> rels_sharedStrings = xlworkbook_Rels.documentRelationships(QStringLiteral("/sharedStrings"));
|
|
if (!rels_sharedStrings.isEmpty()) {
|
|
//In normal case this should be sharedStrings.xml which in xl
|
|
QString name = rels_sharedStrings[0].target;
|
|
QString path = xlworkbook_Dir + QLatin1String("/") + name;
|
|
QSharedPointer<SharedStrings> sst= SharedStrings::loadFromXmlData(zipReader.fileData(path));
|
|
m_document->workbook()->d_ptr->sharedStrings = sst;
|
|
}
|
|
|
|
//load theme
|
|
QList<XlsxRelationship> rels_theme = xlworkbook_Rels.documentRelationships(QStringLiteral("/theme"));
|
|
if (!rels_theme.isEmpty()) {
|
|
//In normal case this should be theme/theme1.xml which in xl
|
|
//:Todo
|
|
}
|
|
|
|
//load worksheets
|
|
QList<XlsxRelationship> rels_worksheets = xlworkbook_Rels.documentRelationships(QStringLiteral("/worksheet"));
|
|
if (rels_worksheets.isEmpty())
|
|
return false;
|
|
|
|
for (int i=0; i<sheetNameIdPairList.size(); ++i) {
|
|
QPair<QString, QString> pair = sheetNameIdPairList[i];
|
|
QString worksheet_rId = pair.second;
|
|
QString name = xlworkbook_Rels.getRelationshipById(worksheet_rId).target;
|
|
QString worksheet_path = xlworkbook_Dir + QLatin1String("/") + name;
|
|
Worksheet *sheet = m_document->workbook()->addWorksheet(pair.first);
|
|
sheet->loadFromXmlData(zipReader.fileData(worksheet_path));
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool Package::createPackage(QIODevice *package)
|
|
{
|
|
ZipWriter zipWriter(package);
|
|
if (zipWriter.error())
|
|
return false;
|
|
|
|
m_workbook->prepareDrawings();
|
|
|
|
writeWorksheetFiles(zipWriter);
|
|
// writeChartsheetFiles(zipWriter);
|
|
writeWorkbookFile(zipWriter);
|
|
// writeChartFiles(zipWriter);
|
|
writeDrawingFiles(zipWriter);
|
|
// writeVmlFiles(zipWriter);
|
|
// writeCommentFiles(zipWriter);
|
|
// writeTableFiles(zipWriter);
|
|
writeSharedStringsFile(zipWriter);
|
|
writeDocPropsAppFile(zipWriter);
|
|
writeDocPropsCoreFile(zipWriter);
|
|
writeContentTypesFile(zipWriter);
|
|
writeStylesFiles(zipWriter);
|
|
writeThemeFile(zipWriter);
|
|
writeRootRelsFile(zipWriter);
|
|
writeWorkbookRelsFile(zipWriter);
|
|
writeWorksheetRelsFiles(zipWriter);
|
|
// writeChartsheetRelsFile(zipWriter);
|
|
writeDrawingRelsFiles(zipWriter);
|
|
writeImageFiles(zipWriter);
|
|
// writeVbaProjectFiles(zipWriter);
|
|
|
|
zipWriter.close();
|
|
return true;
|
|
}
|
|
|
|
void Package::writeWorksheetFiles(ZipWriter &zipWriter)
|
|
{
|
|
int index = 1;
|
|
foreach (WorksheetPtrType sheet, m_workbook->worksheets()) {
|
|
if (sheet->isChartsheet())
|
|
continue;
|
|
|
|
zipWriter.addFile(QStringLiteral("xl/worksheets/sheet%1.xml").arg(index), sheet->saveToXmlData());
|
|
index += 1;
|
|
}
|
|
}
|
|
|
|
void Package::writeWorkbookFile(ZipWriter &zipWriter)
|
|
{
|
|
zipWriter.addFile(QStringLiteral("xl/workbook.xml"), m_workbook->saveToXmlData());
|
|
}
|
|
|
|
void Package::writeDrawingFiles(ZipWriter &zipWriter)
|
|
{
|
|
for (int i=0; i<m_workbook->drawings().size(); ++i) {
|
|
Drawing *drawing = m_workbook->drawings()[i];
|
|
|
|
QByteArray data;
|
|
QBuffer buffer(&data);
|
|
buffer.open(QIODevice::WriteOnly);
|
|
drawing->saveToXmlFile(&buffer);
|
|
zipWriter.addFile(QStringLiteral("xl/drawings/drawing%1.xml").arg(i+1), data);
|
|
}
|
|
}
|
|
|
|
void Package::writeContentTypesFile(ZipWriter &zipWriter)
|
|
{
|
|
ContentTypes content;
|
|
|
|
int worksheet_index = 1;
|
|
foreach (WorksheetPtrType sheet, m_workbook->worksheets()) {
|
|
if (sheet->isChartsheet()) {
|
|
|
|
} else {
|
|
content.addWorksheetName(QStringLiteral("sheet%1").arg(worksheet_index));
|
|
worksheet_index += 1;
|
|
}
|
|
}
|
|
|
|
int drawing_index = 1;
|
|
foreach (Drawing *drawing, m_workbook->drawings()) {
|
|
content.addDrawingName(QStringLiteral("drawing%1").arg(drawing_index));
|
|
drawing_index += 1;
|
|
}
|
|
|
|
if (!m_workbook->images().isEmpty())
|
|
content.addImageTypes(QStringList()<<QStringLiteral("png"));
|
|
|
|
if (m_workbook->sharedStrings()->count())
|
|
content.addSharedString();
|
|
|
|
QByteArray data;
|
|
QBuffer buffer(&data);
|
|
buffer.open(QIODevice::WriteOnly);
|
|
content.saveToXmlFile(&buffer);
|
|
zipWriter.addFile(QStringLiteral("[Content_Types].xml"), data);
|
|
}
|
|
|
|
void Package::writeDocPropsAppFile(ZipWriter &zipWriter)
|
|
{
|
|
DocPropsApp props;
|
|
|
|
foreach (QString name, m_document->documentPropertyNames())
|
|
props.setProperty(name, m_document->documentProperty(name));
|
|
|
|
if (m_worksheet_count)
|
|
props.addHeadingPair(QStringLiteral("Worksheets"), m_worksheet_count);
|
|
if (m_chartsheet_count)
|
|
props.addHeadingPair(QStringLiteral("Chartsheets"), m_chartsheet_count);
|
|
|
|
//Add worksheet parts
|
|
foreach (WorksheetPtrType sheet, m_workbook->worksheets()){
|
|
if (!sheet->isChartsheet())
|
|
props.addPartTitle(sheet->sheetName());
|
|
}
|
|
|
|
//Add the chartsheet parts
|
|
foreach (WorksheetPtrType sheet, m_workbook->worksheets()){
|
|
if (sheet->isChartsheet())
|
|
props.addPartTitle(sheet->sheetName());
|
|
}
|
|
|
|
zipWriter.addFile(QStringLiteral("docProps/app.xml"), props.saveToXmlData());
|
|
}
|
|
|
|
void Package::writeDocPropsCoreFile(ZipWriter &zipWriter)
|
|
{
|
|
DocPropsCore props;
|
|
|
|
foreach (QString name, m_document->documentPropertyNames())
|
|
props.setProperty(name, m_document->documentProperty(name));
|
|
|
|
zipWriter.addFile(QStringLiteral("docProps/core.xml"), props.saveToXmlData());
|
|
}
|
|
|
|
void Package::writeSharedStringsFile(ZipWriter &zipWriter)
|
|
{
|
|
zipWriter.addFile(QStringLiteral("xl/sharedStrings.xml"), m_workbook->sharedStrings()->saveToXmlData());
|
|
}
|
|
|
|
void Package::writeStylesFiles(ZipWriter &zipWriter)
|
|
{
|
|
zipWriter.addFile(QStringLiteral("xl/styles.xml"), m_workbook->styles()->saveToXmlData());
|
|
}
|
|
|
|
void Package::writeThemeFile(ZipWriter &zipWriter)
|
|
{
|
|
QByteArray data;
|
|
QBuffer buffer(&data);
|
|
buffer.open(QIODevice::WriteOnly);
|
|
Theme().saveToXmlFile(&buffer);
|
|
zipWriter.addFile(QStringLiteral("xl/theme/theme1.xml"), data);
|
|
}
|
|
|
|
void Package::writeRootRelsFile(ZipWriter &zipWriter)
|
|
{
|
|
Relationships rels;
|
|
rels.addDocumentRelationship(QStringLiteral("/officeDocument"), QStringLiteral("xl/workbook.xml"));
|
|
rels.addPackageRelationship(QStringLiteral("/metadata/core-properties"), QStringLiteral("docProps/core.xml"));
|
|
rels.addDocumentRelationship(QStringLiteral("/extended-properties"), QStringLiteral("docProps/app.xml"));
|
|
|
|
zipWriter.addFile(QStringLiteral("_rels/.rels"), rels.saveToXmlData());
|
|
}
|
|
|
|
void Package::writeWorkbookRelsFile(ZipWriter &zipWriter)
|
|
{
|
|
Relationships rels;
|
|
|
|
int worksheet_index = 1;
|
|
int chartsheet_index = 1;
|
|
foreach (WorksheetPtrType sheet, m_workbook->worksheets()) {
|
|
if (sheet->isChartsheet()) {
|
|
rels.addDocumentRelationship(QStringLiteral("/chartsheet"), QStringLiteral("chartsheets/sheet%1.xml").arg(chartsheet_index));
|
|
chartsheet_index += 1;
|
|
} else {
|
|
rels.addDocumentRelationship(QStringLiteral("/worksheet"), QStringLiteral("worksheets/sheet%1.xml").arg(worksheet_index));
|
|
worksheet_index += 1;
|
|
}
|
|
}
|
|
|
|
rels.addDocumentRelationship(QStringLiteral("/theme"), QStringLiteral("theme/theme1.xml"));
|
|
rels.addDocumentRelationship(QStringLiteral("/styles"), QStringLiteral("styles.xml"));
|
|
|
|
if (m_workbook->sharedStrings()->count())
|
|
rels.addDocumentRelationship(QStringLiteral("/sharedStrings"), QStringLiteral("sharedStrings.xml"));
|
|
|
|
zipWriter.addFile(QStringLiteral("xl/_rels/workbook.xml.rels"), rels.saveToXmlData());
|
|
}
|
|
|
|
void Package::writeWorksheetRelsFiles(ZipWriter &zipWriter)
|
|
{
|
|
int index = 1;
|
|
foreach (WorksheetPtrType sheet, m_workbook->worksheets()) {
|
|
if (sheet->isChartsheet())
|
|
continue;
|
|
Relationships rels;
|
|
|
|
foreach (QString link, sheet->externUrlList())
|
|
rels.addWorksheetRelationship(QStringLiteral("/hyperlink"), link, QStringLiteral("External"));
|
|
foreach (QString link, sheet->externDrawingList())
|
|
rels.addWorksheetRelationship(QStringLiteral("/drawing"), link);
|
|
|
|
zipWriter.addFile(QStringLiteral("xl/worksheets/_rels/sheet%1.xml.rels").arg(index), rels.saveToXmlData());
|
|
index += 1;
|
|
}
|
|
}
|
|
|
|
void Package::writeDrawingRelsFiles(ZipWriter &zipWriter)
|
|
{
|
|
int index = 1;
|
|
foreach (WorksheetPtrType sheet, m_workbook->worksheets()) {
|
|
if (sheet->drawingLinks().size() == 0)
|
|
continue;
|
|
Relationships rels;
|
|
|
|
typedef QPair<QString, QString> PairType;
|
|
foreach (PairType pair, sheet->drawingLinks())
|
|
rels.addDocumentRelationship(pair.first, pair.second);
|
|
|
|
zipWriter.addFile(QStringLiteral("xl/drawings/_rels/drawing%1.xml.rels").arg(index), rels.saveToXmlData());
|
|
index += 1;
|
|
}
|
|
}
|
|
|
|
void Package::writeImageFiles(ZipWriter &zipWriter)
|
|
{
|
|
for (int i=0; i<m_workbook->images().size(); ++i) {
|
|
QImage image = m_workbook->images()[i];
|
|
|
|
QByteArray data;
|
|
QBuffer buffer(&data);
|
|
buffer.open(QIODevice::WriteOnly);
|
|
image.save(&buffer, "png");
|
|
zipWriter.addFile(QStringLiteral("xl/media/image%1.png").arg(i+1), data);
|
|
}
|
|
}
|
|
|
|
} // namespace QXlsx
|
|
|