#include <errno.h>
#include "wrapper.h"

size_t encode_E2AP_PDU(E2AP_PDU_t* pdu, void* buffer, size_t buf_size)
{
    asn_enc_rval_t encode_result;
    encode_result = aper_encode_to_buffer(&asn_DEF_E2AP_PDU, NULL, pdu, buffer, buf_size);
    if(encode_result.encoded == -1) {
        return -1;
    }
    return encode_result.encoded;
}

E2AP_PDU_t* decode_E2AP_PDU(const void* buffer, size_t buf_size)
{
    asn_dec_rval_t decode_result;
    E2AP_PDU_t *pdu = 0;
    decode_result = aper_decode_complete(NULL, &asn_DEF_E2AP_PDU, (void **)&pdu, buffer, buf_size);
    if(decode_result.code == RC_OK) {
        return pdu;
    } else {
        ASN_STRUCT_FREE(asn_DEF_E2AP_PDU, pdu);
        return 0;
    }
}

ssize_t encode_RIC_subscription_request(RICsubscriptionRequest_t* pdu, void* buffer, size_t buf_size)
{
    asn_enc_rval_t encode_result;
    encode_result = aper_encode_to_buffer(&asn_DEF_RICsubscriptionRequest, NULL, pdu, buffer, buf_size);
    if(encode_result.encoded == -1) {
        return -1;
    }
    return encode_result.encoded;
}

RICsubscriptionRequest_t* decode_RIC_subscription_request(const void *buffer, size_t buf_size)
{
    asn_dec_rval_t decode_result;
    RICsubscriptionRequest_t *pdu = 0;
    decode_result = aper_decode_complete(NULL, &asn_DEF_RICsubscriptionRequest, (void **)&pdu, buffer, buf_size);
    if(decode_result.code == RC_OK) {
        return pdu;
    } else {
        ASN_STRUCT_FREE(asn_DEF_RICsubscriptionRequest, pdu);
        return 0;
    }
}

long e2ap_get_ric_subscription_request_sequence_number(void *buffer, size_t buf_size)
{
    E2AP_PDU_t *pdu = decode_E2AP_PDU(buffer, buf_size);
    if ( pdu != NULL && pdu->present == E2AP_PDU_PR_initiatingMessage)
    {
        InitiatingMessageE2_t* initiatingMessage = pdu->choice.initiatingMessage;
        RICsubscriptionRequest_t *ric_subscription_request = &(initiatingMessage->value.choice.RICsubscriptionRequest);
        for (int i = 0; i < ric_subscription_request->protocolIEs.list.count; ++i )
        {
            if ( ric_subscription_request->protocolIEs.list.array[i]->id == ProtocolIE_ID_id_RICrequestID )
            {
                return ric_subscription_request->protocolIEs.list.array[i]->value.choice.RICrequestID.ricRequestSequenceNumber;
            }
        }
    }
    return -1;
}

ssize_t  e2ap_set_ric_subscription_request_sequence_number(void *buffer, size_t buf_size, long sequence_number)
{
    E2AP_PDU_t *pdu = decode_E2AP_PDU(buffer, buf_size);
    if ( pdu != NULL && pdu->present == E2AP_PDU_PR_initiatingMessage)
    {
        InitiatingMessageE2_t* initiatingMessage = pdu->choice.initiatingMessage;
        RICsubscriptionRequest_t *ric_subscription_request = &(initiatingMessage->value.choice.RICsubscriptionRequest);
        for (int i = 0; i < ric_subscription_request->protocolIEs.list.count; ++i )
        {
            if ( ric_subscription_request->protocolIEs.list.array[i]->id == ProtocolIE_ID_id_RICrequestID )
            {
                ric_subscription_request->protocolIEs.list.array[i]->value.choice.RICrequestID.ricRequestSequenceNumber = sequence_number;
                return encode_E2AP_PDU(pdu, buffer, buf_size);
            }
        }
    }
    return -1;
}

