From 8a398d5060a828dcf0fad91d8f9b2f2e8f8bbc30 Mon Sep 17 00:00:00 2001
From: Vlad Keliy <vekkele@gmail.com>
Date: Wed, 11 Jun 2025 09:57:57 +0200
Subject: [PATCH] refactor(plan) Extract and refactor generic plan utils

---
 .storybook/mock/user.ts                       | 23 ++++---
 src/lib/analytics/amplitude/flow.svelte.ts    |  4 +-
 src/lib/analytics/posthog/flow.svelte.ts      |  4 +-
 src/lib/ui/app/PaymentForm/flow.ts            |  4 +-
 .../PlanChangeDialog/PlanChangeDialog.svelte  |  5 +-
 .../BreakdownTable/breakdown.ts               | 14 ++---
 .../ui/app/SubscriptionPlan/PlanButton.svelte | 22 +++----
 src/lib/ui/app/SubscriptionPlan/api.ts        | 17 ++----
 src/lib/ui/app/SubscriptionPlan/index.ts      | 18 +-----
 src/lib/ui/app/SubscriptionPlan/plans.ts      | 55 ++++-------------
 .../ui/app/SubscriptionPlan/subscription.ts   | 27 +++++----
 src/lib/ui/app/SubscriptionPlan/utils.ts      | 36 +++++------
 src/lib/utils/index.ts                        |  2 +-
 src/lib/utils/object.ts                       |  4 ++
 src/lib/utils/plans/index.ts                  | 16 +++++
 src/lib/utils/plans/plans.ts                  | 60 +++++++++++++++++++
 src/lib/utils/plans/products.ts               | 20 +++++++
 src/lib/utils/plans/subscription.ts           | 32 ++++++++++
 18 files changed, 220 insertions(+), 143 deletions(-)
 create mode 100644 src/lib/utils/plans/index.ts
 create mode 100644 src/lib/utils/plans/plans.ts
 create mode 100644 src/lib/utils/plans/products.ts
 create mode 100644 src/lib/utils/plans/subscription.ts

diff --git a/.storybook/mock/user.ts b/.storybook/mock/user.ts
index 25a2fa1be..3ea2e5cd7 100644
--- a/.storybook/mock/user.ts
+++ b/.storybook/mock/user.ts
@@ -1,10 +1,7 @@
 // import { getTodaysEnd } from '@/utils/dates'
+import { checkIsBusinessPlan, Plan, Product } from '$lib/utils/plans/index.js'
+
 import { getTodaysEnd } from '../../src/lib/utils/dates/index.js'
