/* * Copyright (C) 2008-2020, Juick * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as * published by the Free Software Foundation, either version 3 of the * License, or (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see . */ package com.mitchellbosecke.pebble.extension.filters; import com.juick.model.Message; import com.juick.util.MessageUtils; import io.pebbletemplates.pebble.extension.Filter; import io.pebbletemplates.pebble.extension.escaper.SafeString; import io.pebbletemplates.pebble.template.EvaluationContext; import io.pebbletemplates.pebble.template.PebbleTemplate; import org.apache.commons.lang3.StringUtils; import java.util.List; import java.util.Map; /** * Created by vitalyster on 04.05.2017. */ public class FormatMessageFilter implements Filter { @Override public Object apply(Object input, Map args, PebbleTemplate self, EvaluationContext context, int lineNumber) { if (input instanceof Message) { Message msg = (Message) input; if (msg.isHtml()) { return new SafeString(msg.getText()); } boolean isCode = msg.getTags().stream().anyMatch(t -> t.getName().equals("code")); String formattedMessage = isCode ? MessageUtils.formatMessageCode(StringUtils.defaultString(msg.getText())) : MessageUtils.formatMessage(StringUtils.defaultString(msg.getText())); if (MessageUtils.isPM(msg)) { return new SafeString(formattedMessage); } else { String toUserString = msg.getTo().getUid() == 0 ? String.format("@%s", msg.getTo().getUri().toASCIIString(), msg.getTo().getName()) : String.format("@%s", msg.getTo().getName(), msg.getTo().getName()); String formatString = MessageUtils.replyStartsWithQuote(msg) ? "%s,\n%s" : "%s, %s"; String msgTxt = msg.getRid() > 0 ? String.format(formatString, toUserString, formattedMessage) : formattedMessage; return new SafeString(msgTxt); } } throw new IllegalArgumentException("invalid input"); } @Override public List getArgumentNames() { return null; } }