X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FCore.java;h=06a1467b1364c8a353932927388c1f3cfa22e2c2;hp=585d71fb59257fc4487bc005fd694196655411a9;hb=7dfc9c778882083632f916b178a241f06c2a08fe;hpb=6414c85c8793a01593f0a86c7e569c5c1b8de324

diff --git a/src/main/java/net/pterodactylus/sone/core/Core.java b/src/main/java/net/pterodactylus/sone/core/Core.java
index 585d71f..06a1467 100644
--- a/src/main/java/net/pterodactylus/sone/core/Core.java
+++ b/src/main/java/net/pterodactylus/sone/core/Core.java
@@ -1319,6 +1319,38 @@ public class Core implements IdentityListener, UpdateListener {
 			friends.add(friendId);
 		}
 
+		/* load albums. */
+		List<Album> topLevelAlbums = new ArrayList<Album>();
+		while (true) {
+			String albumPrefix = sonePrefix + "/Albums/" + albums.size();
+			String albumId = configuration.getStringValue(albumPrefix + "/ID").getValue(null);
+			if (albumId == null) {
+				break;
+			}
+			String albumTitle = configuration.getStringValue(albumPrefix + "/Title").getValue(null);
+			String albumDescription = configuration.getStringValue(albumPrefix + "/Description").getValue(null);
+			String albumParentId = configuration.getStringValue(albumPrefix + "/Parent").getValue(null);
+			if ((albumTitle == null) || (albumDescription == null)) {
+				logger.log(Level.WARNING, "Invalid album found, aborting load!");
+				return;
+			}
+			Album album = getAlbum(albumId).setSone(sone).setTitle(albumTitle).setDescription(albumDescription);
+			if (albumParentId != null) {
+				Album parentAlbum = getAlbum(albumParentId, false);
+				if (parentAlbum == null) {
+					logger.log(Level.WARNING, "Invalid parent album ID: " + albumParentId);
+					return;
+				}
+				parentAlbum.addAlbum(album);
+			} else {
+				topLevelAlbums.add(album);
+			}
+		}
+
+		/* load options. */
+		sone.getOptions().addBooleanOption("AutoFollow", new DefaultOption<Boolean>(false));
+		sone.getOptions().getBooleanOption("AutoFollow").set(configuration.getBooleanValue(sonePrefix + "/Options/AutoFollow").getValue(null));
+
 		/* if we’re still here, Sone was loaded successfully. */
 		synchronized (sone) {
 			sone.setTime(soneTime);
@@ -1328,6 +1360,7 @@ public class Core implements IdentityListener, UpdateListener {
 			sone.setLikePostIds(likedPostIds);
 			sone.setLikeReplyIds(likedReplyIds);
 			sone.setFriends(friends);
+			sone.setAlbums(topLevelAlbums);
 			soneInserters.get(sone).setLastInsertFingerprint(lastInsertFingerprint);
 		}
 		synchronized (newSones) {
@@ -1434,6 +1467,31 @@ public class Core implements IdentityListener, UpdateListener {
 			}
 			configuration.getStringValue(sonePrefix + "/Friends/" + friendCounter + "/ID").setValue(null);
 
+			/* save albums. first, collect in a flat structure, top-level first. */
+			List<Album> albums = new ArrayList<Album>();
+			albums.addAll(sone.getAlbums());
+			int lastAlbumIndex = 0;
+			while (lastAlbumIndex < albums.size()) {
+				int previousAlbumCount = albums.size();
+				for (Album album : new ArrayList<Album>(albums.subList(lastAlbumIndex, albums.size()))) {
+					albums.addAll(album.getAlbums());
+				}
+				lastAlbumIndex = previousAlbumCount;
+			}
+
+			int albumCounter = 0;
+			for (Album album : albums) {
+				String albumPrefix = sonePrefix + "/Albums/" + albumCounter++;
+				configuration.getStringValue(albumPrefix + "/ID").setValue(album.getId());
+				configuration.getStringValue(albumPrefix + "/Title").setValue(album.getTitle());
+				configuration.getStringValue(albumPrefix + "/Description").setValue(album.getDescription());
+				configuration.getStringValue(albumPrefix + "/Parent").setValue(album.getParent() == null ? null : album.getParent().getId());
+			}
+			configuration.getStringValue(sonePrefix + "/Albums/" + albumCounter + "/ID").setValue(null);
+
+			/* save options. */
+			configuration.getBooleanValue(sonePrefix + "/Options/AutoFollow").setValue(sone.getOptions().getBooleanOption("AutoFollow").getReal());
+
 			configuration.save();
 			logger.log(Level.INFO, "Sone %s saved.", sone);
 		} catch (ConfigurationException ce1) {
@@ -1710,8 +1768,9 @@ public class Core implements IdentityListener, UpdateListener {
 		album.setSone(sone);
 		if (parent != null) {
 			parent.addAlbum(album);
+		} else {
+			sone.addAlbum(album);
 		}
-		sone.addAlbum(album);
 		return album;
 	}