Return optionals of fields.
[Sone.git] / src / main / java / net / pterodactylus / sone / template / ImageAccessor.java
1 /*
2  * Sone - ImageAccessor.java - Copyright © 2011–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.template;
19
20 import net.pterodactylus.sone.data.Album;
21 import net.pterodactylus.sone.data.Image;
22 import net.pterodactylus.util.template.Accessor;
23 import net.pterodactylus.util.template.ReflectionAccessor;
24 import net.pterodactylus.util.template.TemplateContext;
25
26 /**
27  * {@link Accessor} implementation for {@link Image} objects. It adds the
28  * following properties:
29  * <ul>
30  * <li>{@code previous}: returns the previous image in the image’s album, or
31  * {@code null} if the image is the first image of its album.</li>
32  * <li>{@code next}: returns the next image in the image’s album, or {@code
33  * null} if the image is the last image of its album.</li>
34  * </ul>
35  *
36  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
37  */
38 public class ImageAccessor extends ReflectionAccessor {
39
40         @Override
41         public Object get(TemplateContext templateContext, Object object, String member) {
42                 Image image = (Image) object;
43                 if ("next".equals(member)) {
44                         Album album = image.getAlbum();
45                         int imagePosition = album.getImages().indexOf(image);
46                         if (imagePosition < album.getImages().size() - 1) {
47                                 return album.getImages().get(imagePosition + 1);
48                         }
49                         return null;
50                 } else if ("previous".equals(member)) {
51                         Album album = image.getAlbum();
52                         int imagePosition = album.getImages().indexOf(image);
53                         if (imagePosition > 0) {
54                                 return album.getImages().get(imagePosition - 1);
55                         }
56                         return null;
57                 }
58                 return super.get(templateContext, object, member);
59         }
60
61 }