Skip to content

Commit

Permalink
FlexMeter: Add FlexMeter functionality
Browse files Browse the repository at this point in the history
FlexMeter provides functionality which will allow
users to make custom meters without need of rebuilding
every time htop binary and adding source to the project.
It can be used to print some device status, free disk space
CPU or other specific temeraturer, fan RPM and many more.
Everything that can be fetched from linux shell
with one line result can be printer. For fething information
can be used anything from shell, python, precompiled binary
or simply reading file located somewhere in file system.

New meter will appear uppon restart of htop in list with
available meters.

Configuration folder location where metes should be placed:
- /home/$USER/.config/htop/FlexMeter/
On start folder will be created if does not exist, together
with template file .Template in same folder.
Note: Files starting with '.' (.Template for examlpe) are
ignored

Meter Example:
File name : Template

name=<NAME SHOWN IN AvailableMeters>
command=<COMMAND WHICH WILL BE EXECUTED>
type=<METER TYPE FOR NO ONLY "TEXT_METER">
caption="CAPTION TEXT SHOWN IN THE BEGGINING OF THE METER"

According to this implementation 30 Flex meter can be added
Currently they have hardcoded limit of 30 meter in addition
to all that already exist.

Signed-off-by: Stoyan Bogdanov <[email protected]>
  • Loading branch information
