org.structr.web.converter.ImageConverter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of structr-ui Show documentation
Structr is an open source framework based on the popular Neo4j graph database.
The newest version!
/**
 * Copyright (C) 2010-2016 Structr GmbH
 *
 * This file is part of Structr .
 *
 * Structr 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.
 *
 * Structr 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 Structr.  If not, see .
 */
package org.structr.web.converter;

import net.sf.jmimemagic.Magic;
import net.sf.jmimemagic.MagicMatch;

import org.apache.commons.lang3.StringUtils;

import org.structr.web.common.ImageHelper;
import org.structr.common.KeyAndClass;
import org.structr.common.SecurityContext;
import org.structr.core.GraphObject;
import org.structr.core.converter.PropertyConverter;
import org.structr.web.entity.Image;

//~--- JDK imports ------------------------------------------------------------

import java.util.logging.Level;
import java.util.logging.Logger;

//~--- classes ----------------------------------------------------------------

/**
 * Converts image data into an image node.
 *
 * If a {@link KeyAndClass} object is given, the image will be created with
 * the corresponding type and with setProperty to the given property key.
 *
 * If no {@link KeyAndClass} object is given, the image data will be set on
 * the image node itself.
 *
 *
 */
public class ImageConverter extends PropertyConverter {

	private static final Logger logger = Logger.getLogger(ImageConverter.class.getName());

	private KeyAndClass keyAndClass = null;

	public ImageConverter(SecurityContext securityContext, GraphObject entity, KeyAndClass kc) {

		super(securityContext, entity);

		this.keyAndClass = kc;
	}

	//~--- methods --------------------------------------------------------

	@Override
	public Object convert(final Object source) {

		if (source == null) {

			return false;
		}

		try {

			Image img = null;

			try {
				if (source instanceof byte[]) {

					byte[] data      = (byte[]) source;
					MagicMatch match = Magic.getMagicMatch(data);
					String mimeType  = match.getMimeType();

					if (keyAndClass != null) {

						img = (Image) ImageHelper.createFile(securityContext, data, mimeType, keyAndClass.getCls());

					} else {

						ImageHelper.setImageData((Image) currentObject, data, mimeType);

					}

				} else if (source instanceof String) {

					String sourceString = (String) source;

					if (StringUtils.isNotBlank(sourceString)) {

						if (keyAndClass != null) {

							// UUID?
							if (sourceString.length() == 32) {

								img = (Image) ImageHelper.transformFile(securityContext, sourceString, keyAndClass != null ? keyAndClass.getCls() : null);
							}

							if (img == null) {

								img = (Image) ImageHelper.createFileBase64(securityContext, sourceString, keyAndClass != null ? keyAndClass.getCls() : null);

							}

						} else {

							ImageHelper.decodeAndSetFileData((Image) currentObject, sourceString);

						}
					}

				}

			} catch (Throwable t) {
				logger.log(Level.WARNING, "Cannot create image node from given data", t);
			}

			if (img != null) {

				// manual indexing of UUID needed here to avoid a 404 in the following setProperty call
				img.updateInIndex();
				currentObject.setProperty(keyAndClass.getPropertyKey(), img);
			}


		} catch (Throwable t) {

			logger.log(Level.WARNING, "Cannot create image node from given data", t);
		}

		return null;
	}

	@Override
	public Object revert(Object source) {

		if (currentObject instanceof Image) {
			return ImageHelper.getBase64String((Image) currentObject);
		} else {
			return source;
		}
	}

}