Add login page.
[demoscenemusic.git] / src / main / java / net / pterodactylus / demoscenemusic / data / DataManager.java
index b2a3337..60b8318 100644 (file)
@@ -20,9 +20,13 @@ package net.pterodactylus.demoscenemusic.data;
 import java.sql.ResultSet;
 import java.sql.SQLException;
 import java.util.Collection;
+import java.util.EnumMap;
+import java.util.HashSet;
 import java.util.List;
+import java.util.Map;
 import java.util.concurrent.Callable;
 
+import net.pterodactylus.demoscenemusic.data.Track.Relationship;
 import net.pterodactylus.util.collection.Memoizer;
 import net.pterodactylus.util.database.Database;
 import net.pterodactylus.util.database.DatabaseException;
@@ -30,41 +34,78 @@ import net.pterodactylus.util.database.Field;
 import net.pterodactylus.util.database.Join;
 import net.pterodactylus.util.database.Join.JoinType;
 import net.pterodactylus.util.database.ObjectCreator;
+import net.pterodactylus.util.database.ObjectCreator.StringCreator;
 import net.pterodactylus.util.database.OrderField;
 import net.pterodactylus.util.database.Parameter.StringParameter;
 import net.pterodactylus.util.database.Query;
 import net.pterodactylus.util.database.Query.Type;
+import net.pterodactylus.util.database.ResultProcessor;
 import net.pterodactylus.util.database.ValueField;
 import net.pterodactylus.util.database.ValueFieldWhereClause;
 
 /**
- * TODO
+ * Interface between the database and the application.
  *
  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
 public class DataManager {
 
+       /** The artist object creator. */
        @SuppressWarnings("synthetic-access")
        private final ObjectCreator<Artist> artistCreator = new ArtistCreator();
+
+       /** The group object creator. */
        @SuppressWarnings("synthetic-access")
        private final ObjectCreator<Group> groupCreator = new GroupCreator();
+
+       /** The track object creator. */
        @SuppressWarnings("synthetic-access")
        private final ObjectCreator<Track> trackCreator = new TrackCreator();
+
+       /** The style object creator. */
        @SuppressWarnings("synthetic-access")
        private final ObjectCreator<Style> styleCreator = new StyleCreator();
 
+       /** The {@link User} object creator. */
+       @SuppressWarnings("synthetic-access")
+       private final ObjectCreator<User> userCreator = new UserCreator();
+
+       /** The database. */
        private final Database database;
 
+       /**
+        * Creates a new data manager.
+        *
+        * @param database
+        *            The database to operate on
+        */
        public DataManager(Database database) {
                this.database = database;
        }
 
