TUI: code cleanup + show story info

[fanfix.git] / src / be / nikiroo / fanfix / reader / tui / TuiReaderStoryWindow.java

package be.nikiroo.fanfix.reader.tui;

import java.util.ArrayList;
import java.util.List;

import jexer.TAction;
import jexer.TButton;
import jexer.TLabel;
import jexer.TText;
import jexer.TWindow;
import jexer.event.TResizeEvent;
import jexer.event.TResizeEvent.Type;
import be.nikiroo.fanfix.data.Chapter;
import be.nikiroo.fanfix.data.MetaData;
import be.nikiroo.fanfix.data.Paragraph;
import be.nikiroo.fanfix.data.Paragraph.ParagraphType;
import be.nikiroo.fanfix.data.Story;
import be.nikiroo.fanfix.library.BasicLibrary;

class TuiReaderStoryWindow extends TWindow {
	private BasicLibrary lib;
	private MetaData meta;
	private Story story;
	private TText textField;
	private int chapter = -1;
	private List<TButton> navigationButtons;
	private TLabel currentChapter;

	// chapter: -1 for "none" (0 is desc)
	public TuiReaderStoryWindow(TuiReaderApplication app, BasicLibrary lib,
			MetaData meta, int chapter) {
		super(app, desc(meta), 0, 0, 60, 18, CENTERED | RESIZABLE);

		this.lib = lib;
		this.meta = meta;

		app.setStatusBar(this, desc(meta));

		textField = new TText(this, "", 0, 0, getWidth() - 2, getHeight() - 2);

		navigationButtons = new ArrayList<TButton>(5);

		// -3 because 0-based and 2 for borders
		int row = getHeight() - 3;

		navigationButtons.add(addButton(" ", 0, row, null)); // for bg colour
																// when <<
																// button is
																// pressed
		navigationButtons.add(addButton("<<  ", 0, row, new TAction() {
			@Override
			public void DO() {
				setChapter(0);
			}
		}));
		navigationButtons.add(addButton("<  ", 4, row, new TAction() {
			@Override
			public void DO() {
				setChapter(TuiReaderStoryWindow.this.chapter - 1);
			}
		}));
		navigationButtons.add(addButton(">  ", 7, row, new TAction() {
			@Override
			public void DO() {
				setChapter(TuiReaderStoryWindow.this.chapter + 1);
			}
		}));
		navigationButtons.add(addButton(">>  ", 10, row, new TAction() {
			@Override
			public void DO() {
				setChapter(getStory().getChapters().size());
			}
		}));

		navigationButtons.get(0).setEnabled(false);
		navigationButtons.get(1).setEnabled(false);
		navigationButtons.get(2).setEnabled(false);

		currentChapter = addLabel("", 14, row);
		currentChapter.setWidth(getWidth() - 10);
		setChapter(chapter);
	}

	@Override
	public void onResize(TResizeEvent resize) {
		super.onResize(resize);

		// Resize the text field TODO: why setW/setH/reflow not enough for the
		// scrollbars?
		textField.onResize(new TResizeEvent(Type.WIDGET, resize.getWidth() - 2,
				resize.getHeight() - 2));

		// -3 because 0-based and 2 for borders
		int row = getHeight() - 3;

		String name = currentChapter.getLabel();
		while (name.length() < resize.getWidth() - currentChapter.getX()) {
			name += " ";
		}
		currentChapter.setLabel(name);
		currentChapter.setWidth(resize.getWidth() - 10);
		currentChapter.setY(row);

		for (TButton button : navigationButtons) {
			button.setY(row);
		}
	}

	/**
	 * Display the current chapter in the window, or the {@link Story} info
	 * page.
	 * 
	 * @param chapter
	 *            the chapter (including "0" which is the description) or "-1"
	 *            to display the info page instead
	 */
	private void setChapter(int chapter) {
		if (chapter > getStory().getChapters().size()) {
			chapter = getStory().getChapters().size();
		}

		if (chapter != this.chapter) {
			this.chapter = chapter;

			int max = getStory().getChapters().size();
			navigationButtons.get(0).setEnabled(chapter > -1);
			navigationButtons.get(1).setEnabled(chapter > -1);
			navigationButtons.get(2).setEnabled(chapter > -1);
			navigationButtons.get(3).setEnabled(chapter < max);
			navigationButtons.get(4).setEnabled(chapter < max);

			StringBuilder builder = new StringBuilder();

			if (chapter < 0) {
				appendInfoPage(builder);
			} else {
				appendChapterPage(builder);
			}

			setText(builder.toString());
		}

		setCurrentChapterText();
	}

