From ba5e5e145cb1a7561575fae433da0bb6639d7a11 Mon Sep 17 00:00:00 2001
From: Malte Poll <mp@edgeless.systems>
Date: Wed, 2 Aug 2023 10:23:51 +0200
Subject: [PATCH] config: define lists of valid disk types

---
 internal/config/disktypes/BUILD.bazel | 12 ++++++++++++
 internal/config/disktypes/aws.go      | 17 +++++++++++++++++
 internal/config/disktypes/azure.go    | 16 ++++++++++++++++
 internal/config/disktypes/gcp.go      | 14 ++++++++++++++
 4 files changed, 59 insertions(+)
 create mode 100644 internal/config/disktypes/BUILD.bazel
 create mode 100644 internal/config/disktypes/aws.go
 create mode 100644 internal/config/disktypes/azure.go
 create mode 100644 internal/config/disktypes/gcp.go

diff --git a/internal/config/disktypes/BUILD.bazel b/internal/config/disktypes/BUILD.bazel
new file mode 100644
index 000000000..97aa33daa
--- /dev/null
+++ b/internal/config/disktypes/BUILD.bazel
@@ -0,0 +1,12 @@
+load("@io_bazel_rules_go//go:def.bzl", "go_library")
+
+go_library(
+    name = "disktypes",
+    srcs = [
+        "aws.go",
+        "azure.go",
+        "gcp.go",
+    ],
+    importpath = "github.com/edgelesssys/constellation/v2/internal/config/disktypes",
+    visibility = ["//:__subpackages__"],
+)
diff --git a/internal/config/disktypes/aws.go b/internal/config/disktypes/aws.go
new file mode 100644
index 000000000..8fcdc354a
--- /dev/null
+++ b/internal/config/disktypes/aws.go
@@ -0,0 +1,17 @@
+/*
+Copyright (c) Edgeless Systems GmbH
+
+SPDX-License-Identifier: AGPL-3.0-only
+*/
+
+package disktypes
+
+// AWSDiskTypes is derived from:
+// https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/ebs-volume-types.html (Last updated: August 1st, 2023).
+var AWSDiskTypes = []string{
+	"gp2",
+	"gp3",
+	"st1",
+	"sc1",
+	"io1",
+}
diff --git a/internal/config/disktypes/azure.go b/internal/config/disktypes/azure.go
new file mode 100644
index 000000000..8903a45d2
--- /dev/null
+++ b/internal/config/disktypes/azure.go
@@ -0,0 +1,16 @@
+/*
+Copyright (c) Edgeless Systems GmbH
+
+SPDX-License-Identifier: AGPL-3.0-only
+*/
+
+package disktypes
+
+// AzureDiskTypes are valid Azure disk types.
+var AzureDiskTypes = []string{
+	"Premium_LRS",
+	"Premium_ZRS",
+	"Standard_LRS",
+	"StandardSSD_LRS",
+	"StandardSSD_ZRS",
+}
diff --git a/internal/config/disktypes/gcp.go b/internal/config/disktypes/gcp.go
new file mode 100644
index 000000000..3880b9a2f
--- /dev/null
+++ b/internal/config/disktypes/gcp.go
@@ -0,0 +1,14 @@
+/*
+Copyright (c) Edgeless Systems GmbH
+
+SPDX-License-Identifier: AGPL-3.0-only
+*/
+
+package disktypes
+
+// GCPDiskTypes are valid GCP disk types.
+var GCPDiskTypes = []string{
+	"pd-standard",
+	"pd-balanced",
+	"pd-ssd",
+}