org.jboss.resteasy.plugins.delegates.LinkDelegate Maven / Gradle / Ivy
package org.jboss.resteasy.plugins.delegates;
import org.jboss.resteasy.specimpl.LinkBuilderImpl;
import org.jboss.resteasy.specimpl.MultivaluedMapImpl;
import org.jboss.resteasy.spi.LinkHeader;
import javax.ws.rs.core.Link;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.ext.RuntimeDelegate;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.StringTokenizer;
/**
* @author Bill Burke
* @version $Revision: 1 $
*/
public class LinkDelegate implements RuntimeDelegate.HeaderDelegate
{
private static class Parser
{
private int curr;
private String value;
private Link.Builder builder;
public Parser(String value)
{
this.value = value;
builder = new LinkBuilderImpl();
}
public Link getLink()
{
return builder.build();
}
public void parse()
{
String href = null;
MultivaluedMap attributes = new MultivaluedMapImpl();
while (curr < value.length())
{
char c = value.charAt(curr);
if (c == '<')
{
if (href != null)
throw new IllegalArgumentException("Unable to parse Link header. Too many links in declaration: " + value);
href = parseLink();
}
else if (c == ';' || c == ' ')
{
curr++;
continue;
}
else
{
parseAttribute(attributes);
}
}
populateLink(href, attributes);
}
protected void populateLink(String href, MultivaluedMap attributes)
{
builder.uri(href);
for (String name : attributes.keySet())
{
List values = attributes.get(name);
if (name.equals("rel"))
{
for (String val : values) builder.rel(val);
}
else if (name.equals("title"))
{
for (String val : values) builder.title(val);
}
else if (name.equals("type"))
{
for (String val : values) builder.type(val);
}
else
{
for (String val : values) builder.param(name, val);
}
}
}
public String parseLink()
{
int end = value.indexOf('>', curr);
if (end == -1) throw new IllegalArgumentException("Unable to parse Link header. No end to link: " + value);
String href = value.substring(curr + 1, end);
curr = end + 1;
return href;
}
public void parseAttribute(MultivaluedMap attributes)
{
int end = value.indexOf('=', curr);
if (end == -1 || end + 1 >= value.length())
throw new IllegalArgumentException("Unable to parse Link header. No end to parameter: " + value);
String name = value.substring(curr, end);
name = name.trim();
curr = end + 1;
String val = null;
if (curr >= value.length())
{
val = "";
}
else
{
if (value.charAt(curr) == '"')
{
if (curr + 1 >= value.length())
throw new IllegalArgumentException("Unable to parse Link header. No end to parameter: " + value);
curr++;
end = value.indexOf('"', curr);
if (end == -1)
throw new IllegalArgumentException("Unable to parse Link header. No end to parameter: " + value);
val = value.substring(curr, end);
curr = end + 1;
}
else
{
StringBuffer buf = new StringBuffer();
while (curr < value.length())
{
char c = value.charAt(curr);
if (c == ',' || c == ';') break;
buf.append(value.charAt(curr));
curr++;
}
val = buf.toString();
}
}
attributes.add(name, val);
}
}
@Override
public Link fromString(String value) throws IllegalArgumentException
{
if (value == null) throw new IllegalArgumentException("param was null");
Parser parser = new Parser(value);
parser.parse();
return parser.getLink();
}
@Override
public String toString(Link value) throws IllegalArgumentException
{
if (value == null) throw new IllegalArgumentException("param was null");
StringBuffer buf = new StringBuffer("<");
buf.append(value.getUri().toString()).append(">");
for (Map.Entry entry : value.getParams().entrySet())
{
buf.append("; ").append(entry.getKey()).append("=\"").append(entry.getValue()).append("\"");
}
return buf.toString();
}
}