1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
#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);
iBlock *output = readOutputUntilClosed_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);
}
|
