2 files changed, 97 insertions, 10 deletions

diff --git a/src/Scraping/Scraping.h b/src/Scraping/Scraping.h
index 014285f..be65e5a 100644
--- a/src/Scraping/Scraping.h
+++ b/src/Scraping/Scraping.h
@@ -15,6 +15,7 @@ typedef int (*ParserFunc)(const char *engine_name, xmlDocPtr doc,
                           SearchResult **out_results, int max_results);
 
 typedef struct {
+  const char *id;
   const char *name;
   const char *base_url;
   const char *host_header;
@@ -24,6 +25,7 @@ typedef struct {
   int page_multiplier;
   int page_base;
   ParserFunc parser;
+  int enabled;
 } SearchEngine;
 
 typedef struct {
@@ -54,8 +56,9 @@ typedef struct {
   ScrapeStatus status;
 } ScrapeJob;
 
-extern const SearchEngine ENGINE_REGISTRY[];
+extern SearchEngine ENGINE_REGISTRY[];
 extern const int ENGINE_COUNT;
+void apply_engines_config(const char *engines_str);
 
 size_t write_memory_callback(void *contents, size_t size, size_t nmemb,
                              void *userp);
diff --git a/src/Scraping/ScrapingParsers.c b/src/Scraping/ScrapingParsers.c
index 874cf54..96aaded 100644
--- a/src/Scraping/ScrapingParsers.c
+++ b/src/Scraping/ScrapingParsers.c
@@ -310,38 +310,122 @@ static int parse_yahoo(const char *engine_name, xmlDocPtr doc,
 static int parse_mojeek(const char *engine_name, xmlDocPtr doc,
                         SearchResult **out_results, int max_results);
 
-const SearchEngine ENGINE_REGISTRY[] = {
-    {.name = "DuckDuckGo Lite",
+SearchEngine ENGINE_REGISTRY[] = {
+    {.id = "ddg",
+     .name = "DuckDuckGo Lite",
      .base_url = "https://lite.duckduckgo.com/lite/?q=",
      .host_header = "lite.duckduckgo.com",
      .referer = "https://lite.duckduckgo.com/",
      .page_param = "s",
      .page_multiplier = 30,
      .page_base = 0,
-     .parser = parse_ddg_lite},
-    {.name = "Startpage",
+     .parser = parse_ddg_lite,
+     .enabled = 1},
+    {.id = "startpage",
+     .name = "Startpage",
      .base_url = "https://www.startpage.com/sp/search?q=",
      .host_header = "www.startpage.com",
      .referer = "https://www.startpage.com/",
      .page_param = "page",
      .page_multiplier = 1,
      .page_base = 1,
-     .parser = parse_startpage},
-    {.name = "Yahoo",
+     .parser = parse_startpage,
+     .enabled = 1},
+    {.id = "yahoo",
+     .name = "Yahoo",
      .base_url = "https://search.yahoo.com/search?p=",
      .host_header = "search.yahoo.com",
      .referer = "https://search.yahoo.com/",
      .page_param = "b",
      .page_multiplier = 10,
      .page_base = 1,
-     .parser = parse_yahoo},
-    {.name = "Mojeek",
+     .parser = parse_yahoo,
+     .enabled = 1},
+    {.id = "mojeek",
+     .name = "Mojeek",
      .base_url = "https://www.mojeek.com/search?q=",
      .host_header = "www.mojeek.com",
      .referer = "https://www.mojeek.com/",
      .page_param = "s",
      .page_multiplier = 10,
      .page_base = 1,
-     .parser = parse_mojeek}};
+     .parser = parse_mojeek,
+     .enabled = 1}};
 
 const int ENGINE_COUNT = sizeof(ENGINE_REGISTRY) / sizeof(SearchEngine);
+
+static int engine_id_compare(const char *engine_id, const char *config_id) {
+  while (*engine_id && *config_id) {
+    char e = *engine_id;
+    char c = *config_id;
+    if (e >= 'A' && e <= 'Z')
+      e = e - 'A' + 'a';
+    if (c >= 'A' && c <= 'Z')
+      c = c - 'A' + 'a';
+    if (e != c)
+      return 0;
+    engine_id++;
+    config_id++;
+  }
+  return *engine_id == *config_id;
+}
+
+void apply_engines_config(const char *engines_str) {
+  if (!engines_str || engines_str[0] == '\0') {
+    for (int i = 0; i < ENGINE_COUNT; i++) {
+      ENGINE_REGISTRY[i].enabled = 1;
+    }
+    return;
+  }
+
+  for (int i = 0; i < ENGINE_COUNT; i++) {
+    ENGINE_REGISTRY[i].enabled = 0;
+  }
+
+  char *copy = strdup(engines_str);
+  if (!copy)
+    return;
+
+  char *saveptr;
+  char *token = strtok_r(copy, ",", &saveptr);
+
+  while (token) {
+    while (*token == ' ' || *token == '\t')
+      token++;
+
+    if (strcmp(token, "*") == 0) {
+      for (int i = 0; i < ENGINE_COUNT; i++) {
+        ENGINE_REGISTRY[i].enabled = 1;
+      }
+    } else if (token[0] == '-' && token[1] != '\0') {
+      char *engine_id = token + 1;
+      int found = 0;
+      for (int i = 0; i < ENGINE_COUNT; i++) {
+        if (engine_id_compare(ENGINE_REGISTRY[i].id, engine_id)) {
+          ENGINE_REGISTRY[i].enabled = 0;
+          found = 1;
+          break;
+        }
+      }
+      if (!found) {
+        fprintf(stderr, "[WARN] Unknown engine: %s\n", engine_id);
+      }
+    } else {
+      int found = 0;
+      for (int i = 0; i < ENGINE_COUNT; i++) {
+        if (engine_id_compare(ENGINE_REGISTRY[i].id, token)) {
+          ENGINE_REGISTRY[i].enabled = 1;
+          found = 1;
+          break;
+        }
+      }
+      if (!found) {
+        fprintf(stderr, "[WARN] Unknown engine: %s\n", token);
+      }
+    }
+
+    token = strtok_r(NULL, ",", &saveptr);
+  }
+
+  free(copy);
+}