fix see word count on source/author, step 2

[nikiroo-utils.git] / src / be / nikiroo / fanfix / supported / Text.java

package be.nikiroo.fanfix.supported;

import java.io.File;
import java.io.FileInputStream;
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.MarkableFileInputStream;
import be.nikiroo.utils.Progress;

/**
 * Support class for local stories encoded in textual format, with a few rules:
 * <ul>
 * <li>The title must be on the first line</li>
 * <li>The author (preceded by nothing, "by " or "©") must be on the second
 * line, possibly with the publication date in parenthesis (i.e., "
 * <tt>By Unknown (3rd October 1998)</tt>")</li>
 * <li>Chapters must be declared with "<tt>Chapter x</tt>" or "
 * <tt>Chapter x: NAME OF THE CHAPTER</tt>", where "<tt>x</tt>" is the chapter
 * 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>
 * </ul>
 * 
 * @author niki
 */
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";
	}

	@Override
	protected Document loadDocument(URL source) throws IOException {
		try {
			sourceFile = new File(source.toURI());
			in = new MarkableFileInputStream(new FileInputStream(sourceFile));
		} catch (URISyntaxException e) {
			throw new IOException("Cannot load the text document: " + source);
		}

		return null;
	}

	@Override
	protected MetaData getMeta() throws IOException {
		MetaData meta = new MetaData();

		meta.setTitle(getTitle());
		meta.setAuthor(getAuthor());
		meta.setDate(getDate());
		meta.setTags(new ArrayList<String>());
		meta.setSource(getSourceName());
		meta.setUrl(getSourceFile().toURI().toURL().toString());
		meta.setPublisher("");
		meta.setUuid(getSourceFile().toString());
		meta.setLuid("");
		meta.setLang(getLang()); // default is EN
		meta.setSubject(getSourceFile().getParentFile().getName());
		meta.setType(getType().toString());
		meta.setImageDocument(false);
		meta.setCover(getCover(getSourceFile()));

		return meta;
	}

	private String getLang() {
		@SuppressWarnings("resource")
		Scanner scan = new Scanner(getInput(), "UTF-8");
		scan.useDelimiter("\\n");
		scan.next(); // Title
		scan.next(); // Author (Date)
		String chapter0 = scan.next(); // empty or Chapter 0
		while (chapter0.isEmpty()) {
			chapter0 = scan.next();
		}

		String lang = detectChapter(chapter0, 0);
		if (lang == null) {
			// No description??
			lang = detectChapter(chapter0, 1);
		}

		if (lang == null) {
			lang = "en";
		} else {
			lang = lang.toLowerCase();
		}

		return lang;
	}

	private String getTitle() {
		@SuppressWarnings("resource")
		Scanner scan = new Scanner(getInput(), "UTF-8");
		scan.useDelimiter("\\n");
		return scan.next();
	}

	private String getAuthor() {
		@SuppressWarnings("resource")
		Scanner scan = new Scanner(getInput(), "UTF-8");
		scan.useDelimiter("\\n");
		scan.next();
		String authorDate = scan.next();

		String author = authorDate;
		int pos = authorDate.indexOf('(');
		if (pos >= 0) {
			author = authorDate.substring(0, pos);
		}

		return BasicSupportHelper.fixAuthor(author);
	}

	private String getDate() {
		@SuppressWarnings("resource")
		Scanner scan = new Scanner(getInput(), "UTF-8");
		scan.useDelimiter("\\n");
		scan.next();
		String authorDate = scan.next();

		String date = "";
		int pos = authorDate.indexOf('(');
		if (pos >= 0) {
			date = authorDate.substring(pos + 1).trim();
			pos = date.lastIndexOf(')');
			if (pos >= 0) {
				date = date.substring(0, pos).trim();
			}
		}

		return date;
	}

	@Override
	protected String getDesc() throws IOException {
		return getChapterContent(null, 0, null);
	}

	private Image getCover(File sourceFile) {
		String path = sourceFile.getName();

		for (String ext : new String[] { ".txt", ".text", ".story" }) {
			if (path.endsWith(ext)) {
				path = path.substring(0, path.length() - ext.length());
			}
		}

		Image cover = BasicSupportImages.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<Entry<String, URL>> getChapters(Progress pg)
			throws IOException {
		List<Entry<String, URL>> chaps = new ArrayList<Entry<String, URL>>();
		@SuppressWarnings("resource")
		Scanner scan = new Scanner(getInput(), "UTF-8");
		scan.useDelimiter("\\n");
		boolean prevLineEmpty = false;
		while (scan.hasNext()) {
			String 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();
				}

				chaps.add(new AbstractMap.SimpleEntry<String, URL>(//
						chapName, //
						getSourceFile().toURI().toURL()));
			}

			prevLineEmpty = line.trim().isEmpty();
		}

		return chaps;
	}

	@Override
	protected String getChapterContent(URL source, int number, Progress pg)
			throws IOException {
		StringBuilder builder = new StringBuilder();
		@SuppressWarnings("resource")
		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) {
				inChap = true;
			} else if (inChap && detectChapter(line, number + 1) != null) {
				break;
			} else if (inChap) {
				builder.append(line);
				builder.append("\n");
			}
		}

		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);
	}

	/**
	 * Check if we supports this {@link URL}, that is, if the info file can be
	 * found OR not found.
	 * 
	 * @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) {
		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;
			}

			infoPresent = (file != null && file.exists());
		}

		return infoPresent == info;
	}

	/**
	 * Remove the ".txt" extension if it is present.
	 * 
	 * @param file
	 *            the file to process
	 * 
	 * @return the same file or a copy of it without the ".txt" extension if it
	 *         was present
	 */
	protected File assureNoTxt(File file) {
		if (file.getName().endsWith(".txt")) {
			file = new File(file.getPath().substring(0,
					file.getPath().length() - 4));
		}

		return file;
	}

	/**
	 * 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).
	 * 
	 * @param line
	 *            the line to check
	 * 
	 * @return the language or NULL
	 */
	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);
			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();
					if (test.startsWith(Integer.toString(number))) {
						test = test
								.substring(Integer.toString(number).length())
								.trim();
						if (test.isEmpty() || test.startsWith(":")) {
							return lang;
						}
					}
				}
			}
		}

		return null;
	}
}