//
   //  ========================================================================
   //  Copyright (c) 1995-2013 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.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);
      }
  }