evisum/src/bin/main.c

/*
 * Copyright 2018-2021. Alastair Roy Poole <netstar@gmail.com>
 *
 * See COPYING file for details.
 */

#include "config.h"
#include "evisum_config.h"
#include "evisum_server.h"
#include "ui/evisum_ui.h"
#include "system/filesystems.h"

static Eina_Bool
_shutdown_cb(void *data, int type, void *event EINA_UNUSED)
{
   Evisum_Ui *ui = data;

   if (ui->cpu.win) evas_object_del(ui->cpu.win);
   if (ui->mem.win) evas_object_del(ui->mem.win);
   if (ui->disk.win) evas_object_del(ui->disk.win);
   if (ui->sensors.win) evas_object_del(ui->sensors.win);
   if (ui->proc.win) evas_object_del(ui->proc.win);

   return 0;
}

static void
_signals(Evisum_Ui *ui)
{
   ui->handler_sig = ecore_event_handler_add(ECORE_EVENT_SIGNAL_EXIT, _shutdown_cb, ui);
}

// XXX!!!
static void
_background_poller_cb(void *data, Ecore_Thread *thread)
{
   meminfo_t memory;
   power_t power;
   int32_t poll_count = 0;
   Battery *bat;
   Evisum_Ui *ui = data;

   system_memory_usage_get(&memory);
   ui->mem_total = memory.total;
   ui->mem_used = memory.used;

   system_power_state_get(&power);
   if (power.battery_count)
     {
        ui->have_power = power.have_ac;
        for (int i = 0; i < power.battery_count; i++)
          {
             if (!power.batteries[i]->present) continue;
             bat = calloc(1, sizeof(Battery));
             bat->index = i;
             snprintf(bat->model, sizeof(bat->model), "%s", power.batteries[i]->model);
             snprintf(bat->vendor, sizeof(bat->vendor), "%s", power.batteries[i]->vendor);
             bat->usage = power.batteries[i]->percent;
             ui->batteries = eina_list_append(ui->batteries, bat);
          }
     }
   system_power_state_free(&power);

   while (!ecore_thread_check(thread))
     {
        int ncpu;
        double percent = 0.0;
        cpu_core_t **cores = system_cpu_usage_delayed_get(&ncpu, 250000);
        for (int i = 0; i < ncpu; i++)
          {
             percent += cores[i]->percent;
             free(cores[i]);
          }
        free(cores);

        system_memory_usage_get(&memory);
        ui->mem_used = memory.used;
        if (file_system_in_use("ZFS"))
          ui->mem_used += memory.zfs_arc_used;

        ui->cpu_usage = percent / system_cpu_online_count_get();

        if ((!(poll_count % 4)) && (ui->batteries))
          {
             Eina_List *l;
             system_power_state_get(&power);
             ui->have_power = power.have_ac;
             for (int i = 0; i < power.battery_count; i++)
               {
                  if (!power.batteries[i]->present) continue;
                  l = eina_list_nth_list(ui->batteries, i);
                  if (!l) continue;
                  bat = eina_list_data_get(l);
                  bat->usage = power.batteries[i]->percent;
               }
             system_power_state_free(&power);
          }

        poll_count++;
     }

   EINA_LIST_FREE(ui->batteries, bat)
     free(bat);
}

int
elm_main(int argc, char **argv)
{
   Evisum_Ui *ui;
   int i, pid = -1;
   size_t len;
   Evisum_Action action = EVISUM_ACTION_DEFAULT;

   for (i = 0; i < argc; i++)
     {
        if ((!strcmp(argv[i], "-h")) || (!strcmp(argv[i], "-help")) ||
            (!strcmp(argv[i], "--help") || !strcasecmp(argv[i], "-v")))
          {
             printf("Usage: evisum [OPTIONS] <pid>\n"
                    "   Where OPTIONS can be one of\n"
                    "      -c\n"
                    "        Launch CPU view.\n"
                    "      -m\n"
                    "        Launch memory view.\n"
                    "      -d\n"
                    "        Launch storage view.\n"
                    "      -n\n"
                    "        Launch network view.\n"
                    "      -s | -p\n"
                    "        Launch sensors view.\n"
                    "      -h | -help | --help\n"
                    "        This menu.\n"
                    "   No arguments will launch the process explorer.\n");
             exit(0);
          }
        else if (!strcmp(argv[i], "-c"))
          action = EVISUM_ACTION_CPU;
        else if (!strcmp(argv[i], "-m"))
          action = EVISUM_ACTION_MEM;
        else if (!strcmp(argv[i], "-d"))
          action = EVISUM_ACTION_STORAGE;
        else if ((!strcmp(argv[i], "-s")) || (!strcmp(argv[i], "-p")))
          action = EVISUM_ACTION_SENSORS;
        else if (!strcmp(argv[i], "-n"))
          action = EVISUM_ACTION_NETWORK;
     }

     if ((argc == 2) && (action == EVISUM_ACTION_DEFAULT))
       {
          action = EVISUM_ACTION_PROCESS;
          len = strlen(argv[1]);
          for (int i = 0; i < len; i++)
            {
               if (!isdigit(argv[1][i]))
                 {
                    action = EVISUM_ACTION_DEFAULT;
                    break;
                 }
            }
          if (len > 8) action = EVISUM_ACTION_DEFAULT;
          if (action == EVISUM_ACTION_PROCESS)
            pid = atoi(argv[1]);
       }

   elm_policy_set(ELM_POLICY_QUIT, ELM_POLICY_QUIT_LAST_WINDOW_CLOSED);

#if ENABLE_NLS
   setlocale(LC_ALL, "");
   bindtextdomain(PACKAGE, LOCALEDIR);
   bind_textdomain_codeset(PACKAGE, "UTF-8");
   textdomain(PACKAGE);
#endif

   if (evisum_server_client_add(action, pid))
     return 0;

   ui = evisum_ui_init();
   if (!ui) return 1;

   _signals(ui);

   evisum_server_init(ui);
   evisum_ui_activate(ui, action, pid);

   ui->background_poll_thread = ecore_thread_run(_background_poller_cb, NULL, NULL, ui);

   ecore_main_loop_begin();

   ecore_event_handler_del(ui->handler_sig);
   evisum_ui_shutdown(ui);
   evisum_server_shutdown();

   return 0;
}

ELM_MAIN()