-import {
-  SubscriptionPlan,
-  Product,
-  checkIsBusinessPlan,
-} from '../../src/lib/ui/app/SubscriptionPlan/index.js'
 
 export type CurrentUser = null | {
   /** @default 42 */
@@ -147,14 +144,14 @@ export function mockUser(currentUser: CurrentUser) {
 
     if (pro || max || proPlus || businessPro || businessMax || custom || planName) {
       let name = planName
-      if (pro) name = SubscriptionPlan.PRO.key
-      else if (max) name = SubscriptionPlan.MAX.key
-      else if (proPlus) name = SubscriptionPlan.PRO_PLUS.key
-      else if (businessPro) name = SubscriptionPlan.BUSINESS_PRO.key
-      else if (businessMax) name = SubscriptionPlan.BUSINESS_MAX.key
-      else if (custom) name = SubscriptionPlan.CUSTOM.key
-
-      const id = name && checkIsBusinessPlan({ name }) ? Product.SanAPI.id : Product.Sanbase.id
+      if (pro) name = Plan.PRO
+      else if (max) name = Plan.MAX
+      else if (proPlus) name = Plan.PRO_PLUS
+      else if (businessPro) name = Plan.BUSINESS_PRO
+      else if (businessMax) name = Plan.BUSINESS_MAX
+      else if (custom) name = Plan.CUSTOM
+
+      const id = name && checkIsBusinessPlan(name) ? Product.SANAPI.id : Product.SANBASE.id
 
       subscriptions[0] = {
         status: trial ? 'TRIALING' : 'ACTIVE',
diff --git a/src/lib/analytics/amplitude/flow.svelte.ts b/src/lib/analytics/amplitude/flow.svelte.ts
index 7c0502c7d..c872f3e35 100644
--- a/src/lib/analytics/amplitude/flow.svelte.ts
+++ b/src/lib/analytics/amplitude/flow.svelte.ts
@@ -3,7 +3,7 @@ import * as amplitude from '@amplitude/analytics-browser'
 
 import { useCustomerCtx } from '$lib/ctx/customer/index.svelte.js'
 import { useUiCtx } from '$lib/ctx/ui/index.svelte.js'
-import { SubscriptionPlan } from '$ui/app/SubscriptionPlan/plans.js'
+import { Plan } from '$lib/utils/plans/index.js'
 
 export function useDebouncedFn<GFunction extends (...args: any[]) => void>(
   time: number,
@@ -48,7 +48,7 @@ export function useAmplitudeFlow() {
   )
 
   const updateUserSanbasePlan = useDebouncedFn(1000, (sanbase_plan?: string) =>
-    setAmplitudeUserProperties({ sanbase_plan: sanbase_plan || SubscriptionPlan.FREE.key }),
+    setAmplitudeUserProperties({ sanbase_plan: sanbase_plan || Plan.FREE }),
   )
 
   $effect(() => {
diff --git a/src/lib/analytics/posthog/flow.svelte.ts b/src/lib/analytics/posthog/flow.svelte.ts
index df2f047c5..776cd447a 100644
--- a/src/lib/analytics/posthog/flow.svelte.ts
+++ b/src/lib/analytics/posthog/flow.svelte.ts
@@ -2,7 +2,7 @@ import { posthog } from 'posthog-js'
 import { BROWSER } from 'esm-env'
 
 import { useCustomerCtx } from '$lib/ctx/customer/index.js'
-import { SubscriptionPlan } from '$ui/app/SubscriptionPlan/plans.js'
+import { Plan } from '$lib/utils/plans/index.js'
 
 import { useDebouncedFn } from '../amplitude/flow.svelte.js'
 
@@ -22,7 +22,7 @@ export function usePosthogFlow() {
     (sanbase_plan?: string, featureAccessLevel?: string) =>
       posthog.capture('$set', {
         $set: {
-          sanbase_plan: sanbase_plan || SubscriptionPlan.FREE.key,
+          sanbase_plan: sanbase_plan || Plan.FREE,
           feature_access_level: featureAccessLevel,
         },
       }),
diff --git a/src/lib/ui/app/PaymentForm/flow.ts b/src/lib/ui/app/PaymentForm/flow.ts
index 59a60065d..a37f105b0 100644
--- a/src/lib/ui/app/PaymentForm/flow.ts
+++ b/src/lib/ui/app/PaymentForm/flow.ts
@@ -6,10 +6,10 @@ import { useStripeCtx } from '$lib/ctx/stripe/index.js'
 import { notification } from '$ui/core/Notifications/index.js'
 import { useCustomerCtx } from '$lib/ctx/customer/index.js'
 import { trackEvent } from '$lib/analytics/index.js'
+import { getPlanDisplayName } from '$lib/utils/plans/index.js'
 
 import { mutateSubscribe } from './api.js'
 import { usePaymentFormCtx } from './state.js'
-import { getPlanName } from '../SubscriptionPlan/utils.js'
 
 export type TPaymentFlowResult = undefined | API.ExtractData<typeof mutateSubscribe>
 
@@ -134,7 +134,7 @@ export function usePaymentFlow() {
       return Promise.reject('paymentMethod is missing')
     }
 
-    const planDisplayName = getPlanName(plan)
+    const planDisplayName = getPlanDisplayName(plan.name)
     const isConsumerPlan = !subscriptionPlan.$.formatted?.isBusiness
     const isEligibleForSanbaseTrial = isConsumerPlan && customer.$.isEligibleForSanbaseTrial
 
diff --git a/src/lib/ui/app/PlanChangeDialog/PlanChangeDialog.svelte b/src/lib/ui/app/PlanChangeDialog/PlanChangeDialog.svelte
index 38273711c..449f268cc 100644
--- a/src/lib/ui/app/PlanChangeDialog/PlanChangeDialog.svelte
+++ b/src/lib/ui/app/PlanChangeDialog/PlanChangeDialog.svelte
@@ -16,9 +16,10 @@
   import { useCustomerCtx } from '$lib/ctx/customer/index.svelte.js'
   import { Query } from '$lib/api/executor.js'
   import { getFormattedMonthDayYear } from '$lib/utils/dates/index.js'
+  import { getPlanDisplayName } from '$lib/utils/plans/index.js'
 
   import { mutateUpdateSubscription } from './api.js'
-  import { getFormattedBillingPlan, getPlanName } from '../SubscriptionPlan/utils.js'
+  import { getFormattedBillingPlan } from '../SubscriptionPlan/utils.js'
 
   let {
     source = '',
@@ -42,7 +43,7 @@
     if (!primarySubscription) return
     if (loading) return
 
-    const planDisplayName = getPlanName(newPlan)
+    const planDisplayName = getPlanDisplayName(newPlan.name)
     loading = true
     Controller.lock()
 
diff --git a/src/lib/ui/app/SubscriptionPlan/BreakdownTable/breakdown.ts b/src/lib/ui/app/SubscriptionPlan/BreakdownTable/breakdown.ts
index 7d0700ca0..114b1e5f2 100644
--- a/src/lib/ui/app/SubscriptionPlan/BreakdownTable/breakdown.ts
+++ b/src/lib/ui/app/SubscriptionPlan/BreakdownTable/breakdown.ts
@@ -1,4 +1,4 @@
-import { SubscriptionPlan } from '../plans.js'
+import { Plan } from '$lib/utils/plans/index.js'
 
 export type TBreakdownFeature = {
   name: string
@@ -231,7 +231,7 @@ export const SubscriptionPlanBreakdown: Record<
   string,
   undefined | Record<string, TPlanFeatureBreakdown>
 > = {
-  [SubscriptionPlan.FREE.key]: {
+  [Plan.FREE]: {
     'Browser tabs': 'Up to 4',
     'Bi-weekly report': false,
     'Pro Insights': false,
@@ -273,7 +273,7 @@ export const SubscriptionPlanBreakdown: Record<
     'Direct technical support': false,
   },
 
-  [SubscriptionPlan.PRO.key]: {
+  [Plan.PRO]: {
     'Dedicated account manager': false,
     'Custom onboarding & education': false,
 
@@ -298,7 +298,7 @@ export const SubscriptionPlanBreakdown: Record<
     'Direct technical support': false,
   },
 
-  [SubscriptionPlan.MAX.key]: {
+  [Plan.MAX]: {
     'Custom Alerts': 20,
 
     'Historical data restriction': '1 year',
@@ -315,7 +315,7 @@ export const SubscriptionPlanBreakdown: Record<
     'Private queries': 'No',
   },
 
-  [SubscriptionPlan.CUSTOM.key]: {
+  [Plan.CUSTOM]: {
     'Multi-seat account': 'Custom',
 
     'Historical data restriction': 'No restriction',
@@ -335,7 +335,7 @@ export const SubscriptionPlanBreakdown: Record<
     'Custom Alerts': 'Custom',
   },
 
-  [SubscriptionPlan.BUSINESS_PRO.key]: {
+  [Plan.BUSINESS_PRO]: {
     'Multi-seat account': false,
 
     'Historical data restriction': 'Last 2 years',
@@ -363,7 +363,7 @@ export const SubscriptionPlanBreakdown: Record<
     'Custom data integratoins': false,
   },
 
-  [SubscriptionPlan.BUSINESS_MAX.key]: {
+  [Plan.BUSINESS_MAX]: {
     'Multi-seat account': '3 seats',
 
     'Historical data restriction': 'No restriction',
diff --git a/src/lib/ui/app/SubscriptionPlan/PlanButton.svelte b/src/lib/ui/app/SubscriptionPlan/PlanButton.svelte
index 67bd094d2..e7ea35941 100644
--- a/src/lib/ui/app/SubscriptionPlan/PlanButton.svelte
+++ b/src/lib/ui/app/SubscriptionPlan/PlanButton.svelte
@@ -4,18 +4,18 @@
   import { trackEvent } from '$lib/analytics/index.js'
   import { useCustomerCtx } from '$lib/ctx/customer/index.js'
   import { onSupportClick } from '$lib/utils/support.js'
-  import {
-    BUSINESS_PLANS,
-    CONSUMER_PLANS,
-    SubscriptionPlan,
-  } from '$ui/app/SubscriptionPlan/plans.js'
   import {
     checkIsAlternativeBillingPlan,
     checkIsCurrentPlan,
-    getPlanName,
   } from '$ui/app/SubscriptionPlan/utils.js'
   import Button from '$ui/core/Button/index.js'
   import { cn } from '$ui/utils/index.js'
+  import {
+    checkIsBusinessPlan,
+    checkIsConsumerPlan,
+    getPlanDisplayName,
+    Plan,
+  } from '$lib/utils/plans/index.js'
 
   import { useSubscriptionPlanButtonCtx } from './ctx.js'
 
@@ -34,8 +34,8 @@
   const { customer } = useCustomerCtx()
 
   let source = $derived(_source + '_plan_button')
-  let isBusinessPlan = $derived(BUSINESS_PLANS.has(plan.name))
-  let isConsumerPlan = $derived(CONSUMER_PLANS.has(plan.name))
+  let isBusinessPlan = $derived(checkIsBusinessPlan(plan.name))
+  let isConsumerPlan = $derived(checkIsConsumerPlan(plan.name))
 
   let isCurrentPlan = $derived(checkIsCurrentPlan(customer.$.plan, plan))
   let isSameProductPlan = $derived(
@@ -90,7 +90,7 @@
       Your current plan
     </Button>
   {/if}
-{:else if plan.name === SubscriptionPlan.FREE.key}
+{:else if plan.name === Plan.FREE}
   <Button
     variant="border"
     size="lg"
@@ -102,7 +102,7 @@
   >
     Default plan
   </Button>
-{:else if plan.name === SubscriptionPlan.CUSTOM.key}
+{:else if plan.name === Plan.CUSTOM}
   <Button
     variant="fill"
     size="lg"
@@ -140,7 +140,7 @@
     {...anonymousProps}
     data-type="get_business"
   >
-    Get {getPlanName(plan)}
+    Get {getPlanDisplayName(plan.name)}
   </Button>
 {:else if isConsumerPlan && (customer.$.isEligibleForSanbaseTrial || isAnonymous)}
   <Button
diff --git a/src/lib/ui/app/SubscriptionPlan/api.ts b/src/lib/ui/app/SubscriptionPlan/api.ts
index c211fed0b..de3f8cda1 100644
--- a/src/lib/ui/app/SubscriptionPlan/api.ts
+++ b/src/lib/ui/app/SubscriptionPlan/api.ts
@@ -1,6 +1,5 @@
 import { ApiQuery } from '$lib/api/index.js'
-
-import { BUSINESS_PLANS, CONSUMER_PLANS, Product, SubscriptionPlan } from './plans.js'
+import { checkIsBusinessPlan, checkIsConsumerPlan, Plan, Product } from '$lib/utils/plans/index.js'
 
 export type TProductsWithPlans = readonly {
   id: string
@@ -45,7 +44,7 @@ export type TPlanBillingGroup = Record<string, { month: TProductPlan; year?: TPr
 export function getProductPlans(
   productsWithPlans: TProductsWithPlans,
   productId: string,
-  listOfPlans: Set<string>,
+  planChecker: (planName: string) => boolean,
 ): null | {
   plans: TProductPlan[]
   planBillingGroup: TPlanBillingGroup
@@ -55,13 +54,9 @@ export function getProductPlans(
   if (!productPlans) return null
 
   const plans = productPlans.plans
-    .filter((plan) => listOfPlans.has(plan.name) && !plan.isDeprecated)
+    .filter((plan) => planChecker(plan.name) && !plan.isDeprecated)
     .sort((a, b) =>
-      a.name === SubscriptionPlan.CUSTOM.key
-        ? 1
-        : b.name === SubscriptionPlan.CUSTOM.key
-          ? -1
-          : +a.amount - +b.amount,
+      a.name === Plan.CUSTOM ? 1 : b.name === Plan.CUSTOM ? -1 : +a.amount - +b.amount,
     )
 
   const planBillingGroup = plans.reduce<
@@ -78,7 +73,7 @@ export function getProductPlans(
 export type TProductPlans = ReturnType<typeof getProductPlans>
 
 export const getSanbaseConsumerPlans = (productsWithPlans: TProductsWithPlans) =>
-  getProductPlans(productsWithPlans, Product.Sanbase.id, CONSUMER_PLANS)
+  getProductPlans(productsWithPlans, Product.SANBASE.id, checkIsConsumerPlan)
 
 export const getApiBusinessPlans = (productsWithPlans: TProductsWithPlans) =>
-  getProductPlans(productsWithPlans, Product.SanAPI.id, BUSINESS_PLANS)
+  getProductPlans(productsWithPlans, Product.SANAPI.id, checkIsBusinessPlan)
diff --git a/src/lib/ui/app/SubscriptionPlan/index.ts b/src/lib/ui/app/SubscriptionPlan/index.ts
index 2caa077b6..2e2f1c0c3 100644
--- a/src/lib/ui/app/SubscriptionPlan/index.ts
+++ b/src/lib/ui/app/SubscriptionPlan/index.ts
@@ -5,22 +5,8 @@ export {
   getSanbaseConsumerPlans,
 } from './api.js'
 
-export {
-  Product,
-  SubscriptionPlan,
-  SubscriptionPlanDetails,
-  BUSINESS_PLANS,
-  CONSUMER_PLANS,
-} from './plans.js'
-export {
-  checkIsSanbaseProduct,
-  checkIsSanApiProduct,
-  checkIsBusinessPlan,
-  checkIsAlternativeBillingPlan,
-  checkIsCurrentPlan,
-  getPlanName,
-  getFormattedPlan,
-} from './utils.js'
+export { SubscriptionPlanDetails } from './plans.js'
+export { checkIsAlternativeBillingPlan, checkIsCurrentPlan, getFormattedPlan } from './utils.js'
 
 export { default as PlanButton } from './PlanButton.svelte'
 export { default as PlanCard } from './PlanCard.svelte'
diff --git a/src/lib/ui/app/SubscriptionPlan/plans.ts b/src/lib/ui/app/SubscriptionPlan/plans.ts
index 49381bc97..c94d6513c 100644
--- a/src/lib/ui/app/SubscriptionPlan/plans.ts
+++ b/src/lib/ui/app/SubscriptionPlan/plans.ts
@@ -1,16 +1,7 @@
-import { keyify } from '$lib/utils/object.js'
+import { exactObjectKeys } from '$lib/utils/object.js'
+import { Plan } from '$lib/utils/plans/index.js'
 
-export const Product = keyify(
-  {
-    SanAPI: { id: '1' },
-    Sanbase: { id: '2' },
-  },
-  'name',
-)
-
-export type PlanType = 'business' | 'consumer'
-
-export const PlanTypeDisplayInfo: Record<PlanType, { name: string; className: string }> = {
+export const PlanTypeDisplayInfo = {
   consumer: {
     name: '👨‍🦱 For Individuals',
     className: 'outline-green bg-green-light-1',
@@ -19,21 +10,11 @@ export const PlanTypeDisplayInfo: Record<PlanType, { name: string; className: st
     name: '💼 For Business',
     className: 'outline-blue bg-blue-light-1',
   },
-}
+} satisfies Record<string, { name: string; className: string }>
 
-export const planTypes = Object.keys(PlanTypeDisplayInfo) as PlanType[]
+export const planTypes = exactObjectKeys(PlanTypeDisplayInfo)
 
-export const SubscriptionPlan = keyify({
-  PRO_PLUS: { name: 'Sanbase Pro+' }, // Deprecated
-
-  FREE: { name: 'Free' },
-  PRO: { name: 'Sanbase Pro' },
-  MAX: { name: 'Sanbase Max' },
-
-  BUSINESS_PRO: { name: 'Business Pro' },
-  BUSINESS_MAX: { name: 'Business Max' },
-  CUSTOM: { name: 'Enterprise' },
-})
+export type PlanType = (typeof planTypes)[number]
 
 export const SubscriptionPlanDetails: Record<
   string,
@@ -43,7 +24,7 @@ export const SubscriptionPlanDetails: Record<
       features: string[]
     }
 > = {
-  [SubscriptionPlan.FREE.key]: {
+  [Plan.FREE]: {
     description: 'Ideal for new traders and investors to kickstart crypto analysis',
     features: [
       '30-day lag in Sanbase',
@@ -55,7 +36,7 @@ export const SubscriptionPlanDetails: Record<
     ],
   },
 
-  [SubscriptionPlan.PRO.key]: {
+  [Plan.PRO]: {
     description: 'Ideal for advanced traders and analysts to power decisions with real-time data',
     features: [
       'Present-day and full historical data in Sanbase',
@@ -68,7 +49,7 @@ export const SubscriptionPlanDetails: Record<
     ],
   },
 
-  [SubscriptionPlan.MAX.key]: {
+  [Plan.MAX]: {
     description: 'Ideal for advanced investment strategies with full API access to real-time data',
     features: [
       'Present-day and full historical data in Sanbase and Trending Coins',
@@ -81,7 +62,7 @@ export const SubscriptionPlanDetails: Record<
     ],
   },
 
-  [SubscriptionPlan.BUSINESS_PRO.key]: {
+  [Plan.BUSINESS_PRO]: {
     description: 'Designed for algorithmic trading and investment strategies',
     features: [
       '2 years of historical data through API',
@@ -93,7 +74,7 @@ export const SubscriptionPlanDetails: Record<
     ],
   },
 
-  [SubscriptionPlan.BUSINESS_MAX.key]: {
+  [Plan.BUSINESS_MAX]: {
     description: 'Designed for deep market analysis and large-scale data integration',
     features: [
       'Full historical data and metrics',
@@ -106,7 +87,7 @@ export const SubscriptionPlanDetails: Record<
     ],
   },
 
-  [SubscriptionPlan.CUSTOM.key]: {
+  [Plan.CUSTOM]: {
     description: 'Designed for high-demand professional trading and financial platforms',
     features: [
       'Custom development of data solutions',
@@ -118,15 +99,3 @@ export const SubscriptionPlanDetails: Record<
     ],
   },
 }
-
-export const CONSUMER_PLANS = new Set<string>([
-  SubscriptionPlan.FREE.key,
-  SubscriptionPlan.PRO.key,
-  SubscriptionPlan.MAX.key,
-])
-
-export const BUSINESS_PLANS = new Set<string>([
-  SubscriptionPlan.BUSINESS_MAX.key,
-  SubscriptionPlan.BUSINESS_PRO.key,
-  SubscriptionPlan.CUSTOM.key,
-])
diff --git a/src/lib/ui/app/SubscriptionPlan/subscription.ts b/src/lib/ui/app/SubscriptionPlan/subscription.ts
index 5d0cc9edb..baef74041 100644
--- a/src/lib/ui/app/SubscriptionPlan/subscription.ts
+++ b/src/lib/ui/app/SubscriptionPlan/subscription.ts
@@ -1,13 +1,14 @@
 import type { TSubscriptionPlan } from '$ui/app/SubscriptionPlan/types.js'
 
 import { calculateDaysTo } from '$lib/utils/dates/index.js'
-import { SubscriptionPlan } from '$ui/app/SubscriptionPlan/plans.js'
 import {
   checkIsBusinessPlan,
   checkIsSanApiProduct,
   checkIsSanbaseProduct,
-  getPlanName,
-} from '$ui/app/SubscriptionPlan/utils.js'
+  getPlanDisplayName,
+  isPlanEligibleFor,
+  Plan,
+} from '$lib/utils/plans/index.js'
 
 export enum Status {
   ACTIVE = 'ACTIVE',
@@ -60,7 +61,7 @@ export const getApiSubscription = (subscriptions: null | TSubscription[]) =>
 export function getPrimarySubscription(subscriptions: null | TSubscription[]) {
   const apiSubscription = getApiSubscription(subscriptions)
 
-  if (apiSubscription && checkIsBusinessPlan(apiSubscription.plan)) {
+  if (apiSubscription && checkIsBusinessPlan(apiSubscription.plan.name)) {
     return apiSubscription
   }
 
@@ -100,21 +101,21 @@ export function getCustomerSubscriptionData(subscription: null | TSubscription)
       currentPeriodEnd = Date.now(),
     } = subscription
 
-    const isBusiness = checkIsBusinessPlan(plan)
+    const isBusiness = checkIsBusinessPlan(plan.name)
     const planName = plan.name
     const trialDaysLeft = trialEnd ? calculateDaysTo(trialEnd) : null
 
-    const isCustom = planName === SubscriptionPlan.CUSTOM.key
-    const isBusinessMax = isBusiness && planName === SubscriptionPlan.BUSINESS_MAX.key
-    const isBusinessPro = isBusinessMax || planName === SubscriptionPlan.BUSINESS_PRO.key
-    const isMax = isBusiness || planName === SubscriptionPlan.MAX.key
-    const isProPlus = isBusiness || planName === SubscriptionPlan.PRO_PLUS.key
-    const isPro = isProPlus || isMax || planName === SubscriptionPlan.PRO.key
-    const isFree = !isPro && !isMax && !isBusinessPro && !isBusinessMax
+    const isCustom = isPlanEligibleFor(planName, Plan.CUSTOM)
+    const isBusinessMax = isPlanEligibleFor(planName, Plan.BUSINESS_MAX)
+    const isBusinessPro = isPlanEligibleFor(planName, Plan.BUSINESS_PRO)
+    const isMax = isPlanEligibleFor(planName, Plan.MAX)
+    const isProPlus = isPlanEligibleFor(planName, Plan.PRO_PLUS)
+    const isPro = isPlanEligibleFor(planName, Plan.PRO)
+    const isFree = !isPro
 
     return {
       plan,
-      planName: getPlanName(plan),
+      planName: getPlanDisplayName(plan.name),
 
       isBusinessMax,
       isBusinessPro,
diff --git a/src/lib/ui/app/SubscriptionPlan/utils.ts b/src/lib/ui/app/SubscriptionPlan/utils.ts
index ec2076a81..567b0dc70 100644
--- a/src/lib/ui/app/SubscriptionPlan/utils.ts
+++ b/src/lib/ui/app/SubscriptionPlan/utils.ts
@@ -1,19 +1,13 @@
-import type { TProduct, TSubscriptionPlan } from './types.js'
+import type { TSubscriptionPlan } from './types.js'
 
-import { BUSINESS_PLANS, Product, SubscriptionPlan, SubscriptionPlanDetails } from './plans.js'
+import {
+  checkIsPlan,
+  Plan,
+  checkIsBusinessPlan,
+  getPlanDisplayName,
+} from '$lib/utils/plans/index.js'
 
-export const checkIsSanbaseProduct = (product: Pick<TProduct, 'id'>) =>
-  product.id === Product.Sanbase.id
-
-export const checkIsSanApiProduct = (product: Pick<TProduct, 'id'>) =>
-  product.id === Product.SanAPI.id
-
-export const checkIsBusinessPlan = (plan: null | Pick<TSubscriptionPlan, 'name'>) =>
-  plan ? BUSINESS_PLANS.has(plan.name) : false
-
-type TLooseRecord<T extends Record<string, unknown>> = T & Record<string, undefined | T[keyof T]>
-export const getPlanName = (plan: Pick<TSubscriptionPlan, 'name'>): string =>
-  (SubscriptionPlan as TLooseRecord<typeof SubscriptionPlan>)[plan.name]?.name || plan.name
+import { SubscriptionPlanDetails } from './plans.js'
 
 export function getFormattedBillingPlan(plan: TSubscriptionPlan) {
   const { name, amount, price } = getFormattedPlan(plan)
@@ -29,13 +23,15 @@ export function getFormattedPlan(
   monthlyPlan: TSubscriptionPlan,
   annualPlan?: null | TSubscriptionPlan,
 ) {
-  const name = getPlanName(monthlyPlan)
+  const plan = checkIsPlan(monthlyPlan.name) ? monthlyPlan.name : null
+  const name = getPlanDisplayName(monthlyPlan.name)
   const details = SubscriptionPlanDetails[monthlyPlan.name]
 
   return {
-    isFree: monthlyPlan.name === SubscriptionPlan.FREE.key,
-    isCustom: monthlyPlan.name === SubscriptionPlan.CUSTOM.key,
-    isBusiness: BUSINESS_PLANS.has(monthlyPlan.name),
+    plan,
+    isFree: plan === Plan.FREE,
+    isCustom: plan === Plan.CUSTOM,
+    isBusiness: checkIsBusinessPlan(monthlyPlan.name),
 
     name,
     details,
@@ -62,8 +58,8 @@ export const checkIsCurrentPlan = (
 ) =>
   userPlan
     ? userPlan.name === targetPlan?.name &&
-      (userPlan.name === SubscriptionPlan.CUSTOM.key || userPlan.interval === targetPlan?.interval)
-    : targetPlan?.name === SubscriptionPlan.FREE.key
+      (userPlan.name === Plan.CUSTOM || userPlan.interval === targetPlan?.interval)
+    : targetPlan?.name === Plan.FREE
 
 export const checkIsAlternativeBillingPlan = (
   userPlan: null | TSubscriptionPlan,
diff --git a/src/lib/utils/index.ts b/src/lib/utils/index.ts
index 8e1fd2876..9f4d87421 100644
--- a/src/lib/utils/index.ts
+++ b/src/lib/utils/index.ts
@@ -3,7 +3,7 @@ import { getContext, setContext } from 'svelte'
 
 export type { TNominal } from './types/index.js'
 
-export { keyify, exactObjectKeys } from './object.js'
+export { keyify, exactObjectKeys, createEnumMap } from './object.js'
 
 export { useObserveFnCall, pipeGroupBy } from './observable.svelte.js'
 
diff --git a/src/lib/utils/object.ts b/src/lib/utils/object.ts
index d612f6327..8def3f27d 100644
--- a/src/lib/utils/object.ts
+++ b/src/lib/utils/object.ts
@@ -13,3 +13,7 @@ export function keyify<T, GKeyProp extends string = 'key'>(
 export function exactObjectKeys<T extends object>(obj: T) {
   return Object.keys(obj) as (keyof typeof obj)[]
 }
+
+export function createEnumMap<T extends string[]>(keys: T) {
+  return keys.reduce((res, key) => ({ ...res, [key]: key }), {} as { [K in T[number]]: K })
+}
diff --git a/src/lib/utils/plans/index.ts b/src/lib/utils/plans/index.ts
new file mode 100644
index 000000000..4ce016881
--- /dev/null
+++ b/src/lib/utils/plans/index.ts
@@ -0,0 +1,16 @@
+export {
+  Plan,
+  type TBusinessPlan,
+  type TConsumerPlan,
+  type TPlan,
+  getPlanData,
+  checkIsBusinessPlan,
+  checkIsConsumerPlan,
+  checkIsPlan,
+  isPlanEligibleFor,
+  getPlanDisplayName,
+} from './plans.js'
+
+export { Product, checkIsSanbaseProduct, checkIsSanApiProduct } from './products.js'
+
+export { extractPlanFromSubscriptions } from './subscription.js'
diff --git a/src/lib/utils/plans/plans.ts b/src/lib/utils/plans/plans.ts
new file mode 100644
index 000000000..5df63ee05
--- /dev/null
+++ b/src/lib/utils/plans/plans.ts
@@ -0,0 +1,60 @@
+import { createEnumMap, exactObjectKeys } from '../object.js'
+
+type PlanData = {
+  name: string
+  level: number
+}
+
+const CONSUMER_PLAN_MAP = {
+  FREE: { name: 'Free', level: 0 },
+  PRO: { name: 'Sanbase Pro', level: 10 },
+
+  /** @deprecated */
+  PRO_PLUS: {
+    name: 'Sanbase Pro+',
+    level: 15,
+  },
+
+  MAX: { name: 'Sanbase Max', level: 20 },
+} as const satisfies Record<string, PlanData>
+
+const BUSINESS_PLAN_MAP = {
+  BUSINESS_PRO: { name: 'Business Pro', level: 30 },
+  BUSINESS_MAX: { name: 'Business Max', level: 40 },
+  CUSTOM: { name: 'Enterprise', level: 50 },
+} as const satisfies Record<string, PlanData>
+
+const PLAN_MAP = { ...CONSUMER_PLAN_MAP, ...BUSINESS_PLAN_MAP }
+
+const BUSINESS_PLANS = new Set(exactObjectKeys(BUSINESS_PLAN_MAP))
+const CONSUMER_PLANS = new Set(exactObjectKeys(CONSUMER_PLAN_MAP))
+
+const PLANS = new Set([...CONSUMER_PLANS, ...BUSINESS_PLANS])
+
+export const Plan = createEnumMap(Array.from(PLANS))
+
+export type TBusinessPlan = keyof typeof BUSINESS_PLAN_MAP
+export type TConsumerPlan = keyof typeof CONSUMER_PLAN_MAP
+export type TPlan = keyof typeof PLAN_MAP
+
+export function getPlanData<T extends TPlan>(plan: T) {
+  return PLAN_MAP[plan]
+}
+
+export const checkIsBusinessPlan = (planName: string | undefined): planName is TBusinessPlan =>
+  planName ? BUSINESS_PLANS.has(planName as TBusinessPlan) : false
+
+export const checkIsConsumerPlan = (planName: string | undefined): planName is TConsumerPlan =>
+  planName ? CONSUMER_PLANS.has(planName as TConsumerPlan) : false
+
+export const checkIsPlan = (planName: string | undefined): planName is TPlan =>
+  planName ? checkIsBusinessPlan(planName) || checkIsConsumerPlan(planName) : false
+
+export function isPlanEligibleFor(planName: string, target: TPlan) {
+  if (!checkIsPlan(planName)) return false
+
+  return getPlanData(planName).level >= getPlanData(target).level
+}
+
+export const getPlanDisplayName = (planName: string) =>
+  checkIsPlan(planName) ? getPlanData(planName).name : planName
diff --git a/src/lib/utils/plans/products.ts b/src/lib/utils/plans/products.ts
new file mode 100644
index 000000000..f11d97e24
--- /dev/null
+++ b/src/lib/utils/plans/products.ts
@@ -0,0 +1,20 @@
+import { keyify } from '../object.js'
+
+export const Product = keyify(
+  {
+    SANAPI: { id: '1' },
+    SANBASE: { id: '2' },
+  } as const,
+  'name',
+)
+
+type TProductCheckProps = {
+  id?: string
+  name?: string
+}
+
+export const checkIsSanbaseProduct = ({ id, name }: TProductCheckProps) =>
+  id === Product.SANBASE.id || name === Product.SANBASE.name
+
+export const checkIsSanApiProduct = ({ id, name }: TProductCheckProps) =>
+  id === Product.SANAPI.id || name === Product.SANAPI.name
diff --git a/src/lib/utils/plans/subscription.ts b/src/lib/utils/plans/subscription.ts
new file mode 100644
index 000000000..169fc92ac
--- /dev/null
+++ b/src/lib/utils/plans/subscription.ts
@@ -0,0 +1,32 @@
+import { checkIsBusinessPlan, checkIsConsumerPlan, getPlanData, Plan, type TPlan } from './plans.js'
+import { checkIsSanApiProduct, checkIsSanbaseProduct } from './products.js'
+
+type Subscription = { planName: string; productName?: string; productId?: string }
+
+function getCheckerForProduct({ productId, productName }: Subscription) {
+  if (checkIsSanApiProduct({ id: productId, name: productName })) {
+    return checkIsBusinessPlan
+  }
+
+  if (checkIsSanbaseProduct({ id: productId, name: productName })) {
+    return checkIsConsumerPlan
+  }
+}
+
+function getPlanSubscription(sub: Subscription) {
+  const checker = getCheckerForProduct(sub)
+
+  if (!checker) return null
+
+  return checker(sub.planName) ? sub.planName : null
+}
+
+export function extractPlanFromSubscriptions(subscriptions: Subscription[]): TPlan {
+  const plans = subscriptions.map((sub) => getPlanSubscription(sub)).filter((plan) => !!plan)
+
+  if (plans.length === 0) return Plan.FREE
+
+  return plans.reduce((max, plan) =>
+    getPlanData(plan).level > getPlanData(max).level ? plan : max,
+  )
+}