Rajarshee Chatterjee
132 lines
4.9 KiB
C++
132 lines
4.9 KiB
C++
#include "../import/EpubParser.hpp"
|
|
|
|
#include <algorithm>
|
|
#include <cassert>
|
|
#include <filesystem>
|
|
#include <fstream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
namespace {
|
|
|
|
void writeFile(const std::filesystem::path& path, const std::string& content) {
|
|
std::filesystem::create_directories(path.parent_path());
|
|
std::ofstream file(path);
|
|
file << content;
|
|
}
|
|
|
|
std::string readFile(const std::filesystem::path& path) {
|
|
std::ifstream file(path);
|
|
return {std::istreambuf_iterator<char>(file),
|
|
std::istreambuf_iterator<char>()};
|
|
}
|
|
|
|
bool containsPath(const std::vector<std::string>& paths,
|
|
const std::filesystem::path& expectedPath) {
|
|
return std::find(paths.begin(), paths.end(), expectedPath.string()) !=
|
|
paths.end();
|
|
}
|
|
|
|
} // namespace
|
|
|
|
int main(int argc, char** argv) {
|
|
assert(argc == 2);
|
|
const std::filesystem::path fixtureDirectory = argv[1];
|
|
std::filesystem::remove_all(fixtureDirectory);
|
|
|
|
writeFile(
|
|
fixtureDirectory / "META-INF/container.xml",
|
|
R"xml(<?xml version="1.0"?>
|
|
<ocf:container xmlns:ocf="urn:oasis:names:tc:opendocument:xmlns:container">
|
|
<ocf:rootfiles>
|
|
<ocf:rootfile full-path="OEBPS/content.opf"/>
|
|
</ocf:rootfiles>
|
|
</ocf:container>)xml");
|
|
|
|
writeFile(
|
|
fixtureDirectory / "OEBPS/content.opf",
|
|
R"xml(<?xml version="1.0"?>
|
|
<opf:package xmlns:opf="http://www.idpf.org/2007/opf"
|
|
xmlns:dc="http://purl.org/dc/elements/1.1/" version="3.0">
|
|
<opf:metadata>
|
|
<dc:title>Image fixture</dc:title>
|
|
<opf:meta name="cover" content="cover-document"/>
|
|
</opf:metadata>
|
|
<opf:manifest>
|
|
<opf:item id="nav" href="toc.ncx" media-type="application/x-dtbncx+xml"/>
|
|
<opf:item id="cover-document" href="Text/cover.xhtml" media-type="application/xhtml+xml"/>
|
|
<opf:item id="chapter" href="Text/chapter.xhtml" media-type="application/xhtml+xml"/>
|
|
<opf:item id="jpeg" href="Images/cover.jpg" media-type="image/jpeg"/>
|
|
<opf:item id="png" href="Images/illustration.png" media-type="image/png"/>
|
|
<opf:item id="webp" href="Images/illustration.webp" media-type="image/webp"/>
|
|
<opf:item id="gif" href="Images/illustration.gif" media-type="image/gif"/>
|
|
<opf:item id="svg" href="Images/illustration.svg" media-type="image/svg+xml"/>
|
|
</opf:manifest>
|
|
<opf:spine>
|
|
<opf:itemref idref="cover-document"/>
|
|
<opf:itemref idref="chapter"/>
|
|
</opf:spine>
|
|
</opf:package>)xml");
|
|
|
|
writeFile(
|
|
fixtureDirectory / "OEBPS/toc.ncx",
|
|
R"xml(<?xml version="1.0"?>
|
|
<ncx:ncx xmlns:ncx="http://www.daisy.org/z3986/2005/ncx/">
|
|
<ncx:navMap>
|
|
<ncx:navPoint>
|
|
<ncx:navLabel><ncx:text>Chapter</ncx:text></ncx:navLabel>
|
|
<ncx:content src="Text/chapter.xhtml"/>
|
|
</ncx:navPoint>
|
|
</ncx:navMap>
|
|
</ncx:ncx>)xml");
|
|
writeFile(
|
|
fixtureDirectory / "OEBPS/Text/chapter.xhtml",
|
|
R"xml(<html><body>
|
|
<svg><title>Chapter art</title><image width="1200" height="1600" href="../Images/illustration.png"/></svg>
|
|
<svg xmlns:xlink="http://www.w3.org/1999/xlink"><image xlink:href="../Images/illustration.webp"/><path d="M0 0h10v10z"/></svg>
|
|
</body></html>)xml");
|
|
writeFile(
|
|
fixtureDirectory / "OEBPS/Text/cover.xhtml",
|
|
R"xml(<html><body><svg xmlns:xlink="http://www.w3.org/1999/xlink"><image xlink:href="../Images/cover.jpg"/></svg></body></html>)xml");
|
|
|
|
const EpubMetadata metadata = parseEpub(fixtureDirectory.string());
|
|
const std::filesystem::path imageDirectory = fixtureDirectory / "OEBPS/Images";
|
|
|
|
assert(metadata.name == "Image fixture");
|
|
assert(metadata.imagePaths.size() == 5);
|
|
assert(containsPath(metadata.imagePaths, imageDirectory / "cover.jpg"));
|
|
assert(containsPath(metadata.imagePaths,
|
|
imageDirectory / "illustration.png"));
|
|
assert(containsPath(metadata.imagePaths,
|
|
imageDirectory / "illustration.webp"));
|
|
assert(containsPath(metadata.imagePaths,
|
|
imageDirectory / "illustration.gif"));
|
|
assert(containsPath(metadata.imagePaths,
|
|
imageDirectory / "illustration.svg"));
|
|
assert(metadata.cover == (imageDirectory / "cover.jpg").string());
|
|
assert(metadata.chapters.size() == 2);
|
|
assert(metadata.chapters.front().path ==
|
|
(fixtureDirectory / "OEBPS/Text/cover.xhtml").string());
|
|
assert(metadata.chapters[1].name == "Chapter");
|
|
|
|
const std::string coverChapter =
|
|
readFile(fixtureDirectory / "OEBPS/Text/cover.xhtml");
|
|
assert(coverChapter.find("<svg") == std::string::npos);
|
|
assert(coverChapter.find("<img src=\"../Images/cover.jpg\"") !=
|
|
std::string::npos);
|
|
|
|
const std::string regularChapter =
|
|
readFile(fixtureDirectory / "OEBPS/Text/chapter.xhtml");
|
|
assert(regularChapter.find(
|
|
"<img width=\"1200\" height=\"1600\" "
|
|
"src=\"../Images/illustration.png\" alt=\"Chapter art\"") !=
|
|
std::string::npos);
|
|
assert(regularChapter.find(
|
|
"<image xlink:href=\"../Images/illustration.webp\"") !=
|
|
std::string::npos);
|
|
assert(regularChapter.find("<path d=\"M0 0h10v10z\"") !=
|
|
std::string::npos);
|
|
|
|
return 0;
|
|
}
|