Flag to disable welcome scan UI

Author: Vinnl

src/app/(proper_react)/(redesign)/(authenticated)/user/(dashboard)/dashboard/View.tsx

@@ -538,6 +538,7 @@ export const View = (props: Props) => {
           isEligibleForPremium={canSubscribeToPremium({
             user: props.user,
             countryCode,
+            enabledFeatureFlags: props.enabledFeatureFlags,
           })}
           isEligibleForFreeScan={props.isEligibleForFreeScan}
           hasExposures={hasExposures}

src/app/(proper_react)/(redesign)/(authenticated)/user/(dashboard)/dashboard/[[...slug]]/page.tsx

@@ -140,6 +140,7 @@ export default async function DashboardPage(props: Props) {
         canSubscribeToPremium({
           user: session.user,
           countryCode,
+          enabledFeatureFlags,
         })))
   ) {
     return redirect("/user/welcome");
@@ -153,8 +154,11 @@ export default async function DashboardPage(props: Props) {
   const userIsEligibleForFreeScan = await isEligibleForFreeOnerepScan(
     session.user,
     countryCode,
+    enabledFeatureFlags,
   );
-  const userIsEligibleForPremium = isEligibleForPremium(countryCode);
+  const userIsEligibleForPremium =
+    isEligibleForPremium(countryCode) &&
+    (!enabledFeatureFlags.includes("FreeOnly") || isPremiumUser);
 
   const monthlySubscriptionUrl = getPremiumSubscriptionUrl({
     type: "monthly",

src/app/(proper_react)/(redesign)/(authenticated)/user/(dashboard)/settings/[[...slug]]/page.tsx

@@ -135,6 +135,7 @@ export default async function SettingsPage(props: Props) {
   const isEligibleForPremium = canSubscribeToPremium({
     user: session.user,
     countryCode,
+    enabledFeatureFlags,
   });
 
   const userAnnouncements = await initializeUserAnnouncements(session.user);

src/app/(proper_react)/(redesign)/(authenticated)/user/welcome/[[...slug]]/page.tsx

@@ -68,6 +68,7 @@ export default async function Onboarding(props: Props) {
   const userIsEligible = await isEligibleForFreeOnerepScan(
     session.user,
     countryCode,
+    enabledFeatureFlags,
   );
 
   if (!userIsEligible) {

src/app/(proper_react)/(redesign)/Shell/ShellNavbarList.tsx

@@ -39,23 +39,25 @@ export const NavbarList = (props: {
               {l10n.getString("main-nav-link-dashboard-label")}
             </PageLink>
           </li>
-          {props.enabledFeatureFlags.includes("EditScanProfileDetails") && (
-            <li key="settings-edit-info">
-              <PageLink
-                href="/user/settings/edit-info"
-                activeClassName={styles.isActive}
-                hasTelemetry={{ link_id: "navigation_edit_info" }}
-              >
-                <ScanInfoIcon alt="" />
-                {l10n.getString("settings-tab-label-update-scan-info")}
-              </PageLink>
-            </li>
-          )}
+          {props.enabledFeatureFlags.includes("EditScanProfileDetails") &&
+            !props.enabledFeatureFlags.includes("FreeOnly") && (
+              <li key="settings-edit-info">
+                <PageLink
+                  href="/user/settings/edit-info"
+                  activeClassName={styles.isActive}
+                  hasTelemetry={{ link_id: "navigation_edit_info" }}
+                >
+                  <ScanInfoIcon alt="" />
+                  {l10n.getString("settings-tab-label-update-scan-info")}
+                </PageLink>
+              </li>
+            )}
         </ul>
         <hr />
         <strong>{l10n.getString("main-nav-link-settings-label")}</strong>
         <ul className="noList">
-          {!props.enabledFeatureFlags.includes("EditScanProfileDetails") && (
+          {(!props.enabledFeatureFlags.includes("EditScanProfileDetails") ||
+            props.enabledFeatureFlags.includes("FreeOnly")) && (
             <li key="settings-edit-info">
               <PageLink
                 href="/user/settings/edit-info"

src/app/api/v1/user/welcome-scan/create/route.ts

@@ -30,6 +30,7 @@ import {
   getAcceptLangHeaderInServerComponents,
   getL10n,
 } from "../../../../../functions/l10n/serverComponents";
+import { getEnabledFeatureFlags } from "../../../../../../db/tables/featureFlags";
 
 export interface WelcomeScanBody {
   success: boolean;
@@ -63,7 +64,15 @@ export async function POST(
     countryCode,
     locale: getLocale(getL10n(await getAcceptLangHeaderInServerComponents())),
   });
-  const eligible = await isEligibleForFreeOnerepScan(session.user, countryCode);
+  const enabledFeatureFlags = await getEnabledFeatureFlags({
+    isSignedOut: false,
+    email: session.user.email,
+  });
+  const eligible = await isEligibleForFreeOnerepScan(
+    session.user,
+    countryCode,
+    enabledFeatureFlags,
+  );
   if (!eligible) {
     logger.warn("scan_created_warn", {
       message: "User is not eligible for feature",

src/app/components/client/toolbar/UpsellBadge.tsx

@@ -163,7 +163,14 @@ export function UpsellBadge(props: UpsellBadgeProps) {
 
   const { user } = session.data;
   const userHasPremium = hasPremium(user);
-  if (userHasPremium || canSubscribeToPremium({ user, countryCode })) {
+  if (
+    userHasPremium ||
+    canSubscribeToPremium({
+      user,
+      countryCode,
+      enabledFeatureFlags: props.enabledFeatureFlags,
+    })
+  ) {
     return <UpsellToggleLinkButton {...props} hasPremium={userHasPremium} />;
   }
 

src/app/functions/server/getRelevantGuidedSteps.test.ts

@@ -456,6 +456,27 @@ describe("getNextGuidedStep", () => {
       });
     });
 
+    it("does not link to the scan even if the user has not run a scan yet, if the `FreeOnly` flag is enabled", () => {
+      expect(
+        getNextGuidedStep(
+          {
+            countryCode: "us",
+            latestScanData: {
+              scan: null,
+              results: [],
+            },
+            subscriberBreaches: [],
+            user: {
+              email: "[email protected]",
+            },
+          },
+          ["FreeOnly"],
+        ),
+      ).not.toMatchObject({
+        id: "Scan",
+      });
+    });
+
     it("links to the scan if the user has a scan in progress and not all scan results are resolved", () => {
       expect(
         getNextGuidedStep(

src/app/functions/server/getRelevantGuidedSteps.ts

@@ -7,6 +7,7 @@ import { LatestOnerepScanData } from "../../../db/tables/onerep_scans";
 import { SubscriberBreach } from "../../../utils/subscriberBreaches";
 import { BreachDataTypes, HighRiskDataTypes } from "../universal/breach";
 import { FeatureFlagName } from "../../../db/tables/featureFlags";
+import { hasPremium } from "../universal/user";
 
 export type StepDeterminationData = {
   user: Session["user"];
@@ -164,7 +165,10 @@ export function isEligibleForStep(
   /* c8 ignore stop */
 
   if (stepId === "Scan") {
-    return data.countryCode === "us";
+    return (
+      data.countryCode === "us" &&
+      (!enabledFeatureFlags?.includes("FreeOnly") || hasPremium(data.user))
+    );
   }
 
   if (stepId === "HighRiskSsn") {

src/app/functions/server/onerep.ts

@@ -16,6 +16,7 @@ import { logger } from "./logging";
 import { hasPremium } from "../universal/user.ts";
 import { OnerepProfileAddress } from "knex/types/tables";
 import { isUsingMockONEREPEndpoint } from "../universal/mock.ts";
+import type { FeatureFlagName } from "../../../db/tables/featureFlags";
 
 export const monthlyScansQuota = parseInt(
   (process.env.MONTHLY_SCANS_QUOTA as string) ?? "0",
@@ -460,10 +461,14 @@ async function listScanResults(
 export async function isEligibleForFreeScan(
   user: Session["user"],
   countryCode: string,
+  enabledFeatureFlags: FeatureFlagName[],
 ) {
   if (countryCode !== "us") {
     return false;
   }
+  if (enabledFeatureFlags.includes("FreeOnly") && !hasPremium(user)) {
+    return false;
+  }
 
   if (!user?.subscriber?.id) {
     throw new Error("No session with a known subscriber found");