Refactoring requests and context handling
[src/app-framework-binder.git] / src / afb-ws-json.c
index 6fc6e3b..781722f 100644 (file)
 #include "session.h"
 #include "afb-req-itf.h"
 #include "afb-apis.h"
+#include "afb-context.h"
 
-static void aws_on_close(struct afb_ws_json *ws, uint16_t code, char *text, size_t size);
+static void aws_on_hangup(struct afb_ws_json *ws);
 static void aws_on_text(struct afb_ws_json *ws, char *text, size_t size);
 
 static struct afb_ws_itf aws_itf = {
-       .on_close = (void*)aws_on_close,
-       .on_text = (void*)aws_on_text,
-       .on_binary = NULL,
+       .on_hangup = (void*)aws_on_hangup,
+       .on_text = (void*)aws_on_text
 };
 
 struct afb_wsreq;
@@ -48,24 +48,23 @@ struct afb_ws_json
        void (*cleanup)(void*);
        void *cleanup_closure;
        struct afb_wsreq *requests;
-       struct AFB_clientCtx *context;
+       struct AFB_clientCtx *session;
        struct json_tokener *tokener;
        struct afb_ws *ws;
 };
 
-
 static void aws_send_event(struct afb_ws_json *ws, const char *event, struct json_object *object);
 
