First version (slashdot supported)

[gofetch.git] / src / be / nikiroo / gofetch / output / Gopher.java

package be.nikiroo.gofetch.output;

import java.util.List;

import be.nikiroo.gofetch.StringJustifier;
import be.nikiroo.gofetch.data.Comment;
import be.nikiroo.gofetch.data.Story;
import be.nikiroo.gofetch.support.BasicSupport.Type;

public class Gopher extends Output {
	static private final int LINE_SIZE = 70;

	private String hostname;
	private int port;

	public Gopher(Type type, String hostname, int port) {
		super(type);

		this.hostname = hostname;
		this.port = port;
	}

	@Override
	public String getIndexHeader() {
		return "iHello world!\r\niThis is my news site.\r\ni\r\n";
	}

	@Override
	public String getIndexFooter() {
		return "";
	}

	@Override
	public String export(Story story) {
		return append(new StringBuilder(), story, false).append("i\r\ni\r\n")
				.toString();
	}

	@Override
	public String export(Story story, List<Comment> comments) {
		StringBuilder builder = new StringBuilder();
		append(builder, story, true);

		builder.append("i\r\n");

		for (Comment comment : comments) {
			append(builder, comment, "");
		}

		builder.append("i\r\n");

		return builder.toString();
	}

	private StringBuilder append(StringBuilder builder, Comment comment,
			String space) {

		if (space.length() > LINE_SIZE - 20) {
			space = space.substring(0, LINE_SIZE - 20);
		}

		appendLeft(builder, comment.getTitle(), ">> ", "   ", space);
		appendLeft(builder, "(" + comment.getAuthor() + ")", "   ", "   ",
				space);

		builder.append("i\r\n");

		appendLeft(builder, comment.getContent(), "   ", "   ", space);

		builder.append("i\r\n");
		for (Comment subComment : comment) {
			append(builder, subComment, space + "   ");
			builder.append("i\r\n");
		}

		return builder;
	}

	private StringBuilder append(StringBuilder builder, Story story,
			boolean links) {
		if (links) {
			appendCenter(builder, story.getTitle(), true);
			builder.append("i\r\n");
			appendLeft(builder, story.getDetails(), "", "", "  ");
			builder.append("i\r\n");
			builder.append("i  o News link: ").append(story.getUrlInternal());
			builder.append("i\r\n");
			builder.append("i  o Source link: ").append(story.getUrlExternal());
			builder.append("i\r\n");
		} else {
			builder.append('1').append(story.getTitle()) //
					.append('\t').append("0").append(story.getSelector()) //
					.append('\t').append(hostname) //
					.append('\t').append(port) //
					.append("\r\n");
			appendLeft(builder, story.getDetails(), "", "", "  ");
		}

		builder.append("i\r\n");

		appendLeft(builder, story.getContent(), "", "", "    ");

		builder.append("i\r\n");

		return builder;
	}

	// note: adds "i"
	private static void appendCenter(StringBuilder builder, String text,
			boolean allCaps) {
		if (allCaps) {
			text = text.toUpperCase();
		}

		for (String line : StringJustifier.center(text, LINE_SIZE)) {
			builder.append("i").append(line).append("\r\n");
		}
	}

	// note: adds "i"
	private static void appendLeft(StringBuilder builder, String text,
			String prependFirst, String prependOthers, String space) {
		String prepend = prependFirst;
		for (String line : StringJustifier.left(text,
				LINE_SIZE - space.length())) {
			builder.append("i").append(space).append(prepend).append(line)
					.append("\r\n");
			prepend = prependOthers;
		}
	}
}