From 41ad88e59a3249031c6b546c8189204d99596e26 Mon Sep 17 00:00:00 2001
From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?=
 <bombe@pterodactylus.net>
Date: Sun, 10 Apr 2011 20:13:42 +0200
Subject: [PATCH] Load and save images when loading and saving Sones.

---
 .../java/net/pterodactylus/sone/core/Core.java     | 49 ++++++++++++++++++++++
 1 file changed, 49 insertions(+)

diff --git a/src/main/java/net/pterodactylus/sone/core/Core.java b/src/main/java/net/pterodactylus/sone/core/Core.java
index 767309d..8a6236a 100644
--- a/src/main/java/net/pterodactylus/sone/core/Core.java
+++ b/src/main/java/net/pterodactylus/sone/core/Core.java
@@ -1365,6 +1365,35 @@ public class Core implements IdentityListener, UpdateListener {
 			}
 		}
 
+		/* load images. */
+		int imageCounter = 0;
+		while (true) {
+			String imagePrefix = sonePrefix + "/Images/" + imageCounter++;
+			String imageId = configuration.getStringValue(imagePrefix + "/ID").getValue(null);
+			if (imageId == null) {
+				break;
+			}
+			String albumId = configuration.getStringValue(imagePrefix + "/Album").getValue(null);
+			String key = configuration.getStringValue(imagePrefix + "/Key").getValue(null);
+			String title = configuration.getStringValue(imagePrefix + "/Title").getValue(null);
+			String description = configuration.getStringValue(imagePrefix + "/Description").getValue(null);
+			Long creationTime = configuration.getLongValue(imagePrefix + "/CreationTime").getValue(null);
+			Integer width = configuration.getIntValue(imagePrefix + "/Width").getValue(null);
+			Integer height = configuration.getIntValue(imagePrefix + "/Height").getValue(null);
+			if ((albumId == null) || (key == null) || (title == null) || (description == null) || (creationTime == null) || (width == null) || (height == null)) {
+				logger.log(Level.WARNING, "Invalid image found, aborting load!");
+				return;
+			}
+			Album album = getAlbum(albumId, false);
+			if (album == null) {
+				logger.log(Level.WARNING, "Invalid album image encountered, aborting load!");
+				return;
+			}
+			Image image = getImage(imageId).setSone(sone).setCreationTime(creationTime).setKey(key);
+			image.setTitle(title).setDescription(description).setWidth(width).setHeight(height);
+			album.addImage(image);
+		}
+
 		/* load options. */
 		sone.getOptions().addBooleanOption("AutoFollow", new DefaultOption<Boolean>(false));
 		sone.getOptions().getBooleanOption("AutoFollow").set(configuration.getBooleanValue(sonePrefix + "/Options/AutoFollow").getValue(null));
@@ -1507,6 +1536,26 @@ public class Core implements IdentityListener, UpdateListener {
 			}
 			configuration.getStringValue(sonePrefix + "/Albums/" + albumCounter + "/ID").setValue(null);
 
+			/* save images. */
+			int imageCounter = 0;
+			for (Album album : albums) {
+				for (Image image : album.getImages()) {
+					if (!image.isInserted()) {
+						continue;
+					}
+					String imagePrefix = sonePrefix + "/Images/" + imageCounter++;
+					configuration.getStringValue(imagePrefix + "/ID").setValue(image.getId());
+					configuration.getStringValue(imagePrefix + "/Album").setValue(album.getId());
+					configuration.getStringValue(imagePrefix + "/Key").setValue(image.getKey());
+					configuration.getStringValue(imagePrefix + "/Title").setValue(image.getTitle());
+					configuration.getStringValue(imagePrefix + "/Description").setValue(image.getDescription());
+					configuration.getLongValue(imagePrefix + "/CreationTime").setValue(image.getCreationTime());
+					configuration.getIntValue(imagePrefix + "/Width").setValue(image.getWidth());
+					configuration.getIntValue(imagePrefix + "/Height").setValue(image.getHeight());
+				}
+			}
+			configuration.getStringValue(sonePrefix + "/Images/" + imageCounter + "/ID").setValue(null);
+
 			/* save options. */
 			configuration.getBooleanValue(sonePrefix + "/Options/AutoFollow").setValue(sone.getOptions().getBooleanOption("AutoFollow").getReal());
 
-- 
2.7.4