~ubuntu-branches/ubuntu/natty/htop/natty

/*

htop - Settings.c

(C) 2004-2006 Hisham H. Muhammad

Released under the GNU GPL, see the COPYING file

in the source distribution for its full text.

*/

#include "Settings.h"

#include "String.h"

#include "ProcessList.h"

#include "Header.h"

#include "debug.h"

#define DEFAULT_DELAY 15

/*{

typedef struct Settings_ {

   char* userSettings;

   ProcessList* pl;

   Header* header;

   int colorScheme;

   bool changed;

   int delay;

} Settings;

}*/

void Settings_delete(Settings* this) {

   free(this->userSettings);

   free(this);

}

static void Settings_readMeters(Settings* this, char* line, HeaderSide side) {

   char* trim = String_trim(line);

   char** ids = String_split(trim, ' ');

   free(trim);

   int i;

   for (i = 0; ids[i] != NULL; i++) {

      Header_createMeter(this->header, ids[i], side);

   }

   String_freeArray(ids);

}

static void Settings_readMeterModes(Settings* this, char* line, HeaderSide side) {

   char* trim = String_trim(line);

   char** ids = String_split(trim, ' ');

   free(trim);

   int i;

   for (i = 0; ids[i] != NULL; i++) {

      int mode = atoi(ids[i]);

      Header_setMode(this->header, i, mode, side);

   }

   String_freeArray(ids);

}

static bool Settings_read(Settings* this, char* fileName) {

   // TODO: implement File object and make

   // file I/O object-oriented.

   FILE* fd;

   fd = fopen(fileName, "r");

   if (fd == NULL) {

      return false;

   }

   const int maxLine = 65535;

   char buffer[maxLine];

   bool readMeters = false;

   while (!feof(fd)) {

      buffer[0] = '\0';

      fgets(buffer, maxLine, fd);

      char** option = String_split(buffer, '=');

      if (String_eq(option[0], "fields")) {

         char* trim = String_trim(option[1]);

         char** ids = String_split(trim, ' ');

         free(trim);

         int i, j;

         for (j = 0, i = 0; i < LAST_PROCESSFIELD && ids[i] != NULL; i++) {

            // This "+1" is for compatibility with the older enum format.

            int id = atoi(ids[i]) + 1;

            if (id > 0 && id < LAST_PROCESSFIELD) {

               this->pl->fields[j] = id;

               j++;

            }

         }

         this->pl->fields[j] = (ProcessField) NULL;

         String_freeArray(ids);

      } else if (String_eq(option[0], "sort_key")) {

         // This "+1" is for compatibility with the older enum format.

         this->pl->sortKey = atoi(option[1]) + 1;

      } else if (String_eq(option[0], "sort_direction")) {

         this->pl->direction = atoi(option[1]);

      } else if (String_eq(option[0], "tree_view")) {

         this->pl->treeView = atoi(option[1]);

      } else if (String_eq(option[0], "hide_threads")) {

         this->pl->hideThreads = atoi(option[1]);

      } else if (String_eq(option[0], "hide_kernel_threads")) {

         this->pl->hideKernelThreads = atoi(option[1]);

      } else if (String_eq(option[0], "hide_userland_threads")) {

         this->pl->hideUserlandThreads = atoi(option[1]);

      } else if (String_eq(option[0], "shadow_other_users")) {

         this->pl->shadowOtherUsers = atoi(option[1]);

      } else if (String_eq(option[0], "highlight_base_name")) {

         this->pl->highlightBaseName = atoi(option[1]);

      } else if (String_eq(option[0], "highlight_megabytes")) {

         this->pl->highlightMegabytes = atoi(option[1]);

      } else if (String_eq(option[0], "highlight_threads")) {

         this->pl->highlightThreads = atoi(option[1]);

      } else if (String_eq(option[0], "header_margin")) {

         this->header->margin = atoi(option[1]);

      } else if (String_eq(option[0], "expand_system_time")) {

         // Compatibility option.

         this->pl->detailedCPUTime = atoi(option[1]);

      } else if (String_eq(option[0], "detailed_cpu_time")) {

         this->pl->detailedCPUTime = atoi(option[1]);

      } else if (String_eq(option[0], "delay")) {

         this->delay = atoi(option[1]);

      } else if (String_eq(option[0], "color_scheme")) {

         this->colorScheme = atoi(option[1]);

         if (this->colorScheme < 0) this->colorScheme = 0;

         if (this->colorScheme > 5) this->colorScheme = 5;

      } else if (String_eq(option[0], "left_meters")) {

         Settings_readMeters(this, option[1], LEFT_HEADER);

         readMeters = true;

      } else if (String_eq(option[0], "right_meters")) {

         Settings_readMeters(this, option[1], RIGHT_HEADER);

         readMeters = true;

      } else if (String_eq(option[0], "left_meter_modes")) {

         Settings_readMeterModes(this, option[1], LEFT_HEADER);

         readMeters = true;

      } else if (String_eq(option[0], "right_meter_modes")) {

         Settings_readMeterModes(this, option[1], RIGHT_HEADER);

         readMeters = true;

      }

      String_freeArray(option);

   }

   fclose(fd);

   if (!readMeters) {

      Header_defaultMeters(this->header);

   }

   return true;

}

