Add method to get styles for a track.

[demoscenemusic.git] / src / main / java / net / pterodactylus / demoscenemusic / data / DataManager.java

diff --git a/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java b/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
index c0d458f..f53559e 100644 (file)
--- a/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
+++ b/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
@@ -85,7 +85,7 @@ public class DataManager {
                return database.getMultiple(query, artistCreator);
        }
 
-       public Track trackById(String id) throws DatabaseException {
+       public Track getTrackById(String id) throws DatabaseException {
                Query query = new Query(Type.SELECT, "TRACKS");
                query.addField(new Field("TRACKS.*"));
                query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACKS.ID", new StringParameter(id))));
@@ -108,6 +108,14 @@ public class DataManager {
                return database.getMultiple(query, groupCreator);
        }
 
+       public Collection<Style> getStylesByTrack(String trackId) throws DatabaseException {
+               Query query = new Query(Type.SELECT, "STYLES");
+               query.addField(new Field("STYLES.*"));
+               query.addJoin(new Join(JoinType.INNER, "TRACK_STYLES", new Field("STYLES.ID"), new Field("TRACK_STYLES.STYLE")));
+               query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACK_STYLES.TRACK", new StringParameter(trackId))));
+               return database.getMultiple(query, styleCreator);
+       }
+
        private class LazyArtist extends DefaultArtist {
 
                private final Memoizer<Void> tracksMemoizer = new Memoizer<Void>(new Callable<Void>() {
@@ -218,6 +226,18 @@ public class DataManager {
 
                });
 
+               private final Memoizer<Void> stylesMemoizer = new Memoizer<Void>(new Callable<Void>() {
+
+                       @Override
+                       public Void call() throws Exception {
+                               if (!hasValue("styles")) {
+                                       value("styles", Collection.class).set(getStylesByTrack(id()));
+                               }
+                               return null;
+                       }
+
+               });
+
                /**
                 * @param id
                 */
@@ -239,7 +259,7 @@ public class DataManager {
                 */
                @Override
                public Collection<Style> styles() {
-                       // TODO Auto-generated method stub
+                       stylesMemoizer.get();
                        return super.styles();
                }