X-Git-Url: https://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Ffanfix%2Fsupported%2FText.java;h=cba23bf8c2f1815d2eddba3ff611ed1951b9e4b6;hb=63b9c2077a1e4baf3af3b3bb9187b71b6b706145;hp=14379113f69aee0cee80b6aa95e5474821180344;hpb=0ffa47548f474c1330d8d723300d9aa7a4894736;p=fanfix.git
diff --git a/src/be/nikiroo/fanfix/supported/Text.java b/src/be/nikiroo/fanfix/supported/Text.java
index 1437911..cba23bf 100644
--- a/src/be/nikiroo/fanfix/supported/Text.java
+++ b/src/be/nikiroo/fanfix/supported/Text.java
@@ -1,19 +1,25 @@
package be.nikiroo.fanfix.supported;
import java.io.File;
+import java.io.FileInputStream;
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.MarkableFileInputStream;
import be.nikiroo.utils.Progress;
/**
@@ -28,12 +34,34 @@ import be.nikiroo.utils.Progress;
* number
*
A description of the story must be given as chapter number 0
* A cover may be present, with the same filename but a PNG, JPEG or JPG
- * extension
*
*
* @author niki
*/
-class Text extends BasicSupport_Deprecated {
+class Text extends BasicSupport {
+ private File sourceFile;
+ private InputStream in;
+
+ protected File getSourceFile() {
+ return sourceFile;
+ }
+
+ protected InputStream getInput() {
+ if (in != null) {
+ try {
+ in.reset();
+ } catch (IOException e) {
+ Instance.getTraceHandler().error(
+ new IOException("Cannot reset the Text stream", e));
+ }
+
+ return in;
+ }
+
+ return null;
+ }
+
@Override
protected boolean isHtml() {
return false;
@@ -45,41 +73,42 @@ class Text extends BasicSupport_Deprecated {
}
@Override
- protected MetaData getMeta(URL source, InputStream in) throws IOException {
+ protected Document loadDocument(URL source) throws IOException {
+ try {
+ sourceFile = new File(source.toURI());
+ in = new MarkableFileInputStream(new FileInputStream(sourceFile));
+ } catch (URISyntaxException e) {
+ throw new IOException("Cannot load the text document: " + source);
+ }
+
+ return null;
+ }
+
+ @Override
+ protected MetaData getMeta() throws IOException {
MetaData meta = new MetaData();
- meta.setTitle(getTitle(reset(in)));
- meta.setAuthor(getAuthor(reset(in)));
- meta.setDate(getDate(reset(in)));
+ meta.setTitle(getTitle());
+ meta.setAuthor(getAuthor());
+ meta.setDate(getDate());
meta.setTags(new ArrayList());
meta.setSource(getSourceName());
- meta.setUrl(source.toString());
+ meta.setUrl(getSourceFile().toURI().toURL().toString());
meta.setPublisher("");
- meta.setUuid(source.toString());
+ meta.setUuid(getSourceFile().toString());
meta.setLuid("");
- meta.setLang(getLang(reset(in))); // default is EN
- meta.setSubject(getSubject(source));
+ meta.setLang(getLang()); // default is EN
+ meta.setSubject(getSourceFile().getParentFile().getName());
meta.setType(getType().toString());
meta.setImageDocument(false);
- meta.setCover(getCover(source));
+ meta.setCover(getCover(getSourceFile()));
return meta;
}
- private String getSubject(URL source) throws IOException {
- try {
- File file = new File(source.toURI());
- return file.getParentFile().getName();
- } catch (URISyntaxException e) {
- throw new IOException("Cannot parse the URL to File: "
- + source.toString(), e);
- }
-
- }
-
- private String getLang(InputStream in) {
+ private String getLang() {
@SuppressWarnings("resource")
- Scanner scan = new Scanner(in, "UTF-8");
+ Scanner scan = new Scanner(getInput(), "UTF-8");
scan.useDelimiter("\\n");
scan.next(); // Title
scan.next(); // Author (Date)
@@ -95,24 +124,24 @@ class Text extends BasicSupport_Deprecated {
}
if (lang == null) {
- lang = "EN";
+ lang = "en";
} else {
- lang = lang.toUpperCase();
+ lang = lang.toLowerCase();
}
return lang;
}
- private String getTitle(InputStream in) {
+ private String getTitle() {
@SuppressWarnings("resource")
- Scanner scan = new Scanner(in, "UTF-8");
+ Scanner scan = new Scanner(getInput(), "UTF-8");
scan.useDelimiter("\\n");
return scan.next();
}
- private String getAuthor(InputStream in) {
+ private String getAuthor() {
@SuppressWarnings("resource")
- Scanner scan = new Scanner(in, "UTF-8");
+ Scanner scan = new Scanner(getInput(), "UTF-8");
scan.useDelimiter("\\n");
scan.next();
String authorDate = scan.next();
@@ -126,9 +155,9 @@ class Text extends BasicSupport_Deprecated {
return BasicSupportHelper.fixAuthor(author);
}
- private String getDate(InputStream in) {
+ private String getDate() {
@SuppressWarnings("resource")
- Scanner scan = new Scanner(in, "UTF-8");
+ Scanner scan = new Scanner(getInput(), "UTF-8");
scan.useDelimiter("\\n");
scan.next();
String authorDate = scan.next();
@@ -147,18 +176,12 @@ class Text extends BasicSupport_Deprecated {
}
@Override
- protected String getDesc(URL source, InputStream in) throws IOException {
- return getChapterContent(source, in, 0, null);
+ protected String getDesc() throws IOException {
+ return getChapterContent(null, 0, null);
}
- private Image getCover(URL source) {
- String path;
- try {
- path = new File(source.toURI()).getPath();
- } catch (URISyntaxException e) {
- Instance.getTraceHandler().error(e);
- path = null;
- }
+ private Image getCover(File sourceFile) {
+ String path = sourceFile.getName();
for (String ext : new String[] { ".txt", ".text", ".story" }) {
if (path.endsWith(ext)) {
@@ -166,15 +189,28 @@ class Text extends BasicSupport_Deprecated {
}
}
- return getImage(this, source, path);
+ Image cover = BasicSupportImages.getImage(this,
+ sourceFile.getParentFile(), path);
+ if (cover != null) {
+ try {
+ File tmp = Instance.getTempFiles().createTempFile(
+ "test_cover_image");
+ ImageUtils.getInstance().saveAsImage(cover, tmp, "png");
+ tmp.delete();
+ } catch (IOException e) {
+ cover = null;
+ }
+ }
+
+ return cover;
}
@Override
- protected List> getChapters(URL source, InputStream in,
- Progress pg) throws IOException {
+ protected List> getChapters(Progress pg)
+ throws IOException {
List> chaps = new ArrayList>();
@SuppressWarnings("resource")
- Scanner scan = new Scanner(in, "UTF-8");
+ Scanner scan = new Scanner(getInput(), "UTF-8");
scan.useDelimiter("\\n");
boolean prevLineEmpty = false;
while (scan.hasNext()) {
@@ -185,24 +221,10 @@ class Text extends BasicSupport_Deprecated {
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() {
- @Override
- public URL setValue(URL value) {
- return null;
- }
- @Override
- public URL getValue() {
- return value;
- }
-
- @Override
- public String getKey() {
- return key;
- }
- });
+ chaps.add(new AbstractMap.SimpleEntry(//
+ chapName, //
+ getSourceFile().toURI().toURL()));
}
prevLineEmpty = line.trim().isEmpty();
@@ -212,11 +234,11 @@ class Text extends BasicSupport_Deprecated {
}
@Override
- protected String getChapterContent(URL source, InputStream in, int number,
- Progress pg) throws IOException {
+ protected String getChapterContent(URL source, int number, Progress pg)
+ throws IOException {
StringBuilder builder = new StringBuilder();
@SuppressWarnings("resource")
- Scanner scan = new Scanner(in, "UTF-8");
+ Scanner scan = new Scanner(getInput(), "UTF-8");
scan.useDelimiter("\\n");
boolean inChap = false;
while (scan.hasNext()) {
@@ -234,22 +256,73 @@ class Text extends BasicSupport_Deprecated {
return builder.toString();
}
+ @Override
+ protected void close() {
+ InputStream in = getInput();
+ if (in != null) {
+ try {
+ in.close();
+ } catch (IOException e) {
+ Instance.getTraceHandler().error(
+ new IOException(
+ "Cannot close the text source file input", e));
+ }
+ }
+
+ super.close();
+ }
+
@Override
protected boolean supports(URL url) {
+ return supports(url, false);
+ }
+
+ /**
+ * Check if we supports this {@link URL}, that is, if the info file can be
+ * found OR not found.
+ *
+ * @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) {
+ boolean infoPresent = false;
if ("file".equals(url.getProtocol())) {
File file;
try {
file = new File(url.toURI());
+ file = assureNoTxt(file);
file = new File(file.getPath() + ".info");
} catch (URISyntaxException e) {
Instance.getTraceHandler().error(e);
file = null;
}
- return file == null || !file.exists();
+ infoPresent = (file != null && file.exists());
}
- return false;
+ return infoPresent == info;
+ }
+
+ /**
+ * Remove the ".txt" extension if it is present.
+ *
+ * @param file
+ * the file to process
+ *
+ * @return the same file or a copy of it without the ".txt" extension if it
+ * was present
+ */
+ protected File assureNoTxt(File file) {
+ if (file.getName().endsWith(".txt")) {
+ file = new File(file.getPath().substring(0,
+ file.getPath().length() - 4));
+ }
+
+ return file;
}
/**
@@ -261,7 +334,7 @@ class Text extends BasicSupport_Deprecated {
*
* @return the language or NULL
*/
- private String detectChapter(String line, int number) {
+ static private String detectChapter(String line, int number) {
line = line.toUpperCase();
for (String lang : Instance.getConfig().getString(Config.CHAPTER)
.split(",")) {