/*
* Copyright (C) 2008-2020, Juick
*
* 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 com.juick.util;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.MediaType;
import org.springframework.web.multipart.MultipartFile;
import javax.imageio.ImageIO;
import javax.imageio.ImageReader;
import javax.imageio.stream.ImageInputStream;
import java.io.BufferedInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URI;
import java.net.URL;
import java.net.URLConnection;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Iterator;
import java.util.UUID;
/**
*
* @author Ugnich Anton
*/
public class HttpUtils {
private static final Logger logger = LoggerFactory.getLogger(HttpUtils.class);
public static URI receiveMultiPartFile(MultipartFile attach, String tmpDir) throws IOException {
if (attach != null && !attach.isEmpty()) {
ImageInputStream iis = ImageIO.createImageInputStream(attach.getInputStream());
Iterator<ImageReader> readers = ImageIO.getImageReaders(iis);
String format = StringUtils.EMPTY;
while (readers.hasNext()) {
ImageReader read = readers.next();
format = read.getFormatName();
}
String attachmentType = attachmentTypeFromFormat(format);
if (attachmentType.equals("jpg") || attachmentType.equals("png")) {
String attachmentFName = DigestUtils.md5Hex(UUID.randomUUID().toString()) + "." + attachmentType;
try {
Files.write(Paths.get(tmpDir, attachmentFName),
attach.getBytes());
return URI.create(String.format("juick://%s", attachmentFName));
} catch (IOException e) {
logger.warn("file receive error", e);
}
}
logger.warn("file type is unknown: {}", attach.getOriginalFilename());
}
return URI.create(StringUtils.EMPTY);
}
private static String attachmentTypeFromFormat(String format) throws IOException {
if (format != null && format.equals("JPEG")) {
return "jpg";
} else if (format != null && format.equals("png")) {
return "png";
} else {
throw new IOException("Wrong file type: " + format);
}
}
public static String mediaType(String attachmentType) {
return attachmentType.equals("jpg") ? MediaType.IMAGE_JPEG_VALUE : MediaType.IMAGE_PNG_VALUE;
}
public static URI downloadImage(URL url, String tmpDir) throws IOException {
ImageInputStream iis = ImageIO.createImageInputStream(url.openStream());
Iterator<ImageReader> readers = ImageIO.getImageReaders(iis);
String format = StringUtils.EMPTY;
while (readers.hasNext()) {
ImageReader read = readers.next();
format = read.getFormatName();
}
URLConnection urlConn;
try {
urlConn = url.openConnection();
} catch (IOException e) {
logger.error(String.format("Failed open url: %s", url.toString()));
throw e;
}
try (InputStream is = new BufferedInputStream(urlConn.getInputStream())) {
String attachmentType = attachmentTypeFromFormat(format);
String attachmentFName = DigestUtils.md5Hex(UUID.randomUUID().toString()) + "." + attachmentType;
Files.copy(is, Paths.get(tmpDir, attachmentFName));
return URI.create(String.format("juick://%s", attachmentFName));
} catch (IOException e) {
logger.error(String.format("Failed download image by url: %s", url.toString()), e);
throw e;
}
}
}