ContentService.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2016 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.action;
 
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.imageio.ImageIO;
import javax.servlet.ServletContext;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.turro.collections.parser.ParserException;
import org.turro.string.Strings;
import org.apache.commons.fileupload.FileItemIterator;
import org.apache.commons.fileupload.FileItemStream;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.apache.commons.fileupload.util.Streams;
import org.turro.barcode.Barcode;
import org.turro.collections.KeyValueMap;
import org.turro.elephant.context.Application;
import org.turro.elephant.context.ElephantContext;
import org.turro.elephant.context.IConstructor;
import org.turro.elephant.direct.DirectContent;
import org.turro.elephant.direct.DirectContents;
import org.turro.elephant.direct.EntryPoint;
import org.turro.elephant.direct.IDirectContent;
import org.turro.elephant.direct.IEntryPoint;
import org.turro.elephant.impl.util.CookieUtil;
import org.turro.reflection.Instances;
import org.turro.util.ImageUtil;
 
@DirectContent(identifier="content-service")
public class ContentService implements IDirectContent {
 
  public static String createURL(IConstructor constructor, KeyValueMap values) {
    String exrino = Actions.createRightNowAction(values);
    return ElephantContext.getRootWebPath() +
            DirectContents.DIRECT_CONTENT_PATH + getIdentifier() +
            "?" + exrino;
  }
 
  public static String createURL(IConstructor constructor, String values) {
    String exrino = Actions.createRightNowAction(values);
    return ElephantContext.getRootWebPath() +
            DirectContents.DIRECT_CONTENT_PATH + getIdentifier() +
            "?" + exrino;
  }
  
  public static String createEntryPoint(IConstructor constructor, String entryPoint, String valueStr) {
    try {
      return createEntryPoint(constructor, entryPoint, new KeyValueMap(valueStr));
    } catch (ParserException ex) {
      Logger.getLogger(ContentService.class.getName()).log(Level.SEVERE, null, ex);
    }
    return null;
  }
 
  public static String createEntryPoint(IConstructor constructor, String entryPoint, KeyValueMap values) {
    values.put("type", "entry-point");
    values.put("entry-point", entryPoint);
    String exrino = Actions.createRightNowAction(values);
    return ElephantContext.getRootWebPath() +
            DirectContents.DIRECT_CONTENT_PATH + getIdentifier() +
            "?" + exrino;
  }
 
  public static String getIdentifier() {
    return ContentService.class.getAnnotation(DirectContent.class).identifier();
  }
 
  @Override
  public boolean itsMe(String id) {
    return getIdentifier().equals(id);
  }
 
  @Override
  public boolean myTurn(HttpServletRequest request) {
    return DirectContents.isYourTurn(request, getIdentifier());
  }
 
  @Override
  public void execute(ServletContext context, HttpServletRequest request, HttpServletResponse response) {
    boolean isMultipart = ServletFileUpload.isMultipartContent(request);
    if(isMultipart) {
      try {
        ServletFileUpload upload = new ServletFileUpload();
        upload.setHeaderEncoding(ElephantContext.getEncoding());
        FileItemIterator iter = upload.getItemIterator(request);
        File file = null;
        String redirect = null;
        Double fixWidth = null, fixHeight = null;
        while(iter.hasNext()) {
          FileItemStream item = iter.next();
          String name = item.getFieldName();
          InputStream stream = item.openStream();
          if(item.isFormField()) {
            switch(name) {
              case "fileName":
                file = new File(ElephantContext.getRealPath(Streams.asString(stream, ElephantContext.getEncoding())));
                break;
              case "redirect":
                redirect = Streams.asString(stream, ElephantContext.getEncoding());
                break;
              case "fixWidth":
                fixWidth = Double.valueOf(Streams.asString(stream, ElephantContext.getEncoding()));
                break;
              case "fixHeight":
                fixHeight = Double.valueOf(Streams.asString(stream, ElephantContext.getEncoding()));
                break;
            }
          } else if(file != null) {
            file.getParentFile().mkdirs();
            FileOutputStream fos = new FileOutputStream(file);
            Streams.copy(item.openStream(), fos, true);
          }
        }
        if(file != null && fixWidth != null && fixHeight != null) {
          ImageUtil.smart(file, file, fixWidth, fixHeight);
        }
        if(!Strings.isBlank(redirect)) {
          Application.getApplication().sendRedirect(redirect);
        }
      } catch (FileUploadException | IOException ex) {
        Logger.getLogger(ContentService.class.getName()).log(Level.SEVERE, ElephantContext.logMsg("Multipart"), ex);
      }
    } else {
      KeyValueMap map = Actions.getRightNowAction(ElephantContext.getConstructor(request, response));
      if(map != null) {
        String type = map.get("type");
        if("qrcode".equals(type)) {
          try {
            response.setContentType("image/png");
            Barcode qr = new Barcode();
            qr.writeImageCode(
                    response.getOutputStream(),
                    qr.encodeQrCode(map.get("data"), 300),
                    "png");
          } catch (IOException ex) {
            Logger.getLogger(ContentService.class.getName()).log(Level.SEVERE, ElephantContext.logMsg("QRCode"), ex);
          }
        } else if("image".equals(type)) {
          try {
            response.setContentType("image/png");
            ImageIO.write(ImageIO.read(new File(map.get("data"))), "png", response.getOutputStream());
          } catch (IOException ex) {
            Logger.getLogger(ContentService.class.getName()).log(Level.SEVERE, ElephantContext.logMsg("Image"), ex);
          }
        } else if("cookie".equals(type)) {
          if(map.containsKey("remove")) {
            CookieUtil.deleteCookie(response, map.get("name"), map.getOrDefault("path", "/"));
          } else {
            Application.setCookie(map.get("name"), map.get("value"), 
                    map.getOrDefault("path", "/"), 
                    map.get(Integer.class, "age", 60 * 60 * 24 * 365 * 10));
          }
        } else if("entry-point".equals(type)) {
          if(map.containsKey("entry-point")) {
            getEntryPoint(map.get("entry-point")).execute(map);
          }
        }
      }
    }
  }
  
  private IEntryPoint getEntryPoint(String identifier) {
    return Instances.cached().byAnnotation(EntryPoint.class, IEntryPoint.class).stream()
            .filter(e -> ((EntryPoint) e.getClass().getAnnotation(EntryPoint.class)).identifier().equals(identifier))
            .findFirst().orElse(new IEntryPoint() {});
  }
  
}