refactored websockets

[src/app-framework-binder.git] / src / afb-websock.c

diff --git a/src/afb-websock.c b/src/afb-websock.c
index 1f80071..d92c7b0 100644 (file)
--- a/src/afb-websock.c
+++ b/src/afb-websock.c
@@ -2,8 +2,6 @@
  * Copyright 2016 IoT.bzh
  * Author: José Bollo <jose.bollo@iot.bzh>
  *
- * Inspired by the work of 
- *
  * 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
@@ -30,6 +28,8 @@
 
 #include "websock.h"
 
+#include "afb-ws-json.h"
+
 #include "afb-req-itf.h"
 #include "afb-method.h"
 #include "afb-hreq.h"
@@ -171,7 +171,7 @@ static struct websock_itf aws_itf = {
        .on_text = (void*)aws_on_content,
        .on_binary = (void*)aws_on_content,
        .on_continue = (void*)aws_on_content,
-       .on_extension = NULL,
+       .on_extension = NULL
 };
 
 struct afb_wsreq
@@ -187,12 +187,12 @@ struct afb_wsreq
 struct afb_websock
 {
        int fd;
+       struct afb_wsreq *requests;
        struct MHD_Connection *connection;
+       struct AFB_clientCtx *context;
        struct websock *ws;
        struct upoll *up;
-       struct AFB_clientCtx *context;
        struct json_tokener *tokener;
-       struct afb_wsreq *requests;
 };
 
 static struct afb_arg wsreq_get(struct afb_wsreq *wsreq, const char *name);
@@ -214,6 +214,15 @@ static const struct afb_req_itf wsreq_itf = {
 };
 
 struct afb_websock *afb_websock_create(struct afb_hreq *hreq)
+{
+       return (void*)afb_ws_json_create(
+                       dup(MHD_get_connection_info(hreq->connection,MHD_CONNECTION_INFO_CONNECTION_FD)->connect_fd),
+                       afb_hreq_context(hreq),
+                       (void*)MHD_resume_connection,
+                       hreq->connection);
+}
+
+struct afb_websock *_afb_websock_create(struct afb_hreq *hreq)
 {
        int fd;
        struct afb_websock *result;
@@ -228,8 +237,9 @@ struct afb_websock *afb_websock_create(struct afb_hreq *hreq)
        if (result == NULL)
                goto error;
 
-       result->connection = hreq->connection;
        result->fd = fd;
+       result->requests = NULL;
+       result->connection = hreq->connection;
        result->context = ctxClientGet(afb_hreq_context(hreq));
        if (result->context == NULL)
                goto error2;