]> jfr.im git - irc/quakenet/newserv.git/blame - newsearch/newsearch.h
iirc, changes here were to enable search/patriciasearch etc to use the AST stuff
[irc/quakenet/newserv.git] / newsearch / newsearch.h
CommitLineData
16cc6e32 1#include "../nick/nick.h"
a92bb8e1 2#include "../lib/sstring.h"
56a84a81
IB
3#include "../parser/parser.h"
4#include "../channel/channel.h"
5#include "../lib/flags.h"
e8ad630b 6#include "../authext/authext.h"
a92bb8e1 7#include "../patricia/patricia.h"
c86edd1d 8
1188d606
P
9#ifndef __NEWSEARCH_H
10#define __NEWSEARCH_H
11
f16cabe4 12#define NSMAX_KILL_LIMIT 500
16cc6e32 13#define NSMAX_GLINE_LIMIT 500
22f7c5c6
IB
14#define NSMAX_GLINE_CLONES 5
15
16cc6e32
IB
16/* gline duration, in seconds */
17#define NSGLINE_DURATION 3600
18
96429168 19#define NSMAX_REASON_LEN 120
4622c762 20#define NSMAX_NOTICE_LEN 250
a92bb8e1 21#define NSMAX_COMMAND_LEN 20
16cc6e32 22
c86edd1d
Q
23#define RETURNTYPE_BOOL 0x01
24#define RETURNTYPE_INT 0x02
25#define RETURNTYPE_STRING 0x03
26#define RETURNTYPE_TYPE 0xFF
27#define RETURNTYPE_CONST 0x100
28
b697c21c
CP
29#define VARIABLE_LEN 10
30#define MAX_VARIABLES 10
31
c86edd1d 32struct searchNode;
c8be5183 33struct searchCtx;
b697c21c 34struct coercedata;
c86edd1d 35
f33f3f52 36typedef struct searchNode *(*searchParseFunc)(struct searchCtx *ctx, char *input);
c8be5183 37typedef void (*replyFunc)(nick *np, char *format, ...);
0da2a4ae 38typedef void (*wallFunc)(int level, char *format, ...);
c8be5183 39
f33f3f52 40typedef struct searchNode *(*parseFunc)(struct searchCtx *, int, char **);
b697c21c
CP
41typedef void (*freeFunc)(struct searchCtx *, struct searchNode *);
42typedef void *(*exeFunc)(struct searchCtx *, struct searchNode *, void *);
43typedef void (*ChanDisplayFunc)(struct searchCtx *, nick *, chanindex *);
44typedef void (*NickDisplayFunc)(struct searchCtx *, nick *, nick *);
45typedef void (*UserDisplayFunc)(struct searchCtx *, nick *, authname *);
46typedef void (*HeaderFunc)(void *sender, void *arg);
47
48struct coercedata {
49 struct searchNode *child;
50 union {
51 char *stringbuf;
52 unsigned long val;
53 } u;
54};
55
56typedef struct searchNode {
57 int returntype;
58 exeFunc exe;
59 freeFunc free;
60 void *localdata;
61} searchNode;
62
63struct searchVariable {
64 char name[VARIABLE_LEN];
65 struct searchNode data;
66 struct coercedata cdata;
67};
68
a92bb8e1
P
69typedef struct searchCmd {
70 void *defaultdisplayfunc;
71 sstring *name;
72 CommandHandler handler;
73 struct CommandTree *outputtree;
74 struct CommandTree *searchtree;
75} searchCmd;
76
77typedef struct searchList {
78 void *cmd;
79 sstring *name;
e8ff7b61 80 char *help;
a92bb8e1
P
81 struct searchList *next;
82} searchList;
83
c8be5183
CP
84typedef struct searchCtx {
85 searchParseFunc parser;
86 replyFunc reply;
0da2a4ae 87 wallFunc wall;
0a659cde 88 void *arg;
b697c21c
CP
89 struct searchVariable vars[MAX_VARIABLES];
90 int lastvar;
a92bb8e1 91 struct searchCmd *searchcmd;
e8ff7b61 92 nick *sender;
6b2202d0
CP
93 int limit;
94 void *displayfn;
c8be5183
CP
95} searchCtx;
96
c86edd1d 97/* Core functions */
c7f7a584 98/* Logical (BOOL -> BOOL)*/
f33f3f52
P
99struct searchNode *and_parse(searchCtx *ctx, int argc, char **argv);
100struct searchNode *not_parse(searchCtx *ctx, int argc, char **argv);
101struct searchNode *or_parse(searchCtx *ctx, int argc, char **argv);
c7f7a584 102
103/* Comparison (INT -> BOOL) */
f33f3f52
P
104struct searchNode *eq_parse(searchCtx *ctx, int argc, char **argv);
105struct searchNode *lt_parse(searchCtx *ctx, int argc, char **argv);
106struct searchNode *gt_parse(searchCtx *ctx, int argc, char **argv);
c7f7a584 107
108/* String match (STRING -> BOOL) */
f33f3f52
P
109struct searchNode *match_parse(searchCtx *ctx, int argc, char **argv);
110struct searchNode *regex_parse(searchCtx *ctx, int argc, char **argv);
c7f7a584 111
112/* Length (STRING -> INT) */
f33f3f52 113struct searchNode *length_parse(searchCtx *ctx, int argc, char **argv);
c7f7a584 114
115/* kill/gline actions (BOOL) */
f33f3f52
P
116struct searchNode *kill_parse(searchCtx *ctx, int argc, char **argv);
117struct searchNode *gline_parse(searchCtx *ctx, int argc, char **argv);
c7f7a584 118
4622c762 119/* notice action (BOOL) */
f33f3f52 120struct searchNode *notice_parse(searchCtx *ctx, int argc, char **argv);
4622c762 121
c7f7a584 122/* Nick/Channel functions (various types) */
f33f3f52
P
123struct searchNode *nick_parse(searchCtx *ctx, int argc, char **argv);
124struct searchNode *modes_parse(searchCtx *ctx, int argc, char **argv);
c7f7a584 125
126/* Nick functions (various types) */
f33f3f52
P
127struct searchNode *hostmask_parse(searchCtx *ctx, int argc, char **argv);
128struct searchNode *realname_parse(searchCtx *ctx, int argc, char **argv);
129struct searchNode *authname_parse(searchCtx *ctx, int argc, char **argv);
130struct searchNode *authts_parse(searchCtx *ctx, int argc, char **argv);
131struct searchNode *ident_parse(searchCtx *ctx, int argc, char **argv);
132struct searchNode *host_parse(searchCtx *ctx, int argc, char **argv);
133struct searchNode *channel_parse(searchCtx *ctx, int argc, char **argv);
134struct searchNode *timestamp_parse(searchCtx *ctx, int argc, char **argv);
135struct searchNode *country_parse(searchCtx *ctx, int argc, char **argv);
136struct searchNode *ip_parse(searchCtx *ctx, int argc, char **argv);
137struct searchNode *channels_parse(searchCtx *ctx, int argc, char **argv);
138struct searchNode *server_parse(searchCtx *ctx, int argc, char **argv);
139struct searchNode *authid_parse(searchCtx *ctx, int argc, char **argv);
8a959859 140struct searchNode *cidr_parse(searchCtx *ctx, int argc, char **argv);
b53e3dd8 141struct searchNode *ipv6_parse(searchCtx *ctx, int argc, char **argv);
c7f7a584 142
143/* Channel functions (various types) */
f33f3f52
P
144struct searchNode *exists_parse(searchCtx *ctx, int argc, char **argv);
145struct searchNode *services_parse(searchCtx *ctx, int argc, char **argv);
146struct searchNode *size_parse(searchCtx *ctx, int argc, char **argv);
147struct searchNode *name_parse(searchCtx *ctx, int argc, char **argv);
148struct searchNode *topic_parse(searchCtx *ctx, int argc, char **argv);
149struct searchNode *oppct_parse(searchCtx *ctx, int argc, char **argv);
de8c57b4
C
150struct searchNode *cumodecount_parse(searchCtx *ctx, int argc, char **argv);
151struct searchNode *cumodepct_parse(searchCtx *ctx, int argc, char **argv);
f33f3f52
P
152struct searchNode *hostpct_parse(searchCtx *ctx, int argc, char **argv);
153struct searchNode *authedpct_parse(searchCtx *ctx, int argc, char **argv);
154struct searchNode *kick_parse(searchCtx *ctx, int argc, char **argv);
c86edd1d 155
c7f7a584 156/* Interpret a string to give a node */
f33f3f52 157struct searchNode *search_parse(searchCtx *ctx, char *input);
c86edd1d 158
b697c21c 159/* Iteration functions */
f33f3f52
P
160struct searchNode *any_parse(searchCtx *ctx, int argc, char **argv);
161struct searchNode *all_parse(searchCtx *ctx, int argc, char **argv);
162struct searchNode *var_parse(searchCtx *ctx, int argc, char **argv);
b697c21c
CP
163
164/* Iteraterable functions */
f33f3f52 165struct searchNode *channeliter_parse(searchCtx *ctx, int argc, char **argv);
cb857218 166struct searchNode *nickiter_parse(searchCtx *ctx, int argc, char **argv);
b697c21c 167
90334b4c
CP
168/* Functions that operate on strings */
169struct searchNode *cumodes_parse(searchCtx *ctx, int argc, char **argv);
170
c7f7a584 171/* Force a node to return the thing you want */
c8be5183 172struct searchNode *coerceNode(searchCtx *ctx, struct searchNode *thenode, int type);
c7f7a584 173
c86edd1d 174/* Registration functions */
a92bb8e1
P
175searchCmd *registersearchcommand(char *name, int level, CommandHandler cmd, void *defaultdisplayfunc);
176void deregistersearchcommand(searchCmd *scmd);
e8ff7b61 177void registersearchterm(searchCmd *cmd, char *term, parseFunc parsefunc, int level, char *help);
a92bb8e1 178void deregistersearchterm(searchCmd *cmd, char *term, parseFunc parsefunc);
e8ff7b61
P
179
180void registerglobalsearchterm(char *term, parseFunc parsefunc, char *help);
a92bb8e1 181void deregisterglobalsearchterm(char *term, parseFunc parsefunc);
e8ff7b61
P
182
183void regdisp( searchCmd *cmd, const char *name, void *handler, int level, char *help);
a92bb8e1 184void unregdisp( searchCmd *cmd, const char *name, void *handler);
c86edd1d 185
2ba836f2
CP
186/* Special nick* printf */
187void nssnprintf(char *, size_t, const char *, nick *);
188
b53e3dd8
P
189#ifdef NEWSEARCH_NEWPARSER
190void displaystrerror(replyFunc reply, nick *np, const char *input);
191#endif
192
c86edd1d 193extern const char *parseError;
b697c21c 194extern nick *senderNSExtern;
6f8dfee9 195
3d2bf13b 196void printnick(searchCtx *, nick *, nick *);
e8ad630b 197void printuser(searchCtx *, nick *, authname *);
e1b0e499 198void printchannel(searchCtx *, nick *, chanindex *);
6f8dfee9 199
6b2202d0
CP
200void nicksearch_exe(struct searchNode *search, searchCtx *sctx);
201void chansearch_exe(struct searchNode *search, searchCtx *sctx);
202void usersearch_exe(struct searchNode *search, searchCtx *ctx);
20b9980a
CP
203
204int do_nicksearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, char **cargv);
205int do_chansearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, char **cargv);
e8ad630b 206int do_usersearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, char **cargv);
0a659cde
CP
207
208void *literal_exe(searchCtx *ctx, struct searchNode *thenode, void *theinput);
209void literal_free(searchCtx *ctx, struct searchNode *thenode);
210
f33f3f52
P
211struct searchVariable *var_register(searchCtx *ctx, char *arg, int type);
212searchNode *var_get(searchCtx *ctx, char *arg);
b697c21c
CP
213void var_setstr(struct searchVariable *v, char *data);
214
6b2202d0 215void newsearch_ctxinit(searchCtx *ctx, searchParseFunc searchfn, replyFunc replyfn, wallFunc wallfn, void *arg, searchCmd *cmd, nick *sender, void *displayfn, int limit);
b697c21c 216
0a659cde
CP
217/* AST functions */
218
219struct searchASTNode;
220
221#define AST_NODE_CHILD 1
222#define AST_NODE_LITERAL 2
223
224/* items to store in the ast lookup cache */
225#define AST_RECENT 10
226
68d05acb
CP
227struct searchASTExpr;
228
229typedef struct searchASTNode {
230 parseFunc fn;
231 int argc;
232 struct searchASTExpr *argv;
233} searchASTNode;
234
0a659cde
CP
235typedef struct searchASTExpr {
236 int type;
237 union {
238 char *literal;
68d05acb
CP
239 /* this was originally a pointer, but having it as one means you can't memcpy directly */
240 struct searchASTNode child;
0a659cde 241 } u;
0a659cde
CP
242} searchASTExpr;
243
b53e3dd8
P
244typedef struct searchASTCache {
245 searchASTExpr *tree;
246 searchASTExpr *cache[AST_RECENT];
247 int nextpos;
248} searchASTCache;
249
68d05acb
CP
250#define __NSASTSizeOfArray(x) (sizeof(x) / sizeof(x[0]))
251
252#define __NSASTExpr(x, y, ...) (searchASTExpr){.type = x, .u.y = __VA_ARGS__}
253#define __NSASTRawNode(x, c, y) (searchASTNode){.fn = x, .argc = c, .argv = y}
254
255#define __NSASTSizeNode(x, y) __NSASTRawNode(x, __NSASTSizeOfArray(y), y)
256#define __NSASTNode(x, ...) __NSASTSizeNode(x, ((searchASTExpr []){__VA_ARGS__}))
0a659cde 257
0a659cde
CP
258#define __NSASTChild(...) __NSASTExpr(AST_NODE_CHILD, child, __VA_ARGS__)
259
68d05acb 260#define NSASTLiteral(value) __NSASTExpr(AST_NODE_LITERAL, literal, value)
0a659cde 261#define NSASTNode(fn, ...) __NSASTChild(__NSASTNode(fn, __VA_ARGS__))
68d05acb 262#define NSASTManualNode(fn, count, children) __NSASTChild(__NSASTRawNode(fn, count, children))
0a659cde 263
f33f3f52 264searchNode *search_astparse(searchCtx *, char *);
0a659cde 265
2181a10f
CP
266int ast_nicksearch(searchASTExpr *tree, replyFunc reply, void *sender, wallFunc wall, NickDisplayFunc display, HeaderFunc header, void *headerarg, int limit);
267int ast_chansearch(searchASTExpr *tree, replyFunc reply, void *sender, wallFunc wall, ChanDisplayFunc display, HeaderFunc header, void *headerarg, int limit);
268int ast_usersearch(searchASTExpr *tree, replyFunc reply, void *sender, wallFunc wall, UserDisplayFunc display, HeaderFunc header, void *headerarg, int limit);
0a659cde 269
a92bb8e1
P
270char *ast_printtree(char *buf, size_t bufsize, searchASTExpr *expr, searchCmd *cmd);
271
6cb1b086 272int parseopts(int cargc, char **cargv, int *arg, int *limit, void **subset, void *display, CommandTree *sl, replyFunc reply, void *sender);
0a659cde 273
68d05acb
CP
274typedef int (*ASTFunc)(searchASTExpr *, replyFunc, void *, wallFunc, void *, HeaderFunc, void *, int limit);
275
0a659cde 276/* erk */
a92bb8e1
P
277extern searchList *globalterms;
278
279extern searchCmd *reg_nicksearch;
280extern searchCmd *reg_chansearch;
281extern searchCmd *reg_usersearch;
0a659cde 282
55dd7529
CP
283extern UserDisplayFunc defaultuserfn;
284extern NickDisplayFunc defaultnickfn;
285extern ChanDisplayFunc defaultchanfn;
286
31686847
CP
287struct searchNode *argtoconststr(char *command, searchCtx *ctx, char *arg, char **p);
288
1188d606 289#endif