filter and format call

This commit is contained in:
damage 2025-02-16 19:35:22 +01:00
parent 4c05853c0d
commit 6ac0ea9b35

102
logfoo.c
View File

@ -10,6 +10,11 @@
#define EXIT_LOGFILE_HANDLING 2
#define EXIT_LUAFILE_HANDLING 3
// // TODO: assign to lua
// #define COLOR_RED 1
// #define COLOR_BLUE 2
// #define COLOR_GREEN 3
void trim(char *string) {
if (strlen(string) > 0 && string[strlen(string) - 1] == '\n') {
string[strlen(string) - 1] = '\0';
@ -17,6 +22,66 @@ void trim(char *string) {
}
}
int luaFilterCall(lua_State *L, char *line, size_t len) {
int filter;
lua_getglobal(L, "filter");
lua_pushlstring(L, line, len);
if (lua_pcall(L, 1, 1, 0) != 0) {
fprintf(stderr, "Error running LUA function 'filter': %s\n", lua_tostring(L, -1));
exit(EXIT_LUAFILE_HANDLING);
}
if (!lua_isboolean(L, -1)) {
fprintf(stderr, "LUA function 'filter' did not return a boolean");
exit(EXIT_LUAFILE_HANDLING);
}
filter = lua_toboolean(L, -1);
lua_pop(L, 1);
return filter;
}
const char* luaFormatCall(lua_State *L, char *line, size_t len, size_t *formatLen) {
lua_getglobal(L, "format");
lua_pushlstring(L, line, len);
if (lua_pcall(L, 1, 1, 0) != 0) {
fprintf(stderr, "Error running LUA function 'format': %s\n", lua_tostring(L, -1));
exit(EXIT_LUAFILE_HANDLING);
}
if (!lua_isstring(L, -1)) {
fprintf(stderr, "LUA function 'format' did not return a string\n");
exit(EXIT_LUAFILE_HANDLING);
}
return lua_tolstring(L, -1 , formatLen);
}
// const char* luaColor(lua_State *L) {
// size_t textParaLen;
// const char *textParaStr;
// int colorPara;
// size_t textColorLen;
// char *textColorStr;
// textParaStr = luaL_checklstring(L, 1, &textParaLen);
// colorPara = luaL_checkinteger(L, 2);
// switch (colorPara) {
// case COLOR_RED:
// //lua_pushlstring();
// break;
// default:
// fprintf(stderr, "LUA function 'format' did not return a string\n");
// exit(EXIT_LUAFILE_HANDLING);
// }
// }
int main(int argc, char *argv[]) {
if (argc != 3) {
fprintf(stderr, "Usage: %s <lua file> <log file>\n", argv[0]);
@ -28,7 +93,7 @@ int main(int argc, char *argv[]) {
char *luafile = argv[1];
char *line = NULL;
size_t len = 0;
ssize_t read;
ssize_t bytesRead;
FILE *fp;
lua_State *L;
L = luaL_newstate();
@ -46,45 +111,18 @@ int main(int argc, char *argv[]) {
return EXIT_LUAFILE_HANDLING;
}
while ((read = getline(&line, &len, fp)) != -1) {
int filter;
while ((bytesRead = getline(&line, &len, fp)) != -1) {
size_t formatLen;
const char *formatLine;
trim(line);
lua_getglobal(L, "filter");
lua_pushlstring(L, line, read);
if (lua_pcall(L, 1, 1, 0) != 0) {
fprintf(stderr, "Error running LUA function 'filter': %s\n", lua_tostring(L, -1));
return EXIT_LUAFILE_HANDLING;
}
if (!lua_isboolean(L, -1)) {
fprintf(stderr, "LUA function 'filter' did not return a boolean");
return EXIT_LUAFILE_HANDLING;
}
filter = lua_toboolean(L, -1);
lua_pop(L, 1);
if (filter) {
if (luaFilterCall(L, line, bytesRead)) {
continue;
}
lua_getglobal(L, "format");
lua_pushlstring(L, line, read);
if (lua_pcall(L, 1, 1, 0) != 0) {
fprintf(stderr, "Error running LUA function 'format': %s\n", lua_tostring(L, -1));
return EXIT_LUAFILE_HANDLING;
}
if (!lua_isstring(L, -1)) {
fprintf(stderr, "LUA function 'format' did not return a string");
return EXIT_LUAFILE_HANDLING;
}
formatLine = lua_tolstring(L, -1 , &formatLen);
printf("(%zu) %s\n", formatLen, formatLine);
formatLine = luaFormatCall(L, line, bytesRead, &formatLen);
printf("%s\n", formatLine);
}
if (errno) {
fprintf(stderr, "Error reading file %s: %s\n", logfile, strerror(errno));