billing: promo codes

packages/api/src/controllers/stripe.ts

@@ -1,4 +1,4 @@
-import { Router, Request } from "express";
+import { Router, Request, Response } from "express";
 import { db } from "../store";
 import { products } from "../config";
 import sql from "sql-template-strings";
@@ -186,6 +186,67 @@ const sendUsageRecordToStripe = async (
   );
 };
 
+app.post("/apply-coupon", async (req: Request, res: Response) => {
+  const { coupon, stripeCustomerId, stripeCustomerSubscriptionId } = req.body;
+
+  const user = req.user;
+
+  if (
+    user.stripeCustomerId != stripeCustomerId ||
+    user.stripeCustomerSubscriptionId != stripeCustomerSubscriptionId
+  ) {
+    res.status(400);
+    return res.json({ errors: ["invalid customer or subscription"] });
+  }
+
+  const customer = await req.stripe.customers.retrieve(stripeCustomerId);
+
+  if (!customer) {
+    res.status(404);
+    return res.json({ errors: ["customer not found"] });
+  }
+
+  const subscription = await req.stripe.subscriptions.retrieve(
+    stripeCustomerSubscriptionId
+  );
+
+  if (!subscription) {
+    res.status(404);
+    return res.json({ errors: ["subscription not found"] });
+  }
+
+  if (subscription.status === "canceled") {
+    res.status(400);
+    return res.json({ errors: ["subscription is canceled"] });
+  }
+
+  const couponRes = await req.stripe.coupons.retrieve(coupon);
+
+  if (!couponRes) {
+    res.status(404);
+    return res.json({ errors: ["coupon not found"] });
+  }
+
+  const subscriptionWithCoupon = await req.stripe.subscriptions.update(
+    stripeCustomerSubscriptionId,
+    {
+      coupon: coupon,
+    }
+  );
+
+  if (!subscriptionWithCoupon) {
+    res.status(400);
+    return res.json({ errors: ["failed to apply coupon"] });
+  }
+
+  await db.user.update(user.id, {
+    stripeAppliedCouponId: coupon,
+  });
+
+  res.status(200);
+  return res.json(user);
+});
+
 // Webhook handler for asynchronous events called by stripe on invoice generation
 // https://stripe.com/docs/billing/subscriptions/webhooks
 app.post("/webhook", async (req, res) => {

packages/api/src/controllers/user.ts

@@ -438,7 +438,9 @@ app.post("/", validatePost("user"), async (req, res) => {
   }
 
   let isTest =
-    process.env.NODE_ENV === "test" || process.env.NODE_ENV === "development";
+    process.env.NODE_ENV === "test" ||
+    process.env.NODE_ENV === "development" ||
+    process.env.NODE_ENV === "staging";
 
   if (
     req.config.requireEmailVerification &&
@@ -617,7 +619,9 @@ app.post("/token", validatePost("user"), async (req, res) => {
   );
 
   let isTest =
-    process.env.NODE_ENV === "test" || process.env.NODE_ENV === "development";
+    process.env.NODE_ENV === "test" ||
+    process.env.NODE_ENV === "development" ||
+    process.env.NODE_ENV === "staging";
 
   if (
     req.config.requireEmailVerification &&

packages/api/src/schema/db-schema.yaml

@@ -290,6 +290,9 @@ components:
             - prod_O9XuWMU1Up6QKf
             - prod_OTTbwpzxNLMNSh
             - prod_OTTwpzjA4U8B2P
+        stripeAppliedCouponId:
+          type: string
+          description: stripe coupon id
     update-customer-payment-method:
       type: object
       required:
@@ -1122,6 +1125,9 @@ components:
         stripeCustomerSubscriptionId:
           type: string
           example: sub_I29pdyfOTPBkjb
+        stripeAppliedCouponId:
+          type: string
+          description: stripe coupon id
         ccLast4:
           type: string
           example: 1234

packages/www/components/PaymentMethodDialog/index.tsx

@@ -258,7 +258,6 @@ const PaymentMethodDialog = ({ invalidateQuery }) => {
                   />
                 </Box>
               </Grid>
-
               <Box
                 css={{
                   fontSize: "$1",

packages/www/components/PlanForm/index.tsx

@@ -32,6 +32,7 @@ const PlanForm = ({
   color,
 }) => {
   const { user, updateSubscription } = useApi();
+  const { applyCoupon } = useApi();
   const [status, setStatus] = useState("initial");
   const stripe = useStripe();
   const elements = useElements();
@@ -46,6 +47,7 @@ const PlanForm = ({
 
   function createPaymentMethod({
     cardElement,
+    coupon,
     stripeCustomerId,
     stripeCustomerSubscriptionId,
     stripeProductId,
@@ -63,6 +65,14 @@ const PlanForm = ({
           console.log(result.error);
           setStatus("error");
         } else {
+          if (coupon && coupon !== "") {
+            applyCoupon({
+              stripeCustomerId,
+              stripeCustomerSubscriptionId,
+              stripeProductId,
+              coupon,
+            });
+          }
           updateSubscription({
             stripeCustomerId,
             stripeCustomerPaymentMethodId: paymentMethod.id,
@@ -171,6 +181,7 @@ const PlanForm = ({
         cardElement,
         stripeCustomerId: user.stripeCustomerId,
         stripeCustomerSubscriptionId: user.stripeCustomerSubscriptionId,
+        coupon: data.coupon,
         stripeProductId,
         billingDetails: {
           name: data.name,
@@ -376,7 +387,23 @@ const PlanForm = ({
                       />
                     </Box>
                   </Grid>
-
+                  <Box>
+                    <Label
+                      css={{ mb: "$1", display: "block" }}
+                      htmlFor="coupon">
+                      Promo code
+                    </Label>
+                    <TextField
+                      size="2"
+                      ref={register({ required: false })}
+                      placeholder="Promo code"
+                      id="coupon"
+                      name="coupon"
+                      type="text"
+                      css={{ width: "100%", mb: "$2" }}
+                      required
+                    />
+                  </Box>
                   <Box
                     css={{
                       fontSize: "$1",

packages/www/hooks/use-api/endpoints/user.ts

@@ -373,6 +373,33 @@ export const updateSubscription = async ({
   return res;
 };
 
+export const applyCoupon = async ({
+  coupon,
+  stripeCustomerId,
+  stripeCustomerSubscriptionId,
+  stripeProductId,
+}): Promise<[Response, User | ApiError]> => {
+  const [res, body] = await context.fetch("/stripe/apply-coupon", {
+    method: "POST",
+    body: JSON.stringify({
+      stripeCustomerId,
+      stripeCustomerSubscriptionId,
+      stripeProductId,
+      coupon,
+    }),
+    headers: {
+      "content-type": "application/json",
+    },
+  });
+  setState((state) => ({ ...state, userRefresh: Date.now() }));
+
+  if (res.status !== 201) {
+    return body;
+  }
+
+  return res;
+};
+
 export const getSubscription = async (
   stripeCustomerSubscriptionId: string
 ): Promise<[Response, ApiError]> => {