X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fdemoscenemusic%2Fdata%2FDataManager.java;h=373f80c2fb1b32a61c4d572f5195d798e5067cd2;hb=6044000b16b448b4f1f6df4dd9b101ceed2397ac;hp=ac146e1e018c517a4cd58007550a1e2275b3cbf2;hpb=5fc0bbf713ccc5862e185ad47550dc2dcca2e7f9;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 ac146e1..373f80c 100644
--- a/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
+++ b/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
@@ -20,9 +20,15 @@ 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.Map.Entry;
+import java.util.UUID;
 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,10 +36,12 @@ 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;
 
@@ -60,6 +68,10 @@ public class DataManager {
 	@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;
 
@@ -83,7 +95,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 loadProperties(database.getMultiple(query, artistCreator));
 	}
 
 	/**
@@ -100,7 +112,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 loadProperties(database.getSingle(query, artistCreator));
 	}
 
 	/**
@@ -117,7 +129,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 loadProperties(database.getMultiple(query, artistCreator));
 	}
 
 	/**
@@ -135,7 +147,128 @@ 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 loadProperties(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 loadProperties(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 loadProperties(final Artist artist) throws DatabaseException {
+		Query query = new Query(Type.SELECT, "ARTIST_PROPERTIES");
+		query.addField(new Field("ARTIST_PROPERTIES.PROPERTY"));
+		query.addField(new Field("ARTIST_PROPERTIES.VALUE"));
+		query.addWhereClause(new ValueFieldWhereClause(new ValueField("ARTIST", new StringParameter(artist.getId()))));
+		database.process(query, new ResultProcessor() {
+
+			@Override
+			public void processResult(ResultSet resultSet) throws SQLException {
+				if (resultSet.isFirst()) {
+					artist.getProperties().removeAll();
+				}
+				artist.getProperties().set(resultSet.getString("ARTIST_PROPERTIES.PROPERTY"), resultSet.getString("ARTIST_PROPERTIES.VALUE"));
+			}
+
+		});
+		return 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> loadProperties(List<Artist> artists) throws DatabaseException {
+		for (Artist artist : artists) {
+			loadProperties(artist);
+		}
+		return artists;
 	}
 
 	/**
@@ -157,6 +290,38 @@ public class DataManager {
 	}
 
 	/**
+	 * 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
@@ -225,6 +390,56 @@ public class DataManager {
 	}
 
 	/**
+	 * 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.
@@ -423,6 +638,18 @@ public class DataManager {
 
 		});
 
+		/** 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.
 		 *
@@ -464,6 +691,15 @@ public class DataManager {
 			return super.getRemixArtists();
 		}
 
+		/**
+		 * {@inheritDoc}
+		 */
+		@Override
+		public Map<Relationship, Collection<Track>> getRelatedTracks() {
+			relatedTracksMemoizer.get();
+			return super.getRelatedTracks();
+		}
+
 	}
 
 	/**
@@ -503,4 +739,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"));
+		}
+
+	}
+
 }