bogdanovs committed Jan 10, 2025
1 parent 4b3dfa2 commit 3ff3a68
Show file tree
Hide file tree
Showing 10 changed files with 256 additions and 5 deletions.
6 changes: 6 additions & 0 deletions CRT.c
Original file line number Diff line number Diff line change
Expand Up @@ -129,6 +129,7 @@ static int CRT_colorSchemes[LAST_COLORSCHEME][LAST_COLORELEMENT] = {
[FAILED_READ] = A_BOLD | ColorPair(Red, Black),
[PAUSED] = A_BOLD | ColorPair(Yellow, Cyan),
[UPTIME] = A_BOLD | ColorPair(Cyan, Black),
[FLEX] = A_BOLD | ColorPair(Cyan, Black),
[BATTERY] = A_BOLD | ColorPair(Cyan, Black),
[LARGE_NUMBER] = A_BOLD | ColorPair(Red, Black),
[METER_SHADOW] = A_BOLD | ColorPairGrayBlack,
Expand Down Expand Up @@ -247,6 +248,7 @@ static int CRT_colorSchemes[LAST_COLORSCHEME][LAST_COLORELEMENT] = {
[FAILED_READ] = A_BOLD,
[PAUSED] = A_BOLD | A_REVERSE,
[UPTIME] = A_BOLD,
[FLEX] = A_BOLD,
[BATTERY] = A_BOLD,
[LARGE_NUMBER] = A_BOLD,
[METER_SHADOW] = A_DIM,
Expand Down Expand Up @@ -365,6 +367,7 @@ static int CRT_colorSchemes[LAST_COLORSCHEME][LAST_COLORELEMENT] = {
[FAILED_READ] = ColorPair(Red, White),
[PAUSED] = A_BOLD | ColorPair(Yellow, Cyan),
[UPTIME] = ColorPair(Yellow, White),
[FLEX] = ColorPair(Yellow, White),
[BATTERY] = ColorPair(Yellow, White),
[LARGE_NUMBER] = ColorPair(Red, White),
[METER_SHADOW] = ColorPair(Blue, White),
Expand Down Expand Up @@ -483,6 +486,7 @@ static int CRT_colorSchemes[LAST_COLORSCHEME][LAST_COLORELEMENT] = {
[FAILED_READ] = ColorPair(Red, Black),
[PAUSED] = A_BOLD | ColorPair(Yellow, Cyan),
[UPTIME] = ColorPair(Yellow, Black),
[FLEX] = ColorPair(Yellow, Black),
[BATTERY] = ColorPair(Yellow, Black),
[LARGE_NUMBER] = ColorPair(Red, Black),
[METER_SHADOW] = A_BOLD | ColorPairGrayBlack,
Expand Down Expand Up @@ -601,6 +605,7 @@ static int CRT_colorSchemes[LAST_COLORSCHEME][LAST_COLORELEMENT] = {
[FAILED_READ] = A_BOLD | ColorPair(Red, Blue),
[PAUSED] = A_BOLD | ColorPair(Yellow, Cyan),
[UPTIME] = A_BOLD | ColorPair(Yellow, Blue),
[FLEX] = A_BOLD | ColorPair(Yellow, Blue),
[BATTERY] = A_BOLD | ColorPair(Yellow, Blue),
[LARGE_NUMBER] = A_BOLD | ColorPair(Red, Blue),
[METER_SHADOW] = ColorPair(Cyan, Blue),
Expand Down Expand Up @@ -719,6 +724,7 @@ static int CRT_colorSchemes[LAST_COLORSCHEME][LAST_COLORELEMENT] = {
[FAILED_READ] = A_BOLD | ColorPair(Red, Black),
[PAUSED] = A_BOLD | ColorPair(Yellow, Green),
[UPTIME] = ColorPair(Green, Black),
[FLEX] = ColorPair(Green, Black),
[BATTERY] = ColorPair(Green, Black),
[LARGE_NUMBER] = A_BOLD | ColorPair(Red, Black),
[METER_SHADOW] = A_BOLD | ColorPairGrayBlack,
Expand Down
1 change: 1 addition & 0 deletions CRT.h
Original file line number Diff line number Diff line change
Expand Up @@ -154,6 +154,7 @@ typedef enum ColorElements_ {
DYNAMIC_MAGENTA,
DYNAMIC_YELLOW,
DYNAMIC_WHITE,
FLEX,
LAST_COLORELEMENT
} ColorElements;

Expand Down
203 changes: 203 additions & 0 deletions FlexMeter.c
Original file line number Diff line number Diff line change
@@ -0,0 +1,203 @@
/*
htop - FlexMeter.c
(C) 2024 Stoyan Bogdanov
Released under the GNU GPLv2+, see the COPYING file
in the source distribution for its full text.
*/

#include <dirent.h>
#include <pwd.h>
#include <sys/time.h>
#include <sys/types.h>
#include <time.h>
#include <unistd.h>

#include "CRT.h"
#include "config.h"
#include "FlexMeter.h"
#include "Object.h"


#define FLEX_CFG_FOLDER ".config/htop/FlexMeter"

typedef struct {
char* name;
char* command;
char* type;
char* caption;
char* uiName;
} _flex_meter;

_flex_meter meter_list[METERS_LIST_SIZE];

static int meters_count = 0;

static const int DateMeter_attributes[] = {
FLEX
};

MeterClass* FlexMeter_class = NULL;

static int check_for_meters(void);

static bool parse_input(_flex_meter *meter, char* line)
{
switch(line[0])
{
case 'n':
if (String_startsWith(line, "name=")) {
xAsprintf(&meter->uiName, "Flex: %s", line + 5);
}
break;
case 'c':
if (String_startsWith(line, "command=")) {
meter->command = xStrdup(line + 8);
} else if (String_startsWith(line, "caption=")) {
meter->caption = xStrdup(line + 8);
}
break;
case 't':
if (String_startsWith(line, "type=")) {
meter->type = xStrdup(line + 6);
}
break;
default:
return false;
}

return true;
}

static bool load_config(_flex_meter *meter, char* file)
{
bool ret = false;
FILE* fp = fopen(file, "r");

if (fp != NULL) {
char* buff;
while ((buff = String_readLine(fp)) != NULL) {
ret = parse_input(meter, buff);
if (!ret) {
break;
}
}
free(buff);
buff = NULL;
}

fclose(fp);
return ret;
}

static int check_for_meters(void)
{
char* path;
struct dirent* dir;
struct passwd* pw = getpwuid(getuid());
const char* homedir = pw->pw_dir;
char* home = NULL;
bool ret;

xAsprintf(&home, "%s/%s", homedir, FLEX_CFG_FOLDER);
struct stat fileStat;

if (stat(home, &fileStat) < 0) {
return -1;
}

uint32_t uid = getuid();

if ((fileStat.st_uid == uid) && S_ISDIR(fileStat.st_mode) && ((fileStat.st_mode & 0777) == 0700)) {
DIR* d = opendir(home);
if (d) {
while ((dir = readdir(d)) != NULL) {
if ( dir->d_name[0] == '.') {
/* We are ignoring all files starting with . like ".Template"
* and "." ".." directories
*/
continue;
}

meter_list[meters_count].name = xStrdup(dir->d_name);
xAsprintf(&path, "%s/%s", home, dir->d_name);

if (access(path, R_OK | W_OK | X_OK) == 0) {
ret = load_config(&meter_list[meters_count], path);

if (ret && (meters_count < MAX_METERS_COUNT)) {
meters_count++;
}
}
if (path != NULL) {
free(path);
path=NULL;
}
}
closedir(d);
}
}

if (home != NULL) {
free(home);
home = NULL;
}
return meters_count;
}

static void FlexMeter_updateValues(Meter* this)
{
for (size_t i = 0 ; i < (size_t)meters_count; i++) {
if (this->m_ptr == &FlexMeter_class[i] ) {
char* buff = NULL;
int ret = -1;
if (meter_list[i].command) {
FILE* fd = popen(meter_list[i].command, "r");
if (fd) {
buff = String_readLine(fd);
ret = pclose(fd);
}
}

if (buff && !ret) {
xSnprintf(this->txtBuffer, sizeof(this->txtBuffer), "%s", buff);
} else {
// Once fail, free command pointer and every time print Error message
if (meter_list[i].command != NULL) {
free(meter_list[i].command);
meter_list[i].command = NULL;
}
xSnprintf(this->txtBuffer, sizeof(this->txtBuffer), "%s", "[ERR] Check command");
}
}
}
}

const MeterClass FlexMeter_class_template = {
.super = {
.extends = Class(Meter),
.delete = Meter_delete
},
.updateValues = FlexMeter_updateValues,
.defaultMode = TEXT_METERMODE,
.maxItems = 1,
.total = 100,
.attributes = DateMeter_attributes,
.name = NULL,
.uiName = NULL,
.caption = NULL,
};

int load_flex_modules(void)
{
size_t meters_num = check_for_meters();
if (!FlexMeter_class && meters_num > 0) {
FlexMeter_class = (MeterClass*) xCalloc(meters_num, sizeof(MeterClass));
for (size_t i = 0 ; i < meters_num; i++) {
memcpy(&FlexMeter_class[i], &FlexMeter_class_template, sizeof(MeterClass));
FlexMeter_class[i].name = (const char*) xStrdup(meter_list[i].name);
FlexMeter_class[i].uiName = (const char*) xStrdup(meter_list[i].uiName);
FlexMeter_class[i].caption = (const char*) xStrdup(meter_list[i].caption);
}
}
return meters_num;
}
22 changes: 22 additions & 0 deletions FlexMeter.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
#ifndef HEADER_FlexMeter
#define HEADER_FlexMeter
/*
htop - FlexMeter.c
(C) 2024 Stoyan Bogdanov
Released under the GNU GPLv2+, see the COPYING file
in the source distribution for its full text.
*/
#include <stdbool.h>

#include "Meter.h"


#define METERS_LIST_SIZE 30

#define MAX_METERS_COUNT METERS_LIST_SIZE-1

extern MeterClass *FlexMeter_class ;

int load_flex_modules(void);

#endif
10 changes: 9 additions & 1 deletion Header.c
Original file line number Diff line number Diff line change
Expand Up @@ -20,14 +20,14 @@ in the source distribution for its full text.
#include "CRT.h"
#include "CPUMeter.h"
#include "DynamicMeter.h"
#include "FlexMeter.h"
#include "Macros.h"
#include "Object.h"
#include "Platform.h"
#include "ProvideCurses.h"
#include "Settings.h"
#include "XUtils.h"


Header* Header_new(Machine* host, HeaderLayout hLayout) {
Header* this = xCalloc(1, sizeof(Header));
this->columns = xMallocArray(HeaderLayout_getColumns(hLayout), sizeof(Vector*));
Expand Down Expand Up @@ -121,6 +121,14 @@ void Header_populateFromSettings(Header* this) {
const Settings* settings = this->host->settings;
Header_setLayout(this, settings->hLayout);

int num = load_flex_modules();
int platform_size = 0;

for (platform_size = 0; Platform_meterTypes[platform_size] != NULL; platform_size++);
for (int i = 0; i < num; i++) Platform_meterTypes[platform_size+i]=FlexMeter_class+i;

Platform_meterTypes[platform_size+num]=NULL;

Header_forEachColumn(this, col) {
const MeterColumnSetting* colSettings = &settings->hColumns[col];
Vector_prune(this->columns[col]);
Expand Down
4 changes: 4 additions & 0 deletions Makefile.am
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,7 @@ myhtopsources = \
DynamicScreen.c \
EnvScreen.c \
FileDescriptorMeter.c \
FlexMeter.c \
FunctionBar.c \
Hashtable.c \
Header.c \
Expand Down Expand Up @@ -93,6 +94,7 @@ myhtopsources = \
Vector.c \
XUtils.c


myhtopheaders = \
Action.h \
Affinity.h \
Expand All @@ -118,6 +120,7 @@ myhtopheaders = \
DynamicScreen.h \
EnvScreen.h \
FileDescriptorMeter.h \
FlexMeter.h \
FunctionBar.h \
Hashtable.h \
Header.h \
Expand Down Expand Up @@ -165,6 +168,7 @@ myhtopheaders = \
Vector.h \
XUtils.h


# Linux
# -----

Expand Down
1 change: 1 addition & 0 deletions Meter.c
Original file line number Diff line number Diff line change
Expand Up @@ -377,6 +377,7 @@ Meter* Meter_new(const Machine* host, unsigned int param, const MeterClass* type
this->values = type->maxItems ? xCalloc(type->maxItems, sizeof(double)) : NULL;
this->total = type->total;
this->caption = xStrdup(type->caption);
this->m_ptr = type;
if (Meter_initFn(this)) {
Meter_init(this);
}
Expand Down
9 changes: 5 additions & 4 deletions Meter.h
Original file line number Diff line number Diff line change
Expand Up @@ -67,9 +67,9 @@ typedef struct MeterClass_ {
const MeterModeId defaultMode;
const double total;
const int* const attributes;
const char* const name; /* internal name of the meter, must not contain any space */
const char* const uiName; /* display name in header setup menu */
const char* const caption; /* prefix in the actual header */
const char* name; /* internal name of the meter, must not contain any space */
const char* uiName; /* display name in header setup menu */
const char* caption; /* prefix in the actual header */
const char* const description; /* optional meter description in header setup menu */
const uint8_t maxItems;
const bool isMultiColumn; /* whether the meter draws multiple sub-columns (defaults to false) */
Expand Down Expand Up @@ -102,8 +102,9 @@ typedef struct GraphData_ {
struct Meter_ {
Object super;
Meter_Draw draw;
const Machine* host;

const Machine* host;
const MeterClass* m_ptr;
char* caption;
MeterModeId mode;
unsigned int param;
Expand Down
1 change: 1 addition & 0 deletions configure.ac
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ AM_INIT_AUTOMAKE([-Wall std-options subdir-objects])

# ----------------------------------------------------------------------

AC_DEFINE([MAX_PLATFORM_METERS], [100], [Set max meters for number])

# ----------------------------------------------------------------------
# Checks for platform.
Expand Down
4 changes: 4 additions & 0 deletions docs/FlexMeter/Template
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
name=template
command=echo "`uptime`"
type=TEXT_METERMODE
caption="UPTIME"

0 comments on commit 3ff3a68

Please sign in to comment.