Add aggregation function (#15)

Author: samtstern

.travis.yml

@@ -1,11 +1,9 @@
 language: node_js
 node_js:
   - "7"
-before_install:
-  - openssl aes-256-cbc -K $encrypted_001d217edcb2_key -iv $encrypted_001d217edcb2_iv -in service-account.json.enc -out service-account.json -d
 install:
   - npm install -g lerna
   - npm install -g eslint
   - lerna bootstrap
 script:
-  - ./scripts/test.sh
+  - ./scripts/test.sh

firestore/solution-aggregation/functions/index.js

@@ -0,0 +1,34 @@
+const functions = require('firebase-functions');
+const admin = require('firebase-admin');
+
+const db = admin.firestore();
+
+// [START aggregate_function]
+exports.aggregateRatings = functions.firestore
+    .document('restaurants/{restId}/ratings/{ratingId}')
+    .onWrite((change, context) => {
+      // Get value of the newly added rating
+      var ratingVal = change.after.data().rating;
+
+      // Get a reference to the restaurant
+      var restRef = db.collection('restaurants').document(context.params.restId);
+
+      // Update aggregations in a transaction
+      return db.runTransaction(transaction => {
+        return transaction.get(restRef).then(restDoc => {
+          // Compute new number of ratings
+          var newNumRatings = restDoc.data('numRatings') + 1;
+
+          // Compute new average rating
+          var oldRatingTotal = restDoc.data('avgRating') * restDoc.data('numRatings');
+          var newAvgRating = (oldRatingTotal + ratingVal) / newNumRatings;
+
+          // Update restaurant info
+          return transaction.update(restRef, {
+            avgRating: newAvgRating,
+            numRatings: newNumRatings
+          });
+        });
+      });
+    });
+// [END aggregate_function]

firestore/solution-aggregation/functions/package.json

@@ -0,0 +1,16 @@
+{
+  "name": "solution-aggregation",
+  "description": "Cloud Functions for Firebase",
+  "scripts": {
+    "serve": "firebase serve --only functions",
+    "shell": "firebase experimental:functions:shell",
+    "start": "npm run shell",
+    "deploy": "firebase deploy --only functions",
+    "logs": "firebase functions:log"
+  },
+  "dependencies": {
+    "firebase-admin": "~5.8.1",
+    "firebase-functions": "^1.0.1"
+  },
+  "private": true
+}

scripts/test.sh

@@ -25,6 +25,9 @@ find . -type f -name "*.js" -not -path "*node_modules*" \
 if [ "$TRAVIS_SECURE_ENV_VARS" = false ]; then
   echo "Could not find secure environment variables, skipping integration tests."
 else
+  # Decode secure stuff
+  openssl aes-256-cbc -K $encrypted_001d217edcb2_key -iv $encrypted_001d217edcb2_iv -in service-account.json.enc -out service-account.json -d
+
   # Run all tests
   lerna run test
-fi
+fi