// SPDX-License-Identifier: GPL-2.0
#include <linux/compiler.h>
#include <stdio.h>
#include <string.h>
#include "builtin.h"
#include "debug.h"
#include <subcmd/parse-options.h>
#include "data-convert.h"
#include "util/util.h"

typedef int (*data_cmd_fn_t)(int argc, const char **argv);

struct data_cmd {
        const char      *name;
        const char      *summary;
        data_cmd_fn_t   fn;
};

static struct data_cmd data_cmds[];

#define for_each_cmd(cmd) \
        for (cmd = data_cmds; cmd && cmd->name; cmd++)

static const char * const data_subcommands[] = { "convert", NULL };

static const char *data_usage[] = {
        "perf data convert [<options>]",
        NULL
};

const char *to_json;
const char *to_ctf;
struct perf_data_convert_opts opts = {
        .force = false,
        .all = false,
        .time_str = NULL,
};

const struct option data_options[] = {
                OPT_INCR('v', "verbose", &verbose, "be more verbose"),
                OPT_STRING('i', "input", &input_name, "file", "input file name"),
                OPT_STRING(0, "to-json", &to_json, NULL, "Convert to JSON format"),
#ifdef HAVE_LIBBABELTRACE_SUPPORT
                OPT_STRING(0, "to-ctf", &to_ctf, NULL, "Convert to CTF format"),
                OPT_BOOLEAN(0, "tod", &opts.tod, "Convert time to wall clock time"),
#endif
                OPT_BOOLEAN('f', "force", &opts.force, "don't complain, do it"),
                OPT_BOOLEAN(0, "all", &opts.all, "Convert all events"),
                OPT_STRING(0, "time", &opts.time_str, "str",
                           "Time span of interest (start,stop)"),
                OPT_END()
        };

static int cmd_data_convert(int argc, const char **argv)
{

        argc = parse_options(argc, argv, data_options,
                             data_usage, 0);
        if (argc) {
                usage_with_options(data_usage, data_options);
                return -1;
        }

        if (to_json && to_ctf) {
                pr_err("You cannot specify both --to-ctf and --to-json.\n");
                return -1;
        }
#ifdef HAVE_LIBBABELTRACE_SUPPORT
        if (!to_json && !to_ctf) {
                pr_err("You must specify one of --to-ctf or --to-json.\n");
                return -1;
        }
#else
        if (!to_json) {
                pr_err("You must specify --to-json.\n");
        return -1;
}
#endif

        if (to_json)
                return bt_convert__perf2json(input_name, to_json, &opts);

        if (to_ctf) {
#if defined(HAVE_LIBBABELTRACE_SUPPORT) && defined(HAVE_LIBTRACEEVENT)
                return bt_convert__perf2ctf(input_name, to_ctf, &opts);
#else
                pr_err("The libbabeltrace support is not compiled in. perf should be "
                       "compiled with environment variables LIBBABELTRACE=1 and "
                       "LIBBABELTRACE_DIR=/path/to/libbabeltrace/.\n"
                       "Check also if libbtraceevent devel files are available.\n");
                return -1;
#endif
        }

        return 0;
}

static struct data_cmd data_cmds[] = {
        { "convert", "converts data file between formats", cmd_data_convert },
        { .name = NULL, },
};

int cmd_data(int argc, const char **argv)
{
        struct data_cmd *cmd;
        const char *cmdstr;

        argc = parse_options_subcommand(argc, argv, data_options, data_subcommands, data_usage,
                             PARSE_OPT_STOP_AT_NON_OPTION);

        if (!argc) {
                usage_with_options(data_usage, data_options);
                return -1;
        }

        cmdstr = argv[0];

        for_each_cmd(cmd) {
                if (strcmp(cmd->name, cmdstr))
                        continue;

                return cmd->fn(argc, argv);
        }

        pr_err("Unknown command: %s\n", cmdstr);
        usage_with_options(data_usage, data_options);
        return -1;
}