From 89673d2a028000374a9840b55fb4cb4093635e2e Mon Sep 17 00:00:00 2001
From: Sven Efftinge <sven@gitpod.io>
Date: Sun, 11 Dec 2022 15:32:41 +0000
Subject: [PATCH] [dashboard] allow setting ws class for prebuilds

---
 .../src/projects/ProjectSettings.tsx          | 42 ++++++++++++++++++-
 components/gitpod-protocol/src/protocol.ts    |  4 +-
 2 files changed, 43 insertions(+), 3 deletions(-)

diff --git a/components/dashboard/src/projects/ProjectSettings.tsx b/components/dashboard/src/projects/ProjectSettings.tsx
index d7b9bd643c0109..a4db5f58804ed3 100644
--- a/components/dashboard/src/projects/ProjectSettings.tsx
+++ b/components/dashboard/src/projects/ProjectSettings.tsx
@@ -76,13 +76,53 @@ export default function () {
             return value;
         }
         const before = project.settings?.workspaceClasses?.regular;
-        updateProjectSettings({ workspaceClasses: { prebuild: value, regular: value } });
+        updateProjectSettings({ workspaceClasses: { ...project.settings?.workspaceClasses, regular: value } });
+        return before;
+    };
+
+    const setWorkspaceClassForPrebuild = async (value: string) => {
+        if (!project) {
+            return value;
+        }
+        const before = project.settings?.workspaceClasses?.prebuild;
+        updateProjectSettings({ workspaceClasses: { ...project.settings?.workspaceClasses, prebuild: value } });
         return before;
     };
 
     return (
         <ProjectSettingsPage project={project}>
             <h3>Prebuilds</h3>
+            <p className="text-base text-gray-500 dark:text-gray-400">
+                Choose the workspace machine type for your prebuilds.
+            </p>
+            {BillingMode.canSetWorkspaceClass(billingMode) ? (
+                <SelectWorkspaceClass
+                    workspaceClass={project.settings?.workspaceClasses?.prebuild}
+                    setWorkspaceClass={setWorkspaceClassForPrebuild}
+                />
+            ) : (
+                <Alert type="message" className="mt-4">
+                    <div className="flex flex-col">
+                        <span>
+                            To access{" "}
+                            <a
+                                className="gp-link"
+                                href="https://www.gitpod.io/docs/configure/workspaces/workspace-classes"
+                            >
+                                large workspaces
+                            </a>{" "}
+                            and{" "}
+                            <a className="gp-link" href="https://www.gitpod.io/docs/configure/billing/pay-as-you-go">
+                                pay-as-you-go
+                            </a>
+                            , first cancel your existing plan.
+                        </span>
+                        <Link className="mt-2" to={project.teamId ? "../billing" : "/plans"}>
+                            <button>Go to {project.teamId ? "Team" : "Personal"} Billing</button>
+                        </Link>
+                    </div>
+                </Alert>
+            )}
             <CheckBox
                 title={<span>Enable Incremental Prebuilds </span>}
                 desc={
diff --git a/components/gitpod-protocol/src/protocol.ts b/components/gitpod-protocol/src/protocol.ts
index fd06b191da7f61..f8548a969c0929 100644
--- a/components/gitpod-protocol/src/protocol.ts
+++ b/components/gitpod-protocol/src/protocol.ts
@@ -255,8 +255,8 @@ export type IDESettings = {
 };
 
 export interface WorkspaceClasses {
-    regular: string;
-    prebuild: string;
+    regular?: string;
+    prebuild?: string;
 }
 
 export interface UserPlatform {