cbz_hybrid support (text+images)

[nikiroo-utils.git] / src / be / nikiroo / fanfix / supported / Cbz.java

diff --git a/src/be/nikiroo/fanfix/supported/Cbz.java b/src/be/nikiroo/fanfix/supported/Cbz.java
index 948a2d65ece471813f79af289f59b38a49fb8fec..062adf0fac71550a8946257f9123e2d178ba7d98 100644 (file)
--- a/src/be/nikiroo/fanfix/supported/Cbz.java
+++ b/src/be/nikiroo/fanfix/supported/Cbz.java
@@ -17,6 +17,7 @@ import be.nikiroo.fanfix.Instance;
 import be.nikiroo.fanfix.data.Chapter;
 import be.nikiroo.fanfix.data.MetaData;
 import be.nikiroo.fanfix.data.Paragraph;
+import be.nikiroo.fanfix.data.Paragraph.ParagraphType;
 import be.nikiroo.fanfix.data.Story;
 import be.nikiroo.utils.IOUtils;
 import be.nikiroo.utils.Image;
@@ -126,13 +127,37 @@ class Cbz extends Epub {
 
                        pg.setProgress(90);
 
-                       // include original story
+                       // only the description is kept
                        Story origStory = getStoryFromTxt(tmpDir, basename);
                        if (origStory != null) {
-                               story.setChapters(origStory.getChapters());
+                               if (origStory.getMeta().getCover() == null) {
+                                       origStory.getMeta().setCover(story.getMeta().getCover());
+                               }
                                story.setMeta(origStory.getMeta());
-                       } else {
-                               story.setChapters(new ArrayList<Chapter>());
+                       }
+                       story.setChapters(new ArrayList<Chapter>());
+
+                       // Check if we can find non-images chapters, for hybrid-cbz support
+                       for (Chapter chap : origStory) {
+                               Boolean isImages = null;
+                               for (Paragraph para : chap) {
+                                       ParagraphType t = para.getType();
+                                       if (isImages == null && !t.isText(true)) {
+                                               isImages = true;
+                                       }
+                                       if (t.isText(false)) {
+                                               String line = para.getContent();
+                                               // Images are saved in text mode as "[image-link]"
+                                               if (!(line.startsWith("[") && line.endsWith("]"))) {
+                                                       isImages = false;
+                                               }
+                                       }
+                               }
+
+                               if (isImages != null && !isImages) {
+                                       story.getChapters().add(chap);
+                                       chap.setNumber(story.getChapters().size());
+                               }
                        }
 
                        if (!imagesList.isEmpty()) {
@@ -153,7 +178,6 @@ class Cbz extends Epub {
                                meta.setCover(images.get(imagesList.get(0)));
                                meta.setFakeCover(true);
                        }
-
                } finally {
                        IOUtils.deltree(tmpDir);
                        if (zipIn != null) {