/**
 * 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.config.Configuration;
import net.sf.sahi.request.HttpRequest;
import net.sf.sahi.response.HttpFileResponse;
import net.sf.sahi.response.HttpResponse;
import net.sf.sahi.response.NoCacheHttpResponse;
import net.sf.sahi.util.FileIsDirectoryException;
import net.sf.sahi.util.FileNotFoundRuntimeException;
import org.apache.log4j.Logger;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;


/**
 * User: nraman Date: May 13, 2005 Time: 7:06:11 PM To
 */
public class WebProcessor implements Runnable {
  private Socket client;
  private static Logger logger = Logger.getLogger(WebProcessor.class);

  public WebProcessor(Socket client) {
    this.client = client;
  }

  public void run() {
    String uri = null;
    try {
      HttpRequest requestFromBrowser = getRequestFromBrowser();
      uri = requestFromBrowser.uri();
      if (uri.indexOf("/dyn/stopserver") != -1) {
        sendResponseToBrowser(new NoCacheHttpResponse(200, "OK", "Killing Server"));
        System.exit(1);
      }
      String fileName = fileNamefromURI(uri);
      sendResponseToBrowser(new HttpFileResponse(fileName, null, false, false));
    } catch (FileIsDirectoryException dirEx) {
      try {
        if ("/".equals(uri)) uri = "/demo";
        if (uri.endsWith("/")) uri = uri.substring(0, uri.length() - 1);
        sendResponseToBrowser(new NoCacheHttpResponse(200, "OK", "<script>location.href='" + uri + "/index.htm'</script>"));
      } catch (IOException e) {
        logger.warn(dirEx.getMessage());
      }
      logger.warn(dirEx.getMessage());
    } catch (FileNotFoundRuntimeException fnfre) {
      try {
        sendResponseToBrowser(new NoCacheHttpResponse(404, "FileNotFound", "<html><h2>404 File Not Found</h2></html>"));
      } catch (IOException e) {
        logger.warn(fnfre.getMessage());
      }
      logger.warn(fnfre.getMessage());
    } catch (Exception e) {
      logger.warn(e.getMessage());
    } finally {
      try {
        client.close();
      } catch (IOException e) {
        logger.error(e.getMessage());
      }
    }
  }

  private String fileNamefromURI(String uri) {
    StringBuffer sb = new StringBuffer();
    sb.append(Configuration.getHtdocsRoot());
    sb.append(uri.substring(uri.indexOf("/")));
    logger.debug(sb.toString());
    return sb.toString();
  }

  private HttpRequest getRequestFromBrowser() throws IOException {
    InputStream in = client.getInputStream();
    return new HttpRequest(in);
  }

  protected void sendResponseToBrowser(HttpResponse responseFromHost)
    throws IOException {
    OutputStream outputStreamToBrowser = client.getOutputStream();
    outputStreamToBrowser.write(responseFromHost.rawHeaders());
    outputStreamToBrowser.write(responseFromHost.data());
  }

  protected Socket client() {
    return client;
  }
}