Function profiler

src/profiler.ts:21–122

(options: any)

 */
// eslint-disable-next-line @typescript-eslint/no-explicit-any
export async function profiler(options: any): Promise<unknown> {
  const origin = realtimeOriginOrEmulatorOrCustomUrl(options.instanceDetails.databaseUrl);
  const url = new URL(utils.getDatabaseUrl(origin, options.instance, "/.settings/profile.json?"));
  const rl = readline.createInterface({ input: process.stdin });

  const fileOut = !!options.output;
  const tmpFile = tmp.tmpNameSync();
  const tmpStream = fs.createWriteStream(tmpFile);
  const outStream = fileOut ? fs.createWriteStream(options.output) : process.stdout;
  const spinner = ora({
    text: "0 operations recorded. Press [enter] to stop",
    color: "yellow",
  });
  const outputFormat = options.raw ? "RAW" : options.parent.json ? "JSON" : "TXT";

  // Controller is used to stop the request stream when the user stops the
  // command or the duration passes.
  const controller = new AbortController();

  const generateReport = (): Promise<void> => {
    rl.close();
    spinner.stop();
    controller.abort();
    const dataFile = options.input || tmpFile;
    const reportOptions: ProfileReportOptions = {
      format: outputFormat,
      isFile: fileOut,
      isInput: !!options.input,
      collapse: options.collapse,
    };
    const report = new ProfileReport(dataFile, outStream, reportOptions);
    return report.generate();
  };

  if (options.input) {
    // If there is input, don't contact the server.
    return generateReport();
  }

  const c = new Client({ urlPrefix: url.origin, auth: true });
  const res = await c.request<unknown, NodeJS.ReadableStream>({
    method: "GET",
    path: url.pathname,
    responseType: "stream",
    resolveOnHTTPError: true,
    headers: {
      Accept: "text/event-stream",
    },
    signal: controller.signal,
  });

  if (res.response.status >= 400) {
    throw responseToError(res.response, await res.response.text());
  }

  if (!options.duration) {
    spinner.start();
  }