From 1fcfd8908a7027ab4adaeac6c28181718b1bd9b5 Mon Sep 17 00:00:00 2001
From: Jacob Gillespie <jacobwgillespie@gmail.com>
Date: Thu, 12 Sep 2024 12:14:40 +0100
Subject: [PATCH] Add Fly 8x8 size

---
 proto/depot/cloud/v4/cloud.proto     | 1 +
 src/proto/depot/cloud/v4/cloud_pb.ts | 6 ++++++
 src/utils/fly/reconcile.ts           | 6 ++++--
 3 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/proto/depot/cloud/v4/cloud.proto b/proto/depot/cloud/v4/cloud.proto
index db3b5a9..c7033d4 100644
--- a/proto/depot/cloud/v4/cloud.proto
+++ b/proto/depot/cloud/v4/cloud.proto
@@ -77,6 +77,7 @@ message GetDesiredStateResponse {
     KIND_BUILDKIT_8X16 = 4;
     KIND_BUILDKIT_4X8 = 5;
     KIND_BUILDKIT_4X4 = 6;
+    KIND_BUILDKIT_8X8 = 7;
   }
 
   enum MachineState {
diff --git a/src/proto/depot/cloud/v4/cloud_pb.ts b/src/proto/depot/cloud/v4/cloud_pb.ts
index ab6aaa1..aa702e7 100644
--- a/src/proto/depot/cloud/v4/cloud_pb.ts
+++ b/src/proto/depot/cloud/v4/cloud_pb.ts
@@ -175,6 +175,11 @@ export enum GetDesiredStateResponse_Kind {
    * @generated from enum value: KIND_BUILDKIT_4X4 = 6;
    */
   BUILDKIT_4X4 = 6,
+
+  /**
+   * @generated from enum value: KIND_BUILDKIT_8X8 = 7;
+   */
+  BUILDKIT_8X8 = 7,
 }
 // Retrieve enum metadata with: proto3.getEnumType(GetDesiredStateResponse_Kind)
 proto3.util.setEnumType(GetDesiredStateResponse_Kind, 'depot.cloud.v4.GetDesiredStateResponse.Kind', [
@@ -185,6 +190,7 @@ proto3.util.setEnumType(GetDesiredStateResponse_Kind, 'depot.cloud.v4.GetDesired
   {no: 4, name: 'KIND_BUILDKIT_8X16'},
   {no: 5, name: 'KIND_BUILDKIT_4X8'},
   {no: 6, name: 'KIND_BUILDKIT_4X4'},
+  {no: 7, name: 'KIND_BUILDKIT_8X8'},
 ])
 
 /**
diff --git a/src/utils/fly/reconcile.ts b/src/utils/fly/reconcile.ts
index 39baa66..1b39048 100644
--- a/src/utils/fly/reconcile.ts
+++ b/src/utils/fly/reconcile.ts
@@ -248,9 +248,11 @@ function machineKind(kind: GetDesiredStateResponse_Kind): MachineKind {
     case GetDesiredStateResponse_Kind.BUILDKIT_4X4:
       return {cpuKind: 'shared', cpus: 4, memGBs: 4, needsGPU: false}
     case GetDesiredStateResponse_Kind.BUILDKIT_4X8:
-      return {cpuKind: 'performance', cpus: 4, memGBs: 8, needsGPU: false}
+      return {cpuKind: 'shared', cpus: 4, memGBs: 8, needsGPU: false}
+    case GetDesiredStateResponse_Kind.BUILDKIT_8X8:
+      return {cpuKind: 'shared', cpus: 8, memGBs: 8, needsGPU: false}
     case GetDesiredStateResponse_Kind.BUILDKIT_8X16:
-      return {cpuKind: 'performance', cpus: 8, memGBs: 16, needsGPU: false}
+      return {cpuKind: 'shared', cpus: 8, memGBs: 16, needsGPU: false}
     case GetDesiredStateResponse_Kind.BUILDKIT_16X32:
       return {cpuKind: 'performance', cpus: 16, memGBs: 32, needsGPU: false}
     case GetDesiredStateResponse_Kind.BUILDKIT_16X32_GPU: