1 files changed, 139 insertions, 0 deletions

diff --git a/src/mimehooks.c b/src/mimehooks.c
new file mode 100644
index 00000000..3d999aba
--- /dev/null
+++ b/src/mimehooks.c
@@ -0,0 +1,139 @@
+#include "mimehooks.h"
+
+#include <the_Foundation/file.h>
+#include <the_Foundation/path.h>
+#include <the_Foundation/process.h>
+#include <the_Foundation/stringlist.h>
+
+iDefineTypeConstruction(FilterHook)
+
+void init_FilterHook(iFilterHook *d) {
+    init_String(&d->label);
+    init_String(&d->mimePattern);
+    init_String(&d->command);
+    d->mimeRegex = NULL;
+}
+
+void deinit_FilterHook(iFilterHook *d) {
+    iRelease(d->mimeRegex);
+    deinit_String(&d->command);
+    deinit_String(&d->mimePattern);
+    deinit_String(&d->label);
+}
+
+void setMimePattern_FilterHook(iFilterHook *d, const iString *pattern) {
+    iReleasePtr(&d->mimeRegex);
+    d->mimeRegex = new_RegExp(cstr_String(pattern), caseInsensitive_RegExpOption);
+}
+
+void setCommand_FilterHook(iFilterHook *d, const iString *command) {
+    set_String(&d->command, command);
+}
+
+iBlock *run_FilterHook_(const iFilterHook *d, const iString *mime, const iBlock *body) {
+    iProcess *   proc = new_Process();
+    iStringList *args = new_StringList();
+    iRangecc     seg  = iNullRange;
+    while (nextSplit_Rangecc(range_String(&d->command), ";", &seg)) {
+        pushBackRange_StringList(args, seg);
+    }
+    seg = iNullRange;
+    while (nextSplit_Rangecc(range_String(mime), ";", &seg)) {
+        pushBackRange_StringList(args, seg);
+    }
+    setArguments_Process(proc, args);
+    iRelease(args);
+    start_Process(proc);
+    writeInput_Process(proc, body);
+    waitForFinished_Process(proc);
+    iBlock *output = readOutput_Process(proc);
+    if (!startsWith_Rangecc(range_Block(output), "20")) {
+        /* Didn't produce valid output. */
+        delete_Block(output);
+        output = NULL;
+    }
+    iRelease(proc);
+    return output;
+}
+
+/*----------------------------------------------------------------------------------------------*/
+
+struct Impl_MimeHooks {
+    iPtrArray filters;
+};
+
+iDefineTypeConstruction(MimeHooks)
+
+void init_MimeHooks(iMimeHooks *d) {
+    init_PtrArray(&d->filters);
+}
+
+void deinit_MimeHooks(iMimeHooks *d) {
+    iForEach(PtrArray, i, &d->filters) {
+        delete_FilterHook(i.ptr);
+    }
+    deinit_PtrArray(&d->filters);
+}
+
+iBool willTryFilter_MimeHooks(const iMimeHooks *d, const iString *mime) {
+    /* TODO: Combine this function with tryFilter_MimeHooks? */
+    iRegExpMatch m;
+    iConstForEach(PtrArray, i, &d->filters) {
+        const iFilterHook *xc = i.ptr;
+        init_RegExpMatch(&m);
+        if (matchString_RegExp(xc->mimeRegex, mime, &m)) {
+            return iTrue;
+        }
+    }
+    return iFalse;
+}
+
+iBlock *tryFilter_MimeHooks(const iMimeHooks *d, const iString *mime, const iBlock *body) {
+    iRegExpMatch m;
+    iConstForEach(PtrArray, i, &d->filters) {
+        const iFilterHook *xc = i.ptr;
+        init_RegExpMatch(&m);
+        if (matchString_RegExp(xc->mimeRegex, mime, &m)) {
+            iBlock *result = run_FilterHook_(xc, mime, body);
+            if (result) {
+                return result;
+            }
+        }
+    }
+    return NULL;
+}
+
+static const char *mimeHooksFilename_MimeHooks_ = "mimehooks.txt";
+
+void load_MimeHooks(iMimeHooks *d, const char *saveDir) {
+    iFile *f = newCStr_File(concatPath_CStr(saveDir, mimeHooksFilename_MimeHooks_));
+    if (open_File(f, read_FileMode | text_FileMode)) {
+        iBlock * src     = readAll_File(f);
+        iRangecc srcLine = iNullRange;
+        int      pos     = 0;
+        iRangecc lines[3];
+        iZap(lines);
+        while (nextSplit_Rangecc(range_Block(src), "\n", &srcLine)) {
+            iRangecc line = srcLine;
+            trim_Rangecc(&line);
+            if (isEmpty_Range(&line)) {
+                continue;
+            }
+            lines[pos++] = line;
+            if (pos == 3) {
+                iFilterHook *hook = new_FilterHook();
+                setRange_String(&hook->label, lines[0]);
+                setMimePattern_FilterHook(hook, collect_String(newRange_String(lines[1])));
+                setCommand_FilterHook(hook, collect_String(newRange_String(lines[2])));
+                pushBack_PtrArray(&d->filters, hook);
+            }
+        }
+        delete_Block(src);
+    }
+    iRelease(f);
+}
+
+void save_MimeHooks(const iMimeHooks *d) {
+    iUnused(d);
+}
+