001package org.apache.maven.wagon.tck.http.fixture;
002
003/*
004 * Licensed to the Apache Software Foundation (ASF) under one
005 * or more contributor license agreements.  See the NOTICE file
006 * distributed with this work for additional information
007 * regarding copyright ownership.  The ASF licenses this file
008 * to you under the Apache License, Version 2.0 (the
009 * "License"); you may not use this file except in compliance
010 * with the License.  You may obtain a copy of the License at
011 *
012 *   http://www.apache.org/licenses/LICENSE-2.0
013 *
014 * Unless required by applicable law or agreed to in writing,
015 * software distributed under the License is distributed on an
016 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
017 * KIND, either express or implied.  See the License for the
018 * specific language governing permissions and limitations
019 * under the License.
020 */
021
022import javax.servlet.ServletException;
023import javax.servlet.http.HttpServlet;
024import javax.servlet.http.HttpServletRequest;
025import javax.servlet.http.HttpServletResponse;
026import java.io.IOException;
027
028public class RedirectionServlet
029    extends HttpServlet
030{
031
032    private static final long serialVersionUID = 1L;
033
034    private final String targetPath;
035
036    private final int code;
037
038    private final int maxRedirects;
039
040    private int redirectCount = 0;
041
042    private final String myPath;
043
044    public RedirectionServlet( final int code, final String targetPath )
045    {
046        this.code = code;
047        this.targetPath = targetPath;
048        this.maxRedirects = 1;
049        this.myPath = null;
050    }
051
052    public RedirectionServlet( final int code, final String myPath, final String targetPath, final int maxRedirects )
053    {
054        this.code = code;
055        this.myPath = myPath;
056        this.targetPath = targetPath;
057        this.maxRedirects = maxRedirects;
058    }
059
060    public int getRedirectCount()
061    {
062        return redirectCount;
063    }
064
065    @Override
066    protected void service( final HttpServletRequest req, final HttpServletResponse resp )
067        throws ServletException, IOException
068    {
069        redirectCount++;
070
071        if ( myPath == null )
072        {
073            resp.setStatus( code );
074            resp.setHeader( "Location", targetPath );
075        }
076        else if ( maxRedirects < 0 )
077        {
078            resp.setStatus( code );
079            resp.setHeader( "Location", myPath );
080        }
081        else if ( redirectCount <= maxRedirects )
082        {
083            resp.setStatus( code );
084            resp.setHeader( "Location", myPath + "/" + redirectCount );
085        }
086        else
087        {
088            resp.setStatus( code );
089            resp.setHeader( "Location", targetPath );
090        }
091    }
092
093}