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
|
/*
* lbp.c - bp library for lua
* API:
* bp.match(str, pat[, start_index]) -> nil or (match_text, start_index, match_len_in_source)
* bp.replace(str, pat, replacement, start_index) -> str with replacements
*/
#include <string.h>
#include "lua.h"
#include "lauxlib.h"
#include "../print.h"
#include "../files.h"
#include "../pattern.h"
#include "../match.h"
// The C API changed from 5.1 to 5.2, so these shims help the code compile on >=5.2
#if LUA_VERSION_NUM >= 502
#define lua_objlen(L, i) lua_rawlen(L, i)
#define luaL_register(L, _, R) luaL_setfuncs(L, R, 0)
#endif
static int Lmatch(lua_State *L)
{
size_t textlen, patlen;
const char *text = luaL_checklstring(L, 1, &textlen);
const char *pat_text = luaL_checklstring(L, 2, &patlen);
lua_Integer index = luaL_optinteger(L, 3, 1);
if (index > (lua_Integer)strlen(text)+1)
return 0;
file_t *pat_file = spoof_file(NULL, "<pattern argument>", pat_text, patlen);
pat_t *pat = bp_pattern(pat_file, pat_file->start);
if (!pat) {
destroy_file(&pat_file);
luaL_error(L, "Pattern failed to compile: %s", pat_text);
return 0;
}
file_t *text_file = spoof_file(NULL, "<text argument>", text+(index-1), textlen);
match_t *m = NULL;
int ret = 0;
if (next_match(&m, NULL, text_file, pat, NULL, false)) {
char *buf = NULL;
size_t size = 0;
FILE *out = open_memstream(&buf, &size);
printer_t pr = {
.file = text_file,
.context_before = NO_CONTEXT,
.context_after = NO_CONTEXT,
.use_color = 0,
.lineformat = "",
};
print_match(out, &pr, m);
fflush(out);
lua_pushlstring(L, buf, size);
lua_pushinteger(L, (int)(m->start - text_file->start) + index);
lua_pushinteger(L, (int)(m->end - m->start));
fclose(out);
stop_matching(&m);
ret = 3;
}
destroy_file(&pat_file);
destroy_file(&text_file);
return ret;
}
static int Lreplace(lua_State *L)
{
size_t textlen, patlen, replen;
const char *text = luaL_checklstring(L, 1, &textlen);
const char *pat_text = luaL_checklstring(L, 2, &patlen);
const char *rep_text = luaL_checklstring(L, 3, &replen);
lua_Integer index = luaL_optinteger(L, 4, 1);
if (index > (lua_Integer)strlen(text)+1)
index = (lua_Integer)strlen(text)+1;
file_t *pat_file = spoof_file(NULL, "<pattern argument>", pat_text, patlen);
pat_t *pat = bp_pattern(pat_file, pat_file->start);
if (!pat) {
destroy_file(&pat_file);
luaL_error(L, "Pattern failed to compile: %s", pat_text);
return 0;
}
file_t *rep_file = spoof_file(NULL, "<replacement argument>", rep_text, replen);
pat = bp_replacement(rep_file, pat, rep_file->start);
if (!pat) {
destroy_file(&pat_file);
destroy_file(&rep_file);
luaL_error(L, "Replacement failed to compile: %s", rep_text);
return 0;
}
file_t *text_file = spoof_file(NULL, "<text argument>", text+(index-1), textlen);
char *buf = NULL;
size_t size = 0;
FILE *out = open_memstream(&buf, &size);
printer_t pr = {
.file = text_file,
.context_before = ALL_CONTEXT,
.context_after = ALL_CONTEXT,
.use_color = 0,
.lineformat = "",
};
int replacements = 0;
for (match_t *m = NULL; next_match(&m, NULL, text_file, pat, NULL, false); ) {
print_match(out, &pr, m);
++replacements;
}
print_match(out, &pr, NULL);
fflush(out);
lua_pushlstring(L, buf, size);
lua_pushinteger(L, replacements);
fclose(out);
destroy_file(&pat_file);
destroy_file(&rep_file);
destroy_file(&text_file);
return 2;
}
LUALIB_API int luaopen_bp(lua_State *L)
{
lua_createtable(L, 0, 2);
lua_pushcfunction(L, Lmatch);
lua_setfield(L, -2, "match");
lua_pushcfunction(L, Lreplace);
lua_setfield(L, -2, "replace");
// lua_pushcfunction(L, Leach);
// lua_setfield(L, -1, "each");
return 1;
}
|
