X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FParserFilter.java;h=5f6fd3e5790d7f266e5220c357ca5cd8f6c79ba1;hb=7755990dfedede27bb6422de66e27b484255a1ad;hp=632af02f2613bd9b67c01fc8aec3615f2aca44be;hpb=b1ad350c14abb8614913ef0ff0a64eeb1f402934;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/template/ParserFilter.java b/src/main/java/net/pterodactylus/sone/template/ParserFilter.java
index 632af02..5f6fd3e 100644
--- a/src/main/java/net/pterodactylus/sone/template/ParserFilter.java
+++ b/src/main/java/net/pterodactylus/sone/template/ParserFilter.java
@@ -21,6 +21,8 @@ import java.io.IOException;
 import java.io.StringReader;
 import java.io.StringWriter;
 import java.io.Writer;
+import java.util.ArrayList;
+import java.util.List;
 import java.util.Map;
 
 import net.pterodactylus.sone.core.Core;
@@ -51,7 +53,7 @@ public class ParserFilter implements Filter {
 	private final Core core;
 
 	/** The link parser. */
-	private final SoneTextParser textParser;
+	private final SoneTextParser soneTextParser;
 
 	/** The template context factory. */
 	private final TemplateContextFactory templateContextFactory;
@@ -70,11 +72,13 @@ public class ParserFilter implements Filter {
 	 *            The core
 	 * @param templateContextFactory
 	 *            The context factory for rendering the parts
+	 * @param soneTextParser
+	 *            The Sone text parser
 	 */
-	public ParserFilter(Core core, TemplateContextFactory templateContextFactory) {
+	public ParserFilter(Core core, TemplateContextFactory templateContextFactory, SoneTextParser soneTextParser) {
 		this.core = core;
 		this.templateContextFactory = templateContextFactory;
-		textParser = new SoneTextParser(core);
+		this.soneTextParser = soneTextParser;
 	}
 
 	/**
@@ -83,6 +87,12 @@ public class ParserFilter implements Filter {
 	@Override
 	public Object format(TemplateContext templateContext, Object data, Map<String, String> parameters) {
 		String text = String.valueOf(data);
+		int length = -1;
+		try {
+			length = Integer.parseInt(parameters.get("length"));
+		} catch (NumberFormatException nfe1) {
+			/* ignore. */
+		}
 		String soneKey = parameters.get("sone");
 		if (soneKey == null) {
 			soneKey = "sone";
@@ -95,7 +105,31 @@ public class ParserFilter implements Filter {
 		SoneTextParserContext context = new SoneTextParserContext(request, sone);
 		StringWriter parsedTextWriter = new StringWriter();
 		try {
-			render(parsedTextWriter, textParser.parse(context, new StringReader(text)));
+			Iterable<Part> parts = soneTextParser.parse(context, new StringReader(text));
+			if (length > -1) {
+				List<Part> shortenedParts = new ArrayList<Part>();
+				for (Part part : parts) {
+					if (part instanceof PlainTextPart) {
+						String longText = ((PlainTextPart) part).getText();
+						if (length >= longText.length()) {
+							shortenedParts.add(part);
+						} else {
+							shortenedParts.add(new PlainTextPart(longText.substring(0, length)));
+						}
+						length -= longText.length();
+					} else if (part instanceof LinkPart) {
+						shortenedParts.add(part);
+						length -= ((LinkPart) part).getText().length();
+					} else {
+						shortenedParts.add(part);
+					}
+					if (length <= 0) {
+						break;
+					}
+				}
+				parts = shortenedParts;
+			}
+			render(parsedTextWriter, parts);
 		} catch (IOException ioe1) {
 			/* no exceptions in a StringReader or StringWriter, ignore. */
 		}