-
Notifications
You must be signed in to change notification settings - Fork 138
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'master' into snyk-upgrade-ce692b96cbff5942cd77394b2ab30024
- Loading branch information
Showing
6 changed files
with
421 additions
and
184 deletions.
There are no files selected for viewing
144 changes: 144 additions & 0 deletions
144
core/src/main/java/com/predic8/membrane/core/interceptor/templating/StaticInterceptor.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,144 @@ | ||
| package com.predic8.membrane.core.interceptor.templating; | ||
|
|
||
| import com.predic8.membrane.annot.MCAttribute; | ||
| import com.predic8.membrane.annot.MCElement; | ||
| import com.predic8.membrane.annot.MCTextContent; | ||
| import com.predic8.membrane.core.beautifier.JSONBeautifier; | ||
| import com.predic8.membrane.core.exchange.Exchange; | ||
| import com.predic8.membrane.core.http.Message; | ||
| import com.predic8.membrane.core.interceptor.AbstractInterceptor; | ||
| import com.predic8.membrane.core.interceptor.Outcome; | ||
| import com.predic8.membrane.core.util.TextUtil; | ||
| import org.jetbrains.annotations.NotNull; | ||
| import org.slf4j.Logger; | ||
| import org.slf4j.LoggerFactory; | ||
|
|
||
| import java.io.IOException; | ||
| import java.io.StringReader; | ||
|
|
||
| import static com.predic8.membrane.core.http.MimeType.*; | ||
| import static com.predic8.membrane.core.interceptor.Outcome.CONTINUE; | ||
| import static com.predic8.membrane.core.util.TextUtil.unifyIndent; | ||
| import static java.nio.charset.StandardCharsets.UTF_8; | ||
| import static org.apache.commons.text.StringEscapeUtils.escapeHtml4; | ||
|
|
||
| @MCElement(name = "static", mixed = true) | ||
| public class StaticInterceptor extends AbstractInterceptor { | ||
|
|
||
| protected String location; | ||
|
|
||
| protected String textTemplate; | ||
|
|
||
| protected String contentType = TEXT_PLAIN; | ||
|
|
||
| protected Boolean pretty = false; | ||
|
|
||
| protected final JSONBeautifier jsonBeautifier = new JSONBeautifier(); | ||
|
|
||
| protected static final Logger log = LoggerFactory.getLogger("StaticInterceptor"); | ||
|
|
||
| public StaticInterceptor() { | ||
| name = "Static"; | ||
| } | ||
|
|
||
| @Override | ||
| public Outcome handleRequest(Exchange exc) throws Exception { | ||
| return handleInternal(exc.getRequest()); | ||
| } | ||
|
|
||
| @Override | ||
| public Outcome handleResponse(Exchange exc) throws Exception { | ||
| return handleInternal(exc.getResponse()); | ||
| } | ||
|
|
||
| private Outcome handleInternal(Message msg) { | ||
| msg.setBodyContent(getTemplateBytes()); | ||
| msg.getHeader().setContentType(getContentType()); | ||
| return CONTINUE; | ||
| } | ||
|
|
||
| private byte @NotNull [] getTemplateBytes() { | ||
| if (!pretty) | ||
| return textTemplate.getBytes(UTF_8); | ||
|
|
||
| return switch (contentType) { | ||
| case APPLICATION_JSON -> prettifyJson(textTemplate).getBytes(UTF_8); | ||
| case APPLICATION_XML, APPLICATION_SOAP, TEXT_HTML, TEXT_XML, TEXT_HTML_UTF8, TEXT_XML_UTF8 -> prettifyXML(textTemplate).getBytes(UTF_8); | ||
| default -> unifyIndent(textTemplate).getBytes(UTF_8); | ||
| }; | ||
| } | ||
|
|
||
| private String prettifyXML(String text) { | ||
| try { | ||
| return TextUtil.formatXML(new StringReader(text)); | ||
| } catch (Exception e) { | ||
| log.warn("Failed to format XML", e); | ||
| return text; | ||
| } | ||
| } | ||
|
|
||
| String prettifyJson(String text) { | ||
| try { | ||
| return jsonBeautifier.beautify(text); | ||
| } catch (IOException e) { | ||
| log.warn("Failed to format JSON", e); | ||
| return text; | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public void init() throws Exception { | ||
| if (this.getLocation() != null && (getTextTemplate() != null && !getTextTemplate().isBlank())) { | ||
| throw new IllegalStateException("On <" + getName() + ">, ./text() and ./@location cannot be set at the same time."); | ||
| } | ||
| } | ||
|
|
||
| public String getLocation() { | ||
| return location; | ||
| } | ||
|
|
||
| @MCAttribute | ||
| public void setLocation(String location){ | ||
| this.location = location; | ||
| } | ||
|
|
||
| public String getTextTemplate() { | ||
| return textTemplate; | ||
| } | ||
|
|
||
| @MCTextContent | ||
| public void setTextTemplate(String textTemplate) { | ||
| this.textTemplate = textTemplate; | ||
| } | ||
|
|
||
| protected String getName() { | ||
| return getClass().getAnnotation(MCElement.class).name(); | ||
| } | ||
|
|
||
| public String getContentType() { | ||
| return contentType; | ||
| } | ||
|
|
||
| @MCAttribute | ||
| public void setContentType(String contentType) { | ||
| this.contentType = contentType; | ||
| } | ||
|
|
||
| public Boolean getPretty() { | ||
| return pretty; | ||
| } | ||
|
|
||
| @MCAttribute | ||
| public void setPretty(String pretty) { | ||
| this.pretty = Boolean.valueOf(pretty); | ||
| } | ||
|
|
||
| private String formatAsHtml(String plaintext) { | ||
| return String.join("<br />", escapeHtml4(plaintext).split("\n")); | ||
| } | ||
|
|
||
| @Override | ||
| public String getShortDescription() { | ||
| return formatAsHtml(textTemplate); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.