/*
 * Copyright (C) 2019-2023 by Sukchan Lee <acetcom@gmail.com>
 *
 * This file is part of Open5GS.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#if !defined(OGS_SBI_INSIDE) && !defined(OGS_SBI_COMPILATION)
#error "This header cannot be included directly."
#endif

#ifndef OGS_SBI_CLIENT_H
#define OGS_SBI_CLIENT_H

#ifdef __cplusplus
extern "C" {
#endif

#define OGS_SBI_SETUP_CLIENT(__cTX, __pClient) \
    do { \
        char buf[OGS_ADDRSTRLEN]; \
        ogs_assert((__cTX)); \
        ogs_assert((__pClient)); \
        \
        if ((__cTX)->client) { \
            ogs_sbi_client_t *client = NULL; \
            \
            client = ((__cTX)->client); \
            ogs_assert(client); \
            if (client->fqdn) { \
                ogs_warn("NF EndPoint(fqdn) updated [%s:%d]", \
                        client->fqdn, client->fqdn_port); \
            } \
            if (client->addr) { \
                ogs_warn("NF EndPoint(addr) updated [%s:%d]", \
                    OGS_ADDR(client->addr, buf), OGS_PORT(client->addr)); \
            } \
            if (client->addr6) { \
                ogs_warn("NF EndPoint(addr6) updated [%s:%d]", \
                    OGS_ADDR(client->addr6, buf), OGS_PORT(client->addr6)); \
            } \
            ogs_sbi_client_remove(client); \
        } \
        \
        OGS_OBJECT_REF(__pClient); \
        ((__cTX)->client) = (__pClient); \
        ogs_debug("CLIENT Ref [%d]", (__pClient)->reference_count); \
        if ((__pClient)->fqdn) { \
            ogs_info("NF EndPoint(fqdn) setup [%s:%d]", \
                    (__pClient)->fqdn, (__pClient)->fqdn_port); \
        } \
        if ((__pClient)->addr) { \
            ogs_info("NF EndPoint(addr) setup [%s:%d]", \
                OGS_ADDR((__pClient)->addr, buf), OGS_PORT((__pClient)->addr)); \
        } \
        if ((__pClient)->addr6) { \
            ogs_info("NF EndPoint(addr6) setup [%s:%d]", \
                OGS_ADDR((__pClient)->addr6, buf), \
                OGS_PORT((__pClient)->addr6)); \
        } \
    } while(0)

typedef int (*ogs_sbi_client_cb_f)(
        int status, ogs_sbi_response_t *response, void *data);

typedef struct ogs_sbi_client_s {
    ogs_lnode_t lnode;

    OpenAPI_uri_scheme_e scheme;
    bool insecure_skip_verify;
    char *cacert, *private_key, *cert;

    char *fqdn;
    uint16_t fqdn_port;
    ogs_sockaddr_t  *addr;
    ogs_sockaddr_t  *addr6;

    char *resolve;

    ogs_timer_t     *t_curl;            /* timer for CURL */
    ogs_list_t      connection_list;    /* CURL connection list */

    void            *multi;             /* CURL multi handle */
    int             still_running;      /* number of running CURL handle */

    unsigned int    reference_count;    /* reference count for memory free */
} ogs_sbi_client_t;

typedef struct ogs_sbi_nf_instance_s ogs_sbi_nf_instance_t;

void ogs_sbi_client_init(int num_of_sockinfo_pool, int num_of_connection_pool);
void ogs_sbi_client_final(void);

ogs_sbi_client_t *ogs_sbi_client_add(
        OpenAPI_uri_scheme_e scheme,
        char *fqdn, uint16_t fqdn_port,
        ogs_sockaddr_t *addr, ogs_sockaddr_t *addr6);
void ogs_sbi_client_remove(ogs_sbi_client_t *client);
void ogs_sbi_client_remove_all(void);
ogs_sbi_client_t *ogs_sbi_client_find(
        OpenAPI_uri_scheme_e scheme,
        char *fqdn, uint16_t fqdn_port,
        ogs_sockaddr_t *addr, ogs_sockaddr_t *addr6);

void ogs_sbi_client_stop(ogs_sbi_client_t *client);
void ogs_sbi_client_stop_all(void);

bool ogs_sbi_client_send_request(
        ogs_sbi_client_t *client, ogs_sbi_client_cb_f client_cb,
        ogs_sbi_request_t *request, void *data);
bool ogs_sbi_client_send_via_scp_or_sepp(
        ogs_sbi_client_t *client, ogs_sbi_client_cb_f client_cb,
        ogs_sbi_request_t *request, void *data);

#ifdef __cplusplus
}
#endif

#endif /* OGS_SBI_CLIENT_H */