package org.jboss.resteasy.plugins.delegates;

import org.jboss.resteasy.resteasy_jaxrs.i18n.Messages;
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 $
/** * @author <a href="mailto:bill@burkecentral.com">Bill Burke</a> * @version $Revision: 1 $ */
public class LinkDelegate implements RuntimeDelegate.HeaderDelegate<Link> { 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<String, String> attributes = new MultivaluedMapImpl<String, String>(); while (curr < value.length()) { char c = value.charAt(curr); if (c == '<') { if (href != null) throw new IllegalArgumentException(Messages.MESSAGES.unableToParseLinkHeaderTooManyLinks(value)); href = parseLink(); } else if (c == ';' || c == ' ') { curr++; continue; } else { parseAttribute(attributes); } } populateLink(href, attributes); } protected void populateLink(String href, MultivaluedMap<String, String> attributes) { builder.uri(href); for (String name : attributes.keySet()) { List<String> 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(Messages.MESSAGES.unableToParseLinkHeaderNoEndToLink(value)); String href = value.substring(curr + 1, end); curr = end + 1; return href; } public void parseAttribute(MultivaluedMap<String, String> attributes) { int end = value.indexOf('=', curr); if (end == -1 || end + 1 >= value.length()) throw new IllegalArgumentException(Messages.MESSAGES.unableToParseLinkHeaderNoEndToParameter(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(Messages.MESSAGES.unableToParseLinkHeaderNoEndToParameter(value)); curr++; end = value.indexOf('"', curr); if (end == -1) throw new IllegalArgumentException(Messages.MESSAGES.unableToParseLinkHeaderNoEndToParameter(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(Messages.MESSAGES.paramNull()); Parser parser = new Parser(value); parser.parse(); return parser.getLink(); } @Override public String toString(Link value) throws IllegalArgumentException { if (value == null) throw new IllegalArgumentException(Messages.MESSAGES.paramNull()); StringBuffer buf = new StringBuffer("<"); buf.append(value.getUri().toString()).append(">"); for (Map.Entry<String, String> entry : value.getParams().entrySet()) { buf.append("; ").append(entry.getKey()).append("=\"").append(entry.getValue()).append("\""); } return buf.toString(); } }