X-Git-Url: http://git.nikiroo.be/?p=nikiroo-utils.git;a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Ffanfix%2Fsupported%2FText.java;h=252aca0443fbfe0f71db294c0c8c1d3a816deb72;hp=f1ee71c1195d6337461dc35e6d9fdbfdde7a3ef2;hb=cfdaf6052ddc5ca44cf19f1f6d9f154cc8443024;hpb=08fe2e33007063e30fe22dc1d290f8afaa18eb1d

diff --git a/src/be/nikiroo/fanfix/supported/Text.java b/src/be/nikiroo/fanfix/supported/Text.java
index f1ee71c..252aca0 100644
--- a/src/be/nikiroo/fanfix/supported/Text.java
+++ b/src/be/nikiroo/fanfix/supported/Text.java
@@ -5,13 +5,21 @@ import java.io.IOException;
 import java.io.InputStream;
 import java.net.URISyntaxException;
 import java.net.URL;
+import java.util.AbstractMap;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Map.Entry;
 import java.util.Scanner;
 
+import org.jsoup.nodes.Document;
+
 import be.nikiroo.fanfix.Instance;
 import be.nikiroo.fanfix.bundles.Config;
+import be.nikiroo.fanfix.data.MetaData;
+import be.nikiroo.utils.Image;
+import be.nikiroo.utils.ImageUtils;
+import be.nikiroo.utils.Progress;
+import be.nikiroo.utils.streams.MarkableFileInputStream;
 
 /**
  * Support class for local stories encoded in textual format, with a few rules:
@@ -25,44 +33,72 @@ import be.nikiroo.fanfix.bundles.Config;
  * number</li>
  * <li>A description of the story must be given as chapter number 0</li>
  * <li>A cover may be present, with the same filename but a PNG, JPEG or JPG
- * extension</li<
+ * extension</li>
  * </ul>
  * 
  * @author niki
  */
 class Text extends BasicSupport {
-	@Override
-	protected boolean isHtml() {
-		return false;
+	private File sourceFile;
+	private InputStream in;
+
+	protected File getSourceFile() {
+		return sourceFile;
 	}
 
-	@Override
-	public String getSourceName() {
-		return "text";
+	protected InputStream getInput() {
+		if (in != null) {
+			try {
+				in.reset();
+			} catch (IOException e) {
+				Instance.getInstance().getTraceHandler().error(new IOException("Cannot reset the Text stream", e));
+			}
+
+			return in;
+		}
+
+		return null;
 	}
 
 	@Override
-	protected String getPublisher(URL source, InputStream in)
-			throws IOException {
-		return "";
+	protected boolean isHtml() {
+		return false;
 	}
 
 	@Override
-	protected String getSubject(URL source, InputStream in) throws IOException {
+	protected Document loadDocument(URL source) throws IOException {
 		try {
-			File file = new File(source.toURI());
-			return file.getParentFile().getName();
+			sourceFile = new File(source.toURI());
+			in = new MarkableFileInputStream(sourceFile);
 		} catch (URISyntaxException e) {
-			throw new IOException("Cannot parse the URL to File: "
-					+ source.toString(), e);
+			throw new IOException("Cannot load the text document: " + source);
 		}
 
+		return null;
 	}
 
 	@Override
-	protected String getLang(URL source, InputStream in) throws IOException {
-		@SuppressWarnings("resource")
-		Scanner scan = new Scanner(in, "UTF-8");
+	protected MetaData getMeta() throws IOException {
+		MetaData meta = new MetaData();
+
+		meta.setTitle(getTitle());
+		meta.setAuthor(getAuthor());
+		meta.setDate(bsHelper.formatDate(getDate()));
+		meta.setTags(new ArrayList<String>());
+		meta.setUrl(getSourceFile().toURI().toURL().toString());
+		meta.setUuid(getSourceFile().toString());
+		meta.setLuid("");
+		meta.setLang(getLang()); // default is EN
+		meta.setSubject(getSourceFile().getParentFile().getName());
+		meta.setImageDocument(false);
+		meta.setCover(getCover(getSourceFile()));
+		
+		return meta;
+	}
+
+	private String getLang() {
+		@SuppressWarnings("resource") // cannot close, or we loose getInput()!
+		Scanner scan = new Scanner(getInput(), "UTF-8");
 		scan.useDelimiter("\\n");
 		scan.next(); // Title
 		scan.next(); // Author (Date)
@@ -71,28 +107,31 @@ class Text extends BasicSupport {
 			chapter0 = scan.next();
 		}
 
-		String lang = detectChapter(chapter0);
+		String lang = detectChapter(chapter0, 0);
 		if (lang == null) {
-			lang = super.getLang(source, in);
+			// No description??
+			lang = detectChapter(chapter0, 1);
+		}
+
+		if (lang == null) {
+			lang = "en";
 		} else {
-			lang = lang.toUpperCase();
+			lang = lang.toLowerCase();
 		}
 
 		return lang;
 	}
 
-	@Override
-	protected String getTitle(URL source, InputStream in) throws IOException {
-		@SuppressWarnings("resource")
-		Scanner scan = new Scanner(in, "UTF-8");
+	private String getTitle() {
+		@SuppressWarnings("resource") // cannot close, or we loose getInput()!
+		Scanner scan = new Scanner(getInput(), "UTF-8");
 		scan.useDelimiter("\\n");
 		return scan.next();
 	}
 
-	@Override
-	protected String getAuthor(URL source, InputStream in) throws IOException {
-		@SuppressWarnings("resource")
-		Scanner scan = new Scanner(in, "UTF-8");
+	private String getAuthor() {
+		@SuppressWarnings("resource") // cannot close, or we loose getInput()!
+		Scanner scan = new Scanner(getInput(), "UTF-8");
 		scan.useDelimiter("\\n");
 		scan.next();
 		String authorDate = scan.next();
@@ -103,13 +142,12 @@ class Text extends BasicSupport {
 			author = authorDate.substring(0, pos);
 		}
 
-		return author;
+		return bsHelper.fixAuthor(author);
 	}
 
-	@Override
-	protected String getDate(URL source, InputStream in) throws IOException {
-		@SuppressWarnings("resource")
-		Scanner scan = new Scanner(in, "UTF-8");
+	private String getDate() {
+		@SuppressWarnings("resource") // cannot close, or we loose getInput()!
+		Scanner scan = new Scanner(getInput(), "UTF-8");
 		scan.useDelimiter("\\n");
 		scan.next();
 		String authorDate = scan.next();
@@ -128,19 +166,12 @@ class Text extends BasicSupport {
 	}
 
 	@Override
-	protected String getDesc(URL source, InputStream in) {
-		return getChapterContent(source, in, 0);
+	protected String getDesc() throws IOException {
+		return getChapterContent(null, 0, null).trim();
 	}
 
-	@Override
-	protected URL getCover(URL source, InputStream in) {
-		String path;
-		try {
-			path = new File(source.toURI()).getPath();
-		} catch (URISyntaxException e) {
-			Instance.syserr(e);
-			path = null;
-		}
+	protected Image getCover(File sourceFile) {
+		String path = sourceFile.getName();
 
 		for (String ext : new String[] { ".txt", ".text", ".story" }) {
 			if (path.endsWith(ext)) {
@@ -148,110 +179,152 @@ class Text extends BasicSupport {
 			}
 		}
 
-		return getImage(source, path);
+		Image cover = bsImages.getImage(this, sourceFile.getParentFile(), path);
+		if (cover != null) {
+			try {
+				File tmp = Instance.getInstance().getTempFiles().createTempFile("test_cover_image");
+				ImageUtils.getInstance().saveAsImage(cover, tmp, "png");
+				tmp.delete();
+			} catch (IOException e) {
+				cover = null;
+			}
+		}
+
+		return cover;
 	}
 
 	@Override
-	protected List<Entry<String, URL>> getChapters(URL source, InputStream in) {
+	protected List<Entry<String, URL>> getChapters(Progress pg)
+			throws IOException {
 		List<Entry<String, URL>> chaps = new ArrayList<Entry<String, URL>>();
-		@SuppressWarnings("resource")
-		Scanner scan = new Scanner(in, "UTF-8");
+		@SuppressWarnings("resource") // cannot close, or we loose getInput()!
+		Scanner scan = new Scanner(getInput(), "UTF-8");
 		scan.useDelimiter("\\n");
-		boolean descSkipped = false;
-		boolean prevLineEmpty = false;
+		String line = "first is not empty";
 		while (scan.hasNext()) {
-			String line = scan.next();
-			if (prevLineEmpty && detectChapter(line) != null) {
-				if (descSkipped) {
-					String chapName = Integer.toString(chaps.size());
-					int pos = line.indexOf(':');
-					if (pos >= 0 && pos + 1 < line.length()) {
-						chapName = line.substring(pos + 1).trim();
-					}
-					final URL value = source;
-					final String key = chapName;
-					chaps.add(new Entry<String, URL>() {
-						public URL setValue(URL value) {
-							return null;
-						}
-
-						public URL getValue() {
-							return value;
-						}
-
-						public String getKey() {
-							return key;
-						}
-					});
-				} else {
-					descSkipped = true;
+			boolean prevLineEmpty = line.trim().isEmpty();
+			line = scan.next();
+			if (prevLineEmpty && detectChapter(line, chaps.size() + 1) != null) {
+				String chapName = Integer.toString(chaps.size() + 1);
+				int pos = line.indexOf(':');
+				if (pos >= 0 && pos + 1 < line.length()) {
+					chapName = line.substring(pos + 1).trim();
 				}
-			}
 
-			prevLineEmpty = line.trim().isEmpty();
+				chaps.add(new AbstractMap.SimpleEntry<String, URL>(//
+						chapName, //
+						getSourceFile().toURI().toURL()));
+			}
 		}
-
+		
 		return chaps;
 	}
 
 	@Override
-	protected String getChapterContent(URL source, InputStream in, int number) {
+	protected String getChapterContent(URL source, int number, Progress pg)
+			throws IOException {
 		StringBuilder builder = new StringBuilder();
-		@SuppressWarnings("resource")
-		Scanner scan = new Scanner(in, "UTF-8");
+		@SuppressWarnings("resource") // cannot close, or we loose getInput()!
+		Scanner scan = new Scanner(getInput(), "UTF-8");
 		scan.useDelimiter("\\n");
-		boolean inChap = false;
-		boolean prevLineEmpty = false;
+		scan.next(); // title
+		scan.next(); // author
+		scan.next(); // date or empty
+		Boolean inChap = null;
+		String line = "";
 		while (scan.hasNext()) {
-			String line = scan.next();
-			if (prevLineEmpty) {
-				if (detectChapter(line, number) != null) {
+			if (number == 0 && !line.trim().isEmpty()) {
+				// We found pre-chapter content, we are checking for
+				// Chapter 0 (fake chapter) --> keep the content
+				if (inChap == null)
 					inChap = true;
-				} else if (inChap) {
-					if (prevLineEmpty && detectChapter(line) != null) {
-						break;
-					}
-
-					builder.append(line);
-					builder.append("\n");
-				}
 			}
-
-			prevLineEmpty = line.trim().isEmpty();
+			line = scan.next();
+			if ((inChap == null || !inChap) && detectChapter(line, number) != null) {
+				inChap = true;
+			} else if (detectChapter(line, number + 1) != null) {
+				break;
+			} else if (inChap != null && inChap) {
+				builder.append(line);
+				builder.append("\n");
+			}
 		}
 
 		return builder.toString();
 	}
 
 	@Override
-	protected boolean supports(URL url) {
-		if ("file".equals(url.getProtocol())) {
-			File file;
+	protected void close() {
+		InputStream in = getInput();
+		if (in != null) {
 			try {
-				file = new File(url.toURI());
-				file = new File(file.getPath() + ".info");
-			} catch (URISyntaxException e) {
-				Instance.syserr(e);
-				file = null;
+				in.close();
+			} catch (IOException e) {
+				Instance.getInstance().getTraceHandler()
+						.error(new IOException("Cannot close the text source file input", e));
 			}
+		}
+
+		super.close();
+	}
+
+	@Override
+	protected boolean supports(URL url) {
+		return supports(url, false);
+	}
 
-			return file == null || !file.exists();
+	/**
+	 * Check if we supports this {@link URL}, that is, if the info file can be
+	 * found OR not found.
+	 * <p>
+	 * It must also be a file, not another kind of URL.
+	 * 
+	 * @param url
+	 *            the {@link URL} to check
+	 * @param info
+	 *            TRUE to require the info file, FALSE to forbid the info file
+	 * 
+	 * @return TRUE if it is supported
+	 */
+	protected boolean supports(URL url, boolean info) {
+		if (!"file".equals(url.getProtocol())) {
+			return false;
 		}
 
-		return false;
+		boolean infoPresent = false;
+		File file;
+		try {
+			file = new File(url.toURI());
+			file = assureNoTxt(file);
+			file = new File(file.getPath() + ".info");
+		} catch (URISyntaxException e) {
+			Instance.getInstance().getTraceHandler().error(e);
+			file = null;
+		}
+
+		infoPresent = (file != null && file.exists());
+
+		return infoPresent == info;
 	}
 
 	/**
-	 * Check if the given line looks like a starting chapter in a supported
-	 * language, and return the language if it does (or NULL if not).
+	 * Remove the ".txt" (or ".text") extension if it is present.
 	 * 
-	 * @param line
-	 *            the line to check
+	 * @param file
+	 *            the file to process
 	 * 
-	 * @return the language or NULL
+	 * @return the same file or a copy of it without the ".txt" extension if it
+	 *         was present
 	 */
-	private String detectChapter(String line) {
-		return detectChapter(line, null);
+	protected File assureNoTxt(File file) {
+		for (String ext : new String[] { ".txt", ".text" }) {
+			if (file.getName().endsWith(ext)) {
+				file = new File(file.getPath().substring(0,
+						file.getPath().length() - ext.length()));
+			}
+		}
+
+		return file;
 	}
 
 	/**
@@ -260,31 +333,35 @@ class Text extends BasicSupport {
 	 * 
 	 * @param line
 	 *            the line to check
+	 * @param number
+	 *            the specific chapter number to check for
 	 * 
 	 * @return the language or NULL
 	 */
-	private String detectChapter(String line, Integer number) {
+	static private String detectChapter(String line, int number) {
 		line = line.toUpperCase();
-		for (String lang : Instance.getConfig().getString(Config.CHAPTER)
-				.split(",")) {
-			String chapter = Instance.getConfig().getStringX(Config.CHAPTER,
-					lang);
+		for (String lang : Instance.getInstance().getConfig().getList(Config.CONF_CHAPTER)) {
+			String chapter = Instance.getInstance().getConfig().getStringX(Config.CONF_CHAPTER, lang);
 			if (chapter != null && !chapter.isEmpty()) {
 				chapter = chapter.toUpperCase() + " ";
 				if (line.startsWith(chapter)) {
-					if (number != null) {
-						// We want "[CHAPTER] [number]: [name]", with ": [name]"
-						// optional
-						String test = line.substring(chapter.length()).trim();
-						if (test.startsWith(Integer.toString(number))) {
-							test = test.substring(
-									Integer.toString(number).length()).trim();
-							if (test.isEmpty() || test.startsWith(":")) {
-								return lang;
-							}
+					// We want "[CHAPTER] [number]: [name]", with ": [name]"
+					// optional
+					String test = line.substring(chapter.length()).trim();
+
+					String possibleNum = test.trim();
+					if (possibleNum.indexOf(':') > 0) {
+						possibleNum = possibleNum.substring(0,
+								possibleNum.indexOf(':')).trim();
+					}
+
+					if (test.startsWith(Integer.toString(number))) {
+						test = test
+								.substring(Integer.toString(number).length())
+								.trim();
+						if (test.isEmpty() || test.startsWith(":")) {
+							return lang;
 						}
-					} else {
-						return lang;
 					}
 				}
 			}