X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fdemoscenemusic%2Fdata%2FDataManager.java;h=5f20905ee12d4c6695edc47def6158fbdd1cd3e6;hb=6f3371253d6a4398abea9fcd9c78a042b80c5af2;hp=60b831824e3833cea148bc5cf01eb51e9266f018;hpb=cb96ebd4d6cb9b43fcf9a6bb3e2f335bff243708;p=demoscenemusic.git

diff --git a/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java b/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
index 60b8318..5f20905 100644
--- a/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
+++ b/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
@@ -24,6 +24,8 @@ import java.util.EnumMap;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.Map.Entry;
+import java.util.UUID;
 import java.util.concurrent.Callable;
 
 import net.pterodactylus.demoscenemusic.data.Track.Relationship;
@@ -62,6 +64,10 @@ public class DataManager {
 	@SuppressWarnings("synthetic-access")
 	private final ObjectCreator<Track> trackCreator = new TrackCreator();
 
+	/** The track object creator. */
+	@SuppressWarnings("synthetic-access")
+	private final ObjectCreator<TrackDerivative> trackDerivativeCreator = new TrackDerivativeCreator();
+
 	/** The style object creator. */
 	@SuppressWarnings("synthetic-access")
 	private final ObjectCreator<Style> styleCreator = new StyleCreator();
@@ -93,7 +99,7 @@ public class DataManager {
 	public Collection<Artist> getAllArtists() throws DatabaseException {
 		Query query = new Query(Type.SELECT, "ARTISTS");
 		query.addField(new Field("ARTISTS.*"));
-		return database.getMultiple(query, artistCreator);
+		return loadArtistProperties(database.getMultiple(query, artistCreator));
 	}
 
 	/**
@@ -110,7 +116,7 @@ public class DataManager {
 		Query query = new Query(Type.SELECT, "ARTISTS");
 		query.addField(new Field("ARTISTS.*"));
 		query.addWhereClause(new ValueFieldWhereClause(new ValueField("ARTISTS.ID", new StringParameter(id))));
-		return database.getSingle(query, artistCreator);
+		return loadArtistProperties(database.getSingle(query, artistCreator));
 	}
 
 	/**
@@ -127,7 +133,7 @@ public class DataManager {
 		query.addField(new Field("ARTISTS.*"));
 		query.addJoin(new Join(JoinType.INNER, "GROUP_ARTISTS", new Field("ARTISTS.ID"), new Field("GROUP_ARTISTS.ARTIST")));
 		query.addWhereClause(new ValueFieldWhereClause(new ValueField("GROUP_ARTISTS.GROUP_", new StringParameter(groupId))));
-		return database.getMultiple(query, artistCreator);
+		return loadArtistProperties(database.getMultiple(query, artistCreator));
 	}
 
 	/**
@@ -145,7 +151,113 @@ public class DataManager {
 		query.addJoin(new Join(JoinType.INNER, "TRACK_ARTISTS", new Field("TRACK_ARTISTS.ARTIST"), new Field("ARTISTS.ID")));
 		query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACK_ARTISTS.TRACK", new StringParameter(trackId))));
 		query.addOrderField(new OrderField(new Field("TRACK_ARTISTS.DISPLAY_ORDER")));
-		return database.getMultiple(query, artistCreator);
+		return loadArtistProperties(database.getMultiple(query, artistCreator));
+	}
+
+	/**
+	 * Creates a new artist with the given name.
+	 *
+	 * @param name
+	 *            The name of the artist
+	 * @return The created artist
+	 * @throws DatabaseException
+	 *             if a database error occurs
+	 */
+	public Artist createArtist(String name) throws DatabaseException {
+		Query query = new Query(Type.INSERT, "ARTISTS");
+		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 loadArtistProperties(getArtistById(id));
+	}
+
+	/**
+	 * Saves the given artist.
+	 *
+	 * @param artist
+	 *            The artist to save
+	 * @throws DatabaseException
+	 *             if a database error occurs
+	 */
+	public void saveArtist(Artist artist) throws DatabaseException {
+		Query query = new Query(Type.UPDATE, "ARTISTS");
+		query.addValueField(new ValueField("NAME", new StringParameter(artist.getName())));
+		query.addWhereClause(new ValueFieldWhereClause(new ValueField("ID", new StringParameter(artist.getId()))));
+		database.update(query);
+		saveArtistProperties(artist);
+	}
+
+	/**
+	 * Saves the properties of the given artist.
+	 *
+	 * @param artist
+	 *            The artist whose properties to save
+	 * @throws DatabaseException
+	 *             if a database error occurs
+	 */
+	public void saveArtistProperties(Artist artist) throws DatabaseException {
+		saveProperties(artist.getProperties(), "ARTIST_PROPERTIES", "ARTIST", artist.getId());
+	}
+
+	/**
+	 * Saves the given properties to the given table for the given principal.
+	 *
+	 * @param properties
+	 *            The properties to save
+	 * @param table
+	 *            The table in which to save the properties
+	 * @param type
+	 *            The type of the principal (e. g. “ARTIST” or “TRACK”)
+	 * @param id
+	 *            The ID of the principial
+	 * @throws DatabaseException
+	 *             if a database error occurs
+	 */
+	public void saveProperties(Properties properties, String table, String type, String id) throws DatabaseException {
+		if (!properties.isDirty()) {
+			return;
+		}
+		Query query = new Query(Type.DELETE, table);
+		query.addWhereClause(new ValueFieldWhereClause(new ValueField(type, new StringParameter(id))));
+		database.update(query);
+		for (Entry<String, String> property : properties) {
+			Query insertQuery = new Query(Type.INSERT, table);
+			insertQuery.addValueField(new ValueField(type, new StringParameter(id)));
+			insertQuery.addValueField(new ValueField("PROPERTY", new StringParameter(property.getKey())));
+			insertQuery.addValueField(new ValueField("VALUE", new StringParameter(property.getValue())));
+			database.insert(insertQuery);
+		}
+		properties.resetDirty();
+	}
+
+	/**
+	 * Loads the properties for an artist.
+	 *
+	 * @param artist
+	 *            The artist to load the properties for
+	 * @return The artist
+	 * @throws DatabaseException
+	 *             if a database error occurs
+	 */
+	public Artist loadArtistProperties(final Artist artist) throws DatabaseException {
+		return loadProperties(artist, "ARTIST_PROPERTIES", "ARTIST");
+	}
+
+	/**
+	 * Loads the properties of all given artists.
+	 *
+	 * @param artists
+	 *            The artists to load the properties for
+	 * @return The list of artists
+	 * @throws DatabaseException
+	 *             if a database error occurs
+	 */
+	public List<Artist> loadArtistProperties(List<Artist> artists) throws DatabaseException {
+		for (Artist artist : artists) {
+			loadArtistProperties(artist);
+		}
+		return artists;
 	}
 
 	/**
@@ -212,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));
 	}
 
 	/**
@@ -229,7 +341,133 @@ 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);
+	}
+
+	/**
+	 * Returns the derivative with the given ID.
+	 *
+	 * @param id
+	 *            The ID of the derivatives to load
+	 * @return The derivative with the given ID
+	 * @throws DatabaseException
+	 *             if a database error occurs
+	 */
+	public TrackDerivative getTrackDerivativeById(String id) throws DatabaseException {
+		Query query = new Query(Type.SELECT, "TRACK_DERIVATIVES");
+		query.addField(new Field("TRACK_DERIVATIVES.*"));
+		query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACK_DERIVATIVES.ID", new StringParameter(id))));
+		return loadTrackDerivativeProperties(database.getSingle(query, trackDerivativeCreator));
+	}
+
+	/**
+	 * Returns the derivatives for the given track.
+	 *
+	 * @param trackId
+	 *            The track ID to get the derivatives for
+	 * @return The derivatives for the given track
+	 * @throws DatabaseException
+	 *             if a database error occurs
+	 */
+	public Collection<TrackDerivative> getTrackDerivativesByTrack(String trackId) throws DatabaseException {
+		Query query = new Query(Type.SELECT, "TRACK_DERIVATIVES");
+		query.addField(new Field("TRACK_DERIVATIVES.*"));
+		query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACK_DERIVATIVES.TRACK", new StringParameter(trackId))));
+		return loadTrackDerivativeProperties(database.getMultiple(query, trackDerivativeCreator));
+	}
+
+	/**
+	 * 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
+	 *            The track derivative to load the properties for
+	 * @return The track derivative with its properties loaded
+	 * @throws DatabaseException
+	 *             if a database error occurs
+	 */
+	public TrackDerivative loadTrackDerivativeProperties(TrackDerivative trackDerivative) throws DatabaseException {
+		return loadProperties(trackDerivative, "TRACK_DERIVATIVE_PROPERTIES", "TRACK_DERIVATIVE");
+	}
+
+	/**
+	 * Loads the properties for the given track derivatives.
+	 *
+	 * @param trackDerivatives
+	 *            The track derivatives to load the properties for
+	 * @return The track derivatives with their properties loaded
+	 * @throws DatabaseException
+	 *             if a database error occurs
+	 */
+	public List<TrackDerivative> loadTrackDerivativeProperties(List<TrackDerivative> trackDerivatives) throws DatabaseException {
+		for (TrackDerivative trackDerivative : trackDerivatives) {
+			loadTrackDerivativeProperties(trackDerivative);
+		}
+		return trackDerivatives;
 	}
 
 	/**
@@ -316,6 +554,44 @@ public class DataManager {
 		return database.getMultiple(query, new StringCreator("USER_OPENIDS.OPENID"));
 	}
 
+	//
+	// PRIVATE METHODS
+	//
+
+	/**
+	 * Loads the properties for the given object.
+	 *
+	 * @param <T>
+	 *            The type of the object
+	 * @param object
+	 *            The object
+	 * @param table
+	 *            The table to load the properties from
+	 * @param type
+	 *            The type of the object (“ARTIST,” “TRACK,” etc.)
+	 * @return The object with its properties loaded
+	 * @throws DatabaseException
+	 *             if a database error occurs
+	 */
+	private <T extends Base> T loadProperties(final T object, final String table, String type) throws DatabaseException {
+		Query query = new Query(Type.SELECT, table);
+		query.addField(new Field(table + ".PROPERTY"));
+		query.addField(new Field(table + ".VALUE"));
+		query.addWhereClause(new ValueFieldWhereClause(new ValueField(type, new StringParameter(object.getId()))));
+		database.process(query, new ResultProcessor() {
+
+			@Override
+			public void processResult(ResultSet resultSet) throws SQLException {
+				if (resultSet.isFirst()) {
+					object.getProperties().removeAll();
+				}
+				object.getProperties().set(resultSet.getString(table + ".PROPERTY"), resultSet.getString(table + ".VALUE"));
+			}
+
+		});
+		return object;
+	}
+
 	/**
 	 * {@link Artist} implementation that retrieves some attributes (such as
 	 * {@link #getGroups()}, and {@link #getTracks()}) from the
@@ -515,6 +791,19 @@ public class DataManager {
 
 		});
 
+		/** Memoizer for the track derivatives. */
+		private final Memoizer<Void> derivativesMemoizer = new Memoizer<Void>(new Callable<Void>() {
+
+			@Override
+			public Void call() throws Exception {
+				if (!hasValue("derivatives")) {
+					getValue("derivatives", Collection.class).set(getTrackDerivativesByTrack(getId()));
+				}
+				return null;
+			}
+
+		});
+
 		/** Memoizer for the related tracks. */
 		private final Memoizer<Void> relatedTracksMemoizer = new Memoizer<Void>(new Callable<Void>() {
 
@@ -572,6 +861,15 @@ public class DataManager {
 		 * {@inheritDoc}
 		 */
 		@Override
+		public Collection<TrackDerivative> getDerivatives() {
+			derivativesMemoizer.get();
+			return super.getDerivatives();
+		}
+
+		/**
+		 * {@inheritDoc}
+		 */
+		@Override
 		public Map<Relationship, Collection<Track>> getRelatedTracks() {
 			relatedTracksMemoizer.get();
 			return super.getRelatedTracks();
@@ -599,6 +897,24 @@ public class DataManager {
 	}
 
 	/**
+	 * {@link ObjectCreator} implementation that can create
+	 * {@link TrackDerivative} objects.
+	 *
+	 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+	 */
+	private class TrackDerivativeCreator implements ObjectCreator<TrackDerivative> {
+
+		/**
+		 * {@inheritDoc}
+		 */
+		@Override
+		public TrackDerivative createObject(ResultSet resultSet) throws SQLException {
+			return new DefaultTrackDerivative(resultSet.getString("TRACK_DERIVATIVES.ID"));
+		}
+
+	}
+
+	/**
 	 * {@link ObjectCreator} implementation that can create {@link Style}
 	 * objects.
 	 *