From 8a69bf00a07a5d7aa89f607d08cd5dc84d345f00 Mon Sep 17 00:00:00 2001
From: ernstmul <ernst.mulders@gmail.com>
Date: Wed, 20 Nov 2024 15:09:44 +0100
Subject: [PATCH] Fix missing limits data for non-standard tiers

---
 src/lib/stores/billing.ts                     |  6 ++-
 .../usage/[[invoice]]/+page.svelte            | 51 ++++---------------
 .../usage/[[invoice]]/+page.ts                |  6 ++-
 3 files changed, 18 insertions(+), 45 deletions(-)

diff --git a/src/lib/stores/billing.ts b/src/lib/stores/billing.ts
index e6688ea06..c7ab3752a 100644
--- a/src/lib/stores/billing.ts
+++ b/src/lib/stores/billing.ts
@@ -130,12 +130,14 @@ export type PlanServices =
     | 'usersAddon'
     | 'webhooks';
 
-export function getServiceLimit(serviceId: PlanServices, tier: Tier = null): number {
+export function getServiceLimit(serviceId: PlanServices, tier: Tier = null, plan?: Plan): number {
     if (!isCloud) return 0;
     if (!serviceId) return 0;
     const info = get(plansInfo);
     if (!info) return 0;
-    const plan = info.get(tier ?? get(organization)?.billingPlan);
+    if (!plan) {
+        plan = info.get(tier ?? get(organization)?.billingPlan);
+    }
     return plan?.[serviceId];
 }
 
diff --git a/src/routes/(console)/organization-[organization]/usage/[[invoice]]/+page.svelte b/src/routes/(console)/organization-[organization]/usage/[[invoice]]/+page.svelte
index 8dddbb419..82c211c19 100644
--- a/src/routes/(console)/organization-[organization]/usage/[[invoice]]/+page.svelte
+++ b/src/routes/(console)/organization-[organization]/usage/[[invoice]]/+page.svelte
@@ -4,7 +4,7 @@
     import {
         getServiceLimit,
         showUsageRatesModal,
-        tierToPlan,
+        type Tier,
         upgradeURL
     } from '$lib/stores/billing';
     import { organization } from '$lib/stores/organization';
@@ -21,23 +21,10 @@
 
     export let data;
 
-    const tier = data?.currentInvoice?.plan ?? $organization?.billingPlan;
-    const plan = tierToPlan(tier).name;
-
-    // let invoice = null;
-    // async function handlePeriodChange() {
-    //     const target = invoice
-    //         ? `/console/organization-${$organization.$id}/usage/${invoice}`
-    //         : `/console/organization-${$organization.$id}/usage`;
-    //     if ($page.url.pathname !== target) {
-    //         await goto(target);
-    //     }
-    // }
-
-    // const cycles = data.invoices.invoices.map((invoice) => ({
-    //     label: toLocaleDate(invoice.from),
-    //     value: invoice.$id
-    // }));
+    const tier = data?.plan
+        ? (data.plan.$id as Tier)
+        : (data?.currentInvoice?.plan ?? $organization?.billingPlan);
+    const plan = data?.plan ?? undefined;
 
     $: project = (data.organizationUsage as OrganizationUsage).projects;
 </script>
@@ -78,29 +65,11 @@
             </p>
         {:else if $organization.billingPlan === BillingPlan.FREE}
             <p class="text">
-                If you exceed the limits of the {plan} plan, services for your organization's projects
+                If you exceed the limits of the Free plan, services for your organization's projects
                 may be disrupted.
                 <a href={$upgradeURL} class="link">Upgrade for greater capacity</a>.
             </p>
         {/if}
-
-        <!--<div class="u-flex u-gap-8 u-cross-center">
-            <p class="text">Usage period:</p>
-            <InputSelect
-                wrapperTag="div"
-                id="period"
-                label="Usage period"
-                showLabel={false}
-                bind:value={invoice}
-                on:change={handlePeriodChange}
-                options={[
-                    {
-                        label: 'Current billing cycle',
-                        value: null
-                    },
-                    ...cycles
-                ]} />
-                </div>-->
     </div>
 
     <CardGrid>
@@ -115,7 +84,7 @@
             {#if data.organizationUsage.bandwidth}
                 {@const current = total(data.organizationUsage.bandwidth)}
                 {@const currentHumanized = humanFileSize(current)}
-                {@const max = getServiceLimit('bandwidth', tier)}
+                {@const max = getServiceLimit('bandwidth', tier, plan)}
                 <ProgressBarBig
                     currentUnit={currentHumanized.unit}
                     currentValue={currentHumanized.value}
@@ -178,7 +147,7 @@
         <svelte:fragment slot="aside">
             {#if data.organizationUsage.users}
                 {@const current = data.organizationUsage.usersTotal}
-                {@const max = getServiceLimit('users', tier)}
+                {@const max = getServiceLimit('users', tier, plan)}
                 <ProgressBarBig
                     currentUnit="Users"
                     currentValue={formatNum(current)}
@@ -233,7 +202,7 @@
         <svelte:fragment slot="aside">
             {#if data.organizationUsage.executionsTotal}
                 {@const current = data.organizationUsage.executionsTotal}
-                {@const max = getServiceLimit('executions', tier)}
+                {@const max = getServiceLimit('executions', tier, plan)}
                 <ProgressBarBig
                     currentUnit="Executions"
                     currentValue={formatNum(current)}
@@ -290,7 +259,7 @@
             {#if data.organizationUsage.storageTotal}
                 {@const current = data.organizationUsage.storageTotal}
                 {@const currentHumanized = humanFileSize(current)}
-                {@const max = getServiceLimit('storage', tier)}
+                {@const max = getServiceLimit('storage', tier, plan)}
                 {@const progressBarStorageDate = [
                     {
                         size: bytesToSize(data.organizationUsage.filesStorageTotal, 'GB'),
diff --git a/src/routes/(console)/organization-[organization]/usage/[[invoice]]/+page.ts b/src/routes/(console)/organization-[organization]/usage/[[invoice]]/+page.ts
index f469f97a8..1e6f2cfae 100644
--- a/src/routes/(console)/organization-[organization]/usage/[[invoice]]/+page.ts
+++ b/src/routes/(console)/organization-[organization]/usage/[[invoice]]/+page.ts
@@ -1,7 +1,7 @@
 import { sdk } from '$lib/stores/sdk';
 import { Query, type Models } from '@appwrite.io/console';
 import type { PageLoad } from './$types';
-import type { Organization } from '$lib/stores/organization';
+import { type Organization } from '$lib/stores/organization';
 import type { Invoice } from '$lib/sdk/billing';
 
 export const load: PageLoad = async ({ params, parent }) => {
@@ -41,6 +41,7 @@ export const load: PageLoad = async ({ params, parent }) => {
         startDate = currentInvoice.from;
         endDate = currentInvoice.to;
     }
+    const plan = await sdk.forConsole.billing.getPlan(org.$id);
 
     const [invoices, usage, organizationMembers] = await Promise.all([
         sdk.forConsole.billing.listInvoices(org.$id, [Query.orderDesc('from')]),
@@ -77,6 +78,7 @@ export const load: PageLoad = async ({ params, parent }) => {
         projectNames,
         invoices,
         currentInvoice,
-        organizationMembers
+        organizationMembers,
+        plan
     };
 };