-static const struct afb_event_sender_itf event_sender_itf = {
+static const struct afb_event_listener_itf event_listener_itf = {
        .send = (void*)aws_send_event
 };
 
-struct afb_ws_json *afb_ws_json_create(int fd, struct AFB_clientCtx *context, void (*cleanup)(void*), void *cleanup_closure)
+struct afb_ws_json *afb_ws_json_create(int fd, struct AFB_clientCtx *session, void (*cleanup)(void*), void *cleanup_closure)
 {
        struct afb_ws_json *result;
 
        assert(fd >= 0);
-       assert(context != NULL);
+       assert(session != NULL);
 
        result = malloc(sizeof * result);
        if (result == NULL)
@@ -74,8 +73,8 @@ struct afb_ws_json *afb_ws_json_create(int fd, struct AFB_clientCtx *context, vo
        result->cleanup = cleanup;
        result->cleanup_closure = cleanup_closure;
        result->requests = NULL;
-       result->context = ctxClientGet(context);
-       if (result->context == NULL)
+       result->session = ctxClientAddRef(session);
+       if (result->session == NULL)
                goto error2;
 
        result->tokener = json_tokener_new();
@@ -86,17 +85,17 @@ struct afb_ws_json *afb_ws_json_create(int fd, struct AFB_clientCtx *context, vo
        if (result->ws == NULL)
                goto error4;
 
-       if (0 > ctxClientEventSenderAdd(result->context, (struct afb_event_sender){ .itf = &event_sender_itf, .closure = result }))
+       if (0 > ctxClientEventListenerAdd(result->session, (struct afb_event_listener){ .itf = &event_listener_itf, .closure = result }))
                goto error5;
 
        return result;
 
 error5:
-       /* TODO */
+       afb_ws_destroy(result->ws);
 error4:
        json_tokener_free(result->tokener);
 error3:
-       ctxClientPut(result->context);
+       ctxClientUnref(result->session);
 error2:
        free(result);
 error:
@@ -104,10 +103,15 @@ error:
        return NULL;
 }
 
-static void aws_on_close(struct afb_ws_json *ws, uint16_t code, char *text, size_t size)
+static void aws_on_hangup(struct afb_ws_json *ws)
 {
-       /* do nothing but free the text */
-       free(text);
+       ctxClientEventListenerRemove(ws->session, (struct afb_event_listener){ .itf = &event_listener_itf, .closure = ws });
+       afb_ws_destroy(ws->ws);
+       json_tokener_free(ws->tokener);
+       if (ws->cleanup != NULL)
+               ws->cleanup(ws->cleanup_closure);
+       ctxClientUnref(ws->session);
+       free(ws);
 }
 
 #define CALL 2
@@ -117,6 +121,8 @@ static void aws_on_close(struct afb_ws_json *ws, uint16_t code, char *text, size
 
 struct afb_wsreq
 {
+       struct afb_context context;
+       int refcount;
        struct afb_ws_json *aws;
        struct afb_wsreq *next;
        char *text;
@@ -135,15 +141,14 @@ struct afb_wsreq
        struct json_object *root;
 };
 
+static void wsreq_addref(struct afb_wsreq *wsreq);
+static void wsreq_unref(struct afb_wsreq *wsreq);
 static struct json_object *wsreq_json(struct afb_wsreq *wsreq);
 static struct afb_arg wsreq_get(struct afb_wsreq *wsreq, const char *name);
 static void wsreq_fail(struct afb_wsreq *wsreq, const char *status, const char *info);
 static void wsreq_success(struct afb_wsreq *wsreq, struct json_object *obj, const char *info);
 static const char *wsreq_raw(struct afb_wsreq *wsreq, size_t *size);
-static void wsreq_send(struct afb_wsreq *wsreq, char *buffer, size_t size);
-static int wsreq_session_create(struct afb_wsreq *wsreq);
-static int wsreq_session_check(struct afb_wsreq *wsreq, int refresh);
-static void wsreq_session_close(struct afb_wsreq *wsreq);
+static void wsreq_send(struct afb_wsreq *wsreq, const char *buffer, size_t size);
 
 
 static const struct afb_req_itf wsreq_itf = {
@@ -153,12 +158,10 @@ static const struct afb_req_itf wsreq_itf = {
        .fail = (void*)wsreq_fail,
        .raw = (void*)wsreq_raw,
        .send = (void*)wsreq_send,
-       .session_create = (void*)wsreq_session_create,
-       .session_check = (void*)wsreq_session_check,
-       .session_close = (void*)wsreq_session_close,
        .context_get = (void*)afb_context_get,
-       .context_set = (void*)afb_context_set
-
+       .context_set = (void*)afb_context_set,
+       .addref = (void*)wsreq_addref,
+       .unref = (void*)wsreq_unref
 };
 
 static int aws_wsreq_parse(struct afb_wsreq *r, char *text, size_t size)
@@ -281,14 +284,6 @@ static int aws_wsreq_parse(struct afb_wsreq *r, char *text, size_t size)
        /* done */
        r->text = text;
        r->size = size;
-fprintf(stderr, "\n\nONTEXT([%d, %.*s, %.*s/%.*s, %.*s, %.*s])\n\n",
-       r->code,
-       (int)r->idlen, r->id,
-       (int)r->apilen, r->api,
-       (int)r->verblen, r->verb,
-       (int)r->objlen, r->obj,
-       (int)r->toklen, r->tok
-);
        return 1;
 
 bad_header:
@@ -310,23 +305,44 @@ static void aws_on_text(struct afb_ws_json *ws, char *text, size_t size)
                goto bad_header;
 
        /* fill and record the request */
+       if (wsreq->tok != NULL)
+               wsreq->tok[wsreq->toklen] = 0;
+       afb_context_init(&wsreq->context, ws->session, wsreq->tok);
+       if (!wsreq->context.invalidated)
+               wsreq->context.validated = 1;
+       wsreq->refcount = 1;
        wsreq->aws = ws;
        wsreq->next = ws->requests;
        ws->requests = wsreq;
 
-       r.req_closure = wsreq;
+       r.closure = wsreq;
        r.itf = &wsreq_itf;
-       afb_apis_call(r, ws->context, wsreq->api, wsreq->apilen, wsreq->verb, wsreq->verblen);
+       afb_apis_call(r, &wsreq->context, wsreq->api, wsreq->apilen, wsreq->verb, wsreq->verblen);
+       wsreq_unref(wsreq);
        return;
 
 bad_header:
        free(wsreq);
 alloc_error:
        free(text);
-       afb_ws_close(ws->ws, 1008);
+       afb_ws_close(ws->ws, 1008, NULL);
        return;
 }
 
+static void wsreq_addref(struct afb_wsreq *wsreq)
+{
+       wsreq->refcount++;
+}
+
+static void wsreq_unref(struct afb_wsreq *wsreq)
+{
+       if (--wsreq->refcount == 0) {
+               afb_context_disconnect(&wsreq->context);
+               free(wsreq->text);
+               free(wsreq);
+       }
+}
+
 static struct json_object *wsreq_json(struct afb_wsreq *wsreq)
 {
        struct json_object *root = wsreq->root;
@@ -359,41 +375,9 @@ static struct afb_arg wsreq_get(struct afb_wsreq *wsreq, const char *name)
        return arg;
 }
 
-static int wsreq_session_create(struct afb_wsreq *wsreq)
-{
-       struct AFB_clientCtx *context = wsreq->aws->context;
-       if (context->created)
-               return 0;
-       return wsreq_session_check(wsreq, 1);
-}
-
-static int wsreq_session_check(struct afb_wsreq *wsreq, int refresh)
-{
-       struct AFB_clientCtx *context = wsreq->aws->context;
-
-       if (wsreq->tok == NULL)
-               return 0;
-
-       if (!ctxTokenCheckLen (context, wsreq->tok, wsreq->toklen))
-               return 0;
-
-       if (refresh) {
-               ctxTokenNew (context);
-       }
-
-       return 1;
-}
-
-static void wsreq_session_close(struct afb_wsreq *wsreq)
-{
-       struct AFB_clientCtx *context = wsreq->aws->context;
-       ctxClientClose(context);
-}
-
-static void aws_emit(struct afb_ws_json *aws, int code, const char *id, size_t idlen, struct json_object *data)
+static void aws_emit(struct afb_ws_json *aws, int code, const char *id, size_t idlen, struct json_object *data, const char *token)
 {
        json_object *msg;
-       const char *token;
        const char *txt;
 
        /* pack the message */
@@ -401,7 +385,6 @@ static void aws_emit(struct afb_ws_json *aws, int code, const char *id, size_t i
        json_object_array_add(msg, json_object_new_int(code));
        json_object_array_add(msg, json_object_new_string_len(id, (int)idlen));
        json_object_array_add(msg, data);
-       token = aws->context->token;
        if (token)
                json_object_array_add(msg, json_object_new_string(token));
 
@@ -413,8 +396,8 @@ static void aws_emit(struct afb_ws_json *aws, int code, const char *id, size_t i
 
 static void wsreq_reply(struct afb_wsreq *wsreq, int retcode, const char *status, const char *info, json_object *resp)
 {
-       aws_emit(wsreq->aws, retcode, wsreq->id, wsreq->idlen, afb_msg_json_reply(status, info, resp, NULL, NULL));
-       /* TODO eliminates the wsreq */
+       const char *token = afb_context_sent_token(&wsreq->context);
+       aws_emit(wsreq->aws, retcode, wsreq->id, wsreq->idlen, afb_msg_json_reply(status, info, resp, token, NULL), token);
 }
 
 static void wsreq_fail(struct afb_wsreq *wsreq, const char *status, const char *info)
@@ -433,13 +416,13 @@ static const char *wsreq_raw(struct afb_wsreq *wsreq, size_t *size)
        return wsreq->obj;
 }
 
-static void wsreq_send(struct afb_wsreq *wsreq, char *buffer, size_t size)
+static void wsreq_send(struct afb_wsreq *wsreq, const char *buffer, size_t size)
 {
        afb_ws_text(wsreq->aws->ws, buffer, size);
 }
 
 static void aws_send_event(struct afb_ws_json *aws, const char *event, struct json_object *object)
 {
-       aws_emit(aws, EVENT, event, strlen(event), afb_msg_json_event(event, object));
+       aws_emit(aws, EVENT, event, strlen(event), afb_msg_json_event(event, object), NULL);
 }