*: add --instance-types (#295)

Signed-off-by: Gyuho Lee <[email protected]>

Author: gyuho

avalanche-ops/src/aws/spec.rs

@@ -303,6 +303,7 @@ pub struct DefaultSpecOption {
     pub ingress_ipv4_cidr: String,
     pub instance_mode: String,
     pub instance_size: String,
+    pub instance_types: Vec<String>,
     pub volume_size_in_gb: u32,
 
     pub ip_mode: String,
@@ -671,9 +672,12 @@ impl Spec {
             }
         };
 
-        let instance_types =
-            ec2::default_instance_types(&opts.region, &opts.arch_type, &opts.instance_size)
-                .unwrap();
+        let instance_types = if !opts.instance_types.is_empty() {
+            opts.instance_types.clone()
+        } else {
+            ec2::default_instance_types(&opts.region, &opts.arch_type, &opts.instance_size).unwrap()
+        };
+
         let machine = Machine {
             anchor_nodes,
             non_anchor_nodes,

avalancheup-aws/src/default_spec/mod.rs

@@ -108,6 +108,13 @@ pub fn command() -> Command {
                 .value_parser(["large", "xlarge", "2xlarge", "4xlarge", "8xlarge"])
                 .default_value("xlarge"),
         )
+        .arg(
+            Arg::new("INSTANCE_TYPES")
+                .long("instance-types")
+                .help("Sets the comma-separated instance types (overwrites --instance-size)")
+                .required(false)
+                .num_args(1),
+        )
         .arg(
             Arg::new("IP_MODE")
                 .long("ip-mode")

avalancheup-aws/src/main.rs

@@ -45,6 +45,19 @@ async fn main() -> io::Result<()> {
                 }
             }
 
+            let s = sub_matches
+                .get_one::<String>("INSTANCE_TYPES")
+                .unwrap_or(&String::new())
+                .clone();
+            let ss: Vec<&str> = s.split(',').collect();
+            let mut instance_types = Vec::new();
+            for addr in ss.iter() {
+                let trimmed = addr.trim().to_string();
+                if !trimmed.is_empty() {
+                    instance_types.push(addr.trim().to_string());
+                }
+            }
+
             let opt = avalanche_ops::aws::spec::DefaultSpecOption {
                 log_level: sub_matches
                     .get_one::<String>("LOG_LEVEL")
@@ -96,6 +109,7 @@ async fn main() -> io::Result<()> {
                     .get_one::<String>("INSTANCE_SIZE")
                     .unwrap_or(&String::from("large"))
                     .clone(),
+                instance_types,
 
                 volume_size_in_gb: sub_matches
                     .get_one::<u32>("VOLUME_SIZE_IN_GB")