// ViewFavicon.java
// -----------------------
// part of YaCy
// Copyright 2016 by luccioman; https://github.com/luccioman
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.Map;
import javax.imageio.ImageIO;
import javax.imageio.stream.ImageInputStream;
import net.yacy.cora.document.id.DigestURL;
import net.yacy.cora.document.id.MultiProtocolURL;
import net.yacy.cora.protocol.HeaderFramework;
import net.yacy.cora.protocol.RequestHeader;
import net.yacy.cora.storage.ConcurrentARC;
import net.yacy.cora.util.ConcurrentLog;
import net.yacy.kelondro.util.FileUtils;
import net.yacy.kelondro.util.MemoryControl;
import net.yacy.kelondro.workflow.WorkflowProcessor;
import net.yacy.peers.graphics.EncodedImage;
import net.yacy.search.Switchboard;
import net.yacy.server.serverObjects;
import net.yacy.server.serverSwitch;
import net.yacy.visualization.ImageViewer;
/**
* Extends ViewImage behavior : add a specific favicon cache and use of a
* default image on loading error.
*
* @author luc
*
*/
public class ViewFavicon {
/** Single instance of ImageViewer */
private static final ImageViewer VIEWER = new ImageViewer();
/** Icons cache encoded as png */
private static Map<String, byte[]> pngIconCache = new ConcurrentARC<String, byte[]>(1000,
Math.max(10, Math.min(32, WorkflowProcessor.availableCPU * 2)));
/** Default icon local file */
private static final String defaulticon = "htroot/env/grafics/dfltfvcn.ico";
/**
* Default icon encoded as png : we use a bvte array as it is thread-safe
* instead of a ByteBuffer in EncodedImage
*/
private static byte[] defaultPNGEncodedIcon = null;
/**
* Try parsing image from post "url" parameter (authenticated users) or from
* "code" parameter (non authenticated users). When image could be parsed,
* try encoding to target format specified by header "EXT". When any error
* occurs, return default icon.
*
* @param header
* request header
* @param post
* post parameters
* @param env
* Switchboard instance
* @return an {@link EncodedImage} instance encoded in format specified in
* post, or an InputStream pointing to original image data.
*/
public static Object respond(final RequestHeader header, final serverObjects post, final serverSwitch env) {
final Switchboard sb = (Switchboard) env;
String ext = header.get(HeaderFramework.CONNECTION_PROP_EXT, null);
boolean isPNGTarget = "png".equalsIgnoreCase(ext);
ImageInputStream imageInStream = null;
InputStream inStream = null;
byte[] resultBytes = null;
try {
/* Clear icon cache when running out of memory */
if (MemoryControl.shortStatus()) {
pngIconCache.clear();
}
final boolean auth = ImageViewer.hasFullViewingRights(header, sb); // handle access rights
DigestURL url = VIEWER.parseURL(post, auth);
final String normalizedURL = url.toNormalform(false);
if (isPNGTarget) {
resultBytes = pngIconCache.get(normalizedURL);
}
/* Icon is not already in cache */
if (resultBytes == null) {
String urlExt = MultiProtocolURL.getFileExtension(url.getFileName());
if (ext != null && ext.equalsIgnoreCase(urlExt) && ImageViewer.isBrowserRendered(urlExt)) {
return VIEWER.openInputStream(post, sb.loader, auth, url);
}
/*
* When opening a file, the most efficient is to open
* ImageInputStream directly on file
*/
if (url.isFile()) {
imageInStream = ImageIO.createImageInputStream(url.getFSFile());
} else {
inStream = VIEWER.openInputStream(post, sb.loader, auth, url);
imageInStream = ImageIO.createImageInputStream(inStream);
}
// read image
EncodedImage encodedIcon = VIEWER.parseAndScale(post, auth, url, ext, imageInStream);
if (encodedIcon != null && !encodedIcon.getImage().isEmpty()) {
resultBytes = encodedIcon.getImage().getBytes();
if (isPNGTarget && encodedIcon.getImage().length() <= 10240) {
/* Only store in cache icon images below 10KB, png encoded */
pngIconCache.put(normalizedURL, resultBytes);
}
}
}
} catch (IOException e) {
ConcurrentLog.fine("ViewFavicon", "Error loading favicon, default one wille be used : " + e);
} finally {
/*
* imageInStream.close() method doesn't close source input stream
*/
if (inStream != null) {
try {
inStream.close();
} catch (IOException ignored) {
}
}
if (resultBytes == null) {
/*
* I any error occured when loading icon, return default one
*/
if (ext == null || isPNGTarget) {
/* Load default icon only once */
if (defaultPNGEncodedIcon == null) {
defaultPNGEncodedIcon = loadDefaultIcon(post, sb, ext);
}
resultBytes = defaultPNGEncodedIcon;
} else {
resultBytes = loadDefaultIcon(post, sb, ext);
}
}
}
return new ByteArrayInputStream(resultBytes);
}
/**
* Load default icon and encode it to ext format
*
* @param post
* post parameters
* @param sb
* Switchboard instance
* @param ext
* target image format
* @return icon encoded bytes, empty if and exception occured when loading
* or rendering
*/
private static byte[] loadDefaultIcon(final serverObjects post, final Switchboard sb, String ext) {
byte[] resultBytes;
byte[] defaultBytes = new byte[0];
try {
defaultBytes = FileUtils.read(new File(sb.getAppPath(), defaulticon));
} catch (final IOException initicon) {
defaultBytes = new byte[0];
} finally {
resultBytes = new EncodedImage(defaultBytes, ext, post.getBoolean("isStatic")).getImage().getBytes();
}
return resultBytes;
}
}