From 298d405a94bd30b67ce69add67bd7c764061def4 Mon Sep 17 00:00:00 2001
From: Niki Roo <niki@nikiroo.be>
Date: Thu, 29 Mar 2018 08:52:24 +0200
Subject: [PATCH] Fix cbz reader (bad "main" input stream used)

---
 src/be/nikiroo/fanfix/supported/Cbz.java      | 59 +++++++++++++------
 src/be/nikiroo/fanfix/supported/Epub.java     | 18 ++++--
 .../nikiroo/fanfix/test/ConversionTest.java   | 11 ----
 3 files changed, 55 insertions(+), 33 deletions(-)

diff --git a/src/be/nikiroo/fanfix/supported/Cbz.java b/src/be/nikiroo/fanfix/supported/Cbz.java
index 9d71046..948a2d6 100644
--- a/src/be/nikiroo/fanfix/supported/Cbz.java
+++ b/src/be/nikiroo/fanfix/supported/Cbz.java
@@ -1,7 +1,9 @@
 package be.nikiroo.fanfix.supported;
 
 import java.io.File;
+import java.io.FileInputStream;
 import java.io.IOException;
+import java.io.InputStream;
 import java.net.URL;
 import java.util.ArrayList;
 import java.util.Collections;
@@ -18,6 +20,7 @@ import be.nikiroo.fanfix.data.Paragraph;
 import be.nikiroo.fanfix.data.Story;
 import be.nikiroo.utils.IOUtils;
 import be.nikiroo.utils.Image;
+import be.nikiroo.utils.MarkableFileInputStream;
 import be.nikiroo.utils.Progress;
 
 /**
@@ -76,8 +79,12 @@ class Cbz extends Epub {
 		String basename = null;
 
 		Map<String, Image> images = new HashMap<String, Image>();
+		InputStream cbzIn = null;
+		ZipInputStream zipIn = null;
 		try {
-			ZipInputStream zipIn = new ZipInputStream(getInput());
+			cbzIn = new MarkableFileInputStream(new FileInputStream(
+					getSourceFileOriginal()));
+			zipIn = new ZipInputStream(cbzIn);
 			for (ZipEntry entry = zipIn.getNextEntry(); entry != null; entry = zipIn
 					.getNextEntry()) {
 				if (!entry.isDirectory()
@@ -119,23 +126,12 @@ class Cbz extends Epub {
 
 			pg.setProgress(90);
 
-			File txt = new File(tmpDir, basename + ".txt");
-			if (!txt.exists()) {
-				basename = null;
-			}
-			if (basename != null) {
-				try {
-					BasicSupport support = BasicSupport.getSupport(txt.toURI()
-							.toURL());
-					Story origStory = support.process(null);
-					story.setChapters(origStory.getChapters());
-					story.setMeta(origStory.getMeta());
-				} catch (Exception e) {
-					basename = null;
-				}
-			}
-
-			if (basename == null) {
+			// include original story
+			Story origStory = getStoryFromTxt(tmpDir, basename);
+			if (origStory != null) {
+				story.setChapters(origStory.getChapters());
+				story.setMeta(origStory.getMeta());
+			} else {
 				story.setChapters(new ArrayList<Chapter>());
 			}
 
@@ -160,9 +156,36 @@ class Cbz extends Epub {
 
 		} finally {
 			IOUtils.deltree(tmpDir);
+			if (zipIn != null) {
+				zipIn.close();
+			}
+			if (cbzIn != null) {
+				cbzIn.close();
+			}
 		}
 
 		pg.setProgress(100);
 		return story;
 	}
+
+	private Story getStoryFromTxt(File tmpDir, String basename) {
+		Story origStory = null;
+
+		File txt = new File(tmpDir, basename + ".txt");
+		if (!txt.exists()) {
+			basename = null;
+		}
+		if (basename != null) {
+			try {
+				BasicSupport support = BasicSupport.getSupport(txt.toURI()
+						.toURL());
+				origStory = support.process(null);
+			} catch (Exception e) {
+				basename = null;
+			}
+		}
+
+		return origStory;
+
+	}
 }
diff --git a/src/be/nikiroo/fanfix/supported/Epub.java b/src/be/nikiroo/fanfix/supported/Epub.java
index 7ac3efc..5a1fe43 100644
--- a/src/be/nikiroo/fanfix/supported/Epub.java
+++ b/src/be/nikiroo/fanfix/supported/Epub.java
@@ -27,8 +27,6 @@ import be.nikiroo.utils.StringUtils;
  * @author niki
  */
 class Epub extends InfoText {
-	private File sourceFileOriginal;
-
 	private MetaData meta;
 	private File tmpDir;
 	private String desc;
@@ -42,7 +40,7 @@ class Epub extends InfoText {
 	}
 
 	public File getSourceFileOriginal() {
-		return sourceFileOriginal;
+		return super.getSourceFile();
 	}
 
 	@Override
@@ -61,7 +59,19 @@ class Epub extends InfoText {
 
 	@Override
 	protected InputStream getInput() {
-		return fakeIn;
+		if (fakeIn != null) {
+			try {
+				fakeIn.reset();
+			} catch (IOException e) {
+				Instance.getTraceHandler()
+						.error(new IOException(
+								"Cannot reset the Epub Text stream", e));
+			}
+
+			return fakeIn;
+		}
+
+		return null;
 	}
 
 	@Override
diff --git a/src/be/nikiroo/fanfix/test/ConversionTest.java b/src/be/nikiroo/fanfix/test/ConversionTest.java
index 9d9fac3..51b6deb 100644
--- a/src/be/nikiroo/fanfix/test/ConversionTest.java
+++ b/src/be/nikiroo/fanfix/test/ConversionTest.java
@@ -216,17 +216,6 @@ class ConversionTest extends TestLauncher {
 									.length());
 				}
 
-				if (expectedLines.size() != resultLines.size()) {
-					System.out.println();
-					System.out.println("expected: [");
-					for (String line : expectedLines)
-						System.out.println(line);
-					System.out.println("]");
-					System.out.println("actual: [");
-					for (String line : resultLines)
-						System.out.println(line);
-					System.out.println("]");
-				}
 				testCase.assertEquals(errMess + ": " + name
 						+ ": the number of lines is not the same",
 						expectedLines.size(), resultLines.size());
-- 
2.27.0