1 files changed, 111 insertions, 0 deletions
diff --git a/jetty-rewrite/src/main/java/org/eclipse/jetty/rewrite/handler/VirtualHostRuleContainer.java b/jetty-rewrite/src/main/java/org/eclipse/jetty/rewrite/handler/VirtualHostRuleContainer.java
new file mode 100644
index 0000000000..eabb07837e
--- /dev/null
+++ b/jetty-rewrite/src/main/java/org/eclipse/jetty/rewrite/handler/VirtualHostRuleContainer.java
@@ -0,0 +1,111 @@
+// ========================================================================
+// Copyright (c) 2004-2009 Mort Bay Consulting Pty. Ltd.
+// ------------------------------------------------------------------------
+// All rights reserved. This program and the accompanying materials
+// are made available under the terms of the Eclipse Public License v1.0
+// and Apache License v2.0 which accompanies this distribution.
+// The Eclipse Public License is available at 
+// http://www.eclipse.org/legal/epl-v10.html
+// The Apache License v2.0 is available at
+// http://www.opensource.org/licenses/apache2.0.php
+// You may elect to redistribute this code under either of these licenses. 
+// ========================================================================
+
+package org.eclipse.jetty.rewrite.handler;
+
+import java.io.IOException;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import org.eclipse.jetty.util.LazyList;
+
+/**
+ * Groups rules that apply only to a specific virtual host
+ * or sets of virtual hosts
+ * 
+ *  
+ */
+
+public class VirtualHostRuleContainer extends RuleContainer
+{
+    private String[] _virtualHosts;
+
+    /* ------------------------------------------------------------ */
+    /** Set the virtual hosts that the rules within this container will apply to
+     * @param virtualHosts Array of virtual hosts that the rules within this container are applied to. 
+     * A null hostname or null/empty array means any hostname is acceptable.
+     */
+    public void setVirtualHosts( String[] virtualHosts )
+    {
+        if ( virtualHosts == null )
+        {
+            _virtualHosts = virtualHosts;
+        } 
+        else 
+        {
+            _virtualHosts = new String[virtualHosts.length];
+            for ( int i = 0; i < virtualHosts.length; i++ )
+                _virtualHosts[i] = normalizeHostname( virtualHosts[i]);
+        }
+    }
+
+    /* ------------------------------------------------------------ */
+    /** Get the virtual hosts that the rules within this container will apply to
+     * @param virtualHosts Array of virtual hosts that the rules within this container are applied to. 
+     * A null hostname or null/empty array means any hostname is acceptable.
+     */
+    public String[] getVirtualHosts()
+    {
+        return _virtualHosts;
+    }
+    
+    /* ------------------------------------------------------------ */
+    /**
+     * @param virtualHost add a virtual host to the existing list of virtual hosts
+     * A null hostname means any hostname is acceptable 
+     */
+    public void addVirtualHost(String virtualHost)
+    {
+        _virtualHosts = (String[])LazyList.addToArray(_virtualHosts,virtualHost,String.class);
+    }
+
+    /**
+     * Process the contained rules if the request is applicable to the virtual hosts of this rule
+     * @param target target field to pass on to the contained rules
+     * @param request request object to pass on to the contained rules
+     * @param response response object to pass on to the contained rules
+     */
+    @Override
+    public String matchAndApply(String target, HttpServletRequest request, HttpServletResponse response) throws IOException
+    {
+        if(_virtualHosts != null && _virtualHosts.length > 0 )
+        {
+            String requestHost = normalizeHostname( request.getServerName() );
+            for( String ruleHost : _virtualHosts )
+            {
+                if(ruleHost == null || ruleHost.equalsIgnoreCase(requestHost)
+                        || (ruleHost.startsWith("*.") && ruleHost.regionMatches(true,2,requestHost,requestHost.indexOf(".")+1,ruleHost.length()-2)))
+                    return apply(target, request, response);
+            }
+        }
+        else
+        {
+            return apply(target, request, response);
+        }
+        return null;
+    }
+
+    /* ------------------------------------------------------------ */
+    private String normalizeHostname( String host )
+    {
+        if ( host == null )
+            return null;
+        
+        if ( host.endsWith( "." ) )
+            return host.substring( 0, host.length() -1);
+      
+            return host;
+    }
+
+}