/*
* Copyright (c) 2009 Lockheed Martin Corporation
*
* 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 org.eurekastreams.server.service.restlets;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import net.oauth.OAuth;
import net.oauth.OAuthMessage;
import org.restlet.data.Form;
import org.restlet.data.Parameter;
import org.restlet.data.Request;
/**
* This class represents the OAuth request message for use with restlets.
*/
public class RestletRequestMessage extends OAuthMessage
{
/**
* The restlet request.
*/
private final Request request;
/**
* Constructor.
*
* @param inRequest
* The restlet request.
*/
public RestletRequestMessage(final Request inRequest)
{
super(inRequest.getMethod().toString(),
inRequest.getResourceRef().getBaseRef().toString(),
getParameters(inRequest));
request = inRequest;
copyHeaders(request, getHeaders());
}
/**
* Copies HTTP headers from restlet request and returns them as OAuth parameters.
*
* @param request
* The restlet request.
* @param into
* The Map that will be populated with the retrieved parameters.
*/
private static void copyHeaders(final Request request, final Collection<Map.Entry<String, String>> into)
{
Form requestHeaders = (Form) request.getAttributes().get("org.restlet.http.headers");
Set<String> names = requestHeaders.getNames();
if (names != null)
{
Iterator<String> itr = names.iterator();
while (itr.hasNext())
{
String name = itr.next();
String[] values = requestHeaders.getValuesArray(name);
if (values != null)
{
for (int i = 0; i < values.length; i++)
{
into.add(new OAuth.Parameter(name, values[i]));
}
}
}
}
}
/**
* Gets a list of parameters that are a part of the restlet request.
* @param request
* the restlet request.
* @return the list of OAuth parameters and any form or querystring parameters.
*/
public static List<OAuth.Parameter> getParameters(final Request request)
{
List<OAuth.Parameter> list = new ArrayList<OAuth.Parameter>();
Form requestHeaders = (Form) request.getAttributes().get("org.restlet.http.headers");
String[] values = requestHeaders.getValuesArray("Authorization");
if (values != null)
{
for (int i = 0; i < values.length; i++)
{
String header = values[i];
for (OAuth.Parameter parameter : OAuthMessage.decodeAuthorization(header))
{
if (!"realm".equalsIgnoreCase(parameter.getKey()))
{
list.add(parameter);
}
}
}
}
Form form = request.getEntityAsForm();
for (Parameter parameter : form)
{
list.add(new OAuth.Parameter(parameter.getName(), parameter.getValue()));
}
Form queryString = request.getResourceRef().getQueryAsForm();
for (Parameter parameter : queryString)
{
list.add(new OAuth.Parameter(parameter.getName(), parameter.getValue()));
}
return list;
}
}