labs/smartos-ssh/src/main/java/org/jclouds/smartos/compute/domain/VmSpecification.java

package org.jclouds.smartos.compute.domain;

import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

import com.google.common.collect.ImmutableList;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonElement;
import com.google.gson.JsonPrimitive;
import com.google.gson.JsonSerializationContext;
import com.google.gson.JsonSerializer;
import com.google.gson.annotations.SerializedName;

/**
 * Specification of a VM to build, based on a dataset.
 */
public class VmSpecification {
   protected final String alias;
   protected final String brand;

   @SerializedName("dataset_uuid")
   protected final DataSet dataset;
   protected final String dnsDomain;
   protected final String quota;

   protected final List<VmNIC> nics;

   public static Builder builder() {
      return new Builder();
   }

   public Builder toBuilder() {
      return builder().fromVmSpecification(this);
   }

   public static class Builder {

      protected String alias;
      protected String brand = "joyent";
      protected DataSet dataset;
      protected String dnsDomain = "local";
      protected String quota = "10";

      protected List<VmNIC> nics = new ArrayList<VmNIC>();

      public Builder alias(String alias) {
         this.alias = alias;
         return this;
      }

      public Builder brand(String brand) {
         this.brand = brand;
         return this;
      }

      public Builder dataset(DataSet dataset) {
         this.dataset = dataset;
         return this;
      }

      public Builder dnsDomain(String dnsDomain) {
         this.dnsDomain = dnsDomain;
         return this;
      }

      public Builder quota(String quota) {
         this.quota = quota;
         return this;
      }

      public Builder nics(Collection<VmNIC> nic) {
         this.nics.addAll(nics);
         return this;
      }

      public Builder nic(VmNIC nic) {
         this.nics.add(nic);
         return this;
      }

      public VmSpecification build() {
         return new VmSpecification(alias, brand, dataset, dnsDomain, quota, nics);
      }

      public Builder fromVmSpecification(VmSpecification in) {
         return alias(in.getAlias()).brand(in.getBrand()).dataset(in.getDataset()).dnsDomain(in.getDnsDomain())
                  .quota(in.getQuota()).nics(in.getNics());
      }
   }

   protected VmSpecification(String alias, String brand, DataSet dataset, String dnsDomain, String quota,
            List<VmNIC> nics) {
      this.alias = alias;
      this.brand = brand;
      this.dataset = dataset;
      this.dnsDomain = dnsDomain;
      this.quota = quota;
      this.nics = nics;
   }

   public String getAlias() {
      return alias;
   }

   public String getBrand() {
      return brand;
   }

   public DataSet getDataset() {
      return dataset;
   }

   public String getDnsDomain() {
      return dnsDomain;
   }

   public String getQuota() {
      return quota;
   }

   public List<VmNIC> getNics() {
      return ImmutableList.copyOf(nics);
   }

   public String toJSONSpecification() {
      GsonBuilder gson = new GsonBuilder();
      gson.registerTypeAdapter(DataSet.class, new FlattenDataset());
      Gson g = gson.create();

      return g.toJson(this);
   }

   public class FlattenDataset implements JsonSerializer<DataSet> {
      @Override
      public JsonElement serialize(DataSet vmSpecification, Type type, JsonSerializationContext jsonSerializationContext) {
         return new JsonPrimitive(dataset.getUuid().toString());
      }
   }
}