	/**
	 * Append the info page about the current {@link Story}.
	 * 
	 * @param builder
	 *            the builder to append to
	 */
	private StringBuilder appendInfoPage(StringBuilder builder) {
		MetaData meta = getStory().getMeta();

		// TODO: use a ttable?

		appendTitle(builder, meta.getTitle(), 1).append("\n");

		// i18n
		builder.append("Author: ").append(meta.getAuthor()).append("\n");
		builder.append("Publication date: ").append(meta.getDate())
				.append("\n");
		builder.append("Word count: ").append(meta.getWords()).append("\n");

		return builder;
	}

	/**
	 * Append the current chapter.
	 * 
	 * @param builder
	 *            the builder to append to
	 */
	private void appendChapterPage(StringBuilder builder) {
		Chapter chap = null;
		if (chapter == 0) {
			chap = getStory().getMeta().getResume();
		} else if (chapter > 0) {
			chap = getStory().getChapters().get(chapter - 1);
		}

		// TODO: i18n
		String chapName = chap == null ? "[No RESUME]" : chap.getName();

		appendTitle(builder,
				String.format("Chapter %d: %s", chapter, chapName), 1);
		builder.append("\n");

		if (chap != null) {
			for (Paragraph para : chap) {
				if (para.getType() == ParagraphType.BREAK) {
					builder.append("\n");
				}
				builder.append(para.getContent()).append("\n");
				if (para.getType() == ParagraphType.BREAK) {
					builder.append("\n");
				}
			}
		}
	}

	private Story getStory() {
		if (story == null) {
			// TODO: progress bar?
			story = lib.getStory(meta.getLuid(), null);
		}
		return story;
	}

	/**
	 * Display the given text on the window.
	 * 
	 * @param text
	 *            the text to display
	 */
	private void setText(String text) {
		textField.setText(text);
		textField.reflowData();
		textField.toTop();
	}

	/**
	 * Set the current chapter area to the correct value.
	 */
	private void setCurrentChapterText() {
		String name;
		if (chapter < 0) {
			name = " " + getStory().getMeta().getTitle();
		} else if (chapter == 0) {
			Chapter resume = getStory().getMeta().getResume();
			if (resume != null) {
				name = String.format(" %s", resume.getName());
			} else {
				// TODO: i18n
				name = "[No RESUME]";
			}
		} else {
			int max = getStory().getChapters().size();
			Chapter chap = getStory().getChapters().get(chapter - 1);
			name = String.format(" %d/%d: %s", chapter, max, chap.getName());
		}

		int width = getWidth() - currentChapter.getX();
		while (name.length() < width) {
			name += " ";
		}

		if (name.length() > width) {
			name = name.substring(0, width);
		}

		currentChapter.setLabel(name);

	}

	private static String desc(MetaData meta) {
		return String.format("%s: %s", meta.getLuid(), meta.getTitle());
	}

	/**
	 * Append a title (on its own 2 lines).
	 * 
	 * @param builder
	 *            the {@link StringBuilder} to append to
	 * @param title
	 *            the title itself
	 * @param level
	 *            the title level, 1 being the highest level, 2 second level and
	 *            so on
	 */
	private static StringBuilder appendTitle(StringBuilder builder,
			String title, int level) {
		String hr;
		switch (level) {
		case 1:
			hr = "======";
			break;
		case 2:
			hr = "=-=-=-";
			break;
		case 3:
			hr = "______";
			break;
		case 4:
			hr = "------";
			break;
		default:
			hr = "";
			break;
		}

		int fullPad = title.length() / hr.length();
		int rest = title.length() - (fullPad * hr.length());

		builder.append(title).append("\n");
		for (int i = 0; i < title.length() / hr.length(); i++) {
			builder.append(hr);
		}

		if (rest > 0) {
			builder.append(hr.substring(0, rest));
		}

		builder.append("\n");

		return builder;
	}
}