/*******************************************************************************
    * Copyright (c) 2011 Intalio, Inc.
    * ======================================================================
    * 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.websocket;
  
  import java.io.IOException;
  import javax.servlet.http.HttpServletRequest;
  import javax.servlet.http.HttpServletResponse;
  
  import org.eclipse.jetty.io.EndPoint;
  
  public class WebSocketServletConnectionD06 extends WebSocketConnectionD06 implements WebSocketServletConnection
  {
      private final WebSocketFactory factory;
  
      public WebSocketServletConnectionD06(WebSocketFactory factory, WebSocket websocket, EndPoint endpoint, WebSocketBuffers buffers, long timestamp, int maxIdleTime, String protocol)
              throws IOException
      {
          super(websocket,endpoint,buffers,timestamp,maxIdleTime,protocol);
          this.factory = factory;
      }
  
      /* ------------------------------------------------------------ */
      public void handshake(HttpServletRequest request, HttpServletResponse response, String subprotocol) throws IOException
      {
          String key = request.getHeader("Sec-WebSocket-Key");
  
          response.setHeader("Upgrade","WebSocket");
          response.addHeader("Connection","Upgrade");
          response.addHeader("Sec-WebSocket-Accept",hashKey(key));
          if (subprotocol!=null)
          {
              response.addHeader("Sec-WebSocket-Protocol",subprotocol);
          }
  
          response.sendError(101);
  
          onFrameHandshake();
          onWebSocketOpen();
      }
  
      @Override
      public void onClose()
      {
          super.onClose();
          factory.removeConnection(this);
      }
  }