BadgeInstance method for checking if a user has a badge.

models/badge-instance.js

@@ -13,7 +13,6 @@ var BadgeInstanceSchema = new Schema({
     type: String,
     required: true,
     trim: true,
-    unique: true,
     match: regex.email
   },
   badge: {
@@ -34,7 +33,12 @@ var BadgeInstanceSchema = new Schema({
   hash: {
     type: String,
     required: true,
-  }
+  },
+  userBadgeKey: {
+    type: String,
+    required: true,
+    unique: true
+  },
 });
 var BadgeInstance = db.model('BadgeInstance', BadgeInstanceSchema);
 
@@ -70,4 +74,28 @@ BadgeInstanceSchema.pre('validate', function hashDefault(next) {
   next();
 });
 
+/**
+ * Set the `userBadgeKey` field to be the concatenation of the `user`
+ * and `badge` fields.
+ */
+BadgeInstanceSchema.pre('validate', function userBadgeKeyDefault(next) {
+  if (this.userBadgeKey) return next();
+  this.userBadgeKey = this.user + '.' + this.badge;
+  next();
+});
+
+/**
+ * Check whether a user has a badge.
+ *
+ * @param {String} user Email address for user
+ * @param {String} shortname The badge shortname
+ */
+
+BadgeInstance.userHasBadge = function userHasBadge(user, shortname, callback) {
+  var query = { userBadgeKey: user + '.' + shortname };
+  BadgeInstance.findOne(query, { user: 1 }, function (err, instance) {
+    if (err) return callback(err);
+    return callback(null, !!instance);
+  });
+};
 module.exports = BadgeInstance;

test/badge-instance-model.test.js

@@ -19,13 +19,13 @@ test.applyFixtures({
     behaviors: [{ shortname: 'link', count: 5 }]
   }),
   'dummy-instance': new BadgeInstance({
-    user: 'dummy@example.org',
+    user: 'brian@example.org',
     hash: 'hash',
-    badge: 'link-basic',
+    badge: 'link-advanced',
     assertion: '{ "assertion" : "yep" }',
     seen: true
   }),
-}, function () {
+}, function (fixtures) {
   test('BadgeInstance#save: test defaults', function (t) {
     var instance = new BadgeInstance({
       user: 'brian@example.org',
@@ -40,6 +40,22 @@ test.applyFixtures({
     })
   });
 
+  test('BadgeInstance#userHasBadge', function (t) {
+    var instance = fixtures['dummy-instance'];
+    var user = instance.user;
+    var badge = instance.badge;
+    BadgeInstance.userHasBadge(user, badge, function (err, hasBadge) {
+      t.notOk(err, 'should not have an error');
+      t.same(hasBadge, true, 'user should have badge');
+
+      BadgeInstance.userHasBadge(user, 'non-existent', function (err, hasBadge) {
+        t.notOk(err, 'should not have an error');
+        t.same(hasBadge, false, 'user should have badge');
+        t.end();
+      });
+    });
+  });
+
   // necessary to stop the test runner
   test('shutting down #', function (t) {
     db.close(); t.end();