// Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.
//

import "dart:io";
import "dart:scalarlist";

void testRequestResponseClientCloses(
    int totalConnections, int closeStatus, String closeReason) {
  HttpServer server = new HttpServer();
  HttpClient client = new HttpClient();

  server.listen("127.0.0.1", 0, backlog: totalConnections);

  // Create a web socket handler and set is as the HTTP server default
  // handler.
  WebSocketHandler wsHandler = new WebSocketHandler();
  wsHandler.onOpen = (WebSocketConnection conn) {
    var count = 0;
    conn.onMessage = (Object message) => conn.send(message);
    conn.onClosed = (status, reason) {
      Expect.equals(closeStatus == null
                    ? WebSocketStatus.NO_STATUS_RECEIVED
                    : closeStatus, status);
      Expect.equals(closeReason == null ? "" : closeReason, reason);
    };
  };
  server.defaultRequestHandler = wsHandler.onRequest;

  int closeCount = 0;
  String messageText = "Hello, world!";
  for (int i = 0; i < totalConnections; i++) {
    int messageCount = 0;
    HttpClientConnection conn = client.get("127.0.0.1", server.port, "/");
    WebSocketClientConnection wsconn = new WebSocketClientConnection(conn);
    wsconn.onOpen = () => wsconn.send(messageText);
    wsconn.onMessage = (message) {
      messageCount++;
      if (messageCount < 10) {
        Expect.equals(messageText, message);
        wsconn.send(message);
      } else {
        wsconn.close(closeStatus, closeReason);
      }
    };
    wsconn.onClosed = (status, reason) {
      Expect.equals(closeStatus == null
                    ? WebSocketStatus.NO_STATUS_RECEIVED
                    : closeStatus, status);
      Expect.equals("", reason);
      closeCount++;
      if (closeCount == totalConnections) {
        client.shutdown();
        server.close();
      }
    };
  }
}


void testRequestResponseServerCloses(
    int totalConnections, int closeStatus, String closeReason) {
  HttpServer server = new HttpServer();
  HttpClient client = new HttpClient();

  server.listen("127.0.0.1", 0, backlog: totalConnections);

  // Create a web socket handler and set is as the HTTP server default
  // handler.
  int closeCount = 0;
  WebSocketHandler wsHandler = new WebSocketHandler();
  wsHandler.onOpen = (WebSocketConnection conn) {
    String messageText = "Hello, world!";
    int messageCount = 0;
    conn.onMessage = (Object message) {
      messageCount++;
      if (messageCount < 10) {
        Expect.equals(messageText, message);
        conn.send(message);
      } else {
        conn.close(closeStatus, closeReason);
      }
    };
    conn.onClosed = (status, reason) {
      Expect.equals(closeStatus == null
                    ? WebSocketStatus.NO_STATUS_RECEIVED
                    : closeStatus, status);
      Expect.equals("", reason);
      closeCount++;
      if (closeCount == totalConnections) {
        client.shutdown();
        server.close();
      }
    };
    conn.send(messageText);
  };
  server.defaultRequestHandler = wsHandler.onRequest;

  for (int i = 0; i < totalConnections; i++) {
    HttpClientConnection conn = client.get("127.0.0.1", server.port, "/");
    WebSocketClientConnection wsconn = new WebSocketClientConnection(conn);
    wsconn.onMessage = (message) => wsconn.send(message);
    wsconn.onClosed = (status, reason) {
      Expect.equals(closeStatus == null
                    ? WebSocketStatus.NO_STATUS_RECEIVED
                    : closeStatus, status);
      Expect.equals(closeReason == null ? "" : closeReason, reason);
    };
  }
}


void testMessageLength(int messageLength) {
  HttpServer server = new HttpServer();
  HttpClient client = new HttpClient();

  server.listen("127.0.0.1", 0, backlog: 1);

  // Create a web socket handler and set is as the HTTP server default
  // handler.
  Uint8List originalMessage = new Uint8List(messageLength);
  WebSocketHandler wsHandler = new WebSocketHandler();
  wsHandler.onOpen = (WebSocketConnection conn) {
    conn.onMessage = (Object message) {
      Expect.listEquals(originalMessage, message);
      conn.send(message);
    };
    conn.onClosed = (status, reason) {
    };
  };
  server.defaultRequestHandler = wsHandler.onRequest;

  HttpClientConnection conn = client.get("127.0.0.1", server.port, "/");
  WebSocketClientConnection wsconn = new WebSocketClientConnection(conn);
  wsconn.onMessage = (message) {
    Expect.listEquals(originalMessage, message);
    wsconn.close();
  };
  wsconn.onClosed = (status, reason) {
    server.close();
  };
  wsconn.onOpen = () {
    wsconn.send(originalMessage);
  };
}


void testNoUpgrade() {
  HttpServer server = new HttpServer();
  HttpClient client = new HttpClient();

  server.listen("127.0.0.1", 0, backlog: 5);

  // Create a server which always responds with a redirect.
  server.defaultRequestHandler = (request, response) {
    response.statusCode = HttpStatus.MOVED_PERMANENTLY;
    response.outputStream.close();
  };

  HttpClientConnection conn = client.get("127.0.0.1", server.port, "/");
  conn.followRedirects = false;
  WebSocketClientConnection wsconn = new WebSocketClientConnection(conn);
  wsconn.onNoUpgrade = (response) {
    Expect.equals(HttpStatus.MOVED_PERMANENTLY, response.statusCode);
    client.shutdown();
    server.close();
  };
}


