1 | /** |
2 | * Licensed to jclouds, Inc. (jclouds) under one or more |
3 | * contributor license agreements. See the NOTICE file |
4 | * distributed with this work for additional information |
5 | * regarding copyright ownership. jclouds licenses this file |
6 | * to you under the Apache License, Version 2.0 (the |
7 | * "License"); you may not use this file except in compliance |
8 | * with the License. You may obtain a copy of the License at |
9 | * |
10 | * http://www.apache.org/licenses/LICENSE-2.0 |
11 | * |
12 | * Unless required by applicable law or agreed to in writing, |
13 | * software distributed under the License is distributed on an |
14 | * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
15 | * KIND, either express or implied. See the License for the |
16 | * specific language governing permissions and limitations |
17 | * under the License. |
18 | */ |
19 | package org.jclouds.location.suppliers; |
20 | |
21 | import static com.google.common.base.Preconditions.checkNotNull; |
22 | |
23 | import java.net.URI; |
24 | import java.util.Map; |
25 | import java.util.Set; |
26 | |
27 | import javax.inject.Inject; |
28 | import javax.inject.Singleton; |
29 | |
30 | import org.jclouds.domain.Location; |
31 | import org.jclouds.domain.LocationBuilder; |
32 | import org.jclouds.domain.LocationScope; |
33 | import org.jclouds.location.Iso3166; |
34 | import org.jclouds.location.Provider; |
35 | import org.jclouds.location.Region; |
36 | |
37 | import com.google.common.collect.ImmutableSet; |
38 | import com.google.common.collect.Iterables; |
39 | import com.google.common.collect.ImmutableSet.Builder; |
40 | |
41 | /** |
42 | * |
43 | * @author Adrian Cole |
44 | */ |
45 | @Singleton |
46 | public class RegionToProviderOrJustProvider extends JustProvider { |
47 | private final Set<String> regions; |
48 | private final Map<String, Set<String>> isoCodesById; |
49 | |
50 | @Inject |
51 | public RegionToProviderOrJustProvider(@Iso3166 Set<String> isoCodes, @Provider String providerName, |
52 | @Provider URI endpoint, @Region Set<String> regions, @Iso3166 Map<String, Set<String>> isoCodesById) { |
53 | super(isoCodes, providerName, endpoint); |
54 | this.regions = checkNotNull(regions, "regions"); |
55 | this.isoCodesById = checkNotNull(isoCodesById, "isoCodesById"); |
56 | } |
57 | |
58 | @Override |
59 | public Set<? extends Location> get() { |
60 | return buildJustProviderOrRegions().build(); |
61 | } |
62 | |
63 | protected Builder<Location> buildJustProviderOrRegions() { |
64 | Builder<Location> locations = ImmutableSet.builder(); |
65 | Location provider = Iterables.getOnlyElement(super.get()); |
66 | if (regions.size() == 0) |
67 | return locations.add(provider); |
68 | else |
69 | for (String region : regions) { |
70 | LocationBuilder builder = new LocationBuilder().scope(LocationScope.REGION).id(region).description(region) |
71 | .parent(provider); |
72 | if (isoCodesById.containsKey(region)) |
73 | builder.iso3166Codes(isoCodesById.get(region)); |
74 | locations.add(builder.build()); |
75 | } |
76 | return locations; |
77 | } |
78 | |
79 | } |