fb2save.cpp 8.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313
  1. #include <QtGui>
  2. #include <QtDebug>
  3. #include "fb2save.h"
  4. #include "fb2view.h"
  5. //---------------------------------------------------------------------------
  6. // Fb2SaveWriter
  7. //---------------------------------------------------------------------------
  8. Fb2SaveWriter::Fb2SaveWriter(Fb2WebView &view, QByteArray *array, QList<int> *folds)
  9. : QXmlStreamWriter(array)
  10. , m_folds(folds)
  11. , m_view(view)
  12. , m_line(0)
  13. {
  14. Init();
  15. }
  16. Fb2SaveWriter::Fb2SaveWriter(Fb2WebView &view, QIODevice *device)
  17. : QXmlStreamWriter(device)
  18. , m_folds(0)
  19. , m_view(view)
  20. , m_line(0)
  21. {
  22. Init();
  23. }
  24. Fb2SaveWriter::Fb2SaveWriter(Fb2WebView &view, QString *string)
  25. : QXmlStreamWriter(string)
  26. , m_folds(0)
  27. , m_view(view)
  28. , m_line(0)
  29. {
  30. Init();
  31. }
  32. void Fb2SaveWriter::Init()
  33. {
  34. writeStartDocument();
  35. }
  36. Fb2SaveWriter::~Fb2SaveWriter()
  37. {
  38. writeEndDocument();
  39. }
  40. void Fb2SaveWriter::writeStartElement(const QString &name, int level)
  41. {
  42. if (level) writeLineEnd();
  43. for (int i = 1; i < level; i++) writeCharacters(" ");
  44. QXmlStreamWriter::writeStartElement(name);
  45. }
  46. void Fb2SaveWriter::writeEndElement(int level)
  47. {
  48. if (level) writeLineEnd();
  49. for (int i = 1; i < level; i++) writeCharacters(" ");
  50. QXmlStreamWriter::writeEndElement();
  51. }
  52. void Fb2SaveWriter::writeLineEnd()
  53. {
  54. writeCharacters("\n");
  55. m_line++;
  56. }
  57. QString Fb2SaveWriter::getFile(const QString &path)
  58. {
  59. StringHash::const_iterator it = m_files.find(path);
  60. if (it == m_files.end()) {
  61. QString name = m_view.fileName(path);
  62. if (name.isEmpty()) return QString();
  63. m_files[name] = path;
  64. m_names << name;
  65. return name;
  66. }
  67. return it.value();
  68. }
  69. QString Fb2SaveWriter::getData(const QString &name)
  70. {
  71. return m_view.fileData(name);
  72. }
  73. void Fb2SaveWriter::writeFiles()
  74. {
  75. StringList::const_iterator it;
  76. for (it = m_names.constBegin(); it != m_names.constEnd(); it++) {
  77. QString name = *it;
  78. if (name.isEmpty()) continue;
  79. QString data = getData(name);
  80. if (data.isEmpty()) continue;
  81. writeStartElement("binary", 2);
  82. if (m_folds) m_folds->append(m_line);
  83. writeAttribute("id", name);
  84. writeLineEnd();
  85. int pos = 0;
  86. while (true) {
  87. QString text = data.mid(pos, 76);
  88. if (text.isEmpty()) break;
  89. writeCharacters(text);
  90. writeLineEnd();
  91. pos += 76;
  92. }
  93. writeCharacters(" ");
  94. QXmlStreamWriter::writeEndElement();
  95. }
  96. }
  97. //---------------------------------------------------------------------------
  98. // Fb2SaveHandler::BodyHandler
  99. //---------------------------------------------------------------------------
  100. FB2_BEGIN_KEYHASH(Fb2SaveHandler::BodyHandler)
  101. FB2_KEY( Section , "div" );
  102. FB2_KEY( Anchor , "a" );
  103. FB2_KEY( Image , "img" );
  104. FB2_KEY( Table , "table" );
  105. FB2_KEY( Parag , "p" );
  106. FB2_KEY( Strong , "b" );
  107. FB2_KEY( Emphas , "i" );
  108. FB2_KEY( Strike , "strike" );
  109. FB2_KEY( Sub , "sub" );
  110. FB2_KEY( Sup , "sup" );
  111. FB2_KEY( Code , "tt" );
  112. FB2_END_KEYHASH
  113. Fb2SaveHandler::BodyHandler::BodyHandler(Fb2SaveWriter &writer, const QString &name, const QXmlAttributes &atts, const QString &tag)
  114. : NodeHandler(name)
  115. , m_writer(writer)
  116. , m_tag(tag)
  117. , m_level(1)
  118. , m_hasChild(false)
  119. {
  120. Init(atts);
  121. }
  122. Fb2SaveHandler::BodyHandler::BodyHandler(BodyHandler *parent, const QString &name, const QXmlAttributes &atts, const QString &tag)
  123. : NodeHandler(name)
  124. , m_writer(parent->m_writer)
  125. , m_tag(tag)
  126. , m_level(parent->nextLevel())
  127. , m_hasChild(false)
  128. {
  129. Init(atts);
  130. }
  131. void Fb2SaveHandler::BodyHandler::Init(const QXmlAttributes &atts)
  132. {
  133. if (m_tag.isEmpty()) return;
  134. if (m_tag == "notes") {
  135. m_writer.writeStartElement("section", m_level);
  136. m_writer.writeAttribute("name", "notes");
  137. } else if (m_tag == "note") {
  138. m_writer.writeStartElement("section", m_level);
  139. } else {
  140. m_writer.writeStartElement(m_tag, m_level);
  141. }
  142. int count = atts.count();
  143. for (int i = 0; i < count; i++) {
  144. QString name = atts.qName(i);
  145. if (name == "id") {
  146. m_writer.writeAttribute(name, atts.value(i));
  147. } else if (name.left(4) == "fb2:") {
  148. m_writer.writeAttribute(name.mid(4), atts.value(i));
  149. }
  150. }
  151. }
  152. Fb2XmlHandler::NodeHandler * Fb2SaveHandler::BodyHandler::NewTag(const QString &name, const QXmlAttributes &atts)
  153. {
  154. m_hasChild = true;
  155. QString tag = QString();
  156. switch (toKeyword(name)) {
  157. case Section : tag = atts.value("class") ; break;
  158. case Anchor : return new AnchorHandler(this, name, atts);
  159. case Image : return new ImageHandler(this, name, atts);
  160. case Parag : return new ParagHandler(this, name, atts);
  161. case Strong : tag = "strong" ; break;
  162. case Emphas : tag = "emphasis" ; break;
  163. case Strike : tag = "strikethrough" ; break;
  164. case Code : tag = "code" ; break;
  165. case Sub : tag = "sub" ; break;
  166. case Sup : tag = "sup" ; break;
  167. default: ;
  168. }
  169. return new BodyHandler(this, name, atts, tag);
  170. }
  171. void Fb2SaveHandler::BodyHandler::TxtTag(const QString &text)
  172. {
  173. m_writer.writeCharacters(text);
  174. }
  175. void Fb2SaveHandler::BodyHandler::EndTag(const QString &name)
  176. {
  177. Q_UNUSED(name);
  178. if (m_tag.isEmpty()) return;
  179. m_writer.writeEndElement(m_hasChild ? m_level : 0);
  180. }
  181. int Fb2SaveHandler::BodyHandler::nextLevel() const
  182. {
  183. return m_level ? m_level + 1 : 0;
  184. }
  185. //---------------------------------------------------------------------------
  186. // Fb2SaveHandler::RootHandler
  187. //---------------------------------------------------------------------------
  188. Fb2SaveHandler::RootHandler::RootHandler(Fb2SaveWriter &writer, const QString &name, const QXmlAttributes &atts)
  189. : BodyHandler(writer, name, atts, "FictionBook")
  190. {
  191. m_writer.writeAttribute("xmlns", "http://www.gribuser.ru/xml/fictionbook/2.0");
  192. m_writer.writeAttribute("xmlns:l", "http://www.w3.org/1999/xlink");
  193. }
  194. void Fb2SaveHandler::RootHandler::EndTag(const QString &name)
  195. {
  196. m_writer.writeFiles();
  197. BodyHandler::EndTag(name);
  198. }
  199. //---------------------------------------------------------------------------
  200. // Fb2SaveHandler::AnchorHandler
  201. //---------------------------------------------------------------------------
  202. Fb2SaveHandler::AnchorHandler::AnchorHandler(BodyHandler *parent, const QString &name, const QXmlAttributes &atts)
  203. : BodyHandler(parent, name, atts, "a")
  204. {
  205. QString href = Value(atts, "href");
  206. m_writer.writeAttribute("l:href", href);
  207. }
  208. //---------------------------------------------------------------------------
  209. // Fb2SaveHandler::ImageHandler
  210. //---------------------------------------------------------------------------
  211. Fb2SaveHandler::ImageHandler::ImageHandler(BodyHandler *parent, const QString &name, const QXmlAttributes &atts)
  212. : BodyHandler(parent, name, atts, "image")
  213. {
  214. QString href = Value(atts, "src");
  215. QString file = m_writer.getFile(href);
  216. file.prepend('#');
  217. m_writer.writeAttribute("l:href", file);
  218. m_writer.writeEndElement(0);
  219. }
  220. //---------------------------------------------------------------------------
  221. // Fb2SaveHandler::ParagHandler
  222. //---------------------------------------------------------------------------
  223. Fb2SaveHandler::ParagHandler::ParagHandler(BodyHandler *parent, const QString &name, const QXmlAttributes &atts)
  224. : BodyHandler(parent, name, atts, "")
  225. , m_parent(parent->tag())
  226. , m_empty(true)
  227. {
  228. }
  229. Fb2XmlHandler::NodeHandler * Fb2SaveHandler::ParagHandler::NewTag(const QString &name, const QXmlAttributes &atts)
  230. {
  231. if (m_empty) start();
  232. return BodyHandler::NewTag(name, atts);
  233. }
  234. void Fb2SaveHandler::ParagHandler::TxtTag(const QString &text)
  235. {
  236. if (m_empty) {
  237. if (isWhiteSpace(text)) return;
  238. start();
  239. }
  240. BodyHandler::TxtTag(text);
  241. }
  242. void Fb2SaveHandler::ParagHandler::EndTag(const QString &name)
  243. {
  244. Q_UNUSED(name);
  245. if (m_empty) m_writer.writeStartElement("empty-line", m_level);
  246. m_writer.writeEndElement(0);
  247. }
  248. void Fb2SaveHandler::ParagHandler::start()
  249. {
  250. if (!m_empty) return;
  251. QString tag = m_parent == "stanza" ? "v" : "p";
  252. m_writer.writeStartElement(tag, m_level);
  253. m_empty = false;
  254. }
  255. //---------------------------------------------------------------------------
  256. // Fb2SaveHandler
  257. //---------------------------------------------------------------------------
  258. Fb2SaveHandler::Fb2SaveHandler(Fb2WebView &view, QIODevice *device)
  259. : Fb2XmlHandler()
  260. , m_writer(view, device)
  261. {
  262. }
  263. Fb2SaveHandler::Fb2SaveHandler(Fb2WebView &view, QByteArray *array, QList<int> *folds)
  264. : Fb2XmlHandler()
  265. , m_writer(view, array, folds)
  266. {
  267. }
  268. Fb2XmlHandler::NodeHandler * Fb2SaveHandler::CreateRoot(const QString &name, const QXmlAttributes &atts)
  269. {
  270. if (name == "body") return new RootHandler(m_writer, name, atts);
  271. m_error = QObject::tr("The tag <body> was not found.");
  272. return 0;
  273. }