/*
 * websocket server
 *
 * @build make examples
 * @server bin/websocket_server_test 9999
 * @client bin/websocket_client_test ws://127.0.0.1:9999/
 * @python scripts/websocket_server.py
 * @js html/websocket_client.html
 *
 */

#include "WebSocketServer.h"
#include "EventLoop.h"
#include "htime.h"

using namespace hv;

/*
 * #define TEST_WSS 1
 *
 * @build ./configure --with-openssl && make clean && make
 *
 * @server bin/websocket_server_test 9999
 *
 * @client bin/websocket_client_test ws://127.0.0.1:9999/
 * bin/websocket_client_test wss://127.0.0.1:10000/
 *
 */
#define TEST_WSS 0

using namespace hv;

class MyContext {
public:
 MyContext() {
 printf("MyContext::MyContext()\n");
 timerID = INVALID_TIMER_ID;
 }
 ~MyContext() {
 printf("MyContext::~MyContext()\n");
 }

 int handleMessage(const std::string& msg, enum ws_opcode opcode) {
 printf("onmessage(type=%s len=%d): %.*s\n", opcode == WS_OPCODE_TEXT ? "text" : "binary",
 (int)msg.size(), (int)msg.size(), msg.data());
 return msg.size();
 }

 TimerID timerID;
};

int main(int argc, char** argv) {
 if (argc < 2) {
 printf("Usage: %s port\n", argv[0]);
 return -10;
 }
 int port = atoi(argv[1]);

 HttpService http;
 http.GET("/ping", [](const HttpContextPtr& ctx) {
 return ctx->send("pong");
 });

 WebSocketService ws;
 // ws.setPingInterval(10000);
 ws.onopen = [](const WebSocketChannelPtr& channel, const HttpRequestPtr& req) {
 printf("onopen: GET %s\n", req->Path().c_str());
 auto ctx = channel->newContextPtr<MyContext>();
 // send(time) every 1s
 ctx->timerID = setInterval(1000, [channel](TimerID id) {
 if (channel->isConnected() && channel->isWriteComplete()) {
 char str[DATETIME_FMT_BUFLEN] = {0};
 datetime_t dt = datetime_now();
 datetime_fmt(&dt, str);
 channel->send(str);
 }
 });
 };
 ws.onmessage = [](const WebSocketChannelPtr& channel, const std::string& msg) {
 auto ctx = channel->getContextPtr<MyContext>();
 ctx->handleMessage(msg, channel->opcode);
 };
 ws.onclose = [](const WebSocketChannelPtr& channel) {
 printf("onclose\n");
 auto ctx = channel->getContextPtr<MyContext>();
 if (ctx->timerID != INVALID_TIMER_ID) {
 killTimer(ctx->timerID);
 ctx->timerID = INVALID_TIMER_ID;
 }
 // channel->deleteContextPtr();
 };

 WebSocketServer server;
 server.port = port;
#if TEST_WSS
 server.https_port = port + 1;
 hssl_ctx_opt_t param;
 memset(&param, 0, sizeof(param));
 param.crt_file = "cert/server.crt";
 param.key_file = "cert/server.key";
 param.endpoint = HSSL_SERVER;
 if (server.newSslCtx(&param) != 0) {
 fprintf(stderr, "new SSL_CTX failed!\n");
 return -20;
 }
#endif
 server.registerHttpService(&http);
 server.registerWebSocketService(&ws);

 server.start();

 // press Enter to stop
 while (getchar() != '\n');
 return 0;
}