From fe1fde109cefa140f53eabb08a65bfe5482d9b38 Mon Sep 17 00:00:00 2001
From: phamhieu <phamhieu1998@gmail.com>
Date: Fri, 31 Dec 2021 10:05:35 +0700
Subject: [PATCH] feat: allow user to select pricing plan on project creation

---
 studio/lib/constants/infrastructure.ts |  8 +++++++
 studio/pages/new/[slug].tsx            | 31 ++++++++++++++++++++++++--
 2 files changed, 37 insertions(+), 2 deletions(-)

diff --git a/studio/lib/constants/infrastructure.ts b/studio/lib/constants/infrastructure.ts
index 137e979afd5..00cd4eb3dc8 100644
--- a/studio/lib/constants/infrastructure.ts
+++ b/studio/lib/constants/infrastructure.ts
@@ -20,6 +20,14 @@ export const REGIONS = {
 
 export const REGIONS_DEFAULT = REGIONS.EAST_US
 
+export const PRICING_PLANS = {
+  FREE: 'Free tier',
+  PRO: 'Pro tier',
+  PAYG: 'Pay as you go',
+}
+
+export const PRICING_PLANS_DEFAULT = PRICING_PLANS.PRO
+
 export const PROVIDERS = {
   AWS: {
     id: 'AWS',
diff --git a/studio/pages/new/[slug].tsx b/studio/pages/new/[slug].tsx
index 43c762a46a6..2306474d36a 100644
--- a/studio/pages/new/[slug].tsx
+++ b/studio/pages/new/[slug].tsx
@@ -20,6 +20,8 @@ import {
   PASSWORD_STRENGTH,
   PASSWORD_STRENGTH_COLOR,
   PASSWORD_STRENGTH_PERCENTAGE,
+  PRICING_PLANS,
+  PRICING_PLANS_DEFAULT,
 } from 'lib/constants'
 
 import { useStore, withAuth } from 'hooks'
@@ -75,6 +77,7 @@ export const Wizard = () => {
   const [projectName, setProjectName] = useState('')
   const [dbPass, setDbPass] = useState('')
   const [dbRegion, setDbRegion] = useState(REGIONS_DEFAULT)
+  const [dbPricingPlan, setDbPricingPlan] = useState(PRICING_PLANS_DEFAULT)
   const [newProjectedLoading, setNewProjectLoading] = useState(false)
   const [passwordStrengthMessage, setPasswordStrengthMessage] = useState('')
   const [passwordStrengthWarning, setPasswordStrengthWarning] = useState('')
@@ -83,7 +86,8 @@ export const Wizard = () => {
   const canSubmit =
     projectName != '' &&
     passwordStrengthScore >= DEFAULT_MINIMUM_PASSWORD_STRENGTH &&
-    dbRegion != ''
+    dbRegion != '' &&
+    dbPricingPlan != ''
   const passwordErrorMessage =
     dbPass != '' && passwordStrengthScore < DEFAULT_MINIMUM_PASSWORD_STRENGTH
       ? 'You need a stronger password'
@@ -139,6 +143,10 @@ export const Wizard = () => {
     setDbRegion(e.target.value)
   }
 
+  function onDbPricingPlanChange(e: any) {
+    setDbPricingPlan(e.target.value)
+  }
+
   async function checkPasswordStrength(value: any) {
     let passwordStrength = ''
     if (value && value !== '') {
@@ -166,6 +174,7 @@ export const Wizard = () => {
       name: projectName,
       db_pass: dbPass,
       db_region: dbRegion,
+      db_pricing_plan: dbPricingPlan,
     }
     const response = await post(`${API_URL}/projects`, data)
     if (response.error) {
@@ -333,7 +342,7 @@ export const Wizard = () => {
                 />
               </Panel.Content>
 
-              <Panel.Content className="Form section-block--body has-inputs-centered ">
+              <Panel.Content className="Form section-block--body has-inputs-centered border-b border-panel-border-interior-light dark:border-panel-border-interior-dark">
                 <FormField
                   // @ts-ignore
                   label="Region"
@@ -344,6 +353,24 @@ export const Wizard = () => {
                   description="Select a region close to you for the best performance."
                 />
               </Panel.Content>
+
+              <Panel.Content className="Form section-block--body has-inputs-centered ">
+                <FormField
+                  // @ts-ignore
+                  label="Pricing Plan"
+                  type="select"
+                  choices={PRICING_PLANS}
+                  value={dbPricingPlan}
+                  onChange={onDbPricingPlanChange}
+                  description={[
+                    'Select a pricing plan. ',
+                    <a className="underline" target="_blank" href="https://supabase.com/pricing">
+                      More details
+                    </a>,
+                    '.',
+                  ]}
+                />
+              </Panel.Content>
             </>
           )}
         </>