rbook 0.7.4

A fast, format-agnostic, ergonomic ebook library for reading, building, and editing EPUB 2 and 3.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

use crate::ebook::resource::consts::mime;
use crate::epub::consts::{ocf, xml};
use crate::epub::write::writer::{EpubWriter, EpubWriterContext};
use crate::writer::WriterResult;
use crate::writer::xml::{XmlWriter, write_element};
use std::io::Write;

struct ContainerWriter<'ebook, W> {
    ctx: &'ebook EpubWriterContext<'ebook>,
    writer: XmlWriter<'ebook, W>,
}

impl<'ebook, W: Write> ContainerWriter<'ebook, W> {
    fn new(ctx: &'ebook EpubWriterContext<'ebook>, writer: W) -> Self {
        Self {
            writer: XmlWriter::new(writer),
            ctx,
        }
    }

    fn write_container(mut self) -> WriterResult<()> {
        self.writer.write_utf8_declaration()?;

        write_element! {
            writer: self.writer,
            tag: ocf::CONTAINER,
            attributes: {
                ocf::VERSION => ocf::CONTAINER_VERSION,
                xml::XMLNS   => ocf::CONTAINER_NS,
            }
            inner_content: {
                self.write_root_files()?;
            }
        }
    }

    fn write_root_files(&mut self) -> WriterResult<()> {
        write_element! {
            writer: self.writer,
            tag: ocf::ROOT_FILES,
            inner_content: {
                self.write_root_file()?;
            }
        }
    }

    // When multi-rendition support arrives the parameters will be (&mut self, &Rendition)
    fn write_root_file(&mut self) -> WriterResult<()> {
        write_element! {
            writer: self.writer,
            tag: ocf::ROOT_FILE,
            attributes: {
                // Root file paths must not be prefixed with '/'
                ocf::FULL_PATH  => self.ctx.epub.package.location.trim_start_matches('/'),
                ocf::MEDIA_TYPE => mime::OEBPS_PACKAGE,
            }
        }
    }
}

impl<W: Write> EpubWriter<'_, W> {
    pub(super) fn write_container(&mut self) -> WriterResult<()> {
        self.zip.start_file(ocf::CONTAINER_PATH)?;
        ContainerWriter::new(&self.ctx, &mut self.zip).write_container()
    }
}