bool Settings_write(Settings* this) {

   // TODO: implement File object and make

   // file I/O object-oriented.

   FILE* fd;

   fd = fopen(this->userSettings, "w");

   if (fd == NULL) {

      return false;

   }

   fprintf(fd, "# Beware! This file is rewritten every time htop exits.\n");

   fprintf(fd, "# The parser is also very primitive, and not human-friendly.\n");

   fprintf(fd, "# (I know, it's in the todo list).\n");

   fprintf(fd, "fields=");

   for (int i = 0; this->pl->fields[i]; i++) {

      // This "-1" is for compatibility with the older enum format.

      fprintf(fd, "%d ", (int) this->pl->fields[i]-1);

   }

   fprintf(fd, "\n");

   // This "-1" is for compatibility with the older enum format.

   fprintf(fd, "sort_key=%d\n", (int) this->pl->sortKey-1);

   fprintf(fd, "sort_direction=%d\n", (int) this->pl->direction);

   fprintf(fd, "hide_threads=%d\n", (int) this->pl->hideThreads);

   fprintf(fd, "hide_kernel_threads=%d\n", (int) this->pl->hideKernelThreads);

   fprintf(fd, "hide_userland_threads=%d\n", (int) this->pl->hideUserlandThreads);

   fprintf(fd, "shadow_other_users=%d\n", (int) this->pl->shadowOtherUsers);

   fprintf(fd, "highlight_base_name=%d\n", (int) this->pl->highlightBaseName);

   fprintf(fd, "highlight_megabytes=%d\n", (int) this->pl->highlightMegabytes);

   fprintf(fd, "highlight_threads=%d\n", (int) this->pl->highlightThreads);

   fprintf(fd, "tree_view=%d\n", (int) this->pl->treeView);

   fprintf(fd, "header_margin=%d\n", (int) this->header->margin);

   fprintf(fd, "detailed_cpu_time=%d\n", (int) this->pl->detailedCPUTime);

   fprintf(fd, "color_scheme=%d\n", (int) this->colorScheme);

   fprintf(fd, "delay=%d\n", (int) this->delay);

   fprintf(fd, "left_meters=");

   for (int i = 0; i < Header_size(this->header, LEFT_HEADER); i++) {

      char* name = Header_readMeterName(this->header, i, LEFT_HEADER);

      fprintf(fd, "%s ", name);

      free(name);

   }

   fprintf(fd, "\n");

   fprintf(fd, "left_meter_modes=");

   for (int i = 0; i < Header_size(this->header, LEFT_HEADER); i++)

      fprintf(fd, "%d ", Header_readMeterMode(this->header, i, LEFT_HEADER));

   fprintf(fd, "\n");

   fprintf(fd, "right_meters=");

   for (int i = 0; i < Header_size(this->header, RIGHT_HEADER); i++) {

      char* name = Header_readMeterName(this->header, i, RIGHT_HEADER);

      fprintf(fd, "%s ", name);

      free(name);

   }

   fprintf(fd, "\n");

   fprintf(fd, "right_meter_modes=");

   for (int i = 0; i < Header_size(this->header, RIGHT_HEADER); i++)

      fprintf(fd, "%d ", Header_readMeterMode(this->header, i, RIGHT_HEADER));

   fprintf(fd, "\n");

   fclose(fd);

   return true;

}

Settings* Settings_new(ProcessList* pl, Header* header) {

   Settings* this = malloc(sizeof(Settings));

   this->pl = pl;

   this->header = header;

   char* home;

   char* rcfile;

   home = getenv("HOME_ETC");

   if (!home) home = getenv("HOME");

   if (!home) home = "";

   rcfile = getenv("HOMERC");

   if (!rcfile)

      this->userSettings = String_cat(home, "/.htoprc");

   else

      this->userSettings = String_copy(rcfile);

   this->colorScheme = 0;

   this->changed = false;

   this->delay = DEFAULT_DELAY;

   bool ok = Settings_read(this, this->userSettings);

   if (!ok) {

      this->changed = true;

      // TODO: how to get SYSCONFDIR correctly through Autoconf?

      char* systemSettings = String_cat(SYSCONFDIR, "/htoprc");

      ok = Settings_read(this, systemSettings);

      free(systemSettings);

      if (!ok) {

         Header_defaultMeters(this->header);

         pl->hideKernelThreads = true;

         pl->highlightMegabytes = true;

         pl->highlightThreads = false;

      }

   }

   return this;

}