add button that opens checkout or billing portal

devtodollars · Mar 1, 2024 · a2fc6c3 · a2fc6c3
1 parent dcfb9a4
commit a2fc6c3
Show file tree

Hide file tree

Showing 11 changed files with 176 additions and 2 deletions.
diff --git a/flutter/env.local.json b/flutter/env.local.json
@@ -1,4 +1,6 @@
 {
   "SUPABASE_URL": "http://localhost:54321",
-  "SUPABASE_ANON_KEY": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZS1kZW1vIiwicm9sZSI6ImFub24iLCJleHAiOjE5ODM4MTI5OTZ9.CRXP1A7WOeoJeXxjNni43kdQwgnWNReilDMblYTn_I0"
+  "SUPABASE_ANON_KEY": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZS1kZW1vIiwicm9sZSI6ImFub24iLCJleHAiOjE5ODM4MTI5OTZ9.CRXP1A7WOeoJeXxjNni43kdQwgnWNReilDMblYTn_I0",
+  "STRIPE_PREMIUM_PRICE": "price_1OovJ2FttF99a1NC1hbOKGdg",
+  "STRIPE_BASIC_PRICE": "price_1OovHjFttF99a1NCrlGTo2ue"
 }
diff --git a/flutter/lib/screens/home_screen.dart b/flutter/lib/screens/home_screen.dart
@@ -3,6 +3,7 @@ import 'package:flutter_riverpod/flutter_riverpod.dart';
 import 'package:supabase_flutter/supabase_flutter.dart';
 import 'package:test/components/reset_password_dialog.dart';
 import 'package:test/services/auth_notifier.dart';
