ConsoleProtocol.java

package org.ovirt.mobile.movirt.model.enums;

import android.support.annotation.NonNull;

import org.ovirt.mobile.movirt.model.Console;

import java.util.List;
import java.util.SortedSet;
import java.util.TreeSet;

public enum ConsoleProtocol {
    SPICE,
    VNC;

    @NonNull
    public static ConsoleProtocol mapProtocol(String display) {
        try {
            return valueOf(display.toUpperCase());
        } catch (Exception e) {
            // not particularly nice but same behavior as on the webadmin/userportal
            return VNC;
        }
    }

    public static SortedSet<ConsoleProtocol> getProtocolTypes(List<Console> consoles) {
        SortedSet<ConsoleProtocol> result = new TreeSet<>();

        for (Console console : consoles) {
            result.add(console.getProtocol());
        }

        return result;
    }

    public String getProtocol() {
        return super.toString().toLowerCase();
    }
}