PrintServlet.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2022 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * 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 <http://www.gnu.org/licenses/>.
 */
 
package org.turro.elephant.print;
 
import com.sun.syndication.feed.synd.SyndFeed;
import com.sun.syndication.feed.synd.SyndFeedImpl;
import com.sun.syndication.io.FeedException;
import com.sun.syndication.io.SyndFeedOutput;
import java.io.IOException;
import java.net.URISyntaxException;
import java.net.URLDecoder;
import java.nio.charset.StandardCharsets;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.turro.collections.KeyValueMap;
import org.turro.elephant.context.Application;
import org.turro.elephant.context.ElephantApplication;
import org.turro.elephant.context.ElephantContext;
import org.turro.elephant.context.IConstructor;
import org.turro.elephant.context.IElement;
import org.turro.elephant.feed.IRSSable;
import org.turro.elephant.web.ElContext;
import org.turro.elephant.web.ElContextMap;
import org.turro.html.HtmlToPdf;
import org.turro.log.WebLoggers;
 
@WebServlet(name = "Print", urlPatterns = {"/print_/*"})
public class PrintServlet extends HttpServlet {
  
  private static String PDF = "/pdf_";
  
  private void processPrint(Application application, String path, KeyValueMap map) throws IOException {
    try {
      if(path.startsWith(PDF + "/")) {
        HttpServletResponse response = application.getHttpServletResponse();
        response.setContentType("application/pdf");
        path = "/print_" + path.substring(PDF.length());
        HtmlToPdf.of(ElephantContext.toUri(path).toURL(), ElephantContext.toUri(path))
                .removals("a#toTopButton,style,script,form")
                .pdf(response.getOutputStream());
      } else if(map.containsKey("read")) {
        IConstructor constructor = application.getConstructor();
        ElContext context = ElContextMap.getContext(path);
        constructor.forceIsAContext(context);
        constructor.getResponse().setContentType("text/html");
        constructor.getResponse().setCharacterEncoding(ElephantContext.getEncoding());
        constructor.setStopper(map.get(Boolean.class, "stopper", Boolean.TRUE) ? "<end/>" : null);
        printContext(constructor, context, 1, false);
        constructor.setRenderingContext(null);
      } else if(map.containsKey("rss")) {
        IConstructor constructor = application.getConstructor();
        ElContext context = ElContextMap.getContext(path);
        processRss(constructor, context);
        constructor.setRenderingContext(null);
      } else {
        IConstructor constructor = application.getConstructor();
        ElContext context = ElContextMap.getContext(path);
        constructor.forceIsAContext(context);
        constructor.startBody();
        printContext(constructor, context, map.get(Integer.class, "level", 10), map.get(Boolean.class, "title", Boolean.TRUE));
        constructor.setRenderingContext(null);
        constructor.endBody();
      }
    } catch (URISyntaxException | ServletException ex) {
      WebLoggers.severe(this).exception(ex).log();
    }
  }
 
  private void printContext(IConstructor constructor, ElContext context, int level, boolean title) throws ServletException, IOException {
    if(level <= 0) return;
    if(context.canPrint() && context.getWebContext().isWikiPage()) {
      constructor.setRenderingContext(context);
      if(title) constructor.getOut().print("<div class='context-title'>" + context.getName() + "</div>");
      context.getElement().startConstruction();
      for(ElContext ctx : context.getChildren()) {
        printContext(constructor, ctx, level-1, title);
      }
    }
  }
  
  private void processRss(IConstructor constructor, ElContext context) throws IOException {
    constructor.getResponse().setContentType("application/rss+xml");
    constructor.setRenderingContext(context);
    IElement el = context.getElement();
    if(el instanceof IRSSable) {
      SyndFeed feed = new SyndFeedImpl();
      feed.setFeedType("rss_1.0");
      feed.setEncoding(ElephantContext.getEncoding());
      feed.setTitle(ElephantContext.getSiteName() + " - " + context.getName());
      feed.setLink(ElephantContext.getServerUrl("http"));
      feed.setDescription(context.getName());
      feed.setEntries(((IRSSable) el).getSyndEntries());
      SyndFeedOutput output = new SyndFeedOutput();
      try {
        output.output(feed, constructor.getOut());
      } catch (FeedException ex) {
        WebLoggers.severe(this).exception(ex).log();
      }
    }
  }
 
  protected void processRequest(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
    KeyValueMap map = new KeyValueMap();
    for(String key : request.getParameterMap().keySet()) {
      map.put(key, URLDecoder.decode(request.getParameter(key), StandardCharsets.UTF_8));
    }
    processPrint(new ElephantApplication(request, response), request.getPathInfo(), map);
  }
  
  @Override
  protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
    processRequest(request, response);
  }
  
  @Override
  protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
    processRequest(request, response);
  }
  
  @Override
  public String getServletInfo() {
    return "Print servlet";
  }
 
}