wip scores

Author: gerad

lib/node-mongodb-native/lib/mongodb/collection.js

@@ -434,7 +434,7 @@ Collection.prototype.group = function(keys, condition, initial, reduce, command,
     var reduceFunction = reduce != null && reduce instanceof this.db.bson_serializer.Code ? reduce : new this.db.bson_serializer.Code(reduce);
     var selector = {'group': {
                       'ns':this.collectionName,
-                      '$reduce': reduce,
+                      '$reduce': reduceFunction,
                       'key':hash,
                       'cond':condition,
                       'initial': initial}};

models/models.coffee

@@ -99,9 +99,53 @@ class Team
   generateDeploySlugs: ->
     @joyentSlug or= @slug.replace(/^(\d)/, 'ko-$1').replace(/_/g, '-').substring(0, 30)
     @herokuSlug or= 'nko-' + @slug.replace(/_/g, '-').substring(0, 26)
-
 nko.Team = Team
 
+class ScoreCalculator
+  calculate: (fn) ->
+    @select => @merge fn
+
+  select: (fn) ->
+    threads = 4
+    @where { confirmed: false }, (error, unconfirmed) =>
+      @unconfirmed = unconfirmed
+      fn() if --threads is 0
+    @where { confirmed: true }, (error, confirmed) =>
+      @confirmed = confirmed
+      fn() if --threads is 0
+
+    Person.all { type: 'Judge' }, (error, judges) =>
+      judge_ids = _.pluck(judges, '_id');
+      @where { 'person._id': { $in: judge_ids }}, (error, judged) =>
+        @judged = judged
+        fn() if --threads is 0
+
+  merge: (fn) ->
+    scores = {}
+    for type in ['unconfirmed', 'confirmed', 'judged']
+      for score in this[type]
+        scores[score['team._id']] ||= {}
+        scores[score['team._id']][type] = score
+    fn(scores)
+
+  where: (cond, fn) ->
+    Vote.group {
+      cond: cond
+      keys: ['team._id']
+      initial: { popularity: 0, utility: 0, design: 0, innovation: 0, completeness: 0 }
+      reduce: (row, memo) ->
+        memo.popularity += 1
+        for dimension in ['utility', 'design', 'innovation', 'completeness']
+          memo[dimension] += parseInt row[dimension]
+      }, fn
+
+_.extend ScoreCalculator, {
+  calculate: (fn) ->
+    (new ScoreCalculator).calculate fn
+}
+
+nko.ScoreCalculator = ScoreCalculator
+
 class Person
   constructor: (options) ->
     @name = options?.name or ''

models/mongo.coffee

@@ -113,6 +113,11 @@ _.extend Mongo,
         return fn error if error?
         collection.update Mongo.queryify(query), cmd, {upsert: false, multi: true}, fn
 
+    group: (opts, fn) ->
+      @prototype.collection (error, collection) ->
+        return fn error if error?
+        collection.group(opts.keys, opts.cond, (opts.initial || {}), opts.reduce, true, fn)
+
   InstanceMethods:
     collection: (fn) ->
       Mongo.db.collection @serializer.name, fn

nodeko.coffee

@@ -7,7 +7,7 @@ connect = require 'connect'
 express = require 'express'
 
 models = require './models/models'
-[Team, Person, Vote] = [models.Team, models.Person, models.Vote]
+[Team, Person, Vote, ScoreCalculator] = [models.Team, models.Person, models.Vote, models.ScoreCalculator]
 
 pub = __dirname + '/public'
 app = express.createServer(
@@ -130,6 +130,15 @@ get '/register', ->
 get '/error', ->
   throw new Error('Foo')
 
+get '/scores', ->
+  ScoreCalculator.calculate (scores) =>
+    Team.all (error, teams) =>
+      for team in teams
+        team.score = scores[team.id()]
+      sys.log sys.inspect teams
+      @teams = teams
+      @render 'teams/scores.html.haml'
+
 # list teams
 get '/teams', ->
   q = if @req.param('invalid') then { url: /\w/, validDeploy: false } else { validDeploy: true }