search should now be OK (CLI, Fanfiction only)

[fanfix.git] / src / be / nikiroo / fanfix / reader / cli / CliReader.java

package be.nikiroo.fanfix.reader.cli;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import be.nikiroo.fanfix.Instance;
import be.nikiroo.fanfix.bundles.StringId;
import be.nikiroo.fanfix.data.Chapter;
import be.nikiroo.fanfix.data.MetaData;
import be.nikiroo.fanfix.data.Paragraph;
import be.nikiroo.fanfix.data.Story;
import be.nikiroo.fanfix.reader.BasicReader;
import be.nikiroo.fanfix.searchable.BasicSearchable;
import be.nikiroo.fanfix.searchable.SearchableTag;
import be.nikiroo.fanfix.supported.SupportType;
import be.nikiroo.utils.StringUtils;

/**
 * Command line {@link Story} reader.
 * <p>
 * Will output stories to the console.
 * 
 * @author niki
 */
class CliReader extends BasicReader {
	@Override
	public void read(boolean sync) throws IOException {
		MetaData meta = getMeta();

		if (meta == null) {
			throw new IOException("No story to read");
		}

		String title = "";
		String author = "";

		if (meta.getTitle() != null) {
			title = meta.getTitle();
		}

		if (meta.getAuthor() != null) {
			author = "©" + meta.getAuthor();
			if (meta.getDate() != null && !meta.getDate().isEmpty()) {
				author = author + " (" + meta.getDate() + ")";
			}
		}

		System.out.println(title);
		System.out.println(author);
		System.out.println("");

		// TODO: progress?
		for (Chapter chap : getStory(null)) {
			if (chap.getName() != null && !chap.getName().isEmpty()) {
				System.out.println(Instance.getTrans().getString(
						StringId.CHAPTER_NAMED, chap.getNumber(),
						chap.getName()));
			} else {
				System.out.println(Instance.getTrans().getString(
						StringId.CHAPTER_UNNAMED, chap.getNumber()));
			}
		}
	}

	public void read(int chapter) throws IOException {
		MetaData meta = getMeta();

		if (meta == null) {
			throw new IOException("No story to read");
		}

		// TODO: progress?
		if (chapter > getStory(null).getChapters().size()) {
			System.err.println("Chapter " + chapter + ": no such chapter");
		} else {
			Chapter chap = getStory(null).getChapters().get(chapter - 1);
			System.out.println("Chapter " + chap.getNumber() + ": "
					+ chap.getName());

			for (Paragraph para : chap) {
				System.out.println(para.getContent());
				System.out.println("");
			}
		}
	}

	@Override
	public void browse(String source) {
		List<MetaData> stories;
		stories = getLibrary().getListBySource(source);

		for (MetaData story : stories) {
			String author = "";
			if (story.getAuthor() != null && !story.getAuthor().isEmpty()) {
				author = " (" + story.getAuthor() + ")";
			}

			System.out.println(story.getLuid() + ": " + story.getTitle()
					+ author);
		}
	}

	@Override
	public void search(SupportType searchOn, String keywords, int page,
			int item, boolean sync) throws IOException {
		BasicSearchable search = BasicSearchable.getSearchable(searchOn);

		if (page == 0) {
			System.out.println(search.searchPages(keywords));
		} else {
			List<MetaData> metas = search.search(keywords, page);

			if (item == 0) {
				System.out.println("Page " + page + " of stories for: "
						+ keywords);
				displayStories(metas);
			} else {
				// ! 1-based index !
				if (item <= 0 | item > metas.size()) {
					throw new IOException("Index out of bounds: " + item);
				}

				MetaData meta = metas.get(item - 1);
				displayStory(meta);
			}
		}
	}

	@Override
	public void searchTag(SupportType searchOn, int page, int item,
			boolean sync, Integer... tags) throws IOException {
		BasicSearchable search = BasicSearchable.getSearchable(searchOn);
		List<SearchableTag> stags = search.getTags();
		String fqnTag = "";

		SearchableTag stag = null;
		for (Integer tagIndex : tags) {
			// ! 1-based index !
			if (tagIndex == null || tagIndex <= 0 | tagIndex > stags.size()) {
				throw new IOException("Index out of bounds: " + tagIndex);
			}

			stag = stags.get(tagIndex - 1);
			if (stag != null) {
				search.fillTag(stag);
				stags = stag.getChildren();
				if (!fqnTag.isEmpty()) {
					fqnTag += " / ";
				}
				fqnTag += stag.getName();
			} else {
				stags = new ArrayList<SearchableTag>();
				break;
			}
		}

		if (stag != null) {
			if (page <= 0) {
				if (stag.isLeaf()) {
					search.search(stag, 1);
					System.out.println(stag.getPages());
				} else {
					System.out.println(stag.getCount());
				}
			} else {
				List<MetaData> metas = null;
				List<SearchableTag> subtags = null;
				int count;

				if (stag.isLeaf()) {
					metas = search.search(stag, page);
					count = metas.size();
				} else {
					subtags = stag.getChildren();
					count = subtags.size();
				}

				if (item > 0) {
					if (item <= count) {
						if (metas != null) {
							MetaData meta = metas.get(item - 1);
							displayStory(meta);
						} else {
							SearchableTag subtag = subtags.get(item - 1);

							// TODO: i18n
							String stories = "stories";
							String num = StringUtils.formatNumber(subtag
									.getCount());
							System.out.println(String.format("%s (%s), %s %s",
									subtag.getName(), fqnTag, num, stories));
						}
					} else {
						System.out.println("Invalid item: only " + count
								+ " items found");
					}
				} else {
					if (metas != null) {
						// TODO i18n
						System.out.println(String.format("Content of %s: ",
								fqnTag));
						displayStories(metas);
					} else {
						// TODO i18n
						System.out.println(String.format("Subtags of %s: ",
								fqnTag));
						int i = 1;
						for (SearchableTag subtag : subtags) {
							String total = "";
							if (subtag.getCount() > 0) {
								total = StringUtils.formatNumber(subtag
										.getCount());
							}

							if (total.isEmpty()) {
								System.out.println(String.format("%d: %s", i,
										subtag.getName()));
							} else {
								System.out.println(String.format("%d: %s (%s)",
										i, subtag.getName(), total));
							}

							i++;
						}
					}
				}
			}
		} else {
			// TODO i18n
			System.out.println("Known tags: ");
			int i = 1;
			for (SearchableTag s : stags) {
				System.out.println(String.format("%d: %s", i, s.getName()));
				i++;
			}
		}
	}

	private void displayStory(MetaData meta) {
		System.out.println(meta.getTitle());
		System.out.println();
		System.out.println(meta.getUrl());
		System.out.println();
		System.out.println("Tags: " + meta.getTags());
		System.out.println();
		for (Paragraph para : meta.getResume()) {
			System.out.println(para.getContent());
			System.out.println("");
		}
	}

	private void displayStories(List<MetaData> metas) {
		int i = 1;
		for (MetaData meta : metas) {
			System.out.println(i + ": " + meta.getTitle());
			i++;
		}
	}
}