-
-
Save steinarb/ec1c22513a64870a880f21e134ea33fb to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
File downloadImageUrlToTempFile(AlbumEntry albumEntry, Path tempDir) { | |
var imageUrl = albumEntry.getImageUrl(); | |
if (imageUrl == null || imageUrl.isEmpty()) { | |
throw new OldAlbumException(String.format("Unable to download album entry matching id=%d, imageUrl is missing", albumEntry.getId())); | |
} | |
var fileName = findFileNamePartOfUrl(imageUrl); | |
var tempfile = tempDir.resolve(fileName).toFile(); | |
IIOImage image = null; | |
ImageWriter writer = null; | |
try { | |
HttpURLConnection connection = getConnectionFactory().connect(imageUrl); | |
connection.setRequestMethod("GET"); | |
try(var inputStream = ImageIO.createImageInputStream(connection.getInputStream())) { | |
var readers = ImageIO.getImageReaders(inputStream); | |
if (readers.hasNext()) { | |
var reader = readers.next(); | |
writer = ImageIO.getImageWriter(reader); | |
reader.setInput(inputStream); | |
image = reader.readAll(0, null); | |
} else { | |
throw new OldAlbumException(String.format("Album entry matching id=%d with url=\"%s\" not recognizable as an image. Download failed", albumEntry.getId(), albumEntry.getImageUrl())); | |
} | |
} | |
} catch (IOException e) { | |
throw new OldAlbumException(String.format("Unable to download album entry matching id=%d from url=\"%s\"", albumEntry.getId(), albumEntry.getImageUrl()), e); | |
} | |
return writeImageWithModifiedMetadataToTempFile(tempfile, albumEntry, image, writer); | |
} | |
File writeImageWithModifiedMetadataToTempFile(File tempfile, AlbumEntry albumEntry, IIOImage image, ImageWriter writer) { | |
var metadataAsTree = (IIOMetadataNode) image.getMetadata().getAsTree("javax_imageio_jpeg_image_1.0"); | |
var markerSequence = findMarkerSequenceAndCreateIfNotFound(metadataAsTree); | |
setJfifCommentFromAlbumEntryDescription(markerSequence, albumEntry); | |
try { | |
writeDateTitleAndDescriptionToExifDataStructure(markerSequence, albumEntry); | |
try (var outputStream = ImageIO.createImageOutputStream(new FileOutputStream(tempfile))){ | |
writer.setOutput(outputStream); | |
var param = writer.getDefaultWriteParam(); | |
var modifiedMetadata = writer.getDefaultImageMetadata(ImageTypeSpecifiers.createFromRenderedImage(image.getRenderedImage()), param); | |
modifiedMetadata.setFromTree("javax_imageio_jpeg_image_1.0", metadataAsTree); | |
image.setMetadata(modifiedMetadata); | |
writer.write(image); | |
Files.setLastModifiedTime(tempfile.toPath(), FileTime.from(albumEntry.getLastModified().toInstant())); | |
return tempfile; | |
} | |
} catch (IOException e) { | |
throw new OldAlbumException(String.format("Unable to save local copy of album entry matching id=%d from url=\"%s\"", albumEntry.getId(), albumEntry.getImageUrl()), e); | |
} | |
} | |
void writeDateTitleAndDescriptionToExifDataStructure(IIOMetadataNode markerSequence, AlbumEntry albumEntry) throws IOException { | |
Collection<Entry> entries = new ArrayList<>(); | |
entries.add(new TIFFEntry(TIFF.TAG_DATE_TIME, formatLastModifiedTimeAsExifDateString(albumEntry))); | |
entries.add(new TIFFEntry(TIFF.TAG_IMAGE_DESCRIPTION, albumEntry.getTitle())); | |
try (var bytes = new ByteArrayOutputStream()) { | |
bytes.write("Exif".getBytes(StandardCharsets.US_ASCII)); | |
bytes.write(new byte[2]); | |
try(var imageOutputStream = new MemoryCacheImageOutputStream(bytes)) { | |
new TIFFWriter().write(entries, imageOutputStream); | |
} | |
IIOMetadataNode exif = new IIOMetadataNode("unknown"); | |
exif.setAttribute("MarkerTag", String.valueOf(0xE1)); // APP1 or "225" | |
exif.setUserObject(bytes.toByteArray()); | |
markerSequence.appendChild(exif); | |
} | |
} | |
String formatLastModifiedTimeAsExifDateString(AlbumEntry albumEntry) { | |
var exifDateTimeFormat = new SimpleDateFormat("yyyy:MM:dd HH:mm:ss"); | |
exifDateTimeFormat.setTimeZone(TimeZone.getTimeZone("Europe/Oslo")); | |
var datetime = exifDateTimeFormat.format(albumEntry.getLastModified()); | |
return datetime; | |
} | |
IIOMetadataNode findMarkerSequenceAndCreateIfNotFound(IIOMetadataNode metadataAsTree) { | |
var markerSequence = (IIOMetadataNode) metadataAsTree.getElementsByTagName("markerSequence").item(0); | |
if (markerSequence == null) { | |
markerSequence = new IIOMetadataNode("markerSequence"); | |
metadataAsTree.appendChild(markerSequence); | |
} | |
return markerSequence; | |
} | |
void setJfifCommentFromAlbumEntryDescription(IIOMetadataNode markerSequence, AlbumEntry albumEntry) { | |
var comList = markerSequence.getElementsByTagName("com"); | |
if (comList.getLength() > 0) { | |
var com = (IIOMetadataNode) comList.item(0); | |
com.setAttribute("comment", albumEntry.getDescription()); | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment