Website: Add primaryBuyingSituation migration script. (#31494)

Related to https://github.com/fleetdm/confidential/issues/11600 Changes: - Added a script to update user records to only use the new `primaryBuyingSituation` values added in https://github.com/fleetdm/fleet/pull/31301
2025-08-01 17:07:27 -05:00
parent 9438cf1346
commit cba94083c3
1 changed files with 90 additions and 0 deletions
@@ -0,0 +1,90 @@
+module.exports = {
+
+
+  friendlyName: 'Migrate old primary buying situation values',
+
+
+  description: 'Updates the primaryBuyingSituation value of user records that have older values set.',
+
+  inputs: {
+    dry: { type: 'boolean', description: 'Whether to make this a dry run. (No database changes will be made.)' },
+  },
+
+  fn: async function ({dry}) {
+    sails.log('Running custom shell script... (`sails run migrate-old-primary-buying-situation-values`)');
+
+    let numberOfUsersWhoWouldHaveBeenUpdated = 0;
+    sails.log('Migrating users with primaryBuyingSituation === mdm.');
+    await User.stream({primaryBuyingSituation: 'mdm'})
+    .eachRecord(async (thisUser)=>{
+      // Default to 'it-major-mdm' for users with an mdm primaryBuyingSituation.
+      let newPrimaryBuyingSituationForThisUser = 'it-major-mdm';
+      // If this user has filled out the get started questionnaire, check their answers to determine if they were interested in Linux MDM.
+      if(thisUser.getStartedQuestionnaireAnswers !== {}){
+        if(thisUser.getStartedQuestionnaireAnswers['what-do-you-manage-mdm']) {
+          // If they previosuly selected linux as a response to the "What do you manage?" question, set their new primaryBuyingSituation to be 'it-gap-filler-mdm'
+          if(thisUser.getStartedQuestionnaireAnswers['what-do-you-manage-mdm'].mdmUseCase === 'linux') {
+            newPrimaryBuyingSituationForThisUser = 'it-gap-filler-mdm';
+          }
+        }
+      }
+      if(dry) {
+        numberOfUsersWhoWouldHaveBeenUpdated++;
+      } else {
+        // Update this user's primary buying situation.
+        await User.updateOne({id: thisUser.id}).set({
+          primaryBuyingSituation: newPrimaryBuyingSituationForThisUser,
+        });
+      }
+    });
+
+    sails.log('Migrating users with primaryBuyingSituation === eo-it.');
+    await User.stream({primaryBuyingSituation: 'eo-it'})
+    .eachRecord(async (thisUser)=>{
+      let newPrimaryBuyingSituationForThisUser = 'it-misc';
+      if(dry) {
+        numberOfUsersWhoWouldHaveBeenUpdated++;
+      } else {
+        // Update this user's primary buying situation.
+        await User.updateOne({id: thisUser.id}).set({
+          primaryBuyingSituation: newPrimaryBuyingSituationForThisUser,
+        });
+      }
+    });
+
+    sails.log('Migrating users with primaryBuyingSituation === eo-security.');
+    await User.stream({primaryBuyingSituation: 'eo-security'})
+    .eachRecord(async (thisUser)=>{
+      let newPrimaryBuyingSituationForThisUser = 'security-misc';
+      if(dry) {
+        numberOfUsersWhoWouldHaveBeenUpdated++;
+      } else {
+        // Update this user's primary buying situation.
+        await User.updateOne({id: thisUser.id}).set({
+          primaryBuyingSituation: newPrimaryBuyingSituationForThisUser,
+        });
+      }
+    });
+
+    sails.log('Migrating users with primaryBuyingSituation === vm.');
+    await User.stream({primaryBuyingSituation: 'vm'})
+    .eachRecord(async (thisUser)=>{
+      let newPrimaryBuyingSituationForThisUser = 'security-vm';
+      if(dry) {
+        numberOfUsersWhoWouldHaveBeenUpdated++;
+      } else {
+        // Update this user's primary buying situation.
+        await User.updateOne({id: thisUser.id}).set({
+          primaryBuyingSituation: newPrimaryBuyingSituationForThisUser,
+        });
+      }
+    });
+
+    if(dry){
+      sails.log(`Dry run: would have migrated the primaryBuyingSituation values for ${numberOfUsersWhoWouldHaveBeenUpdated} user records.`);
+    }
+  }
+
+
+};
+