Store DefaultAlbum in DefaultImage.
[Sone.git] / src / main / java / net / pterodactylus / sone / database / memory / MemoryImage.java
1 /*
2  * Sone - MemoryImage.java - Copyright © 2013 David Roden
3  *
4  * This program is free software: you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 3 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
16  */
17
18 package net.pterodactylus.sone.database.memory;
19
20 import net.pterodactylus.sone.data.Album;
21 import net.pterodactylus.sone.data.Image;
22 import net.pterodactylus.sone.data.Sone;
23 import net.pterodactylus.sone.data.impl.AbstractImage;
24
25 /**
26  * In-memory implementation of an {@link Image}.
27  *
28  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
29  */
30 public class MemoryImage extends AbstractImage {
31
32         private final MemoryDatabase memoryDatabase;
33         private final Sone sone; /* TODO - store sone ID only. */
34         private final String albumId;
35
36         public MemoryImage(MemoryDatabase memoryDatabase, String id, Sone sone, String albumId, String key, long creationTime, int width, int height) {
37                 super(id, key, creationTime, width, height);
38                 this.memoryDatabase = memoryDatabase;
39                 this.sone = sone;
40                 this.albumId = albumId;
41         }
42
43         @Override
44         public Sone getSone() {
45                 return sone;
46         }
47
48         @Override
49         public Album getAlbum() {
50                 return memoryDatabase.getAlbum(albumId).get();
51         }
52
53 }