X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Ffanfix%2Fsupported%2FText.java;h=c54b6a5d44ce0a328d63451420c6a9c00a82bd3c;hb=d66deb8d8b30cff6b54db352eef34a3508939f84;hp=a6105947faab29f348f158edb32d008bc2301b5b;hpb=276f95c6ca9a79d2fc6879877b240cefb4ceb598;p=nikiroo-utils.git diff --git a/src/be/nikiroo/fanfix/supported/Text.java b/src/be/nikiroo/fanfix/supported/Text.java index a610594..c54b6a5 100644 --- a/src/be/nikiroo/fanfix/supported/Text.java +++ b/src/be/nikiroo/fanfix/supported/Text.java @@ -5,16 +5,23 @@ 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.Chapter; import be.nikiroo.fanfix.data.MetaData; +import be.nikiroo.fanfix.data.Paragraph; 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: @@ -33,53 +40,70 @@ import be.nikiroo.utils.Progress; * * @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.getInstance().getTraceHandler().error(new IOException("Cannot reset the Text stream", e)); + } + + return in; + } + + return null; + } + @Override protected boolean isHtml() { return false; } @Override - public String getSourceName() { - return "text"; + protected Document loadDocument(URL source) throws IOException { + try { + sourceFile = new File(source.toURI()); + in = new MarkableFileInputStream(sourceFile); + } catch (URISyntaxException e) { + throw new IOException("Cannot load the text document: " + source); + } + + return null; } @Override - protected MetaData getMeta(URL source, InputStream in) throws IOException { + 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.setSource(getType().getSourceName()); + 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) @@ -103,16 +127,16 @@ class Text extends BasicSupport_Deprecated { 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(); @@ -123,12 +147,12 @@ class Text extends BasicSupport_Deprecated { author = authorDate.substring(0, pos); } - return BasicSupportHelper.fixAuthor(author); + return bsHelper.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 +171,25 @@ 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 { + String content = getChapterContent(null, 0, null).trim(); + if (!content.isEmpty()) { + Chapter desc = bsPara.makeChapter(this, null, 0, "Description", + content, isHtml(), null); + StringBuilder builder = new StringBuilder(); + for (Paragraph para : desc) { + if (builder.length() > 0) { + builder.append("\n"); + } + builder.append(para.getContent()); + } + } + + return content; } - 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 +197,26 @@ class Text extends BasicSupport_Deprecated { } } - return getImage(this, 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> 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 +227,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,18 +240,18 @@ 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()) { String line = scan.next(); - if (detectChapter(line, number) != null) { + if (!inChap && detectChapter(line, number) != null) { inChap = true; - } else if (inChap && detectChapter(line, number + 1) != null) { + } else if (detectChapter(line, number + 1) != null) { break; } else if (inChap) { builder.append(line); @@ -234,6 +262,21 @@ 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.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); @@ -242,6 +285,8 @@ class Text extends BasicSupport_Deprecated { /** * Check if we supports this {@link URL}, that is, if the info file can be * found OR not found. + *

+ * It must also be a file, not another kind of URL. * * @param url * the {@link URL} to check @@ -251,21 +296,23 @@ class Text extends BasicSupport_Deprecated { * @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; - } + if (!"file".equals(url.getProtocol())) { + return false; + } - infoPresent = (file != null && file.exists()); + 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; } @@ -293,21 +340,28 @@ class Text extends BasicSupport_Deprecated { * * @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, int 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)) { // 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())