X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Ffanfix%2Fsupported%2FHtml.java;h=900fa0a1772f0ffd5ecfcaf2fecfbd8b248168a0;hb=d66deb8d8b30cff6b54db352eef34a3508939f84;hp=8dec5f75d55eac093a8808df1a5023da81251dc0;hpb=0ffa47548f474c1330d8d723300d9aa7a4894736;p=nikiroo-utils.git

diff --git a/src/be/nikiroo/fanfix/supported/Html.java b/src/be/nikiroo/fanfix/supported/Html.java
index 8dec5f7..900fa0a 100644
--- a/src/be/nikiroo/fanfix/supported/Html.java
+++ b/src/be/nikiroo/fanfix/supported/Html.java
@@ -3,7 +3,6 @@ package be.nikiroo.fanfix.supported;
 import java.io.File;
 import java.io.IOException;
 import java.net.MalformedURLException;
-import java.net.URISyntaxException;
 import java.net.URL;
 
 import be.nikiroo.fanfix.Instance;
@@ -15,41 +14,76 @@ import be.nikiroo.fanfix.Instance;
  * @author niki
  */
 class Html extends InfoText {
-	@Override
-	public String getSourceName() {
-		return "html";
-	}
-
 	@Override
 	protected boolean supports(URL url) {
-		if (url.getPath().toLowerCase()
-				.endsWith(File.separatorChar + "index.html")) {
-			try {
-				File file = new File(url.toURI()).getParentFile();
-				return super.supports(file.toURI().toURL());
-			} catch (URISyntaxException e) {
-			} catch (MalformedURLException e) {
+		try {
+			File txt = getTxt(url);
+			if (txt != null) {
+				return super.supports(txt.toURI().toURL());
 			}
+		} catch (MalformedURLException e) {
 		}
 
 		return false;
 	}
 
+	@Override
+	protected File getInfoFile() {
+		File source = getSourceFile();
+		if ("index.html".equals(source.getName())) {
+			source = source.getParentFile();
+		}
+
+		String src = source.getPath();
+		File infoFile = new File(src + ".info");
+		if (!infoFile.exists() && src.endsWith(".txt")) {
+			infoFile = new File(
+					src.substring(0, src.length() - ".txt".length()) + ".info");
+		}
+
+		return infoFile;
+	}
+
 	@Override
 	public URL getCanonicalUrl(URL source) {
-		if (source.toString().endsWith(File.separator + "index.html")) {
+		File txt = getTxt(source);
+		if (txt != null) {
 			try {
-				File fakeFile = new File(source.toURI()); // "story/index.html"
-				fakeFile = new File(fakeFile.getParent()); // "story"
-				fakeFile = new File(fakeFile, fakeFile.getName()); // "story/story"
-				return fakeFile.toURI().toURL();
-			} catch (Exception e) {
-				Instance.getTraceHandler().error(
-						new IOException("Cannot find the right URL for "
-								+ source, e));
+				source = txt.toURI().toURL();
+			} catch (MalformedURLException e) {
+				Instance.getInstance().getTraceHandler()
+						.error(new IOException("Cannot convert the right URL for " + source, e));
 			}
 		}
 
 		return source;
 	}
+
+	/**
+	 * Return the associated TXT source file if it can be found.
+	 * 
+	 * @param source
+	 *            the source URL
+	 * 
+	 * @return the supported source text file or NULL
+	 */
+	private static File getTxt(URL source) {
+		try {
+			File fakeFile = new File(source.toURI());
+			if (fakeFile.getName().equals("index.html")) { // "story/index.html"
+				fakeFile = new File(fakeFile.getParent()); // -> "story/"
+			}
+
+			if (fakeFile.isDirectory()) { // "story/"
+				fakeFile = new File(fakeFile, fakeFile.getName() + ".txt"); // "story/story.txt"
+			}
+
+			if (fakeFile.getName().endsWith(".txt")) {
+				return fakeFile;
+			}
+		} catch (Exception e) {
+		}
+
+		return null;
+	}
 }