/* RICsubscriptionResponse */
ssize_t encode_RIC_subscription_response(RICsubscriptionResponse_t* pdu, void* buffer, size_t buf_size)
{
    asn_enc_rval_t encode_result;
    encode_result = aper_encode_to_buffer(&asn_DEF_RICsubscriptionResponse, NULL, pdu, buffer, buf_size);
    if(encode_result.encoded == -1) {
        return -1;
    }
    return encode_result.encoded;
}

RICsubscriptionResponse_t* decode_RIC_subscription_response(const void *buffer, size_t buf_size)
{
    asn_dec_rval_t decode_result;
    RICsubscriptionResponse_t *pdu = 0;
    decode_result = aper_decode_complete(NULL, &asn_DEF_RICsubscriptionResponse, (void **)&pdu, buffer, buf_size);
    if(decode_result.code == RC_OK) {
        fprintf(stdout, "decoded bytes: %ld\n", decode_result.consumed);
        return pdu;
    } else {
        ASN_STRUCT_FREE(asn_DEF_RICsubscriptionResponse, pdu);
        return 0;
    }
}

long e2ap_get_ric_subscription_response_sequence_number(void *buffer, size_t buf_size)
{
    RICsubscriptionResponse_t *pdu = decode_RIC_subscription_response(buffer, buf_size);
    if ( pdu != NULL )
    {
        for (int i = 0; i < pdu->protocolIEs.list.count; ++i )
        {
            if ( pdu->protocolIEs.list.array[i]->id == ProtocolIE_ID_id_RICrequestID )
            {
                return pdu->protocolIEs.list.array[i]->value.choice.RICrequestID.ricRequestSequenceNumber;
            }
        }
    }
    return -1;
}

ssize_t  e2ap_set_ric_subscription_response_sequence_number(void *buffer, size_t buf_size, long sequence_number)
{
    RICsubscriptionResponse_t *pdu = decode_RIC_subscription_response(buffer, buf_size);
    if ( pdu != NULL )
    {
        for (int i = 0; i < pdu->protocolIEs.list.count; ++i )
        {
            if ( pdu->protocolIEs.list.array[i]->id == ProtocolIE_ID_id_RICrequestID )
            {
                pdu->protocolIEs.list.array[i]->value.choice.RICrequestID.ricRequestSequenceNumber = sequence_number;
                return encode_RIC_subscription_response(pdu, buffer, buf_size);
            }
        }
    }
    return -1;
}

/* RICsubscriptionDeleteRequest */
ssize_t encode_RIC_subscription_delete_request(RICsubscriptionDeleteRequest_t* pdu, void* buffer, size_t buf_size)
{
    asn_enc_rval_t encode_result;
    encode_result = aper_encode_to_buffer(&asn_DEF_RICsubscriptionDeleteRequest, NULL, pdu, buffer, buf_size);
    if(encode_result.encoded == -1) {
        return -1;
    }
    return encode_result.encoded;
}

RICsubscriptionDeleteRequest_t* decode_RIC_subscription_delete_request(const void *buffer, size_t buf_size)
{
    asn_dec_rval_t decode_result;
    RICsubscriptionDeleteRequest_t *pdu = 0;
    decode_result = aper_decode_complete(NULL, &asn_DEF_RICsubscriptionDeleteRequest, (void **)&pdu, buffer, buf_size);
    if(decode_result.code == RC_OK) {
        fprintf(stdout, "decoded bytes: %ld\n", decode_result.consumed);
        return pdu;
    } else {
        ASN_STRUCT_FREE(asn_DEF_RICsubscriptionDeleteRequest, pdu);
        return 0;
    }
}

