search and filter for the strace and lsof screens!

1 files changed, 213 insertions, 0 deletions

diff --git a/IncSet.c b/IncSet.c
new file mode 100644
index 00000000..2b9e052b
--- /dev/null
+++ b/IncSet.c
@@ -0,0 +1,213 @@
+/*
+htop - IncSet.c
+(C) 2005-2012 Hisham H. Muhammad
+Released under the GNU GPL, see the COPYING file
+in the source distribution for its full text.
+*/
+
+#include "IncSet.h"
+#include "String.h"
+#include "Panel.h"
+#include "ListItem.h"
+#include "CRT.h"
+#include <string.h>
+#include <stdlib.h>
+
+/*{
+
+#include "FunctionBar.h"
+#include "Panel.h"
+#include <stdbool.h>
+
+#define INCMODE_MAX 40
+
+typedef enum {
+   INC_SEARCH = 0,
+   INC_FILTER = 1
+} IncType;
+
+#define IncSet_filter(inc_) (inc_->filtering ? inc_->modes[INC_FILTER].buffer : NULL)
+
+typedef struct IncMode_ {
+   char buffer[INCMODE_MAX];
+   int index;
+   FunctionBar* bar;
+   bool isFilter;
+} IncMode;
+
+typedef struct IncSet_ {
+   IncMode modes[2];
+   IncMode* active;
+   FunctionBar* bar;
+   FunctionBar* defaultBar;
+   bool filtering;
+} IncSet;
+
+typedef const char* (*IncMode_GetPanelValue)(Panel*, int);
+
+}*/
+
+static void IncMode_reset(IncMode* mode) {
+   mode->index = 0;
+   mode->buffer[0] = 0;
+}
+
+static const char* searchFunctions[] = {"Next  ", "Cancel ", " Search: ", NULL};
+static const char* searchKeys[] = {"F3", "Esc", "  "};
+static int searchEvents[] = {KEY_F(3), 27, ERR};
+
+static inline void IncMode_initSearch(IncMode* search) {
+   memset(search, 0, sizeof(IncMode));
+   search->bar = FunctionBar_new(searchFunctions, searchKeys, searchEvents);
+   search->isFilter = false;
+}
+
+static const char* filterFunctions[] = {"Done  ", "Clear ", " Filter: ", NULL};
+static const char* filterKeys[] = {"Enter", "Esc", "  "};
+static int filterEvents[] = {13, 27, ERR};
+
+static inline void IncMode_initFilter(IncMode* filter) {
+   memset(filter, 0, sizeof(IncMode));
+   filter->bar = FunctionBar_new(filterFunctions, filterKeys, filterEvents);
+   filter->isFilter = true;
+}
+
+static inline void IncMode_done(IncMode* mode) {
+   FunctionBar_delete((Object*)mode->bar);
+}
+
+IncSet* IncSet_new(FunctionBar* bar) {
+   IncSet* this = calloc(sizeof(IncSet), 1);
+   IncMode_initSearch(&(this->modes[INC_SEARCH]));
+   IncMode_initFilter(&(this->modes[INC_FILTER]));
+   this->active = NULL;
+   this->filtering = false;
+   this->bar = bar;
+   this->defaultBar = bar;
+   return this;
+}
+
+void IncSet_delete(IncSet* this) {
+   IncMode_done(&(this->modes[0]));
+   IncMode_done(&(this->modes[1]));
+   free(this);
+}
+
+static void updateWeakPanel(IncSet* this, Panel* panel, Vector* lines) {
+   Object* selected = Panel_getSelected(panel);
+   Panel_prune(panel);
+   if (this->filtering) {
+      int n = 0;
+      const char* incFilter = this->modes[INC_FILTER].buffer;
+      for (int i = 0; i < Vector_size(lines); i++) {
+         ListItem* line = (ListItem*)Vector_get(lines, i);
+         if (String_contains_i(line->value, incFilter)) {
+            Panel_add(panel, (Object*)line);
+            if (selected == (Object*)line) Panel_setSelected(panel, n);
+            n++;
+         }
+      }
+   } else {
+      for (int i = 0; i < Vector_size(lines); i++) {
+         Object* line = Vector_get(lines, i);
+         Panel_add(panel, line);
+         if (selected == line) Panel_setSelected(panel, i);
+      }
+   }
+}
+
+static void search(IncMode* mode, Panel* panel, IncMode_GetPanelValue getPanelValue) {
+   int size = Panel_size(panel);
+   bool found = false;
+   for (int i = 0; i < size; i++) {
+      if (String_contains_i(getPanelValue(panel, i), mode->buffer)) {
+         Panel_setSelected(panel, i);
+         found = true;
+         break;
+      }
+   }
+   if (found)
+      FunctionBar_draw(mode->bar, mode->buffer);
+   else
+      FunctionBar_drawAttr(mode->bar, mode->buffer, CRT_colors[FAILED_SEARCH]);
+}
+
+bool IncSet_handleKey(IncSet* this, int ch, Panel* panel, IncMode_GetPanelValue getPanelValue, Vector* lines) {
+   if (ch == ERR)
+      return true;
+   IncMode* mode = this->active;
+   int size = Panel_size(panel);
+   bool filterChange = false;
+   bool doSearch = true;
+   if (ch == KEY_F(3)) {
+      if (size == 0) return true;
+      int here = Panel_getSelectedIndex(panel);
+      int i = here;
+      for(;;) {
+         i++;
+         if (i == size) i = 0;
+         if (i == here) break;
+         if (String_contains_i(getPanelValue(panel, i), mode->buffer)) {
+            Panel_setSelected(panel, i);
+            break;
+         }
+      }
+      doSearch = false;
+   } else if (isprint((char)ch) && (mode->index < INCMODE_MAX)) {
+      mode->buffer[mode->index] = ch;
+      mode->index++;
+      mode->buffer[mode->index] = 0;
+      if (mode->isFilter) {
+         filterChange = true;
+         if (mode->index == 1) this->filtering = true;
+      }
+   } else if ((ch == KEY_BACKSPACE || ch == 127) && (mode->index > 0)) {
+      mode->index--;
+      mode->buffer[mode->index] = 0;
+      if (mode->isFilter) {
+         filterChange = true;
+         if (mode->index == 0) {
+            this->filtering = false;
+            IncMode_reset(mode);
+         }
+      }
+   } else {
+      if (mode->isFilter) {
+         filterChange = true;
+         if (ch == 27) {
+            this->filtering = false;
+            IncMode_reset(mode);
+         }
+      } else {
+         IncMode_reset(mode);
+      }
+      this->active = NULL;
+      this->bar = this->defaultBar;
+      FunctionBar_draw(this->defaultBar, NULL);
+      doSearch = false;
+   }
+   if (doSearch) {
+      search(mode, panel, getPanelValue);
+   }
+   if (filterChange && lines) {
+      updateWeakPanel(this, panel, lines);
+   }
+   return filterChange;
+}
+
+const char* IncSet_getListItemValue(Panel* panel, int i) {
+   ListItem* l = (ListItem*) Panel_get(panel, i);
+   if (l)
+      return l->value;
+   return "";
+}
+
+void IncSet_activate(IncSet* this, IncType type) {
+   this->active = &(this->modes[type]);
+   this->bar = this->active->bar;
+   FunctionBar_draw(this->active->bar, this->active->buffer);
+}
+
+void IncSet_drawBar(IncSet* this) {
+   FunctionBar_draw(this->bar, this->active ? this->active->buffer : NULL);
+}