/* * 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; } } }