Add MACHINE_TYPE to template parameter type

metadata/src/main/java/com/google/cloud/teleport/metadata/TemplateParameter.java

@@ -667,6 +667,44 @@ public final class TemplateParameter {
     boolean hiddenUi() default false;
   }
 
+  /**
+   * Template Parameter containing machine type. See
+   * https://cloud.google.com/compute/docs/machine-types for a list of valid options.
+   */
+  @Retention(RetentionPolicy.RUNTIME)
+  @Target({ElementType.FIELD, ElementType.METHOD})
+  public @interface MachineType {
+    /** Order of appearance. */
+    int order() default 999;
+
+    /** Name of the parameter. */
+    String name() default "";
+
+    /** Group Name of the parameter. */
+    String groupName() default "";
+
+    /** Parent Name of the parameter. */
+    String parentName() default "";
+
+    /** List of parent trigger values. */
+    String[] parentTriggerValues() default "";
+
+    /** If parameter is optional. */
+    boolean optional() default false;
+
+    /** Description of the parameter. */
+    String description();
+
+    /** Help text of the parameter. */
+    String helpText();
+
+    /** Example of the parameter. */
+    String example() default "";
+
+    /** Parameter visibility in the UI. */
+    boolean hiddenUi() default false;
+  }
+
   /** Template Parameter containing a date/time input. */
   @Retention(RetentionPolicy.RUNTIME)
   @Target({ElementType.FIELD, ElementType.METHOD})

metadata/src/main/java/com/google/cloud/teleport/metadata/util/MetadataUtils.java

@@ -45,6 +45,7 @@ public final class MetadataUtils {
         TemplateParameter.Integer.class,
         TemplateParameter.KafkaTopic.class,
         TemplateParameter.KmsEncryptionKey.class,
+        TemplateParameter.MachineType.class,
         TemplateParameter.Long.class,
         TemplateParameter.Password.class,
         TemplateParameter.ProjectId.class,
@@ -209,6 +210,10 @@ public static List<String> getRegexes(Annotation parameterAnnotation) {
                 + "\\/]+\\/cryptoKeys\\/[^\\n"
                 + "\\r"
                 + "\\/]+$");
+      case "MachineType":
+        TemplateParameter.MachineType machineTypeParam =
+            (TemplateParameter.MachineType) parameterAnnotation;
+        return List.of("^[a-z0-9]+(-[a-z0-9]+)+$");
       case "Duration":
         TemplateParameter.Duration durationParam = (TemplateParameter.Duration) parameterAnnotation;
         return List.of("^[1-9][0-9]*[s|m|h]$");

plugins/core-plugin/src/main/java/com/google/cloud/teleport/plugin/model/ImageSpecParameter.java

@@ -469,6 +469,23 @@ public void processParamType(Annotation parameterAnnotation) {
         this.setHiddenUi(kmsEncryptionKeyParam.hiddenUi());
         this.setParamType(ImageSpecParameterType.TEXT);
         break;
+      case "MachineType":
+        TemplateParameter.MachineType machineTypeParam =
+            (TemplateParameter.MachineType) parameterAnnotation;
+        if (!machineTypeParam.name().isEmpty()) {
+          this.setName(machineTypeParam.name());
+        }
+        processDescriptions(
+            machineTypeParam.groupName(),
+            machineTypeParam.description(),
+            machineTypeParam.helpText(),
+            machineTypeParam.example());
+        this.setParentName(machineTypeParam.parentName());
+        this.setParentTriggerValues(machineTypeParam.parentTriggerValues());
+        this.setOptional(machineTypeParam.optional());
+        this.setHiddenUi(machineTypeParam.hiddenUi());
+        this.setParamType(ImageSpecParameterType.MACHINE_TYPE);
+        break;
       case "Duration":
         TemplateParameter.Duration durationParam = (TemplateParameter.Duration) parameterAnnotation;
         if (!durationParam.name().isEmpty()) {

plugins/core-plugin/src/main/java/com/google/cloud/teleport/plugin/model/ImageSpecParameterType.java

@@ -45,6 +45,9 @@ public enum ImageSpecParameterType {
   /** Pub/Sub Topic to read or write. */
   PUBSUB_TOPIC,
 
+  /** Machine Type parameter. */
+  MACHINE_TYPE,
+
   /** Boolean parameter. */
   BOOLEAN,
 

plugins/core-plugin/src/test/java/com/google/cloud/teleport/plugin/model/TemplateDefinitionsTest.java

@@ -70,6 +70,9 @@ public void testSampleAtoBOk() {
     assertEquals(ImageSpecParameterType.BOOLEAN, logical.getParamType());
     assertEquals("^(true|false)$", logical.getRegexes().get(0));
 
+    ImageSpecParameter machineType = metadata.getParameter("machineType").get();
+    assertEquals(ImageSpecParameterType.MACHINE_TYPE, machineType.getParamType());
+
     ImageSpecParameter json = metadata.getParameter("JSON").get();
     assertEquals(ImageSpecParameterType.BOOLEAN, json.getParamType());
 

plugins/core-plugin/src/test/java/com/google/cloud/teleport/plugin/sample/AtoBOk.java

@@ -98,15 +98,22 @@ public interface AtoBOptions {
     @Default.Boolean(false)
     Boolean getParamWithGroupName();
 
-    @TemplateParameter.Text(
+    @TemplateParameter.Boolean(
         order = 9,
         parentName = "paramWithGroupName",
         parentTriggerValues = {"true"},
         description = "N/A",
-        helpText = "Text that has parent name and parent trigger value")
+        helpText = "Boolean that has parent name and parent trigger value")
     @Default.Boolean(false)
     Boolean getParamWithParentName();
 
+    @TemplateParameter.MachineType(
+        order = 15,
+        description = "N/A",
+        helpText = "String that is machine type")
+    @Default.String("")
+    String getMachineType();
+
     @TemplateParameter.KafkaTopic(
         order = 10,
         description = "Kafka input topic",