X-Git-Url: http://git.nikiroo.be/?p=nikiroo-utils.git;a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Ffanfix%2Fsupported%2FText.java;h=2f614fca83529ea1f5e7ae27f08706c4963089e9;hp=a6105947faab29f348f158edb32d008bc2301b5b;hb=8d59ce0748baeeea0458bab49716ab4543aae439;hpb=276f95c6ca9a79d2fc6879877b240cefb4ceb598 diff --git a/src/be/nikiroo/fanfix/supported/Text.java b/src/be/nikiroo/fanfix/supported/Text.java index a610594..2f614fc 100644 --- a/src/be/nikiroo/fanfix/supported/Text.java +++ b/src/be/nikiroo/fanfix/supported/Text.java @@ -5,16 +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: @@ -33,53 +38,71 @@ 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.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 +126,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 +146,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 +170,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 +183,28 @@ 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.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 +215,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 +228,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,6 +250,22 @@ 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); @@ -296,11 +328,10 @@ 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(",")) { - String chapter = Instance.getConfig().getStringX(Config.CHAPTER, + for (String lang : Instance.getConfig().getList(Config.CONF_CHAPTER)) { + String chapter = Instance.getConfig().getStringX(Config.CONF_CHAPTER, lang); if (chapter != null && !chapter.isEmpty()) { chapter = chapter.toUpperCase() + " ";