aboutsummaryrefslogtreecommitdiff
path: root/Lua
diff options
context:
space:
mode:
Diffstat (limited to 'Lua')
-rw-r--r--Lua/lbp.c157
1 files changed, 56 insertions, 101 deletions
diff --git a/Lua/lbp.c b/Lua/lbp.c
index 0f59653..3ea18e8 100644
--- a/Lua/lbp.c
+++ b/Lua/lbp.c
@@ -1,14 +1,14 @@
/*
-* lbp.c - bp library for lua
-* API:
-* bp.match(pat, str, [start_index]) -> nil or match_table
-* bp.replace(pat, replacement, str, [start_index]) -> str with replacements, num_replacements
-* for match_table in bp.matches(pat, str, [start_index]) do ... end
-* bp.compile(pat) -> pattern object
-* pat:match(str, [start_index])
-* pat:replace(replacement, str, [start_index])
-* for match in pat:matches(str, [start_index]) do ... end
-*/
+ * lbp.c - bp library for lua
+ * API:
+ * bp.match(pat, str, [start_index]) -> nil or match_table
+ * bp.replace(pat, replacement, str, [start_index]) -> str with replacements, num_replacements
+ * for match_table in bp.matches(pat, str, [start_index]) do ... end
+ * bp.compile(pat) -> pattern object
+ * pat:match(str, [start_index])
+ * pat:replace(replacement, str, [start_index])
+ * for match in pat:matches(str, [start_index]) do ... end
+ */
#include <fcntl.h>
#include <stdlib.h>
@@ -16,11 +16,11 @@
#include <sys/mman.h>
#include <sys/stat.h>
-#include "lua.h"
#include "lauxlib.h"
+#include "lua.h"
-#include "../pattern.h"
#include "../match.h"
+#include "../pattern.h"
#include "../printmatch.h"
#include "../utils.h"
@@ -34,25 +34,22 @@ static void push_match(lua_State *L, bp_match_t *m, const char *start);
lua_State *cur_state = NULL;
-static void match_error(char **msg)
-{
+static void match_error(char **msg) {
lua_pushstring(cur_state, *msg);
free(*msg);
*msg = NULL;
lua_error(cur_state);
}
-static inline void raise_parse_error(lua_State *L, maybe_pat_t m)
-{
+static inline void raise_parse_error(lua_State *L, maybe_pat_t m) {
size_t err_len = (size_t)(m.value.error.end - m.value.error.start);
- char *buf = calloc(err_len+1, sizeof(char));
+ char *buf = calloc(err_len + 1, sizeof(char));
memcpy(buf, m.value.error.start, err_len);
luaL_error(L, "%s: \"%s\"", m.value.error.msg, buf);
free(buf);
}
-static int Lcompile(lua_State *L)
-{
+static int Lcompile(lua_State *L) {
size_t patlen;
const char *pat_text = luaL_checklstring(L, 1, &patlen);
maybe_pat_t maybe_pat = bp_pattern(pat_text, pat_text + patlen);
@@ -60,18 +57,17 @@ static int Lcompile(lua_State *L)
raise_parse_error(L, maybe_pat);
return 0;
}
- bp_pat_t **pat_storage = (bp_pat_t**)lua_newuserdatauv(L, sizeof(bp_pat_t*), 1);
+ bp_pat_t **pat_storage = (bp_pat_t **)lua_newuserdatauv(L, sizeof(bp_pat_t *), 1);
*pat_storage = maybe_pat.value.pat;
lua_pushvalue(L, 1);
lua_setiuservalue(L, -2, 1);
- lua_pushlightuserdata(L, (void*)&PAT_METATABLE);
+ lua_pushlightuserdata(L, (void *)&PAT_METATABLE);
lua_gettable(L, LUA_REGISTRYINDEX);
lua_setmetatable(L, -2);
return 1;
}
-static void push_matchstring(lua_State *L, bp_match_t *m)
-{
+static void push_matchstring(lua_State *L, bp_match_t *m) {
char *buf = NULL;
size_t size = 0;
FILE *out = open_memstream(&buf, &size);
@@ -81,8 +77,7 @@ static void push_matchstring(lua_State *L, bp_match_t *m)
fclose(out);
}
-static bp_match_t *get_first_capture(bp_match_t *m)
-{
+static bp_match_t *get_first_capture(bp_match_t *m) {
if (m->pat->type == BP_TAGGED) {
return m;
} else if (m->pat->type == BP_CAPTURE && !When(m->pat, BP_CAPTURE)->name) {
@@ -96,8 +91,7 @@ static bp_match_t *get_first_capture(bp_match_t *m)
return NULL;
}
-static void set_capture_fields(lua_State *L, bp_match_t *m, int *n, const char *start)
-{
+static void set_capture_fields(lua_State *L, bp_match_t *m, int *n, const char *start) {
if (m->pat->type == BP_CAPTURE) {
bp_match_t *cap = get_first_capture(m->children[0]);
if (!cap) cap = m->children[0];
@@ -119,10 +113,9 @@ static void set_capture_fields(lua_State *L, bp_match_t *m, int *n, const char *
}
}
-static void push_match(lua_State *L, bp_match_t *m, const char *start)
-{
+static void push_match(lua_State *L, bp_match_t *m, const char *start) {
lua_createtable(L, 1, 2);
- lua_pushlightuserdata(L, (void*)&MATCH_METATABLE);
+ lua_pushlightuserdata(L, (void *)&MATCH_METATABLE);
lua_gettable(L, LUA_REGISTRYINDEX);
lua_setmetatable(L, -2);
push_matchstring(L, m);
@@ -144,11 +137,9 @@ static void push_match(lua_State *L, bp_match_t *m, const char *start)
lua_setfield(L, -2, "after");
}
-static int Lmatch(lua_State *L)
-{
+static int Lmatch(lua_State *L) {
if (lua_isstring(L, 1)) {
- if (Lcompile(L) != 1)
- return 0;
+ if (Lcompile(L) != 1) return 0;
lua_replace(L, 1);
}
bp_pat_t **at_pat = lua_touserdata(L, 1);
@@ -162,19 +153,17 @@ static int Lmatch(lua_State *L)
lua_getfield(L, 3, "start");
lua_getfield(L, 3, "after");
index = luaL_optinteger(L, -1, 1);
- if (lua_rawequal(L, -1, -2))
- ++index;
+ if (lua_rawequal(L, -1, -2)) ++index;
} else {
index = luaL_optinteger(L, 3, 1);
}
- if (index > (lua_Integer)strlen(text)+1)
- return 0;
+ if (index > (lua_Integer)strlen(text) + 1) return 0;
bp_match_t *m = NULL;
int ret = 0;
cur_state = L;
bp_errhand_t old = bp_set_error_handler(match_error);
- if (next_match(&m, text+index-1, &text[textlen], pat, builtins, NULL, false)) {
+ if (next_match(&m, text + index - 1, &text[textlen], pat, builtins, NULL, false)) {
push_match(L, m, text);
stop_matching(&m);
ret = 1;
@@ -183,11 +172,9 @@ static int Lmatch(lua_State *L)
return ret;
}
-static int Lreplace(lua_State *L)
-{
+static int Lreplace(lua_State *L) {
if (lua_isstring(L, 1)) {
- if (Lcompile(L) != 1)
- return 0;
+ if (Lcompile(L) != 1) return 0;
lua_replace(L, 1);
}
bp_pat_t **at_pat = lua_touserdata(L, 1);
@@ -198,8 +185,7 @@ static int Lreplace(lua_State *L)
const char *rep_text = luaL_checklstring(L, 2, &replen);
const char *text = luaL_checklstring(L, 3, &textlen);
lua_Integer index = luaL_optinteger(L, 4, 1);
- if (index > (lua_Integer)strlen(text)+1)
- index = (lua_Integer)strlen(text)+1;
+ if (index > (lua_Integer)strlen(text) + 1) index = (lua_Integer)strlen(text) + 1;
maybe_pat_t maybe_replacement = bp_replacement(pat, rep_text, rep_text + replen);
if (!maybe_replacement.success) {
@@ -215,7 +201,7 @@ static int Lreplace(lua_State *L)
bp_pat_t *rep_pat = maybe_replacement.value.pat;
cur_state = L;
bp_errhand_t old = bp_set_error_handler(match_error);
- for (bp_match_t *m = NULL; next_match(&m, text, &text[textlen], rep_pat, builtins, NULL, false); ) {
+ for (bp_match_t *m = NULL; next_match(&m, text, &text[textlen], rep_pat, builtins, NULL, false);) {
fwrite(prev, sizeof(char), (size_t)(m->start - prev), out);
fprint_match(out, text, m, NULL);
prev = m->end;
@@ -233,8 +219,7 @@ static int Lreplace(lua_State *L)
return 2;
}
-static int iter(lua_State *L)
-{
+static int iter(lua_State *L) {
lua_geti(L, 1, 1);
lua_geti(L, 1, 2);
lua_replace(L, 1);
@@ -242,14 +227,12 @@ static int iter(lua_State *L)
return Lmatch(L);
}
-static int Lmatches(lua_State *L)
-{
+static int Lmatches(lua_State *L) {
int nargs = lua_gettop(L);
lua_pushcfunction(L, iter); // iter
lua_createtable(L, 2, 0); // state: {pat, str}
if (lua_isstring(L, 1)) {
- if (Lcompile(L) != 1)
- return 0;
+ if (Lcompile(L) != 1) return 0;
} else {
lua_pushvalue(L, 1);
}
@@ -262,20 +245,17 @@ static int Lmatches(lua_State *L)
return 3;
}
-static int Lmatch_tostring(lua_State *L)
-{
+static int Lmatch_tostring(lua_State *L) {
lua_geti(L, 1, 0);
return 1;
}
-static int Lpat_source(lua_State *L)
-{
+static int Lpat_source(lua_State *L) {
lua_getiuservalue(L, 1, 1);
return 1;
}
-static int Lpat_tostring(lua_State *L)
-{
+static int Lpat_tostring(lua_State *L) {
luaL_Buffer b;
luaL_buffinit(L, &b);
luaL_addstring(&b, "Pattern [[");
@@ -286,8 +266,7 @@ static int Lpat_tostring(lua_State *L)
return 1;
}
-static int Lpat_gc(lua_State *L)
-{
+static int Lpat_gc(lua_State *L) {
(void)L;
bp_pat_t **at_pat = lua_touserdata(L, 1);
bp_pat_t *pat = *at_pat;
@@ -296,8 +275,7 @@ static int Lpat_gc(lua_State *L)
return 0;
}
-static int Lpat_join(lua_State *L, const char *joiner)
-{
+static int Lpat_join(lua_State *L, const char *joiner) {
if (!lua_isstring(L, 1)) {
lua_pushcfunction(L, Lpat_source);
lua_pushvalue(L, 1);
@@ -330,62 +308,39 @@ static int Lpat_join(lua_State *L, const char *joiner)
return 1;
}
-static int Lpat_concat(lua_State *L)
-{
- return Lpat_join(L, " ");
-}
+static int Lpat_concat(lua_State *L) { return Lpat_join(L, " "); }
-static int Lpat_div(lua_State *L)
-{
- return Lpat_join(L, " / ");
-}
+static int Lpat_div(lua_State *L) { return Lpat_join(L, " / "); }
-static const luaL_Reg match_metamethods[] = {
- {"__tostring", Lmatch_tostring},
- {NULL, NULL}
-};
+static const luaL_Reg match_metamethods[] = {{"__tostring", Lmatch_tostring}, {NULL, NULL}};
static const luaL_Reg pat_methods[] = {
- {"match", Lmatch},
- {"replace", Lreplace},
- {"matches", Lmatches},
- {"getsource", Lpat_source},
- {NULL, NULL}
-};
-
-static const luaL_Reg pat_metamethods[] = {
- {"__gc", Lpat_gc},
- {"__concat", Lpat_concat},
- {"__div", Lpat_div},
- {"__tostring", Lpat_tostring},
- {"__index", NULL}, // placeholder for pat_methods
- {NULL, NULL}
-};
+ {"match", Lmatch}, {"replace", Lreplace}, {"matches", Lmatches}, {"getsource", Lpat_source}, {NULL, NULL}};
+
+static const luaL_Reg pat_metamethods[] = {{"__gc", Lpat_gc}, {"__concat", Lpat_concat},
+ {"__div", Lpat_div}, {"__tostring", Lpat_tostring},
+ {"__index", NULL}, // placeholder for pat_methods
+ {NULL, NULL}};
static const luaL_Reg bp_methods[] = {
- {"match", Lmatch},
- {"replace", Lreplace},
- {"compile", Lcompile},
- {"matches", Lmatches},
- {NULL, NULL}
-};
-
-public LUALIB_API int luaopen_bp(lua_State *L)
-{
- maybe_pat_t maybe_pat = bp_pattern(builtins_source, builtins_source+strlen(builtins_source));
+ {"match", Lmatch}, {"replace", Lreplace}, {"compile", Lcompile}, {"matches", Lmatches}, {NULL, NULL}};
+
+public
+LUALIB_API int luaopen_bp(lua_State *L) {
+ maybe_pat_t maybe_pat = bp_pattern(builtins_source, builtins_source + strlen(builtins_source));
if (!maybe_pat.success) {
raise_parse_error(L, maybe_pat);
return 0;
}
builtins = maybe_pat.value.pat;
- lua_pushlightuserdata(L, (void*)&PAT_METATABLE);
+ lua_pushlightuserdata(L, (void *)&PAT_METATABLE);
luaL_newlib(L, pat_metamethods);
luaL_newlib(L, pat_methods);
lua_setfield(L, -2, "__index");
lua_settable(L, LUA_REGISTRYINDEX);
- lua_pushlightuserdata(L, (void*)&MATCH_METATABLE);
+ lua_pushlightuserdata(L, (void *)&MATCH_METATABLE);
luaL_newlib(L, match_metamethods);
lua_settable(L, LUA_REGISTRYINDEX);
generated by cgit v1.2.3 (git 2.25.1) at 2026-03-01 04:22:25 +0000