#include "facilities.h"
#include "cam.h"
#include "denm.h"

#include <itss-transport/BTPDataRequest.h>
#include <itss-transport/BTPDataIndication.h>
#include <itss-facilities/FacilitiesDataIndication.h>
#include <itss-facilities/FacilitiesDataRequest.h>

#include <camv2/CAM.h>
#include <denmv2/DENM.h>
#include <ivim/IVIM.h>

#include <it2s-btp.h>

#include <zmq.h>
#include <unistd.h>
#include <syslog.h>

#define syslog_info(msg, ...) syslog(LOG_INFO, msg, ##__VA_ARGS__)
#define syslog_emerg(msg, ...) syslog(LOG_EMERG, "%s:%d [" msg "]",  __func__, __LINE__, ##__VA_ARGS__)
#define syslog_err(msg, ...) syslog(LOG_ERR, "%s:%d [" msg "]",  __func__, __LINE__, ##__VA_ARGS__)

#ifndef NDEBUG
#define syslog_debug(msg, ...) syslog(LOG_DEBUG, "%s:%d " msg "", __func__, __LINE__, ##__VA_ARGS__)
#else
#define syslog_debug(msg, ...)
#endif

static int transport_indication(facilities_t *facilities, uint8_t *msg, uint32_t msg_len) {
    int rv = 0, code = 0;
    FacilitiesDataIndication_t *fdi = NULL;

    BTPDataIndication_t *bdi = calloc(1, sizeof(BTPDataIndication_t));  

    asn_dec_rval_t dec = oer_decode(NULL, &asn_DEF_BTPDataIndication, (void**) &bdi, msg, msg_len);
    if (dec.code) {
        syslog_err("[facilities] <- invalid BDI received");
        rv = 1;
        code = 1;
        zmq_send(facilities->responder, &code, 1, 0);
        goto cleanup;
    }

    zmq_send(facilities->responder, &code, 1, 0);
    syslog_debug("[facilities] <- BDI (%ldB)", bdi->data.size);

    // Parse message
    asn_TYPE_descriptor_t *its_msg_descriptor;
    void *its_msg;
    switch (bdi->destinationPort) {
        case Port_cam:
            its_msg_descriptor = &asn_DEF_CAM;
            its_msg = calloc(1, sizeof(CAM_t));
            break;
        case Port_denm:
            its_msg_descriptor = &asn_DEF_DENM;
            its_msg = calloc(1, sizeof(DENM_t));
            break;
        case Port_ivim:
            its_msg_descriptor = &asn_DEF_IVIM;
            its_msg = calloc(1, sizeof(IVIM_t));
            break;
        default:
            syslog_debug("[facilities] messsage with unhandled BTP port received, ignoring");
            goto cleanup;
    }

    dec = uper_decode_complete(NULL, its_msg_descriptor, (void**) &its_msg, bdi->data.buf, bdi->data.size);
    if (dec.code) {
        syslog_debug("[facilities] invalid %s received", its_msg_descriptor->name);
        rv = 1;
        goto cleanup;
    }

    // Manage message
    if (bdi->destinationPort == Port_denm) {
        #ifdef DEBUG
            uint8_t* xml_denm = malloc(4096);
            asn_enc_rval_t rve = xer_encode_to_buffer(xml_denm, 4096, 0x02, &asn_DEF_DENM, its_msg);
            syslog_debug("DENM XER %d: %.*s", (int)rve.encoded, (int)rve.encoded , xml_denm);
            free(xml_denm);
        #endif

        event_manage(facilities->den, its_msg);
    }


    // Forward to app
    fdi = calloc(1, sizeof(FacilitiesDataIndication_t));

    fdi->itssMessageType = bdi->destinationPort;

    fdi->data.size = bdi->data.size;
    fdi->data.buf = malloc(bdi->data.size);
    memcpy(fdi->data.buf, bdi->data.buf, bdi->data.size);

    uint8_t buffer[512];
    buffer[0] = 4; // Facilities
    asn_enc_rval_t enc = oer_encode_to_buffer(&asn_DEF_FacilitiesDataIndication, NULL, fdi, buffer+1, 511);
    syslog_debug("[facilities] -> [app] FDI (%ldB)", enc.encoded);
    zmq_send(facilities->app_socket, buffer, enc.encoded, 0);
    zmq_recv(facilities->app_socket, &code, 1, 0);

cleanup:
    if (bdi->destinationPort != Port_denm) ASN_STRUCT_FREE(*its_msg_descriptor, its_msg);
    ASN_STRUCT_FREE(asn_DEF_BTPDataIndication, bdi);
    ASN_STRUCT_FREE(asn_DEF_FacilitiesDataIndication, fdi);

    return rv;
}

int main() {
    syslog_info("[facilities] starting");

    facilities_t facilities;
    facilities.exit = false;

    void *context = zmq_ctx_new();
    facilities.responder = zmq_socket(context, ZMQ_REP);
    int rc = zmq_bind(facilities.responder, "ipc:///tmp/itss/facilities");

    facilities.app_socket = zmq_socket(context, ZMQ_REQ);
    rc = zmq_connect(facilities.app_socket, "ipc:///tmp/itss/application");

    facilities.transport_socket = zmq_socket(context, ZMQ_REQ);
    rc = zmq_connect(facilities.transport_socket, "ipc:///tmp/itss/transport");

    facilities.den = calloc(1, sizeof(den_t));

    pthread_create(&facilities.ca_service, NULL, ca_service, (void*) &facilities);
    pthread_create(&facilities.den_service, NULL, den_service, (void*) &facilities);

    uint8_t buffer[512];
    uint8_t code;
    syslog_info("[facilities] listening");

    while (!facilities.exit) {
        zmq_recv(facilities.responder, buffer, 512, 0);

        switch (buffer[0]) {
            case 3:
                code = transport_indication(&facilities, buffer+1, 511);
                break;
            default:
                syslog_debug("[facilities] unrecognized service");
                continue;
        }
    }


    return 0;
}