added modules to tofi (currently math(documented) and search(undocumented) with firefox only)

Author: flloschy

doc/tofi.5.md

@@ -630,6 +630,10 @@ options.
 >
 > Default: true
 
+## MODULES
+**module-math**=true|false
+> Calculate Math equations inside the Input field. The Calculation result will appear inside the result-suggestions, when Selecting this result it will be copied to clipboard and a notification will be sent
+
 ## COLORS
 
 Colors can be specified in the form *RGB*, *RGBA*, *RRGGBB* or

meson.build

@@ -116,6 +116,7 @@ common_sources = files(
   'src/surface.c',
   'src/unicode.c',
   'src/xmalloc.c',
+  'src/modules.c',
 )
 
 compgen_sources = files(

src/config.c

@@ -766,6 +766,16 @@ bool parse_option(struct tofi *tofi, const char *filename, size_t lineno, const
 		if (!err) {
 			tofi->use_scale = val;
 		}
+    } else if (strcasecmp(option, "module-math") == 0) {
+        bool val = parse_bool(filename, lineno, value, &err);
+        if (!err) {
+        tofi->module_math = val;
+        }
+    } else if (strcasecmp(option, "module-search") == 0) {
+        bool val = parse_bool(filename, lineno, value, &err);
+        if (!err) {
+            tofi->module_search = val;
+        }
 	} else {
 		PARSE_ERROR(filename, lineno, "Unknown option \"%s\"\n", option);
 		err = true;

src/desktop_vec.c

@@ -6,6 +6,7 @@
 #include "string_vec.h"
 #include "unicode.h"
 #include "xmalloc.h"
+#include "modules.h"
 
 static bool match_current_desktop(char * const *desktop_list, gsize length);
 
@@ -148,6 +149,7 @@ struct desktop_entry *desktop_vec_find_sorted(struct desktop_vec *restrict vec,
 }
 
 struct string_ref_vec desktop_vec_filter(
+        struct tofi *tofi,
 		const struct desktop_vec *restrict vec,
 		const char *restrict substr,
 		enum matching_algorithm algorithm)
@@ -175,6 +177,10 @@ struct string_ref_vec desktop_vec_filter(
 			}
 		}
 	}