long e2ap_get_ric_subscription_delete_request_sequence_number(void *buffer, size_t buf_size)
{
    RICsubscriptionDeleteRequest_t *pdu = decode_RIC_subscription_delete_request(buffer, buf_size);
    if ( pdu != NULL )
    {
        for (int i = 0; i < pdu->protocolIEs.list.count; ++i )
        {
            /* TODO */
            if ( pdu->protocolIEs.list.array[i]->id == ProtocolIE_ID_id_RICrequestID )
            {
                return pdu->protocolIEs.list.array[i]->value.choice.RICrequestID.ricRequestSequenceNumber;
            }
        }
    }
    return -1;
}

ssize_t  e2ap_set_ric_subscription_delete_request_sequence_number(void *buffer, size_t buf_size, long sequence_number)
{
    RICsubscriptionDeleteRequest_t *pdu = decode_RIC_subscription_delete_request(buffer, buf_size);
    if ( pdu != NULL )
    {
        for (int i = 0; i < pdu->protocolIEs.list.count; ++i )
        {
            /* TODO */
            if ( pdu->protocolIEs.list.array[i]->id == ProtocolIE_ID_id_RICrequestID )
            {
                pdu->protocolIEs.list.array[i]->value.choice.RICrequestID.ricRequestSequenceNumber = sequence_number;
                return encode_RIC_subscription_delete_request(pdu, buffer, buf_size);
            }
        }
    }
    return -1;
}

/* RICsubscriptionDeleteResponse */
ssize_t encode_RIC_subscription_delete_response(RICsubscriptionDeleteResponse_t* pdu, void* buffer, size_t buf_size)
{
    asn_enc_rval_t encode_result;
    encode_result = aper_encode_to_buffer(&asn_DEF_RICsubscriptionDeleteResponse, NULL, pdu, buffer, buf_size);
    if(encode_result.encoded == -1) {
        return -1;
    }
    return encode_result.encoded;
}

RICsubscriptionDeleteResponse_t* decode_RIC_subscription_delete_response(const void *buffer, size_t buf_size)
{
    asn_dec_rval_t decode_result;
    RICsubscriptionDeleteResponse_t *pdu = 0;
    decode_result = aper_decode_complete(NULL, &asn_DEF_RICsubscriptionDeleteResponse, (void **)&pdu, buffer, buf_size);
    if(decode_result.code == RC_OK) {
        fprintf(stdout, "decoded bytes: %ld\n", decode_result.consumed);
        return pdu;
    } else {
        ASN_STRUCT_FREE(asn_DEF_RICsubscriptionDeleteResponse, pdu);
        return 0;
    }
}

long e2ap_get_ric_subscription_delete_response_sequence_number(void *buffer, size_t buf_size)
{
    RICsubscriptionDeleteResponse_t *pdu = decode_RIC_subscription_delete_response(buffer, buf_size);
    if ( pdu != NULL )
    {
        for (int i = 0; i < pdu->protocolIEs.list.count; ++i )
        {
            if ( pdu->protocolIEs.list.array[i]->id == ProtocolIE_ID_id_RICrequestID )
            {
                /* TODO */
                return pdu->protocolIEs.list.array[i]->value.choice.RICrequestID.ricRequestSequenceNumber;
            }
        }
    }
    return -1;
}

ssize_t  e2ap_set_ric_subscription_delete_response_sequence_number(void *buffer, size_t buf_size, long sequence_number)
{
    RICsubscriptionDeleteResponse_t *pdu = decode_RIC_subscription_delete_response(buffer, buf_size);
    if ( pdu != NULL )
    {
        for (int i = 0; i < pdu->protocolIEs.list.count; ++i )
        {
            if ( pdu->protocolIEs.list.array[i]->id == ProtocolIE_ID_id_RICrequestID )
            {
                /* todo */
                pdu->protocolIEs.list.array[i]->value.choice.RICrequestID.ricRequestSequenceNumber = sequence_number;
                return encode_RIC_subscription_delete_response(pdu, buffer, buf_size);
            }
        }
    }
    return -1;
}