2 * Sone - RequestChangeFilter.java - Copyright © 2010 David Roden
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 package net.pterodactylus.sone.template;
20 import java.io.UnsupportedEncodingException;
22 import java.net.URISyntaxException;
23 import java.net.URLEncoder;
24 import java.util.Collection;
25 import java.util.HashMap;
27 import java.util.Map.Entry;
29 import net.pterodactylus.sone.web.page.Page.Request;
30 import net.pterodactylus.util.template.DataProvider;
31 import net.pterodactylus.util.template.Filter;
34 * This filter expects a {@link Request} as input and outputs a {@link URI} that
35 * is modified by the parameters. The name of the parameter is handed in as
36 * “name”, the value may either be stored in “value”, or in a template variable
37 * whose key is stored in “key”.
39 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
41 public class RequestChangeFilter implements Filter {
47 public Object format(DataProvider dataProvider, Object data, Map<String, String> parameters) {
48 Request request = (Request) data;
49 String name = parameters.get("name");
50 String nameKey = parameters.get("nameKey");
51 if (nameKey != null) {
52 name = String.valueOf(dataProvider.getData(nameKey));
54 String key = parameters.get("key");
57 value = String.valueOf(dataProvider.getData(key));
60 value = parameters.get("value");
63 return request.getUri();
66 Map<String, String> values = new HashMap<String, String>();
67 Collection<String> parameterNames = request.getHttpRequest().getParameterNames();
68 for (String parameterName : parameterNames) {
69 values.put(parameterName, request.getHttpRequest().getParam(parameterName));
71 values.put(name, value);
73 StringBuilder query = new StringBuilder();
75 for (Entry<String, String> parameterEntry : values.entrySet()) {
76 query.append((query.length() == 0) ? '?' : '&');
77 query.append(URLEncoder.encode(parameterEntry.getKey(), "UTF-8"));
79 query.append(URLEncoder.encode(parameterEntry.getValue(), "UTF-8"));
81 String oldUri = request.getUri().toString();
82 int questionMark = oldUri.indexOf('?');
83 if (questionMark == -1) {
84 questionMark = oldUri.length();
86 URI u = new URI(oldUri.substring(0, questionMark) + query.toString());
88 } catch (UnsupportedEncodingException uee1) {
89 /* UTF-8 not supported? I don’t think so. */
90 } catch (URISyntaxException use1) {
91 use1.printStackTrace();