feat: Add audit trial length optimizely experiment

- Add the experiment and variation keys
- And some logic for upgrade eligibility

src/data/optimizely.js

@@ -23,13 +23,21 @@ const getOptimizely = () => {
   return instance;
 };
 
+const OPTIMIZELY_AUDIT_TRIAL_LENGTH_EXPERIMENT_KEY = 'xpert_audit_trial_experiment';
+const OPTIMIZELY_AUDIT_TRIAL_LENGTH_EXPERIMENT_VARIATION_KEYS = {
+  XPERT_AUDIT_14_DAY_TRIAL: 'xpert_audit_14_day_trial',
+  XPERT_AUDIT_28_DAY_TRIAL: 'xpert_audit_28_day_trial',
+};
+
 const OPTIMIZELY_PROMPT_EXPERIMENT_KEY = '_cosmo__xpert_gpt_4_0_prompt';
 const OPTIMIZELY_PROMPT_EXPERIMENT_VARIATION_KEYS = {
   UPDATED_PROMPT: 'updated_prompt',
 };
 
 export {
   getOptimizely,
+  OPTIMIZELY_AUDIT_TRIAL_LENGTH_EXPERIMENT_KEY,
+  OPTIMIZELY_AUDIT_TRIAL_LENGTH_EXPERIMENT_VARIATION_KEYS,
   OPTIMIZELY_PROMPT_EXPERIMENT_KEY,
   OPTIMIZELY_PROMPT_EXPERIMENT_VARIATION_KEYS,
 };

src/experiments/experimentHooks.js

@@ -1,7 +1,7 @@
 import { getAuthenticatedUser } from '@edx/frontend-platform/auth';
 import { useDecision } from '@optimizely/react-sdk';
 
-import { OPTIMIZELY_PROMPT_EXPERIMENT_KEY } from '../data/optimizely';
+import { OPTIMIZELY_AUDIT_TRIAL_LENGTH_EXPERIMENT_KEY, OPTIMIZELY_PROMPT_EXPERIMENT_KEY } from '../data/optimizely';
 
 // eslint-disable-next-line import/prefer-default-export
 export function usePromptExperimentDecision() {
@@ -14,3 +14,15 @@ export function usePromptExperimentDecision() {
 
   return [decision];
 }
+
+// eslint-disable-next-line import/prefer-default-export
+export function useAuditTrialExperimentDecision() {
+  const { userId } = getAuthenticatedUser();
+
+  const [decision] = useDecision(
+    OPTIMIZELY_AUDIT_TRIAL_LENGTH_EXPERIMENT_KEY,
+    { overrideUserId: userId.toString() }, // This override is just to make sure the userId is up to date.
+  );
+
+  return [decision];
+}

src/hooks/use-course-upgrade.js

@@ -3,6 +3,14 @@ import { useModel } from '@src/generic/model-store'; // eslint-disable-line impo
 import { useSelector } from 'react-redux';
 import { CourseInfoContext } from '../context';
 
+import { sendTrackEvent } from '@edx/frontend-platform/analytics';
+
+import {
+  OPTIMIZELY_AUDIT_TRIAL_LENGTH_EXPERIMENT_KEY,
+  OPTIMIZELY_AUDIT_TRIAL_LENGTH_EXPERIMENT_VARIATION_KEYS,
+} from '../../data/optimizely';
+import { useAuditTrialExperimentDecision } from '../../experiments';
+
 const millisecondsInOneDay = 24 * 60 * 60 * 1000; // hours*minutes*seconds*milliseconds
 
 /**
@@ -28,6 +36,7 @@ const millisecondsInOneDay = 24 * 60 * 60 * 1000; // hours*minutes*seconds*milli
  *
  * @returns {CourseUpgradeInfo}
  */
+// Update logic here to make upgrade eligible if in experiment variation
 export default function useCourseUpgrade() {
   const { courseId, isUpgradeEligible } = useContext(CourseInfoContext);
   const {
@@ -41,6 +50,30 @@ export default function useCourseUpgrade() {
     auditTrial,
   } = useSelector(state => state.learningAssistant);
 
+  const [decision] = useAuditTrialExperimentDecision();
+  const { enabled, variationKey } = decision || {};
+  const experimentPayload = enabled ? {
+    experiment_name: OPTIMIZELY_AUDIT_TRIAL_LENGTH_EXPERIMENT_KEY,
+    variation_key: variationKey,
+  } : {};
+
+  // Make upgrade eligible if in experiment variation
+  if (OPTIMIZELY_AUDIT_TRIAL_LENGTH_EXPERIMENT_VARIATION_KEYS.includes(variationKey)) {
+    isUpgradeEligible = true;
+
+    // Do I need to add a sendTrackEvent call like this?
+    // sendTrackEvent(
+    //   'edx.ui.lms.learning_assistant.audit_trial_started',
+    //   {
+    //     course_id: courseId,
+    //     user_id: userId,
+    //     source: event.target.id === 'toggle-button' ? 'toggle' : 'cta',
+    //     ...experimentPayload,
+    //   },
+    // );
+  }
+
+
   const upgradeUrl = offer?.upgradeUrl || verifiedMode?.upgradeUrl;
 
   if (!isUpgradeEligible || !upgradeUrl) { return { upgradeable: false }; }