Simplify album removal.
[Sone.git] / src / main / java / net / pterodactylus / sone / data / impl / DefaultAlbum.java
index 70fac0d..cbef5da 100644 (file)
@@ -17,8 +17,6 @@
 
 package net.pterodactylus.sone.data.impl;
 
-import static com.google.common.base.Optional.absent;
-import static com.google.common.base.Optional.fromNullable;
 import static com.google.common.base.Preconditions.checkArgument;
 import static com.google.common.base.Preconditions.checkNotNull;
 import static com.google.common.base.Preconditions.checkState;
@@ -32,52 +30,39 @@ import java.util.UUID;
 import net.pterodactylus.sone.data.Album;
 import net.pterodactylus.sone.data.Image;
 import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.database.AlbumBuilder;
 import net.pterodactylus.sone.database.ImageBuilder;
 
 import com.google.common.base.Function;
 import com.google.common.base.Optional;
 import com.google.common.base.Predicates;
 import com.google.common.collect.Collections2;
-import com.google.common.hash.Hasher;
-import com.google.common.hash.Hashing;
 
 /**
- * Container for images that can also contain nested {@link Album}s.
+ * Dumb, store-everything-in-memory implementation of an {@link Album}.
  *
  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
-public class DefaultAlbum implements Album {
-
-       /** The ID of this album. */
-       private final String id;
+public class DefaultAlbum extends AbstractAlbum {
 
        /** The Sone this album belongs to. */
        private Sone sone;
 
+       /** The parent album. */
+       private final DefaultAlbum parent;
+
        /** Nested albums. */
        private final List<Album> albums = new ArrayList<Album>();
 
        /** The image IDs in order. */
-       private final List<String> imageIds = new ArrayList<String>();
+       final List<String> imageIds = new ArrayList<String>();
 
        /** The images in this album. */
-       private final Map<String, Image> images = new HashMap<String, Image>();
-
-       /** The parent album. */
-       private Album parent;
-
-       /** The title of this album. */
-       private String title;
-
-       /** The description of this album. */
-       private String description;
-
-       /** The ID of the album picture. */
-       private String albumImage;
+       final Map<String, Image> images = new HashMap<String, Image>();
 
        /** Creates a new album with a random ID. */
-       public DefaultAlbum() {
-               this(UUID.randomUUID().toString());
+       public DefaultAlbum(Sone sone, DefaultAlbum parent) {
+               this(UUID.randomUUID().toString(), sone, parent);
        }
 
        /**
@@ -86,8 +71,10 @@ public class DefaultAlbum implements Album {
         * @param id
         *              The ID of the album
         */
-       public DefaultAlbum(String id) {
-               this.id = checkNotNull(id, "id must not be null");
+       public DefaultAlbum(String id, Sone sone, DefaultAlbum parent) {
+               super(id);
+               this.sone = sone;
+               this.parent = parent;
        }
 
        //
@@ -95,48 +82,16 @@ public class DefaultAlbum implements Album {
        //
 
        @Override
-       public String getId() {
-               return id;
-       }
-
-       @Override
        public Sone getSone() {
                return sone;
        }
 
        @Override
-       public Album setSone(Sone sone) {
-               checkNotNull(sone, "sone must not be null");
-               checkState((this.sone == null) || (this.sone.equals(sone)), "album owner must not already be set to some other Sone");
-               this.sone = sone;
-               return this;
-       }
-
-       @Override
        public List<Album> getAlbums() {
                return new ArrayList<Album>(albums);
        }
 
        @Override
-       public void addAlbum(Album album) {
-               checkNotNull(album, "album must not be null");
-               checkArgument(album.getSone().equals(sone), "album must belong to the same Sone as this album");
-               album.setParent(this);
-               if (!albums.contains(album)) {
-                       albums.add(album);
-               }
-       }
-
-       @Override
-       public void removeAlbum(Album album) {
-               checkNotNull(album, "album must not be null");
-               checkArgument(album.getSone().equals(sone), "album must belong this album’s Sone");
-               checkArgument(equals(album.getParent()), "album must belong to this album");
-               albums.remove(album);
-               album.removeParent();
-       }
-
-       @Override
        public Album moveAlbumUp(Album album) {
                checkNotNull(album, "album must not be null");
                checkArgument(album.getSone().equals(sone), "album must belong to the same Sone as this album");
@@ -177,22 +132,6 @@ public class DefaultAlbum implements Album {
        }
 
        @Override
-       public void removeImage(Image image) {
-               checkNotNull(image, "image must not be null");
-               checkNotNull(image.getSone(), "image must have an owner");
-               checkArgument(image.getSone().equals(sone), "image must belong to the same Sone as this album");
-               imageIds.remove(image.getId());
-               images.remove(image.getId());
-               if (image.getId().equals(albumImage)) {
-                       if (images.isEmpty()) {
-                               albumImage = null;
-                       } else {
-                               albumImage = images.values().iterator().next().getId();
-                       }
-               }
-       }
-
-       @Override
        public Image moveImageUp(Image image) {
                checkNotNull(image, "image must not be null");
                checkNotNull(image.getSone(), "image must have an owner");
@@ -231,45 +170,25 @@ public class DefaultAlbum implements Album {
        }
 
        @Override
-       public boolean isEmpty() {
-               return albums.isEmpty() && images.isEmpty();
-       }
-
-       @Override
-       public boolean isRoot() {
-               return parent == null;
-       }
-
-       @Override
        public Album getParent() {
                return parent;
        }
 
        @Override
-       public Album setParent(Album parent) {
-               this.parent = checkNotNull(parent, "parent must not be null");
-               return this;
-       }
-
-       @Override
-       public Album removeParent() {
-               this.parent = null;
-               return this;
-       }
-
-       @Override
-       public String getTitle() {
-               return title;
-       }
-
-       @Override
-       public String getDescription() {
-               return description;
+       public AlbumBuilder newAlbumBuilder() {
+               return new DefaultAlbumBuilder(sone, this) {
+                       @Override
+                       public Album build() throws IllegalStateException {
+                               Album album = super.build();
+                               albums.add(album);
+                               return album;
+                       }
+               };
        }
 
        @Override
        public ImageBuilder newImageBuilder() throws IllegalStateException {
-               return new DefaultImageBuilder(this) {
+               return new DefaultImageBuilder(sone, this) {
                        @Override
                        public Image build() throws IllegalStateException {
                                Image image = super.build();
@@ -284,100 +203,23 @@ public class DefaultAlbum implements Album {
        }
 
        @Override
-       public Modifier modify() throws IllegalStateException {
-               // TODO: reenable check for local Sones
-               return new Modifier() {
-                       private Optional<String> title = absent();
-
-                       private Optional<String> description = absent();
-
-                       private Optional<String> albumImage = absent();
-
-                       @Override
-                       public Modifier setTitle(String title) {
-                               this.title = fromNullable(title);
-                               return this;
-                       }
-
-                       @Override
-                       public Modifier setDescription(String description) {
-                               this.description = fromNullable(description);
-                               return this;
-                       }
-
-                       @Override
-                       public Modifier setAlbumImage(String imageId) {
-                               this.albumImage = fromNullable(imageId);
-                               return this;
-                       }
-
-                       @Override
-                       public Album update() throws IllegalStateException {
-                               if (title.isPresent()) {
-                                       DefaultAlbum.this.title = title.get();
-                               }
-                               if (description.isPresent()) {
-                                       DefaultAlbum.this.description = description.get();
-                               }
-                               if (albumImage.isPresent()) {
-                                       DefaultAlbum.this.albumImage = albumImage.get();
-                               }
-                               return DefaultAlbum.this;
-                       }
-               };
+       public void remove() throws IllegalStateException {
+               checkState(!isRoot(), "can not remove root album");
+               removeAllAlbums();
+               removeAllImages();
+               parent.albums.remove(this);
        }
 
-       //
-       // FINGERPRINTABLE METHODS
-       //
-
-       @Override
-       public String getFingerprint() {
-               Hasher hash = Hashing.sha256().newHasher();
-               hash.putString("Album(");
-               hash.putString("ID(").putString(id).putString(")");
-               hash.putString("Title(").putString(title).putString(")");
-               hash.putString("Description(").putString(description).putString(")");
-               if (albumImage != null) {
-                       hash.putString("AlbumImage(").putString(albumImage).putString(")");
-               }
-
-               /* add nested albums. */
-               hash.putString("Albums(");
-               for (Album album : albums) {
-                       hash.putString(album.getFingerprint());
+       private void removeAllImages() {
+               for (Image image : images.values()) {
+                       image.remove();
                }
-               hash.putString(")");
-
-               /* add images. */
-               hash.putString("Images(");
-               for (Image image : getImages()) {
-                       if (image.isInserted()) {
-                               hash.putString(image.getFingerprint());
-                       }
-               }
-               hash.putString(")");
-
-               hash.putString(")");
-               return hash.hash().toString();
        }
 
-       //
-       // OBJECT METHODS
-       //
-
-       @Override
-       public int hashCode() {
-               return id.hashCode();
-       }
-
-       @Override
-       public boolean equals(Object object) {
-               if (!(object instanceof DefaultAlbum)) {
-                       return false;
+       private void removeAllAlbums() {
+               for (Album album: albums) {
+                       album.remove();
                }
-               DefaultAlbum album = (DefaultAlbum) object;
-               return id.equals(album.id);
        }
 
 }