/*
* Copyright 2010-2013 Amazon.com, Inc. or its affiliates. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License").
* You may not use this file except in compliance with the License.
* A copy of the License is located at
*
* http://aws.amazon.com/apache2.0
*
* or in the "license" file accompanying this file. This file 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.amazonaws.services.ec2.model.transform;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
import org.apache.commons.codec.binary.Base64;
import com.amazonaws.AmazonWebServiceRequest;
import com.amazonaws.Request;
import com.amazonaws.handlers.AbstractRequestHandler;
import com.amazonaws.services.ec2.model.DescribeInstancesResult;
import com.amazonaws.services.ec2.model.DescribeSpotInstanceRequestsResult;
import com.amazonaws.services.ec2.model.GroupIdentifier;
import com.amazonaws.services.ec2.model.ImportKeyPairRequest;
import com.amazonaws.services.ec2.model.LaunchSpecification;
import com.amazonaws.services.ec2.model.RequestSpotInstancesRequest;
import com.amazonaws.services.ec2.model.RequestSpotInstancesResult;
import com.amazonaws.services.ec2.model.Reservation;
import com.amazonaws.services.ec2.model.RunInstancesRequest;
import com.amazonaws.services.ec2.model.RunInstancesResult;
import com.amazonaws.services.ec2.model.SpotInstanceRequest;
import com.amazonaws.util.TimingInfo;
public class EC2RequestHandler extends AbstractRequestHandler {
@Override
public void beforeRequest(Request<?> request) {
AmazonWebServiceRequest originalRequest = request.getOriginalRequest();
if (originalRequest instanceof ImportKeyPairRequest) {
ImportKeyPairRequest importKeyPairRequest = (ImportKeyPairRequest)originalRequest;
String publicKeyMaterial = importKeyPairRequest.getPublicKeyMaterial();
String encodedKeyMaterial = new String(Base64.encodeBase64(publicKeyMaterial.getBytes()));
request.addParameter("PublicKeyMaterial", encodedKeyMaterial);
}
// Request -> Query string marshalling for RequestSpotInstancesRequest is a little tricky since
// the query string params follow a different form than the XML responses, so we manually set the parameters here.
else if (originalRequest instanceof RequestSpotInstancesRequest) {
RequestSpotInstancesRequest requestSpotInstancesRequest = (RequestSpotInstancesRequest)originalRequest;
int count = 1;
for (GroupIdentifier group : requestSpotInstancesRequest.getLaunchSpecification().getAllSecurityGroups()) {
if (group.getGroupId() != null) {
request.addParameter("LaunchSpecification.SecurityGroupId." + count++, group.getGroupId());
}
}
// Remove any of the incorrect parameters.
List<String> keysToRemove = new ArrayList<String>();
for (String parameter : request.getParameters().keySet()) {
if (parameter.startsWith("LaunchSpecification.GroupSet.")) keysToRemove.add(parameter);
}
for (String key : keysToRemove) {
request.getParameters().remove(key);
}
}
// If a RunInstancesRequest doesn't specify a ClientToken, fill one in, otherwise
// retries could result in unwanted instances being launched in the customer's account.
else if (originalRequest instanceof RunInstancesRequest) {
RunInstancesRequest runInstancesRequest = (RunInstancesRequest)originalRequest;
if (runInstancesRequest.getClientToken() == null) {
request.getParameters().put("ClientToken", UUID.randomUUID().toString());
}
}
}
@Override
public void afterResponse(Request<?> request, Object response, TimingInfo timingInfo) {
/*
* For backwards compatibility, we preserve the existing List<String> of
* security group names by explicitly populating it from the full list
* of security group info.
*/
if (response instanceof DescribeSpotInstanceRequestsResult) {
DescribeSpotInstanceRequestsResult result = (DescribeSpotInstanceRequestsResult)response;
for (SpotInstanceRequest spotInstanceRequest : result.getSpotInstanceRequests()) {
LaunchSpecification launchSpecification = spotInstanceRequest.getLaunchSpecification();
populateLaunchSpecificationSecurityGroupNames(launchSpecification);
}
} else if (response instanceof RequestSpotInstancesResult) {
RequestSpotInstancesResult result = (RequestSpotInstancesResult)response;
for (SpotInstanceRequest spotInstanceRequest : result.getSpotInstanceRequests()) {
LaunchSpecification launchSpecification = spotInstanceRequest.getLaunchSpecification();
populateLaunchSpecificationSecurityGroupNames(launchSpecification);
}
} else if (response instanceof DescribeInstancesResult) {
DescribeInstancesResult result = (DescribeInstancesResult)response;
for (Reservation reservation : result.getReservations()) {
populateReservationSecurityGroupNames(reservation);
}
} else if (response instanceof RunInstancesResult) {
RunInstancesResult result = (RunInstancesResult)response;
populateReservationSecurityGroupNames(result.getReservation());
}
}
private void populateReservationSecurityGroupNames(Reservation reservation) {
List<String> groupNames = new ArrayList<String>();
for (GroupIdentifier group : reservation.getGroups()) {
groupNames.add(group.getGroupName());
}
reservation.setGroupNames(groupNames);
}
private void populateLaunchSpecificationSecurityGroupNames(LaunchSpecification launchSpecification) {
List<String> groupNames = new ArrayList<String>();
for (GroupIdentifier group : launchSpecification.getAllSecurityGroups()) {
groupNames.add(group.getGroupName());
}
launchSpecification.setSecurityGroups(groupNames);
}
}