-
Notifications
You must be signed in to change notification settings - Fork 306
Expand file tree
/
Copy pathOptions.java
More file actions
61 lines (54 loc) · 1.91 KB
/
Options.java
File metadata and controls
61 lines (54 loc) · 1.91 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
/*
* Copyright (C) 2016 Google Inc.
*
* Licensed 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 com.google.copybara;
import com.google.common.collect.ImmutableCollection;
import com.google.common.collect.ImmutableMap;
import java.util.Map.Entry;
/**
* A class that groups all the options used in the program
*/
public class Options {
private final ImmutableMap<Class<? extends Option>, Option> config;
public Options(Iterable<? extends Option> options) {
ImmutableMap.Builder<Class<? extends Option>, Option> builder = ImmutableMap.builder();
for (Option option : options) {
builder.put(option.getClass(), option);
}
config = builder.buildOrThrow();
}
/**
* Get an option for a given class.
*
* @throws IllegalStateException if the configuration cannot be found
*/
@SuppressWarnings("unchecked")
public <T extends Option> T get(Class<? extends T> optionClass) {
Option option = config.get(optionClass);
if (option == null) {
// If we didn't find the exact class, look for a subclass.
for (Entry<Class<? extends Option>, Option> entry : config.entrySet()) {
if (optionClass.isAssignableFrom(entry.getKey())) {
return (T) entry.getValue();
}
}
throw new IllegalStateException("No option type found for " + optionClass);
}
return (T) option;
}
public ImmutableCollection<Option> getAll() {
return config.values();
}
}