org.jclouds.virtualbox.domain.StorageController.java Source code

Introduction

Here is the source code for org.jclouds.virtualbox.domain.StorageController.java

Source

/*
 * 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.
 * <p/>
 * 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<HardDisk> hardDisks;
    private Set<IsoImage> isoImages;

    public StorageController(String name, StorageBus bus, Set<HardDisk> hardDisks, Set<IsoImage> 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<HardDisk> hardDisks = filter(getHardDisks(), new HardDiskPredicate(diskName));
        return Iterables.getFirst(hardDisks,
                HardDisk.builder().diskpath("notfound").controllerPort(0).deviceSlot(0).build());
    }

    public Set<HardDisk> getHardDisks() {
        return hardDisks;
    }

    public Set<IsoImage> 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<HardDisk> hardDisks = Sets.newHashSet();
        private Set<IsoImage> 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<HardDisk> {

        private String diskName;

        public HardDiskPredicate(String diskName) {
            this.diskName = diskName;
        }

        @Override
        public boolean apply(@Nullable HardDisk hardDisk) {
            return hardDisk.getName().equals(diskName);
        }
    };
}