+       /**
+        * Returns all artists.
+        *
+        * @return All artists
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
        public Collection<Artist> getAllArtists() throws DatabaseException {
                Query query = new Query(Type.SELECT, "ARTISTS");
                query.addField(new Field("ARTISTS.*"));
                return database.getMultiple(query, artistCreator);
        }
 
+       /**
+        * Returns the artist with the given ID.
+        *
+        * @param id
+        *            The ID of the artist
+        * @return The artist with the given ID, or {@code null} if there is no
+        *         artist with the given ID
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
        public Artist getArtistById(String id) throws DatabaseException {
                Query query = new Query(Type.SELECT, "ARTISTS");
                query.addField(new Field("ARTISTS.*"));
@@ -72,6 +113,15 @@ public class DataManager {
                return database.getSingle(query, artistCreator);
        }
 
+       /**
+        * Returns all artists that belong to the group with the given ID.
+        *
+        * @param groupId
+        *            The ID of the group
+        * @return All artists belonging to the given group
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
        public Collection<Artist> getArtistsByGroup(String groupId) throws DatabaseException {
                Query query = new Query(Type.SELECT, "ARTISTS");
                query.addField(new Field("ARTISTS.*"));
@@ -80,6 +130,15 @@ public class DataManager {
                return database.getMultiple(query, artistCreator);
        }
 
+       /**
+        * Returns all artists involved in the track with the given ID.
+        *
+        * @param trackId
+        *            The ID of the track
+        * @return All artists involved in the track, in preferred order
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
        public List<Artist> getArtistsByTrack(String trackId) throws DatabaseException {
                Query query = new Query(Type.SELECT, "ARTISTS");
                query.addField(new Field("ARTISTS.*"));
@@ -89,6 +148,66 @@ public class DataManager {
                return database.getMultiple(query, artistCreator);
        }
 
+       /**
+        * Returns all remix artists involved in the track with the given ID.
+        *
+        * @param trackId
+        *            The ID of the track
+        * @return All remix artists involved in the track, in preferred order
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
+       public List<Artist> getRemixArtistsByTrack(String trackId) throws DatabaseException {
+               Query query = new Query(Type.SELECT, "ARTISTS");
+               query.addField(new Field("ARTISTS.*"));
+               query.addJoin(new Join(JoinType.INNER, "TRACK_REMIX_ARTISTS", new Field("TRACK_REMIX_ARTISTS.ARTIST"), new Field("ARTISTS.ID")));
+               query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACK_REMIX_ARTISTS.TRACK", new StringParameter(trackId))));
+               query.addOrderField(new OrderField(new Field("TRACK_REMIX_ARTISTS.DISPLAY_ORDER")));
+               return database.getMultiple(query, artistCreator);
+       }
+
+       /**
+        * Returns all related tracks for the track with the given ID.
+        *
+        * @param trackId
+        *            The ID of the tracks
+        * @return A mapping from relationship to all tracks that match the relation
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
+       public Map<Relationship, Collection<Track>> getRelatedTracksByTrack(String trackId) throws DatabaseException {
+               Query query = new Query(Type.SELECT, "TRACKS");
+               query.addField(new Field("TRACKS.*"));
+               query.addField(new Field("TRACK_RELATIONS.*"));
+               query.addJoin(new Join(JoinType.INNER, "TRACK_RELATIONS", new Field("TRACK_RELATIONS.TRACK"), new Field("TRACK_RELATIONS.RELATED_TRACK")));
+               query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACK_RELATIONS.TRACK", new StringParameter(trackId))));
+               final Map<Relationship, Collection<Track>> relatedTracks = new EnumMap<Relationship, Collection<Track>>(Relationship.class);
+               database.process(query, new ResultProcessor() {
+
+                       @Override
+                       @SuppressWarnings("synthetic-access")
+                       public void processResult(ResultSet resultSet) throws SQLException {
+                               Track track = trackCreator.createObject(resultSet);
+                               Relationship relationship = Relationship.valueOf(resultSet.getString("TRACK_RELATIONS.RELATIONSHIP"));
+                               if (!relatedTracks.containsKey(relationship)) {
+                                       relatedTracks.put(relationship, new HashSet<Track>());
+                               }
+                               relatedTracks.get(relationship).add(track);
+                       }
+               });
+               return relatedTracks;
+       }
+
+       /**
+        * Returns the track with the given ID.
+        *
+        * @param id
+        *            The ID of the track
+        * @return The track with the given ID, or {@code null} if there is no such
+        *         track
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
        public Track getTrackById(String id) throws DatabaseException {
                Query query = new Query(Type.SELECT, "TRACKS");
                query.addField(new Field("TRACKS.*"));
@@ -96,6 +215,15 @@ public class DataManager {
                return database.getSingle(query, trackCreator);
        }
 
+       /**
+        * Returns all tracks by the artist with the given ID.
+        *
+        * @param artistId
+        *            The ID of the artist
+        * @return All tracks by the given artist
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
        public Collection<Track> getTracksByArtist(String artistId) throws DatabaseException {
                Query query = new Query(Type.SELECT, "TRACKS");
                query.addField(new Field("TRACKS.*"));
@@ -104,6 +232,15 @@ public class DataManager {
                return database.getMultiple(query, trackCreator);
        }
 
+       /**
+        * Returns all groups the artist with the given ID belongs to.
+        *
+        * @param artistId
+        *            The ID of the artist
+        * @return All groups the artist belongs to
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
        public Collection<Group> getGroupsByArtist(String artistId) throws DatabaseException {
                Query query = new Query(Type.SELECT, "GROUPS");
                query.addField(new Field("GROUPS.*"));
@@ -112,6 +249,15 @@ public class DataManager {
                return database.getMultiple(query, groupCreator);
        }
 
+       /**
+        * Returns all styles for the track with the given ID.
+        *
+        * @param trackId
+        *            The ID of the track
+        * @return All styles for the given track
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
        public Collection<Style> getStylesByTrack(String trackId) throws DatabaseException {
                Query query = new Query(Type.SELECT, "STYLES");
                query.addField(new Field("STYLES.*"));
@@ -120,8 +266,66 @@ public class DataManager {
                return database.getMultiple(query, styleCreator);
        }
 
+       /**
+        * Returns the user with the given name.
+        *
+        * @param username
+        *            The name of the user
+        * @return The user, or {@code null} if the user does not exist
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
+       public User getUserByName(String username) throws DatabaseException {
+               Query query = new Query(Type.SELECT, "USERS");
+               query.addField(new Field("USERS.*"));
+               query.addWhereClause(new ValueFieldWhereClause(new ValueField("USERS.NAME", new StringParameter(username))));
+               return database.getSingle(query, userCreator);
+       }
+
+       /**
+        * Returns the user connected with the given OpenID.
+        *
+        * @param openId
+        *            The OpenID to find the user for
+        * @return The user connected with the given OpenID, or {@code null} if
+        *         there is no such user
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
+       public User getUserByOpenId(String openId) throws DatabaseException {
+               Query query = new Query(Type.SELECT, "USERS");
+               query.addField(new Field("USERS.*"));
+               query.addJoin(new Join(JoinType.INNER, "USER_OPENIDS", new Field("USER_OPENIDS.USER"), new Field("USERS.ID")));
+               query.addWhereClause(new ValueFieldWhereClause(new ValueField("USER_OPENIDS.OPENID", new StringParameter(openId))));
+               return database.getSingle(query, userCreator);
+       }
+
+       /**
+        * Returns all OpenIDs connected with the user with the given ID.
+        *
+        * @param userId
+        *            The ID of the user
+        * @return All OpenIDs connected with the given user
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
+       public Collection<String> getOpenIdsByUser(String userId) throws DatabaseException {
+               Query query = new Query(Type.SELECT, "USER_OPENIDS");
+               query.addField(new Field("USER_OPENIDS.*"));
+               query.addWhereClause(new ValueFieldWhereClause(new ValueField("USER_OPENIDS.USER", new StringParameter(userId))));
+               return database.getMultiple(query, new StringCreator("USER_OPENIDS.OPENID"));
+       }
+
+       /**
+        * {@link Artist} implementation that retrieves some attributes (such as
+        * {@link #getGroups()}, and {@link #getTracks()}) from the
+        * {@link DataManager} on demand.
+        *
+        * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+        */
        private class LazyArtist extends DefaultArtist {
 
+               /** Memoizer for the tracks by this artist. */
                private final Memoizer<Void> tracksMemoizer = new Memoizer<Void>(new Callable<Void>() {
                        @Override
                        public Void call() throws DatabaseException {
@@ -132,6 +336,7 @@ public class DataManager {
                        }
                });
 
+               /** Memoizer for the groups of this artist. */
                private final Memoizer<Void> groupsMemoizer = new Memoizer<Void>(new Callable<Void>() {
 
                        @Override
@@ -144,10 +349,20 @@ public class DataManager {
 
                });
 
+               /**
+                * Creates a new lazy artist.
+                *
+                * @param id
+                *            The ID of the artist
+                */
                public LazyArtist(String id) {
                        super(id);
                }
 
+               //
+               // DEFAULTARTIST METHODS
+               //
+
                /**
                 * {@inheritDoc}
                 */
@@ -157,6 +372,9 @@ public class DataManager {
                        return super.getGroups();
                }
 
+               /**
+                * {@inheritDoc}
+                */
                @Override
                public Collection<Track> getTracks() {
                        tracksMemoizer.get();
@@ -165,6 +383,13 @@ public class DataManager {
 
        }
 
+       /**
+        * {@link ObjectCreator} implementation that can create {@link Artist}
+        * objects. This specific class actually creates {@link LazyArtist}
+        * instances.
+        *
+        * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+        */
        private class ArtistCreator implements ObjectCreator<Artist> {
 
                /**
@@ -177,8 +402,15 @@ public class DataManager {
 
        }
 
+       /**
+        * {@link Group} implementation that retrieves some attributes (such as
+        * {@link #getArtists()}) from the {@link DataManager} on demand.
+        *
+        * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+        */
        private class LazyGroup extends DefaultGroup {
 
+               /** Memoizer for the artist. */
                private final Memoizer<Void> artistsMemoizer = new Memoizer<Void>(new Callable<Void>() {
 
                        @Override
@@ -191,10 +423,20 @@ public class DataManager {
 
                });
 
+               /**
+                * Creates a new lazy group.
+                *
+                * @param id
+                *            The ID of the group
+                */
                public LazyGroup(String id) {
                        super(id);
                }
 
+               //
+               // DEFAULTGROUP METHODS
+               //
+
                /**
                 * {@inheritDoc}
                 */
@@ -206,6 +448,13 @@ public class DataManager {
 
        }
 
+       /**
+        * {@link ObjectCreator} implementation that can create {@link Group}
+        * objects. This specific implementation creates {@link LazyGroup}
+        * instances.
+        *
+        * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+        */
        private class GroupCreator implements ObjectCreator<Group> {
 
                /**
@@ -218,8 +467,16 @@ public class DataManager {
 
        }
 
+       /**
+        * {@link Track} implementation that retrieves some attributes (such as
+        * {@link #getArtists()}, and {@link #getStyles()}) from the
+        * {@link DataManager} on demand.
+        *
+        * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+        */
        private class LazyTrack extends DefaultTrack {
 
+               /** Memoizer for the artists. */
                private final Memoizer<Void> artistsMemoizer = new Memoizer<Void>(new Callable<Void>() {
 
                        @Override
@@ -232,6 +489,7 @@ public class DataManager {
 
                });
 
+               /** Memoizer for the styles. */
                private final Memoizer<Void> stylesMemoizer = new Memoizer<Void>(new Callable<Void>() {
 
                        @Override
@@ -244,13 +502,45 @@ public class DataManager {
 
                });
 
+               /** Memoizer for the remix artists. */
+               private final Memoizer<Void> remixArtistsMemoizer = new Memoizer<Void>(new Callable<Void>() {
+
+                       @Override
+                       public Void call() throws Exception {
+                               if (!hasValue("remixArtists")) {
+                                       getValue("remixArtists", List.class).set(getRemixArtistsByTrack(getId()));
+                               }
+                               return null;
+                       }
+
+               });
+
+               /** Memoizer for the related tracks. */
+               private final Memoizer<Void> relatedTracksMemoizer = new Memoizer<Void>(new Callable<Void>() {
+
+                       @Override
+                       public Void call() throws Exception {
+                               if (!hasValue("relatedTracks")) {
+                                       getValue("relatedTracks", Map.class).set(getRelatedTracksByTrack(getId()));
+                               }
+                               return null;
+                       }
+               });
+
                /**
+                * Creates a new track.
+                *
                 * @param id
+                *            The ID of the track
                 */
                public LazyTrack(String id) {
                        super(id);
                }
 
+               //
+               // DEFAULTTRACK METHODS
+               //
+
                /**
                 * {@inheritDoc}
                 */
@@ -269,8 +559,33 @@ public class DataManager {
                        return super.getStyles();
                }
 
+               /**
+                * {@inheritDoc}
+                */
+               @Override
+               public List<Artist> getRemixArtists() {
+                       remixArtistsMemoizer.get();
+                       return super.getRemixArtists();
+               }
+
+               /**
+                * {@inheritDoc}
+                */
+               @Override
+               public Map<Relationship, Collection<Track>> getRelatedTracks() {
+                       relatedTracksMemoizer.get();
+                       return super.getRelatedTracks();
+               }
+
        }
 
+       /**
+        * {@link ObjectCreator} implementation that can create {@link Track}
+        * objects. This specific implementation creates {@link LazyTrack}
+        * instances.
+        *
+        * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+        */
        private class TrackCreator implements ObjectCreator<Track> {
 
                /**
@@ -278,11 +593,17 @@ public class DataManager {
                 */
                @Override
                public Track createObject(ResultSet resultSet) throws SQLException {
-                       return new LazyTrack(resultSet.getString("TRACKS.ID")).setName(resultSet.getString("TRACKS.NAME"));
+                       return new LazyTrack(resultSet.getString("TRACKS.ID")).setName(resultSet.getString("TRACKS.NAME")).setRemix(resultSet.getString("TRACKS.REMIX"));
                }
 
        }
 
+       /**
+        * {@link ObjectCreator} implementation that can create {@link Style}
+        * objects.
+        *
+        * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+        */
        private class StyleCreator implements ObjectCreator<Style> {
 
                /**
@@ -295,4 +616,63 @@ public class DataManager {
 
        }
 
+       /**
+        * {@link User} implementation that retrieves some attributes (such as
+        * {@link #getOpenIds()}) from the {@link DataManager} on demand.
+        *
+        * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+        */
+       private class LazyUser extends DefaultUser {
+
+               /** Memoizer for a user’s OpenIDs. */
+               private final Memoizer<Void> openIdMemoizer = new Memoizer<Void>(new Callable<Void>() {
+
+                       @Override
+                       public Void call() throws Exception {
+                               if (!hasValue("openIds")) {
+                                       getValue("openIds", Collection.class).set(getOpenIdsByUser(getId()));
+                               }
+                               return null;
+                       }
+               });
+
+               /**
+                * Creates a new user.
+                *
+                * @param id
+                *            The ID of the user
+                */
+               public LazyUser(String id) {
+                       super(id);
+               }
+
+               /**
+                * {@inheritDoc}
+                */
+               @Override
+               public Collection<String> getOpenIds() {
+                       openIdMemoizer.get();
+                       return super.getOpenIds();
+               }
+
+       }
+
+       /**
+        * {@link ObjectCreator} implementation that can create {@link User}
+        * objects. This specific implementation creates {@link LazyUser} instances.
+        *
+        * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+        */
+       private class UserCreator implements ObjectCreator<User> {
+
+               /**
+                * {@inheritDoc}
+                */
+               @Override
+               public User createObject(ResultSet resultSet) throws SQLException {
+                       return new LazyUser(resultSet.getString("USERS.ID")).setName(resultSet.getString("USERS.NAME")).setPasswordHash(resultSet.getString("USERS.PASSWORD")).setLevel(resultSet.getInt("USERS.LEVEL"));
+               }
+
+       }
+
 }