1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.eclipse.jetty.rewrite.handler;
20
21 import java.io.IOException;
22
23 import javax.servlet.http.HttpServletRequest;
24 import javax.servlet.http.HttpServletResponse;
25
26 import org.eclipse.jetty.server.Request;
27 import org.eclipse.jetty.util.ArrayUtil;
28 import org.eclipse.jetty.util.URIUtil;
29 import org.eclipse.jetty.util.log.Log;
30 import org.eclipse.jetty.util.log.Logger;
31
32
33
34
35
36 public class RuleContainer extends Rule
37 {
38 private static final Logger LOG = Log.getLogger(RuleContainer.class);
39
40 protected Rule[] _rules;
41
42 protected String _originalPathAttribute;
43 protected boolean _rewriteRequestURI=true;
44 protected boolean _rewritePathInfo=true;
45
46
47
48
49
50
51 public Rule[] getRules()
52 {
53 return _rules;
54 }
55
56
57
58
59
60
61 public void setRules(Rule[] rules)
62 {
63 _rules = rules;
64 }
65
66
67
68
69
70
71 public void addRule(Rule rule)
72 {
73 _rules = ArrayUtil.addToArray(_rules,rule,Rule.class);
74 }
75
76
77
78
79
80
81
82 public boolean isRewriteRequestURI()
83 {
84 return _rewriteRequestURI;
85 }
86
87
88
89
90
91
92 public void setRewriteRequestURI(boolean rewriteRequestURI)
93 {
94 _rewriteRequestURI=rewriteRequestURI;
95 }
96
97
98
99
100
101
102 public boolean isRewritePathInfo()
103 {
104 return _rewritePathInfo;
105 }
106
107
108
109
110
111
112 public void setRewritePathInfo(boolean rewritePathInfo)
113 {
114 _rewritePathInfo=rewritePathInfo;
115 }
116
117
118
119
120
121
122 public String getOriginalPathAttribute()
123 {
124 return _originalPathAttribute;
125 }
126
127
128
129
130
131
132 public void setOriginalPathAttribute(String originalPathAttribte)
133 {
134 _originalPathAttribute=originalPathAttribte;
135 }
136
137
138
139
140
141
142
143 @Override
144 public String matchAndApply(String target, HttpServletRequest request, HttpServletResponse response) throws IOException
145 {
146 return apply(target, request, response);
147 }
148
149
150
151
152
153
154
155
156
157 protected String apply(String target, HttpServletRequest request, HttpServletResponse response) throws IOException
158 {
159 boolean original_set=_originalPathAttribute==null;
160
161 for (Rule rule : _rules)
162 {
163 String applied=rule.matchAndApply(target,request, response);
164 if (applied!=null)
165 {
166 LOG.debug("applied {}",rule);
167 LOG.debug("rewrote {} to {}",target,applied);
168 if (!original_set)
169 {
170 original_set=true;
171 request.setAttribute(_originalPathAttribute, target);
172 }
173
174 if (_rewriteRequestURI)
175 {
176 String encoded=URIUtil.encodePath(applied);
177 if (rule instanceof Rule.ApplyURI)
178 ((Rule.ApplyURI)rule).applyURI((Request)request,((Request)request).getRequestURI(), encoded);
179 else
180 ((Request)request).setURIPathQuery(encoded);
181 }
182
183 if (_rewritePathInfo)
184 ((Request)request).setPathInfo(applied);
185
186 target=applied;
187
188 if (rule.isHandling())
189 {
190 LOG.debug("handling {}",rule);
191 Request.getBaseRequest(request).setHandled(true);
192 }
193
194 if (rule.isTerminating())
195 {
196 LOG.debug("terminating {}",rule);
197 break;
198 }
199 }
200 }
201
202 return target;
203 }
204 }