+import 'package:url_launcher/url_launcher.dart';
 
 class HomeScreen extends ConsumerStatefulWidget {
   const HomeScreen({super.key, required this.title});
@@ -15,13 +16,22 @@ class HomeScreen extends ConsumerStatefulWidget {
 
 class _HomeScreenState extends ConsumerState<HomeScreen> {
   int _counter = 0;
+  bool loading = false;
 
   void _incrementCounter() {
     setState(() {
       _counter++;
     });
   }
 
+  void goToPayments() async {
+    setState(() => loading = true);
+    final authNotif = ref.read(authProvider.notifier);
+    final url = await authNotif.getUserStripeLink();
+    if (url != null) launchUrl(url);
+    setState(() => loading = false);
+  }
+
   @override
   Widget build(BuildContext context) {
     ref.listen(authProvider, (previous, next) {
@@ -41,6 +51,10 @@ class _HomeScreenState extends ConsumerState<HomeScreen> {
         backgroundColor: Theme.of(context).colorScheme.inversePrimary,
         title: Text(widget.title),
         actions: [
+          TextButton(
+            onPressed: (loading) ? null : goToPayments,
+            child: const Text("Payments"),
+          ),
           TextButton(onPressed: authNotif.signOut, child: const Text("Logout")),
         ],
       ),

diff --git a/flutter/lib/services/auth_notifier.dart b/flutter/lib/services/auth_notifier.dart
@@ -68,6 +68,25 @@ class Auth extends _$Auth {
     );
   }
 
+  Future<Uri?> getUserStripeLink() async {
+    Uri baseUri = Uri.base;
+    String baseUrl = Uri(
+            scheme: baseUri.scheme,
+            host: baseUri.host,
+            port: baseUri.hasPort ? baseUri.port : null)
+        .toString();
+
+    var res = await client.functions.invoke("get_stripe_url", body: {
+      "return_url": baseUrl,
+      "price": const String.fromEnvironment("STRIPE_PREMIUM_PRICE"),
+    });
+    String? redirectUrl = res.data["redirect_url"];
+    if (redirectUrl is String) {
+      return Uri.parse(redirectUrl);
+    }
+    return null;
+  }
+
   Future<void> signOut() async {
     await client.auth.signOut();
   }

diff --git a/flutter/pubspec.lock b/flutter/pubspec.lock
@@ -870,7 +870,7 @@ packages:
     source: hosted
     version: "1.3.2"
   url_launcher:
-    dependency: transitive
+    dependency: "direct main"
     description:
       name: url_launcher
       sha256: "0ecc004c62fd3ed36a2ffcbe0dd9700aee63bd7532d0b642a488b1ec310f492e"

diff --git a/flutter/pubspec.yaml b/flutter/pubspec.yaml
@@ -41,6 +41,7 @@ dependencies:
   supabase_flutter: ^2.3.4
   flutter_riverpod: ^2.4.10
   riverpod_annotation: ^2.3.4
+  url_launcher: ^6.2.5
 
 dev_dependencies:
   flutter_test:

diff --git a/supabase/.gitignore b/supabase/.gitignore
@@ -2,3 +2,4 @@
 .branches
 .temp
 .env
+.env.local
diff --git a/supabase/functions/_shared/request.ts b/supabase/functions/_shared/request.ts
@@ -0,0 +1,48 @@
+import { User } from "supabase";
+import { supabase } from "./supabase.ts";
+
+async function getUserFromRequest(req: Request) {
+  const token = req.headers.get("Authorization")!.replace("Bearer ", "");
+  const { data, error } = await supabase.auth.getUser(token);
+  if (error) throw error;
+  const user = data?.user;
+  return user;
+}
+
+// handles cors and returns the user if function called from client
+export const corsHeaders = {
+  "Access-Control-Allow-Origin": "*",
+  "Access-Control-Allow-Headers":
+    "authorization, x-client-info, apikey, content-type",
+};
+
+export function resFromError(e: { message?: string }): Response {
+  return new Response(
+    JSON.stringify({ message: e?.message }),
+    {
+      status: 400,
+      headers: { "Content-Type": "application/json", ...corsHeaders },
+    },
+  );
+}
+
+export function clientRequestHandler(
+  handler: (req: Request, user: User) => Promise<Response> | Response,
+) {
+  const enhancedHandler = async (req: Request) => {
+    // Handle CORS preflight requests
+    if (req.method === "OPTIONS") {
+      return new Response("ok", { headers: corsHeaders });
+    }
+
+    try {
+      const user = await getUserFromRequest(req);
+      if (!user) throw Error("Authentication Error");
+      return await handler(req, user);
+    } catch (e) {
+      console.error(e);
+      return resFromError(e);
+    }
+  };
+  Deno.serve(enhancedHandler);
+}
diff --git a/supabase/functions/_shared/stripe.ts b/supabase/functions/_shared/stripe.ts
@@ -0,0 +1,6 @@
+import Stripe from "stripe";
+
+export const stripe = Stripe(Deno.env.get("STRIPE_SK")!, {
+  httpClient: Stripe.createFetchHttpClient(),
+  apiVersion: "2022-08-01",
+});
diff --git a/supabase/functions/_shared/supabase.ts b/supabase/functions/_shared/supabase.ts
@@ -0,0 +1,11 @@
+import { createClient } from "supabase";
+
+export const supabase = createClient(
+  Deno.env.get("SUPABASE_URL")!,
+  Deno.env.get("SUPABASE_SERVICE_ROLE_KEY")!,
+  {
+    auth: {
+      persistSession: false,
+    },
+  },
+);
diff --git a/supabase/functions/get_stripe_url/index.ts b/supabase/functions/get_stripe_url/index.ts
@@ -0,0 +1,66 @@
+import { clientRequestHandler, corsHeaders } from "../_shared/request.ts";
+import { supabase } from "../_shared/supabase.ts";
+import { stripe } from "../_shared/stripe.ts";
+
+clientRequestHandler(async (req, user) => {
+  const { price, return_url } = await req.json();
+  // get stripe information from user_metadata table!
+  const { data: metadata } = await supabase.from("user_metadata").select(
+    "stripe_customer_id,tier",
+  ).eq(
+    "user_id",
+    user.id,
+  ).maybeSingle();
+  let stripeCustomerId = metadata?.stripe_customer_id;
+  const tier = metadata?.tier;
+  if (!stripeCustomerId) {
+    // create stripe customer if doesn't exist
+    console.log("create stripe customer");
+    const customer = await stripe.customers.create({
+      phone: user.phone,
+      email: user.email,
+      metadata: {
+        uid: user.id,
+      },
+    });
+    stripeCustomerId = customer.id;
+    await supabase.from("user_metadata").upsert({
+      user_id: user.id,
+      stripe_customer_id: customer.id,
+    });
+  }
+  let redirect_url: string | undefined;
+  if (tier) {
+    // open billing portal
+    console.log("open billing portal session");
+    const session = await stripe.billingPortal.sessions.create({
+      customer: stripeCustomerId,
+      return_url,
+    });
+    redirect_url = session?.url;
+  } else {
+    // if tier is null go to checkout page
+    console.log("open checkout session");
+    const session = await stripe.checkout.sessions.create({
+      customer: stripeCustomerId,
+      mode: "payment",
+      line_items: [{
+        price,
+        quantity: 1,
+      }],
+      allow_promotion_codes: true,
+      success_url: return_url,
+      cancel_url: return_url,
+    });
+    redirect_url = session?.url;
+  }
+  return new Response(
+    JSON.stringify({ redirect_url }),
+    {
+      headers: {
+        ...corsHeaders,
+        "Content-Type": "application/json",
+      },
+    },
+  );
+});
diff --git a/supabase/functions/import_map.json b/supabase/functions/import_map.json
@@ -0,0 +1,6 @@
+{
+  "imports": {
+    "supabase": "https://esm.sh/@supabase/supabase-js@2.39.7",
+    "stripe": "https://esm.sh/stripe@10.12.0?target=deno"
+  }
+}