Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
io.webfolder.micro4j.mvc.jaxrs.ViewWriter Maven / Gradle / Ivy
/**
* The MIT License
* Copyright © 2016 - 2020 WebFolder OÜ
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
package io.webfolder.micro4j.mvc.jaxrs;
import static javax.ws.rs.core.HttpHeaders.CACHE_CONTROL;
import static javax.ws.rs.core.MediaType.TEXT_HTML_TYPE;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.StringWriter;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import java.util.LinkedHashMap;
import java.util.Map;
import javax.ws.rs.GET;
import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.core.UriInfo;
import javax.ws.rs.ext.MessageBodyWriter;
import io.webfolder.micro4j.mvc.View;
import io.webfolder.micro4j.mvc.ViewModel;
import io.webfolder.micro4j.mvc.template.TemplateEngine;
public class ViewWriter implements MessageBodyWriter {
@Context
private HttpHeaders requestHttpHeaders;
@Context
private UriInfo uriInfo;
private final TemplateEngine engine;
private final ViewInterceptor viewInterceptor;
private static final String PJAX = "X-PJAX";
private static final String HTMX = "X-HX-Request";
public ViewWriter(TemplateEngine engine) {
this.engine = engine;
this.viewInterceptor = engine.getConfiguration().getViewInterceptor();
}
@Override
public boolean isWriteable(Class type, Type genericType, Annotation[] annotations, MediaType mediaType) {
return isCompatibleMediaType(mediaType, annotations) && isView(type, annotations);
}
protected boolean isCompatibleMediaType(MediaType mediaType, Annotation[] annotations) {
if (annotations != null) {
for (Annotation next : annotations) {
if (GET.class.equals(next.getClass()) && ! TEXT_HTML_TYPE.isCompatible(mediaType) ) {
return false;
}
}
}
return true;
}
protected boolean isView(Class type, Annotation[] annotations) {
return ViewModel.class.isAssignableFrom(type) || hasAnnotation(annotations);
}
protected boolean hasAnnotation(Annotation[] annotations) {
if (annotations == null || annotations.length <= 0) {
return false;
}
for (Annotation annotation : annotations) {
if (View.class.isAssignableFrom(annotation.annotationType())) {
return true;
}
}
return false;
}
@Override
public long getSize(Object data, Class type, Type genericType, Annotation[] annotations,
MediaType mediaType) {
return -1;
}
@Override
public void writeTo(Object data, Class type, Type genericType, Annotation[] annotations, MediaType mediaType,
MultivaluedMap httpHeaders, OutputStream entityStream)
throws IOException, WebApplicationException {
String name = null;
String container = null;
Object context = null;
if (data != null && ViewModel.class.isAssignableFrom(data.getClass())) {
ViewModel model = (ViewModel) data;
name = model.getName();
container = model.getContainer();
context = model.getEntity();
} else {
for (Annotation annotation : annotations) {
if (View.class.isAssignableFrom(annotation.annotationType())) {
View view = (View) annotation;
name = view.value();
if ( ! view.container().trim().isEmpty() ) {
container = view.container();
}
context = data;
break;
}
}
}
writeTo(name, container, context, httpHeaders, entityStream);
}
protected void writeTo(String name, String container, Object context, MultivaluedMap responseHttpHeaders,
OutputStream entityStream) throws IOException {
String containerName = container;
String defaultContainer = engine.getConfiguration().getContainer();
if (container == null || container.isEmpty() && ! defaultContainer.isEmpty()) {
containerName = engine.getConfiguration().getContainer();
}
setHeader(name, context, responseHttpHeaders);
Map parentContext = new LinkedHashMap<>();
boolean isPjax = isPjaxRequest(responseHttpHeaders);
if (isPjax || containerName.trim().isEmpty()) {
try (StringWriter stringWriter = new StringWriter();
OutputStreamWriter containerWriter = new OutputStreamWriter(entityStream, engine.getConfiguration().getCharset())) {
if ( viewInterceptor != null ) {
String content = stringWriter.toString();
String modifiedContent = viewInterceptor.intercept(uriInfo, name,
isPjax, requestHttpHeaders,
responseHttpHeaders, content);
containerWriter.write(modifiedContent);
} else {
engine.execute(name, context, parentContext, containerWriter);
}
}
} else {
StringWriter pageWriter = new StringWriter();
engine.execute(name, context, parentContext, pageWriter);
Object contentFunction = engine.createFunction(pageWriter.toString());
try (StringWriter stringWriter = new StringWriter();
OutputStreamWriter containerWriter = new OutputStreamWriter(entityStream, engine.getConfiguration().getCharset())) {
parentContext.put(engine.getConfiguration().getBodyName(), contentFunction);
if ( viewInterceptor != null ) {
engine.execute(containerName, context, parentContext, stringWriter);
String content = stringWriter.toString();
String modifiedContent = viewInterceptor.intercept(uriInfo, name,
isPjax, requestHttpHeaders,
responseHttpHeaders, content);
containerWriter.write(modifiedContent);
} else {
engine.execute(containerName, context, parentContext, stringWriter);
}
}
}
}
protected void setHeader(String name, Object context, MultivaluedMap httpHeaders) {
if ( ! httpHeaders.containsKey(CACHE_CONTROL) ) {
httpHeaders.add(CACHE_CONTROL, "no-cache, no-store, must-revalidate");
}
}
protected boolean isPjaxRequest(MultivaluedMap httpHeaders) {
return requestHttpHeaders.getHeaderString(PJAX) != null ||
"true".equals(requestHttpHeaders.getHeaderString(HTMX));
}
}