void testUsePOST() {
  HttpServer server = new HttpServer();
  HttpClient client = new HttpClient();

  server.listen("127.0.0.1", 0, backlog: 5);

  // Create a web socket handler and set is as the HTTP server default
  // handler.
  int closeCount = 0;
  WebSocketHandler wsHandler = new WebSocketHandler();
  wsHandler.onOpen = (WebSocketConnection conn) {
    Expect.fail("No connection expected");
  };
  server.defaultRequestHandler = wsHandler.onRequest;

  HttpClientConnection conn = client.post("127.0.0.1", server.port, "/");
  WebSocketClientConnection wsconn = new WebSocketClientConnection(conn);
  wsconn.onNoUpgrade = (response) {
    Expect.equals(HttpStatus.BAD_REQUEST, response.statusCode);
    client.shutdown();
    server.close();
  };
}


class WebSocketInfo {
  int messageCount = 0;
}


void testHashCode(int totalConnections) {
  HttpServer server = new HttpServer();
  HttpClient client = new HttpClient();
  Map connections = new Map();

  server.listen("127.0.0.1", 0, backlog: totalConnections);

  void handleMessage(conn, message) {
    var info = connections[conn];
    Expect.isNotNull(info);
    info.messageCount++;
    if (info.messageCount < 10) {
      conn.send(message);
    } else {
      conn.close();
    }
  }

  // Create a web socket handler and set is as the HTTP server default
  // handler.
  int closeCount = 0;
  WebSocketHandler wsHandler = new WebSocketHandler();
  wsHandler.onOpen = (WebSocketConnection conn) {
    connections[conn] = new WebSocketInfo();
    String messageText = "Hello, world!";
    conn.onMessage = (Object message) {
      handleMessage(conn, message);
    };
    conn.onClosed = (status, reason) {
      closeCount++;
      var info = connections[conn];
      Expect.equals(10, info.messageCount);
      if (closeCount == totalConnections) {
        client.shutdown();
        server.close();
      }
    };
    conn.send(messageText);
  };
  server.defaultRequestHandler = wsHandler.onRequest;

  for (int i = 0; i < totalConnections; i++) {
    HttpClientConnection conn = client.get("127.0.0.1", server.port, "/");
    WebSocketClientConnection wsconn = new WebSocketClientConnection(conn);
    wsconn.onMessage = (message) => wsconn.send(message);
  }
}


void testW3CInterface(
    int totalConnections, int closeStatus, String closeReason) {
  HttpServer server = new HttpServer();

  server.listen("127.0.0.1", 0, backlog: totalConnections);

  // Create a web socket handler and set is as the HTTP server default
  // handler.
  int closeCount = 0;
  WebSocketHandler wsHandler = new WebSocketHandler();
  wsHandler.onOpen = (WebSocketConnection conn) {
    String messageText = "Hello, world!";
    int messageCount = 0;
    conn.onMessage = (Object message) {
      messageCount++;
      if (messageCount < 10) {
        Expect.equals(messageText, message);
        conn.send(message);
      } else {
        conn.close(closeStatus, closeReason);
      }
    };
    conn.onClosed = (status, reason) {
      Expect.equals(closeStatus, status);
      Expect.equals("", reason);
      closeCount++;
      if (closeCount == totalConnections) {
        server.close();
      }
    };
    conn.send(messageText);
  };
  server.defaultRequestHandler = wsHandler.onRequest;

  void webSocketConnection() {
    bool onopenCalled = false;
    int onmessageCalled = 0;
    bool oncloseCalled = false;

    var websocket = new WebSocket("ws://127.0.0.1:${server.port}");
    Expect.equals(WebSocket.CONNECTING, websocket.readyState);
    websocket.onopen = () {
      Expect.isFalse(onopenCalled);
      Expect.equals(0, onmessageCalled);
      Expect.isFalse(oncloseCalled);
      onopenCalled = true;
      Expect.equals(WebSocket.OPEN, websocket.readyState);
    };
    websocket.onmessage = (event) {
      onmessageCalled++;
      Expect.isTrue(onopenCalled);
      Expect.isFalse(oncloseCalled);
      Expect.equals(WebSocket.OPEN, websocket.readyState);
      websocket.send(event.data);
    };
    websocket.onclose = (event) {
      Expect.isTrue(onopenCalled);
      Expect.equals(10, onmessageCalled);
      Expect.isFalse(oncloseCalled);
      oncloseCalled = true;
      Expect.isTrue(event.wasClean);
      Expect.equals(3002, event.code);
      Expect.equals("Got tired", event.reason);
      Expect.equals(WebSocket.CLOSED, websocket.readyState);
    };
  }

  for (int i = 0; i < totalConnections; i++) {
    webSocketConnection();
  }
}


main() {
  testRequestResponseClientCloses(2, null, null);
  testRequestResponseClientCloses(2, 3001, null);
  testRequestResponseClientCloses(2, 3002, "Got tired");
  testRequestResponseServerCloses(2, null, null);
  testRequestResponseServerCloses(2, 3001, null);
  testRequestResponseServerCloses(2, 3002, "Got tired");
  testMessageLength(125);
  testMessageLength(126);
  testMessageLength(127);
  testMessageLength(65535);
  testMessageLength(65536);
  testNoUpgrade();
  testUsePOST();
  testHashCode(2);

  testW3CInterface(2, 3002, "Got tired");
}