| /** |
| * Licensed to jclouds, Inc. (jclouds) under one or more |
| * contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. jclouds 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.s3.binders; |
| |
| import static org.jclouds.s3.reference.S3Constants.PROPERTY_S3_SERVICE_PATH; |
| import static org.jclouds.s3.reference.S3Constants.PROPERTY_S3_VIRTUAL_HOST_BUCKETS; |
| |
| import javax.inject.Inject; |
| import javax.inject.Named; |
| import javax.inject.Provider; |
| import javax.inject.Singleton; |
| import javax.ws.rs.core.HttpHeaders; |
| import javax.ws.rs.core.UriBuilder; |
| |
| import org.jclouds.http.HttpRequest; |
| import org.jclouds.http.utils.ModifyRequest; |
| import org.jclouds.rest.Binder; |
| import org.jclouds.rest.annotations.SkipEncoding; |
| import org.jclouds.rest.binders.BindAsHostPrefix; |
| import org.jclouds.s3.S3AsyncClient; |
| import org.jclouds.util.Strings2; |
| |
| import com.google.common.collect.ImmutableMap; |
| |
| /** |
| * |
| * @author Adrian Cole |
| */ |
| @Singleton |
| public class BindAsHostPrefixIfConfigured implements Binder { |
| |
| protected final Provider<UriBuilder> uriBuilderProvider; |
| protected final BindAsHostPrefix bindAsHostPrefix; |
| protected final boolean isVhostStyle; |
| protected final String servicePath; |
| |
| @Inject |
| public BindAsHostPrefixIfConfigured(BindAsHostPrefix bindAsHostPrefix, |
| @Named(PROPERTY_S3_VIRTUAL_HOST_BUCKETS) boolean isVhostStyle, |
| @Named(PROPERTY_S3_SERVICE_PATH) String servicePath, Provider<UriBuilder> uriBuilderProvider) { |
| this.bindAsHostPrefix = bindAsHostPrefix; |
| this.isVhostStyle = isVhostStyle; |
| this.servicePath = servicePath; |
| this.uriBuilderProvider = uriBuilderProvider; |
| } |
| |
| @SuppressWarnings("unchecked") |
| @Override |
| public <R extends HttpRequest> R bindToRequest(R request, Object payload) { |
| if (isVhostStyle) { |
| request = bindAsHostPrefix.bindToRequest(request, payload); |
| String host = request.getEndpoint().getHost(); |
| if (request.getEndpoint().getPort() != -1) { |
| host += ":" + request.getEndpoint().getPort(); |
| } |
| return ModifyRequest.replaceHeader(request, HttpHeaders.HOST, host); |
| } else { |
| UriBuilder builder = uriBuilderProvider.get().uri(request.getEndpoint()); |
| StringBuilder path = new StringBuilder(Strings2.urlEncode(request.getEndpoint().getPath(), S3AsyncClient.class |
| .getAnnotation(SkipEncoding.class).value())); |
| int indexToInsert = 0; |
| if (!servicePath.equals("/")) { |
| indexToInsert = path.indexOf(servicePath); |
| indexToInsert = indexToInsert == -1 ? 0 : indexToInsert; |
| indexToInsert += servicePath.length(); |
| } |
| path.insert(indexToInsert, "/" + payload.toString()); |
| builder.replacePath(path.toString()); |
| return (R) request.toBuilder().endpoint(builder.buildFromEncodedMap(ImmutableMap.<String, Object> of())) |
| .build(); |
| } |
| } |
| } |