From 9ceb27f6e31932447ee061ed8979353ea7b03a58 Mon Sep 17 00:00:00 2001
From: Luke Vella <me@lukevella.com>
Date: Fri, 24 Nov 2023 22:25:47 +0700
Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20Refactor=20logic=20for=20s?=
 =?UTF-8?q?howing=20upgrade=20button?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/web/src/components/layouts/standard-layout.tsx | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/apps/web/src/components/layouts/standard-layout.tsx b/apps/web/src/components/layouts/standard-layout.tsx
index b9dbd1e47..3063625ba 100644
--- a/apps/web/src/components/layouts/standard-layout.tsx
+++ b/apps/web/src/components/layouts/standard-layout.tsx
@@ -21,7 +21,7 @@ import { Logo } from "@/components/logo";
 import { Trans } from "@/components/trans";
 import { UserDropdown } from "@/components/user-dropdown";
 import { IfCloudHosted } from "@/contexts/environment";
-import { IfFreeUser } from "@/contexts/plan";
+import { useSubscription } from "@/contexts/plan";
 import { appVersion, isFeedbackEnabled } from "@/utils/constants";
 
 import { IconComponent, NextPageWithLayout } from "../../types";
@@ -97,6 +97,7 @@ const Changelog = () => {
 };
 
 const MainNav = () => {
+  const subscription = useSubscription();
   return (
     <m.div
       variants={{
@@ -122,9 +123,7 @@ const MainNav = () => {
         <div className="flex items-center gap-x-2.5">
           <nav className="flex items-center gap-x-1 sm:gap-x-1.5">
             <IfCloudHosted>
-              <IfFreeUser>
-                <Upgrade />
-              </IfFreeUser>
+              {subscription?.active === false ? <Upgrade /> : null}
             </IfCloudHosted>
             <IfGuest>
               <Button