/* SPDX-License-Identifier: LGPL-2.1+ */

#include "nm-default.h"

#include <glib-unix.h>

#include "devices/bluetooth/nm-bluez5-dun.h"

#include "nm-test-utils-core.h"

/*****************************************************************************/

#define _NMLOG_DOMAIN LOGD_BT
#define _NMLOG(level, ...)                                             \
    nm_log((level),                                                    \
           _NMLOG_DOMAIN,                                              \
           NULL,                                                       \
           NULL,                                                       \
           "bt%s%s%s: " _NM_UTILS_MACRO_FIRST(__VA_ARGS__),            \
           NM_PRINT_FMT_QUOTED(gl.argv_cmd, "[", gl.argv_cmd, "]", "") \
               _NM_UTILS_MACRO_REST(__VA_ARGS__))

/*****************************************************************************/

struct {
    int                argc;
    const char *const *argv;
    const char *       argv_cmd;
    GMainLoop *        loop;
} gl;

typedef struct _MainCmdInfo {
    const char *name;
    int (*main_func)(const struct _MainCmdInfo *main_cmd_info);
} MainCmdInfo;

/*****************************************************************************/

#if WITH_BLUEZ5_DUN

typedef struct {
    NMBluez5DunContext *dun_context;
    GCancellable *      cancellable;
    guint               timeout_id;
    guint               sig_term_id;
    guint               sig_int_id;
} DunConnectData;

static void
_dun_connect_cb(NMBluez5DunContext *context,
                const char *        rfcomm_dev,
                GError *            error,
                gpointer            user_data)
{
    DunConnectData *dun_connect_data = user_data;

    g_assert(dun_connect_data);
    g_assert(!dun_connect_data->dun_context);
    g_assert((!!error) != (!!rfcomm_dev));

    if (rfcomm_dev && !context) {
        _LOGI("dun-connect notifies path \"%s\". Wait longer...", rfcomm_dev);
        return;
    }

    if (rfcomm_dev) {
        g_assert(context);
        _LOGI("dun-connect completed with path \"%s\"", rfcomm_dev);
    } else {
        g_assert(!context);
        _LOGI("dun-connect failed with error: %s", error->message);
    }

    dun_connect_data->dun_context = context;

    g_main_loop_quit(gl.loop);
}

static void
_dun_notify_tty_hangup_cb(NMBluez5DunContext *context, gpointer user_data)
{
    _LOGI("dun-connect: notified TTY hangup");
}

static gboolean
_timeout_cb(gpointer user_data)
{
    DunConnectData *dun_connect_data = user_data;

    _LOGI("timeout");
    dun_connect_data->timeout_id = 0;
    if (dun_connect_data->cancellable)
        g_cancellable_cancel(dun_connect_data->cancellable);
    return G_SOURCE_REMOVE;
}

static gboolean
_sig_xxx_cb(DunConnectData *dun_connect_data, int sigid)
{
    _LOGI("signal %s received", sigid == SIGTERM ? "SIGTERM" : "SIGINT");
    g_main_loop_quit(gl.loop);
    return G_SOURCE_CONTINUE;
}

static gboolean
_sig_term_cb(gpointer user_data)
{
    return _sig_xxx_cb(user_data, SIGTERM);
}

static gboolean
_sig_int_cb(gpointer user_data)
{
    return _sig_xxx_cb(user_data, SIGINT);
}
#endif

static int
do_dun_connect(const MainCmdInfo *main_cmd_info)
{
#if WITH_BLUEZ5_DUN
    gs_unref_object GCancellable *cancellable = NULL;
    gs_free_error GError *error               = NULL;
    const char *          adapter;
    const char *          remote;
    DunConnectData        dun_connect_data = {};

    if (gl.argc < 4) {
        _LOGE("missing arguments \"adapter\" and \"remote\"");
        return -1;
    }

    adapter = gl.argv[2];
    remote  = gl.argv[3];

    cancellable                  = g_cancellable_new();
    dun_connect_data.cancellable = cancellable;

    if (!nm_bluez5_dun_connect(adapter,
                               remote,
                               cancellable,
                               _dun_connect_cb,
                               &dun_connect_data,
                               _dun_notify_tty_hangup_cb,
                               &dun_connect_data,
                               &error)) {
        _LOGE("connect failed to start: %s", error->message);
        return -1;
    }

    dun_connect_data.timeout_id = g_timeout_add(60000, _timeout_cb, &dun_connect_data);

    g_main_loop_run(gl.loop);

    nm_clear_g_source(&dun_connect_data.timeout_id);

    if (dun_connect_data.dun_context) {
        dun_connect_data.sig_term_id = g_unix_signal_add(SIGTERM, _sig_term_cb, &dun_connect_data);
        dun_connect_data.sig_int_id  = g_unix_signal_add(SIGINT, _sig_int_cb, &dun_connect_data);

        g_main_loop_run(gl.loop);

        nm_clear_g_source(&dun_connect_data.sig_term_id);
        nm_clear_g_source(&dun_connect_data.sig_int_id);

        nm_bluez5_dun_disconnect(g_steal_pointer(&dun_connect_data.dun_context));
    }

    return 0;
#else
    _LOGE("compiled without bluetooth DUN support");
    return 1;
#endif
}

/*****************************************************************************/

NMTST_DEFINE();

int
main(int argc, char **argv)
{
    static const MainCmdInfo main_cmd_infos[] = {
        {
            .name      = "dun-connect",
            .main_func = do_dun_connect,
        },
    };
    int   exit_code = 0;
    guint i;

    if (!g_getenv("G_MESSAGES_DEBUG"))
        g_setenv("G_MESSAGES_DEBUG", "all", TRUE);

    nmtst_init_with_logging(&argc, &argv, "DEBUG", "ALL");

    nm_logging_init(NULL, TRUE);

    gl.argv = (const char *const *) argv;
    gl.argc = argc;
    gl.loop = g_main_loop_new(NULL, FALSE);

    _LOGI("bluetooth test util start");

    gl.argv_cmd = argc >= 2 ? argv[1] : NULL;

    for (i = 0; i < G_N_ELEMENTS(main_cmd_infos); i++) {
        if (nm_streq0(main_cmd_infos[i].name, gl.argv_cmd)) {
            _LOGD("start \"%s\"", gl.argv_cmd);
            exit_code = main_cmd_infos[i].main_func(&main_cmd_infos[i]);
            _LOGD("completed with %d", exit_code);
            break;
        }
    }
    if (gl.argv_cmd && i >= G_N_ELEMENTS(main_cmd_infos)) {
        nm_log_err(LOGD_BT, "invalid command \"%s\"", gl.argv_cmd);
        exit_code = -1;
    }

    nm_clear_pointer(&gl.loop, g_main_loop_unref);

    return exit_code;
}