X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fdemoscenemusic%2Fdata%2FDataManager.java;h=48c5ace8a3edc72106a084bbd3e1f2af71cfa512;hb=efcd05460aea3a517d03d1b36fd6ff81ac7cf12e;hp=6bec2a80c007b876cd6d817022b9f7cf1d75d578;hpb=904ede237e9dd7961ff660a407b43a9191656d2e;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 6bec2a8..48c5ace 100644
--- a/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
+++ b/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
@@ -34,6 +34,7 @@ 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;
@@ -65,6 +66,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;
 
@@ -262,6 +267,40 @@ public class DataManager {
 	}
 
 	/**
+	 * 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.
@@ -561,4 +600,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")).setLevel(resultSet.getInt("USERS.LEVEL"));
+		}
+
+	}
+
 }