Load the properties of a track together with the 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 03ec2d9..c87df73 100644 (file)
--- a/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
+++ b/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
@@ -324,7 +324,7 @@ public class DataManager {
                Query query = new Query(Type.SELECT, "TRACKS");
                query.addField(new Field("TRACKS.*"));
                query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACKS.ID", new StringParameter(id))));
-               return database.getSingle(query, trackCreator);
+               return loadTrackProperties(database.getSingle(query, trackCreator));
        }
 
        /**
@@ -341,7 +341,54 @@ public class DataManager {
                query.addField(new Field("TRACKS.*"));
                query.addJoin(new Join(JoinType.INNER, "TRACK_ARTISTS", new Field("TRACKS.ID"), new Field("TRACK_ARTISTS.TRACK")));
                query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACK_ARTISTS.ARTIST", new StringParameter(artistId))));
-               return database.getMultiple(query, trackCreator);
+               return loadTrackProperties(database.getMultiple(query, trackCreator));
+       }
+
+       /**
+        * Loads the properties for the given track.
+        *
+        * @param track
+        *            The track for which to load the properties
+        * @return The track with the properties loaded
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
+       public Track loadTrackProperties(Track track) throws DatabaseException {
+               return loadProperties(track, "TRACK_PROPERTIES", "TRACK");
+       }
+
+       /**
+        * Loads the properties for the given tracks.
+        *
+        * @param tracks
+        *            The tracks for which to load the properties
+        * @return The tracks with the properties loaded
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
+       public List<Track> loadTrackProperties(List<Track> tracks) throws DatabaseException {
+               for (Track track : tracks) {
+                       loadTrackProperties(track);
+               }
+               return tracks;
+       }
+
+       /**
+        * Creates a new track with the given name.
+        *
+        * @param name
+        *            The name of the track
+        * @return The created track
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
+       public Track createTrack(String name) throws DatabaseException {
+               Query query = new Query(Type.INSERT, "TRACKS");
+               String id = UUID.randomUUID().toString();
+               query.addValueField(new ValueField("ID", new StringParameter(id)));
+               query.addValueField(new ValueField("NAME", new StringParameter(name)));
+               database.insert(query);
+               return getTrackById(id);
        }
 
        /**
@@ -377,6 +424,24 @@ public class DataManager {
        }
 
        /**
+        * Creates a new derivative for the given track.
+        *
+        * @param track
+        *            The track to create the derivative for
+        * @return The created derivative
+        * @throws DatabaseException
+        *             if a database error occurs
+        */
+       public TrackDerivative createTrackDerivative(Track track) throws DatabaseException {
+               Query query = new Query(Type.INSERT, "TRACK_DERIVATIVES");
+               String id = UUID.randomUUID().toString();
+               query.addValueField(new ValueField("TRACK_DERIVATIVES.ID", new StringParameter(id)));
+               query.addValueField(new ValueField("TRACK_DERIVATIVES.TRACK", new StringParameter(track.getId())));
+               database.insert(query);
+               return getTrackDerivativeById(id);
+       }
+
+       /**
         * Loads the properties for the given track derivative.
         *
         * @param trackDerivative