1
2
3
4
5
6
7
8
9
10
11 package org.mule.transport.http.transformers;
12
13 import org.mule.api.MuleMessage;
14 import org.mule.api.transformer.TransformerException;
15 import org.mule.transformer.AbstractMessageTransformer;
16 import org.mule.transformer.types.DataTypeFactory;
17 import org.mule.transport.http.HttpConstants;
18 import org.mule.util.StringUtils;
19
20 import java.net.URI;
21 import java.util.Arrays;
22 import java.util.HashMap;
23 import java.util.List;
24 import java.util.Map;
25
26 import org.apache.commons.codec.net.URLCodec;
27
28 public class HttpRequestBodyToParamMap extends AbstractMessageTransformer
29 {
30 public HttpRequestBodyToParamMap()
31 {
32 registerSourceType(DataTypeFactory.OBJECT);
33 setReturnDataType(DataTypeFactory.OBJECT);
34 }
35
36 @Override
37 public Object transformMessage(MuleMessage message, String outputEncoding) throws TransformerException
38 {
39 Map<String, Object> paramMap = new HashMap<String, Object>();
40
41 try
42 {
43 String httpMethod = message.getInboundProperty("http.method");
44 String contentType = message.getInboundProperty("Content-Type");
45
46 boolean isGet = HttpConstants.METHOD_GET.equalsIgnoreCase(httpMethod);
47 boolean isPost = HttpConstants.METHOD_POST.equalsIgnoreCase(httpMethod);
48 boolean isUrlEncoded = false;
49 if (contentType != null)
50 {
51 isUrlEncoded = contentType.startsWith("application/x-www-form-urlencoded");
52 }
53
54 if (!(isGet || (isPost && isUrlEncoded)))
55 {
56 throw new Exception("The HTTP method or content type is unsupported!");
57 }
58
59 String queryString = null;
60 if (isGet)
61 {
62 URI uri = new URI(message.getPayloadAsString(outputEncoding));
63 queryString = uri.getRawQuery();
64 }
65 else if (isPost)
66 {
67 queryString = new String(message.getPayloadAsBytes());
68 }
69
70 if (StringUtils.isNotBlank(queryString))
71 {
72 addQueryStringToParameterMap(queryString, paramMap, outputEncoding);
73 }
74 }
75 catch (Exception e)
76 {
77 throw new TransformerException(this, e);
78 }
79
80 return paramMap;
81 }
82
83 protected void addQueryStringToParameterMap(String queryString, Map<String, Object> paramMap,
84 String outputEncoding) throws Exception
85 {
86 String[] pairs = queryString.split("&");
87 for (String pair : pairs)
88 {
89 String[] nameValue = pair.split("=");
90 if (nameValue.length == 2)
91 {
92 URLCodec codec = new URLCodec(outputEncoding);
93 String key = codec.decode(nameValue[0]);
94 String value = codec.decode(nameValue[1]);
95 addToParameterMap(paramMap, key, value);
96 }
97 }
98 }
99
100 @SuppressWarnings("unchecked")
101 protected void addToParameterMap(Map<String, Object> paramMap, String key, String value)
102 {
103 Object existingValue = paramMap.get(key);
104 if (existingValue != null)
105 {
106 List<Object> values;
107 if (existingValue instanceof List<?>)
108 {
109 values = (List<Object>) existingValue;
110 }
111 else
112 {
113 values = Arrays.asList(existingValue);
114 }
115
116 values.add(value);
117 paramMap.put(key, values);
118 }
119 else
120 {
121 paramMap.put(key, value);
122 }
123 }
124
125 @Override
126 public boolean isAcceptNull()
127 {
128 return false;
129 }
130 }