Version 1.2.4: fixes, new "Re-download" UI option
[nikiroo-utils.git] / src / be / nikiroo / fanfix / supported / Text.java
index f1ee71c1195d6337461dc35e6d9fdbfdde7a3ef2..3b486ce8d0c4a36b8610a3bd10da9943e3dbfb6c 100644 (file)
@@ -1,5 +1,6 @@
 package be.nikiroo.fanfix.supported;
 
+import java.awt.image.BufferedImage;
 import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
@@ -12,6 +13,7 @@ import java.util.Scanner;
 
 import be.nikiroo.fanfix.Instance;
 import be.nikiroo.fanfix.bundles.Config;
+import be.nikiroo.fanfix.data.MetaData;
 
 /**
  * Support class for local stories encoded in textual format, with a few rules:
@@ -42,13 +44,28 @@ class Text extends BasicSupport {
        }
 
        @Override
-       protected String getPublisher(URL source, InputStream in)
-                       throws IOException {
-               return "";
+       protected MetaData getMeta(URL source, InputStream in) throws IOException {
+               MetaData meta = new MetaData();
+
+               meta.setTitle(getTitle(reset(in)));
+               meta.setAuthor(getAuthor(reset(in)));
+               meta.setDate(getDate(reset(in)));
+               meta.setTags(new ArrayList<String>());
+               meta.setSource(getSourceName());
+               meta.setUrl(source.toString());
+               meta.setPublisher("");
+               meta.setUuid(source.toString());
+               meta.setLuid("");
+               meta.setLang(getLang(source, reset(in))); // default is EN
+               meta.setSubject(getSubject(source));
+               meta.setType(getType().toString());
+               meta.setImageDocument(false);
+               meta.setCover(getCover(source));
+
+               return meta;
        }
 
-       @Override
-       protected String getSubject(URL source, InputStream in) throws IOException {
+       private String getSubject(URL source) throws IOException {
                try {
                        File file = new File(source.toURI());
                        return file.getParentFile().getName();
@@ -59,8 +76,7 @@ class Text extends BasicSupport {
 
        }
 
-       @Override
-       protected String getLang(URL source, InputStream in) throws IOException {
+       private String getLang(URL source, InputStream in) throws IOException {
                @SuppressWarnings("resource")
                Scanner scan = new Scanner(in, "UTF-8");
                scan.useDelimiter("\\n");
@@ -71,9 +87,14 @@ 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();
                }
@@ -81,16 +102,14 @@ class Text extends BasicSupport {
                return lang;
        }
 
-       @Override
-       protected String getTitle(URL source, InputStream in) throws IOException {
+       private String getTitle(InputStream in) throws IOException {
                @SuppressWarnings("resource")
                Scanner scan = new Scanner(in, "UTF-8");
                scan.useDelimiter("\\n");
                return scan.next();
        }
 
-       @Override
-       protected String getAuthor(URL source, InputStream in) throws IOException {
+       private String getAuthor(InputStream in) throws IOException {
                @SuppressWarnings("resource")
                Scanner scan = new Scanner(in, "UTF-8");
                scan.useDelimiter("\\n");
@@ -103,11 +122,10 @@ class Text extends BasicSupport {
                        author = authorDate.substring(0, pos);
                }
 
-               return author;
+               return fixAuthor(author);
        }
 
-       @Override
-       protected String getDate(URL source, InputStream in) throws IOException {
+       private String getDate(InputStream in) throws IOException {
                @SuppressWarnings("resource")
                Scanner scan = new Scanner(in, "UTF-8");
                scan.useDelimiter("\\n");
@@ -128,12 +146,11 @@ class Text extends BasicSupport {
        }
 
        @Override
-       protected String getDesc(URL source, InputStream in) {
+       protected String getDesc(URL source, InputStream in) throws IOException {
                return getChapterContent(source, in, 0);
        }
 
-       @Override
-       protected URL getCover(URL source, InputStream in) {
+       private BufferedImage getCover(URL source) throws IOException {
                String path;
                try {
                        path = new File(source.toURI()).getPath();
@@ -148,44 +165,40 @@ class Text extends BasicSupport {
                        }
                }
 
-               return getImage(source, path);
+               return getImage(this, source, path);
        }
 
        @Override
-       protected List<Entry<String, URL>> getChapters(URL source, InputStream in) {
+       protected List<Entry<String, URL>> getChapters(URL source, InputStream in)
+                       throws IOException {
                List<Entry<String, URL>> chaps = new ArrayList<Entry<String, URL>>();
                @SuppressWarnings("resource")
                Scanner scan = new Scanner(in, "UTF-8");
                scan.useDelimiter("\\n");
-               boolean descSkipped = false;
                boolean prevLineEmpty = false;
                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();
+                       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();
+                               }
+                               final URL value = source;
+                               final String key = chapName;
+                               chaps.add(new Entry<String, URL>() {
+                                       public URL setValue(URL value) {
+                                               return null;
                                        }
-                                       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 URL getValue() {
+                                               return value;
+                                       }
 
-                                               public String getKey() {
-                                                       return key;
-                                               }
-                                       });
-                               } else {
-                                       descSkipped = true;
-                               }
+                                       public String getKey() {
+                                               return key;
+                                       }
+                               });
                        }
 
                        prevLineEmpty = line.trim().isEmpty();
@@ -195,29 +208,23 @@ class Text extends BasicSupport {
        }
 
        @Override
-       protected String getChapterContent(URL source, InputStream in, int number) {
+       protected String getChapterContent(URL source, InputStream in, int number)
+                       throws IOException {
                StringBuilder builder = new StringBuilder();
                @SuppressWarnings("resource")
                Scanner scan = new Scanner(in, "UTF-8");
                scan.useDelimiter("\\n");
                boolean inChap = false;
-               boolean prevLineEmpty = false;
                while (scan.hasNext()) {
                        String line = scan.next();
-                       if (prevLineEmpty) {
-                               if (detectChapter(line, number) != null) {
-                                       inChap = true;
-                               } else if (inChap) {
-                                       if (prevLineEmpty && detectChapter(line) != null) {
-                                               break;
-                                       }
-
-                                       builder.append(line);
-                                       builder.append("\n");
-                               }
+                       if (detectChapter(line, number) != null) {
+                               inChap = true;
+                       } else if (inChap && detectChapter(line, number + 1) != null) {
+                               break;
+                       } else if (inChap) {
+                               builder.append(line);
+                               builder.append("\n");
                        }
-
-                       prevLineEmpty = line.trim().isEmpty();
                }
 
                return builder.toString();
@@ -241,19 +248,6 @@ class Text extends BasicSupport {
                return false;
        }
 
-       /**
-        * 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).
-        * 
-        * @param line
-        *            the line to check
-        * 
-        * @return the language or NULL
-        */
-       private String detectChapter(String line) {
-               return detectChapter(line, null);
-       }
-
        /**
         * Check if the given line looks like the given starting chapter in a
         * supported language, and return the language if it does (or NULL if not).
@@ -263,7 +257,7 @@ class Text extends BasicSupport {
         * 
         * @return the language or NULL
         */
-       private String detectChapter(String line, Integer number) {
+       private String detectChapter(String line, int number) {
                line = line.toUpperCase();
                for (String lang : Instance.getConfig().getString(Config.CHAPTER)
                                .split(",")) {
@@ -272,19 +266,16 @@ class Text extends BasicSupport {
                        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();
+                                       if (test.startsWith(Integer.toString(number))) {
+                                               test = test
+                                                               .substring(Integer.toString(number).length())
+                                                               .trim();
+                                               if (test.isEmpty() || test.startsWith(":")) {
+                                                       return lang;
                                                }
-                                       } else {
-                                               return lang;
                                        }
                                }
                        }