// SPDX-License-Identifier: GPL-2.0
/* Display a menu with individual samples to browse with perf script */
#include "hist.h"
#include "evsel.h"
#include "hists.h"
#include "sort.h"
#include "config.h"
#include "time-utils.h"
#include "../util.h"
#include "../../util/util.h" // perf_exe()
#include "../../perf.h"
#include <stdlib.h>
#include <string.h>
#include <linux/time64.h>
#include <linux/zalloc.h>

static u64 context_len = 10 * NSEC_PER_MSEC;

static int res_sample_config(const char *var, const char *value, void *data __maybe_unused)
{
 if (!strcmp(var, "samples.context"))
  return perf_config_u64(&context_len, var, value);
 return 0;
}

void res_sample_init(void)
{
 perf_config(res_sample_config, NULL);
}

int res_sample_browse(struct res_sample *res_samples, int num_res,
        struct evsel *evsel, enum rstype rstype)
{
 char **names;
 int i, n;
 int choice;
 char *cmd;
 char pbuf[256], tidbuf[32], cpubuf[32];
 const char *perf = perf_exe(pbuf, sizeof pbuf);
 char trange[128], tsample[64];
 struct res_sample *r;
 char extra_format[256];

 names = calloc(num_res, sizeof(char *));
 if (!names)
  return -1;
 for (i = 0; i < num_res; i++) {
  char tbuf[64];

  timestamp__scnprintf_nsec(res_samples[i].time, tbuf, sizeof tbuf);
  if (asprintf(&names[i], "%s: CPU %d tid %d", tbuf,
        res_samples[i].cpu, res_samples[i].tid) < 0) {
   while (--i >= 0)
    zfree(&names[i]);
   free(names);
   return -1;
  }
 }
 choice = ui__popup_menu(num_res, names, NULL);
 for (i = 0; i < num_res; i++)
  zfree(&names[i]);
 free(names);

 if (choice < 0 || choice >= num_res)
  return -1;
 r = &res_samples[choice];

 n = timestamp__scnprintf_nsec(r->time - context_len, trange, sizeof trange);
 trange[n++] = ',';
 timestamp__scnprintf_nsec(r->time + context_len, trange + n, sizeof trange - n);

 timestamp__scnprintf_nsec(r->time, tsample, sizeof tsample);

 attr_to_script(extra_format, &evsel->core.attr);

 if (asprintf(&cmd, "%s script %s%s --time %s %s%s %s%s --ns %s %s %s %s %s | less +/%s",
       perf,
       input_name ? "-i " : "",
       input_name ? input_name : "",
       trange,
       r->cpu >= 0 ? "--cpu " : "",
       r->cpu >= 0 ? (sprintf(cpubuf, "%d", r->cpu), cpubuf) : "",
       r->tid ? "--tid " : "",
       r->tid ? (sprintf(tidbuf, "%d", r->tid), tidbuf) : "",
       extra_format,
       rstype == A_ASM ? "-F +disasm" :
       rstype == A_SOURCE ? "-F +srcline,+srccode" : "",
       symbol_conf.inline_name ? "--inline" : "",
       "--show-lost-events ",
       r->tid ? "--show-switch-events --show-task-events " : "",
       tsample) < 0)
  return -1;
 run_script(cmd);
 free(cmd);
 return 0;
}