Move method to remove image from Album to Image.

[Sone.git] / src / main / java / net / pterodactylus / sone / data / impl / DefaultImage.java

diff --git a/src/main/java/net/pterodactylus/sone/data/impl/DefaultImage.java b/src/main/java/net/pterodactylus/sone/data/impl/DefaultImage.java
index 54e6495..156eb81 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/impl/DefaultImage.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/DefaultImage.java
@@ -17,6 +17,8 @@
 
 package net.pterodactylus.sone.data.impl;
 
+import static com.google.common.collect.FluentIterable.from;
+
 import net.pterodactylus.sone.data.Album;
 import net.pterodactylus.sone.data.Image;
 import net.pterodactylus.sone.data.Sone;
@@ -47,4 +49,15 @@ public class DefaultImage extends AbstractImage {
                return album;
        }
 
+       @Override
+       public void remove() throws IllegalStateException {
+               synchronized (album) {
+                       album.images.remove(getId());
+                       album.imageIds.remove(getId());
+                       if (getId().equals(album.albumImage)) {
+                               album.albumImage = from(album.images.values()).transform(GET_ID).first().orNull();
+                       }
+               }
+       }
+
 }