+
+    // add suggestions form enabled modules
+    modules_suggest(tofi, substr, &filt);
+
 	/*
 	 * Sort the results by this search_score. This moves matches at the beginnings
 	 * of words to the front of the result list.

src/desktop_vec.h

@@ -35,8 +35,10 @@ void desktop_vec_add_file(struct desktop_vec *desktop, const char *id, const cha
 
 void desktop_vec_sort(struct desktop_vec *restrict vec);
 struct desktop_entry *desktop_vec_find_sorted(struct desktop_vec *restrict vec, const char *name);
+struct tofi;
 struct string_ref_vec desktop_vec_filter(
-		const struct desktop_vec *restrict vec,
+		struct tofi *tofi,
+        const struct desktop_vec *restrict vec,
 		const char *restrict substr,
 		enum matching_algorithm algorithm);
 

src/input.c

@@ -203,7 +203,7 @@ void add_character(struct tofi *tofi, xkb_keycode_t keycode)
 		entry->input_utf8_length += len;
 
 		if (entry->mode == TOFI_MODE_DRUN) {
-			struct string_ref_vec results = desktop_vec_filter(&entry->apps, entry->input_utf8, tofi->matching_algorithm);
+			struct string_ref_vec results = desktop_vec_filter(tofi, &entry->apps, entry->input_utf8, tofi->matching_algorithm);
 			string_ref_vec_destroy(&entry->results);
 			entry->results = results;
 		} else {
@@ -241,7 +241,7 @@ void input_refresh_results(struct tofi *tofi)
 	entry->input_utf8_length = bytes_written;
 	string_ref_vec_destroy(&entry->results);
 	if (entry->mode == TOFI_MODE_DRUN) {
-		entry->results = desktop_vec_filter(&entry->apps, entry->input_utf8, tofi->matching_algorithm);
+		entry->results = desktop_vec_filter(tofi, &entry->apps, entry->input_utf8, tofi->matching_algorithm);
 	} else {
 		entry->results = string_ref_vec_filter(&entry->commands, entry->input_utf8, tofi->matching_algorithm);
 	}

src/main.c

@@ -13,6 +13,7 @@
 #include <wayland-client.h>
 #include <wayland-util.h>
 #include <xkbcommon/xkbcommon.h>
+#include "src/modules.h"
 #include "tofi.h"
 #include "compgen.h"
 #include "drun.h"
@@ -1030,6 +1031,8 @@ static bool do_submit(struct tofi *tofi)
 			}
 		}
 		if (app == NULL) {
+            if (modules_try_execute(res, entry->input_utf8))
+                return true;
 			log_error("Couldn't find application file! This shouldn't happen.\n");
 			return false;
 		}
@@ -1200,6 +1203,8 @@ int main(int argc, char *argv[])
 		.require_match = true,
 		.use_scale = true,
 		.physical_keybindings = true,
+        .module_math = true,
+        .module_search = false
 	};
 	wl_list_init(&tofi.output_list);
 	if (getenv("TERMINAL") != NULL) {

src/modules.c

@@ -0,0 +1,86 @@
+#include "modules.h"
+#include "history.h"
+#include "string_vec.h"
+#include "unicode.h"
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+void module_math_suggestion(const char *restrict query,
+                            struct string_ref_vec *filt) {
+  char command[128] = "qalc -t -m 1000  \"";
+  command[sizeof(command) - 1] = '\0'; // Ensure null termination
+  strncat(command, query, sizeof(command) - 1);
+  strncat(command, "\"", sizeof(command) - 1);
+
+  char result[128] = "";
+
+  // execute command
+  FILE *fp = popen(command, "r");
+  // limit output to 50 characters
+  fgets(result, 50, fp);
+  pclose(fp);
+
+  // add module prefix
+  char finalresult[128] = "=";
+  strncat(finalresult, result, sizeof(finalresult) - 1);
+  finalresult[strlen(finalresult) - 1] = '\0'; // Ensure null termination
+
+  string_ref_vec_add(filt, utf8_normalize(finalresult));
+}
+
+bool module_math_selected(char *suggestion, char *query) {
+  char notify[128] = "notify-send \"Calculation: ";
+  // set notification title to be the calculation
+  strncat(notify, query, sizeof(notify) - 1);
+  strncat(notify, "\" \"", sizeof(notify) - 1);
+  // set notification body to be result
+  strncat(notify, suggestion, sizeof(notify) - 1);
+  strncat(notify, "\"", sizeof(notify) - 1);
+
+  char copy[128] = "wl-copy \"";
+  strncat(copy, suggestion, sizeof(notify) - 1);
+  strncat(copy, "\"", sizeof(notify) - 1);
+
+  system(copy);
+  system(notify);
+
+  // return is here to avoid an ugly switch
+  return true;
+}
+
+void module_search_suggestion(char *restrict query,
+                              struct string_ref_vec *restrict filt) {
+  // always display this to avoid duplicate string
+  string_ref_vec_add(filt, "?search");
+}
+
+bool module_search_selected(char *restrict query) {
+  char command[128] = "firefox --search \"";
+  strncat(command, query, sizeof(command) - 1);
+  strncat(command, "\"", sizeof(command) - 1);
+
+  system(command);
+  // return is here to avoid an ugly switch
+  return true;
+}
+
+void modules_suggest(struct tofi *tofi, char *restrict query,
+                     struct string_ref_vec *filt) {
+  if (tofi->module_math)
+    module_math_suggestion(query, filt);
+  if (tofi->module_search)
+    module_search_suggestion(query, filt);
+}
+
+bool modules_try_execute(char *suggestion, char *restrict query) {
+  char prefix = suggestion[0];
+  // remove first character (prefix)
+  suggestion++;
+  switch (prefix) {
+  case '=':
+    return module_math_selected(suggestion, query);
+  case '?':
+    return module_search_selected(query);
+  }
+}

src/modules.h

@@ -0,0 +1,10 @@
+#ifndef MODULES_H
+#define MODULES_H
+
+#include "tofi.h"
+
+void modules_suggest(struct tofi *tofi, char *restrict query,
+                     struct string_ref_vec *filt);
+bool modules_try_execute(char *suggestion, char *restrict query);
+
+#endif /* MKDIRP_H */

src/tofi.h

@@ -104,6 +104,8 @@ struct tofi {
 	bool print_index;
 	bool multiple_instance;
 	bool physical_keybindings;
+    bool module_math;
+    bool module_search;
 	char target_output_name[MAX_OUTPUT_NAME_LEN];
 	char default_terminal[MAX_TERMINAL_NAME_LEN];
 	char history_file[MAX_HISTORY_FILE_NAME_LEN];