org.jclouds.virtualbox.domain.StorageController Maven / Gradle / Ivy

Go to download
/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.jclouds.virtualbox.domain;

import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.collect.Iterables.filter;

import java.util.Set;

import org.jclouds.javax.annotation.Nullable;
import org.virtualbox_4_2.DeviceType;
import org.virtualbox_4_2.StorageBus;

import com.google.common.base.Objects;
import com.google.common.base.Predicate;
import com.google.common.collect.Iterables;
import com.google.common.collect.Sets;

/**
 * Represents a storage controller in a VirtualBox VM.
 * 
 * name is the unique name of the controller.
 * bus is the type of connection bus for the controller
 * hardDisks contains the hard disks that are attached (or should be attached) to this controller
 * isoImages contains the ISOs that are attached (or should be attached) to this controller
 *
 * @see StorageBus
 */
public class StorageController {
   private final String name;
   private final StorageBus bus;
   private Set hardDisks;
   private Set isoImages;

   public StorageController(String name, StorageBus bus, Set hardDisks, Set isoImages) {
      this.name = checkNotNull(name, "storage name can't be null");
      this.bus = checkNotNull(bus, "bus can't be null");
      this.hardDisks = checkNotNull(hardDisks, "hardDisks can't be null");
      this.isoImages = checkNotNull(isoImages, "isoImages can't be null");
   }

   public String getName() {
      return name;
   }

   public StorageBus getBus() {
      return bus;
   }

   public HardDisk getHardDisk(String diskName) {

      final Iterable hardDisks = filter(getHardDisks(), new HardDiskPredicate(diskName));
      return Iterables.getFirst(hardDisks, HardDisk.builder().diskpath("notfound").controllerPort(0).deviceSlot(0).build());
   }

   public Set getHardDisks() {
      return hardDisks;
   }

   public Set getIsoImages() {
      return isoImages;
   }

   @Override
   public boolean equals(Object o) {
      if (this == o) return true;
      if (o instanceof StorageController) {
         StorageController other = (StorageController) o;
         return Objects.equal(name, other.name) &&
                 Objects.equal(bus, other.bus) &&
                 Objects.equal(hardDisks, other.hardDisks) &&
                 Objects.equal(isoImages, other.isoImages);
      }
      return false;
   }

   @Override
   public int hashCode() {
      return Objects.hashCode(name, bus, hardDisks, isoImages);
   }

   @Override
   public String toString() {
      return "StorageController{" +
              "name='" + name + '\'' +
              ", bus=" + bus +
              ", hardDisks=" + hardDisks +
              ", isoImages=" + isoImages +
              '}';
   }

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

   public static class Builder {

      private String name;
      private StorageBus bus;
      private Set hardDisks = Sets.newHashSet();
      private Set dvds = Sets.newHashSet();

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

      public Builder bus(StorageBus bus) {
         this.bus = bus;
         return this;
      }

      public Builder attachISO(int controllerPort, int deviceSlot, String sourcePath) {
         dvds.add(new IsoImage(new DeviceDetails(controllerPort, deviceSlot, DeviceType.DVD), sourcePath));
         return this;
      }

      public Builder attachHardDisk(HardDisk hardDisk) {
         hardDisks.add(hardDisk);
         return this;
      }

      public StorageController build() {
         checkNotNull(name);
         checkNotNull(bus);
         return new StorageController(name, bus, hardDisks, dvds);
      }
   }
   
   private static class HardDiskPredicate implements Predicate  {
   	
   	private String diskName;
   	
      public HardDiskPredicate(String diskName) {
         this.diskName = diskName;
      }
   	
      @Override
      public boolean apply(@Nullable HardDisk hardDisk) {
         return hardDisk.getName().equals(diskName);
      }
   };
}