From 5a6993689975fcd53a397c07014936bc59a5c820 Mon Sep 17 00:00:00 2001
From: Scott Erickson <sderickson@gmail.com>
Date: Wed, 13 Apr 2016 09:43:05 -0700
Subject: [PATCH] Update init-school-roles

---
 .../2016-03-18-init-school-roles.js           | 58 ++++++++++++++-----
 1 file changed, 43 insertions(+), 15 deletions(-)

diff --git a/scripts/mongodb/migrations/2016-03-18-init-school-roles.js b/scripts/mongodb/migrations/2016-03-18-init-school-roles.js
index c57c54545..a08395f20 100644
--- a/scripts/mongodb/migrations/2016-03-18-init-school-roles.js
+++ b/scripts/mongodb/migrations/2016-03-18-init-school-roles.js
@@ -3,30 +3,57 @@
 
 // Set all users with trial requests to a teacher or teacher-like role, depending on trial request.
 
-var hasTrialRequest = {};
+var project = {role:1, name:1, email:1, permissions: 1};
 
 db.trial.requests.find().forEach(function(trialRequest) {
+    print('Inspecting trial request', trialRequest._id);
     var role = trialRequest.properties.role || 'teacher';
-    var user = db.users.findOne({_id: trialRequest.applicant}, {role:1, name:1, email:1});
-    print(JSON.stringify(user), JSON.stringify(trialRequest.properties), role);
-    if (!user.role) {
-        print(db.users.update({_id: trialRequest.applicant}, {$set: {role: role}}));
+    var user = null;
+    if(!trialRequest.applicant) {
+        print('\tNO APPLICANT INCLUDED', JSON.stringify(trialRequest));
+        if(!trialRequest.properties.email) {
+            print('\tNO EMAIL EITHER');
+            return;
+        }
+        user = db.users.findOne({emailLower: trialRequest.properties.email.toLowerCase()}, project);
+        if(!user) {
+            print('\tUSER WITH EMAIL NOT FOUND, CONTINUE');
+            return;
+        }
+        else {
+            print("\tOKAY GOT USER, UPDATE TRIAL REQUEST", JSON.stringify(user));
+            db.trial.requests.update({_id: trialRequest._id}, {$set: {applicant: user._id}});
+        }
+    }
+    else {
+        user = db.users.findOne({_id: trialRequest.applicant}, project);
+    }
+    if (!user.role && (user.permissions||[]).indexOf('admin') === -1) {
+        print('\tUpdating', JSON.stringify(user), 'to', role);
+        print(db.users.update({_id: user._id}, {$set: {role: role}}));
     }
-    hasTrialRequest[user._id.str] = true;
 });
 
-var teacherRoles = ['teacher', 'technology coordinator', 'advisor', 'principal', 'superintendent'];
-
 // Unset all teacher-like roles for users without a trial request.
 // AND removes all remaining users with a teacher-like role from classroom membership (after conversion period)
 
+var hasTrialRequest = {};
+var teacherRoles = ['teacher', 'technology coordinator', 'advisor', 'principal', 'superintendent'];
+
+db.trial.requests.find().forEach(function(trialRequest) {
+    if(!trialRequest.applicant) { return; }
+    hasTrialRequest[trialRequest.applicant.str] = true;
+});
+print(Object.keys(hasTrialRequest).length);
+
 db.users.find({'role': {$in: teacherRoles}}, {_id: 1, name: 1, email: 1, role: 1}).forEach(function(user) {
-    print('Updating user', JSON.stringify(user));
-    if (!hasTrialRequest.user._id.str) {
-        print('\tunset role');
-        //db.users.update({_id: user._id}, {$unset: {role: ''}});
+    print('Got user with teacher role', user._id);
+    if (!hasTrialRequest[user._id.str]) {
+        print('\tUnset role', JSON.stringify(user));
+        db.users.update({_id: user._id}, {$unset: {role: ''}});
     }
     else {
+        return; // TODO: Run when we've moved completely to separate user roles
         var count = db.classrooms.count({members: user._id}, {name: 1});
         if (count) {
             print('\tWill remove from classrooms');
@@ -44,9 +71,10 @@ db.classrooms.find({}, {members: 1}).forEach(function(classroom) {
     if(!classroom.members) {
         return;
     }
+    print('Updating for classroom', classroom._id, 'with members', classroom.members.length);
     for (var i in classroom.members) {
         var memberID = classroom.members[i];
-        print('updating member', memberID);
+        print('\tupdating member', memberID);
         print(db.users.update({_id: memberID, role: {$exists: false}}, {$set: {role: 'student'}}));
-    } 
-});
\ No newline at end of file
+    }
+});