add manage subscription button

brendan-kellam · brendan-kellam · commit 97bb79316caa · 2026-04-13T18:45:05.000-07:00
diff --git a/packages/web/src/app/(app)/settings/license/manageSubscriptionButton.tsx b/packages/web/src/app/(app)/settings/license/manageSubscriptionButton.tsx
@@ -0,0 +1,43 @@
+"use client";
+
+import { useState, useCallback } from "react";
+import { LoadingButton } from "@/components/ui/loading-button";
+import { createPortalSession } from "@/ee/features/lighthouse/actions";
+import { isServiceError } from "@/lib/utils";
+import { useToast } from "@/components/hooks/use-toast";
+import { useRouter } from "next/navigation";
+
+export function ManageSubscriptionButton() {
+    const [isLoading, setIsLoading] = useState(false);
+    const { toast } = useToast();
+    const router = useRouter();
+
+    const handleClick = useCallback(() => {
+        setIsLoading(true);
+
+        const returnUrl = `${window.location.origin}/settings/license`;
+
+        createPortalSession(returnUrl)
+            .then((response) => {
+                if (isServiceError(response)) {
+                    toast({
+                        description: `Failed to open subscription portal: ${response.message}`,
+                        variant: "destructive",
+                    });
+                    setIsLoading(false);
+                } else {
+                    router.push(response.url);
+                }
+            });
+    }, [router, toast]);
+
+    return (
+        <LoadingButton
+            variant="outline"
+            onClick={handleClick}
+            loading={isLoading}
+        >
+            Manage subscription
+        </LoadingButton>
+    );
+}
diff --git a/packages/web/src/app/(app)/settings/license/page.tsx b/packages/web/src/app/(app)/settings/license/page.tsx
@@ -2,6 +2,7 @@ import { authenticatedPage } from "@/middleware/authenticatedPage";
 import { OrgRole } from "@sourcebot/db";
 import { ActivationCodeCard } from "./activationCodeCard";
 import { PurchaseButton } from "./purchaseButton";
+import { ManageSubscriptionButton } from "./manageSubscriptionButton";
 import { BasicSettingsCard } from "../components/settingsCard";
 import { getPlan } from "@/lib/entitlements";
 
@@ -25,7 +26,10 @@ export default authenticatedPage(async ({ prisma, org }) => {
                 <span className="text-sm font-medium">{plan}</span>
             </BasicSettingsCard>
             <ActivationCodeCard isActivated={!!license} />
-            <PurchaseButton />
+            <div className="flex gap-3">
+                <PurchaseButton />
+                {license && <ManageSubscriptionButton />}
+            </div>
         </div>
     );
 }, {
diff --git a/packages/web/src/ee/features/lighthouse/actions.ts b/packages/web/src/ee/features/lighthouse/actions.ts
@@ -7,10 +7,10 @@ import { OrgRole } from "@sourcebot/db";
 import { ServiceError } from "@/lib/serviceError";
 import { StatusCodes } from "http-status-codes";
 import { ErrorCode } from "@/lib/errorCodes";
-import { env, encryptActivationCode } from "@sourcebot/shared";
+import { env, encryptActivationCode, decryptActivationCode } from "@sourcebot/shared";
 import { sendServicePing } from "@/ee/features/lighthouse/servicePing";
 import { fetchWithRetry } from "@/lib/utils";
-import { checkoutResponseSchema } from "./types";
+import { checkoutResponseSchema, portalResponseSchema } from "./types";
 
 export const activateLicense = async (activationCode: string): Promise<{ success: boolean } | ServiceError> => sew(() =>
     withAuth(async ({ org, role, prisma }) =>
@@ -124,3 +124,53 @@ export const deactivateLicense = async (): Promise<{ success: boolean } | Servic
         })
     )
 );
+
+export const createPortalSession = async (returnUrl: string): Promise<{ url: string } | ServiceError> => sew(() =>
+    withAuth(async ({ org, role, prisma }) =>
+        withMinimumOrgRole(role, OrgRole.OWNER, async () => {
+            const license = await prisma.license.findUnique({
+                where: { orgId: org.id },
+            });
+
+            if (!license) {
+                return {
+                    statusCode: StatusCodes.NOT_FOUND,
+                    errorCode: ErrorCode.NOT_FOUND,
+                    message: "No license found.",
+                } satisfies ServiceError;
+            }
+
+            const activationCode = decryptActivationCode(license.activationCode);
+
+            const response = await fetchWithRetry(`${env.SOURCEBOT_LIGHTHOUSE_URL}/portal`, {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({
+                    activationCode,
+                    returnUrl,
+                }),
+            });
+
+            if (!response.ok) {
+                return {
+                    statusCode: StatusCodes.BAD_GATEWAY,
+                    errorCode: ErrorCode.UNEXPECTED_ERROR,
+                    message: "Failed to create portal session.",
+                } satisfies ServiceError;
+            }
+
+            const body = await response.json();
+            const result = portalResponseSchema.safeParse(body);
+
+            if (!result.success) {
+                return {
+                    statusCode: StatusCodes.BAD_GATEWAY,
+                    errorCode: ErrorCode.UNEXPECTED_ERROR,
+                    message: "Invalid response from Lighthouse.",
+                } satisfies ServiceError;
+            }
+
+            return { url: result.data.url };
+        })
+    )
+);
diff --git a/packages/web/src/ee/features/lighthouse/types.ts b/packages/web/src/ee/features/lighthouse/types.ts
@@ -9,3 +9,7 @@ export const lighthouseResponseSchema = z.object({
 export const checkoutResponseSchema = z.object({
     url: z.string(),
 });
+
+export const portalResponseSchema = z.object({
+    url: z.string(),
+});
