tweak usage of str replacement code

1 files changed, 22 insertions, 36 deletions

diff --git a/plugins/shortcuts/shortcuts.c b/plugins/shortcuts/shortcuts.c
index 9bf4a25..f4ab13f 100644
--- a/plugins/shortcuts/shortcuts.c
+++ b/plugins/shortcuts/shortcuts.c
@@ -5,30 +5,18 @@
 #include "../strings/strings.h"
 
 #define SHORTCUT_N 41
-
-/* Uncomment for debug */
-/* #define DEBUG */
+#define DEBUG false
 
 /* Inspired by https://duckduckgo.com/bangs */
-
 int shortcut_expand(const char* uri, char* output)
 {
-    printf("SHORTCUT EXPAND!\n");
-    int l1 = strlen(uri);
-    int l2 = strlen(output);
-    int len;
-    char tmp_uri[l2++];
-    char tmp_output[l2++];
+    int len_uri = strlen(uri);
+    int len_output = strlen(output);
 
-    if ((l2 - l1) < SHORTCUT_N) {
-#ifdef DEBUG
-        printf("Not enough memory\n");
-#endif
+    if ((len_output - len_uri) < SHORTCUT_N) {
+        fprintf(stderr, "Not enough memory\n");
         return 1; // not enough memory.
     } else {
-        strcpy(tmp_uri, uri); // strcpy also copies the terminating '\0'
-        strcpy(tmp_output, output);
-
         char* shortcuts[] = {
             "!aa",
             "!blog",
@@ -50,30 +38,28 @@ int shortcut_expand(const char* uri, char* output)
         };
 
         // len = sizeof(shortcuts) / sizeof(shortcuts[0]);
-        len = sizeof(shortcuts) / sizeof(char*);
+        int len = sizeof(shortcuts) / sizeof(char*);
 
         for (int i = 0; i < len; i++) {
-            int replace_check = str_replace_start(tmp_uri, shortcuts[i],
+            str_init(output, len_output);
+            int replace_check = str_replace_start(uri, shortcuts[i],
                 expansions[i], output);
-            if (replace_check == 2) {
-#ifdef DEBUG
-                printf("tmp_uri: %s\n", tmp_uri);
-                printf("output: %s\n", output);
-#endif
-                return 2;
-            } else if (replace_check == 1) {
-#ifdef DEBUG
-                printf("replace_check failed\n");
-#endif
-                return 1;
+            switch (replace_check) {
+                case 0: // no match found
+                    break;
+                case 1: // str_replace_start somehow failed
+                    fprintf(stderr, "str_replace_start failed\n");
+                    return 1;
+                    break;
+                case 2: // match succeeded
+                    return 2;
+                    break;
+                default:
+                    fprintf(stderr, "Unreachable state\n");
             }
-            strcpy(tmp_uri, output);
-            str_init(output, l2);
         }
-        strcpy(output, tmp_uri);
+        strcpy(output, uri);
     }
-#ifdef DEBUG
-    printf("No match found\n\n");
-#endif
+    if(DEBUG) printf("No match found\n\n");
     return 0;
 }