rpm-ostree/src/libpriv/rpmostree-output.c

/*
* Copyright (C) 2016 Red Hat, Inc.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 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, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/

#include "config.h"

#include <ostree.h>
#include <libglnx.h>

#include "rpmostree-output.h"

/* These are helper functions that automatically determine whether data should
 * be sent through an appropriate D-Bus signal or sent directly to the local
 * terminal. This is helpful in situations in which code may be executed both
 * from the daemon and daemon-less. */

static GLnxConsoleRef console;

void
rpmostree_output_default_handler (RpmOstreeOutputType type,
                                  void *data,
                                  void *opaque)
{
  switch (type)
  {
  case RPMOSTREE_OUTPUT_TASK_BEGIN:
    /* XXX: move to libglnx spinner once it's implemented */
    g_print ("%s... ", ((RpmOstreeOutputTaskBegin*)data)->text);
    break;
  case RPMOSTREE_OUTPUT_TASK_END:
    g_print ("%s\n", ((RpmOstreeOutputTaskEnd*)data)->text);
    break;
  case RPMOSTREE_OUTPUT_PERCENT_PROGRESS:
    if (!console.locked)
      glnx_console_lock (&console);
    glnx_console_progress_text_percent (
      ((RpmOstreeOutputPercentProgress*)data)->text,
      ((RpmOstreeOutputPercentProgress*)data)->percentage);
    break;
  case RPMOSTREE_OUTPUT_PERCENT_PROGRESS_END:
    if (console.locked)
      glnx_console_unlock (&console);
    break;
  }
}

static void (*active_cb)(RpmOstreeOutputType, void*, void*) =
  rpmostree_output_default_handler;

static void *active_cb_opaque;

void
rpmostree_output_set_callback (void (*cb)(RpmOstreeOutputType, void*, void*),
                               void* opaque)
{
  active_cb = cb ?: rpmostree_output_default_handler;
  active_cb_opaque = opaque;
}

void
rpmostree_output_task_begin (const char *format, ...)
{
  g_autofree char *final = NULL;
  va_list args;

  va_start (args, format);
  final = g_strdup_vprintf (format, args);
  va_end (args);

  {
    RpmOstreeOutputTaskBegin task = { final };
    active_cb (RPMOSTREE_OUTPUT_TASK_BEGIN, &task, active_cb_opaque);
  }
}

void
rpmostree_output_task_end (const char *format, ...)
{
  g_autofree char *final = NULL;
  va_list args;

  va_start (args, format);
  final = g_strdup_vprintf (format, args);
  va_end (args);

  {
    RpmOstreeOutputTaskEnd task = { final };
    active_cb (RPMOSTREE_OUTPUT_TASK_END, &task, active_cb_opaque);
  }
}

void
rpmostree_output_percent_progress (const char *text, int percentage)
{
  RpmOstreeOutputPercentProgress progress = { text, percentage };
  active_cb (RPMOSTREE_OUTPUT_PERCENT_PROGRESS, &progress, active_cb_opaque);
}

void
rpmostree_output_percent_progress_end (void)
{
  active_cb (RPMOSTREE_OUTPUT_PERCENT_PROGRESS_END, NULL, active_cb_opaque);
}