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:
* 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)
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();
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();
}
@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)) {
}
}
- 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;
}
/**
*
* @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;
}
}
}
git://git.nikiroo.be / nikiroo-utils.git / blobdiff