LocalRequestProcessor.java

/**
 * Sahi - Web Automation and Test Tool
 *
 * Copyright  2006  V Narayan Raman
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package net.sf.sahi;

import net.sf.sahi.command.CommandExecuter;
import net.sf.sahi.command.Driver;
import net.sf.sahi.command.Hits;
import net.sf.sahi.config.Configuration;
import net.sf.sahi.report.LogViewer;
import net.sf.sahi.request.HttpRequest;
import net.sf.sahi.response.HttpFileResponse;
import net.sf.sahi.response.HttpModifiedResponse2;
import net.sf.sahi.response.HttpResponse;
import net.sf.sahi.response.NoCacheHttpResponse;
import net.sf.sahi.response.SimpleHttpResponse;
import net.sf.sahi.util.URLParser;

public class LocalRequestProcessor {

  // FIXME there has to be a better way to do this, this does not scale well and is hard to maintain
  public HttpResponse getLocalResponse(String uri, HttpRequest requestFromBrowser) throws Exception {
    HttpResponse httpResponse = new NoCacheHttpResponse("");
    if (uri.indexOf("/dyn/pro/") != -1) {
//        	httpResponse = handleDyn(uri, requestFromBrowser, httpResponse, true);
      httpResponse = new SimpleHttpResponse("This feature is available only in Sahi Pro");
    } else if (uri.indexOf("/dyn/") != -1) {
      httpResponse = handleDyn(uri, requestFromBrowser, httpResponse, false);
    } else if (uri.indexOf("/scripts/") != -1) {
      String fileName = URLParser.scriptFileNamefromURI(
        requestFromBrowser.uri(), "/scripts/");
      httpResponse = new HttpFileResponse(fileName, null, false, false);
    } else if (uri.indexOf("/spr/") != -1) {
      String fileName = URLParser.fileNamefromURI(requestFromBrowser.uri());
      httpResponse = new HttpFileResponse(fileName, null, true, true);
    } else if (uri.indexOf("/sprm/") != -1) {
      String fileName = URLParser.fileNamefromURI(requestFromBrowser.uri().replace("/sprm/", "/spr/"));
      httpResponse = new HttpModifiedResponse2(new HttpFileResponse(fileName, null, true, true), requestFromBrowser.isSSL(), requestFromBrowser.fileExtension());
    } else if (uri.indexOf("/logs") != -1) {
      httpResponse = new NoCacheHttpResponse(LogViewer.getLogsList(Configuration.getPlayBackLogsRoot()));
    } else if (uri.indexOf("/userdata") != -1) {
      httpResponse = new HttpFileResponse(Configuration.getRootCaPath());
    } else {
      httpResponse = getLocalResponse(Driver.INITIALIZE_CALL, requestFromBrowser);
    }
    return httpResponse;
  }

  private HttpResponse handleDyn(String uri, HttpRequest requestFromBrowser,
                                 HttpResponse httpResponse, boolean isPro) throws Exception {
    String command = URLParser.getCommandFromUri(uri, isPro ? "/dyn/pro/" : "/dyn/");
    Hits.increment(command);
    if (uri.indexOf("/stopserver") != -1) {
//                Proxy.stopCurrentIntance();
      System.exit(1);
    } else if (command != null) {
      httpResponse = new CommandExecuter(command, requestFromBrowser, isPro).execute();
    }
    return httpResponse;
  }
}