136 lines
3.3 KiB
C
136 lines
3.3 KiB
C
#include <stdio.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <stdlib.h>
|
|
#include <lua.h>
|
|
#include <lualib.h>
|
|
#include <lauxlib.h>
|
|
|
|
#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';
|
|
trim(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]);
|
|
return 1;
|
|
}
|
|
|
|
int status;
|
|
char *logfile = argv[2];
|
|
char *luafile = argv[1];
|
|
char *line = NULL;
|
|
size_t len = 0;
|
|
ssize_t bytesRead;
|
|
FILE *fp;
|
|
lua_State *L;
|
|
L = luaL_newstate();
|
|
luaL_openlibs(L);
|
|
|
|
fp = fopen(logfile, "r");
|
|
if (errno) {
|
|
fprintf(stderr, "Error opening file %s: %s\n", logfile, strerror(errno));
|
|
return EXIT_LOGFILE_HANDLING;
|
|
}
|
|
|
|
status = luaL_dofile(L, luafile);
|
|
if (status) {
|
|
fprintf(stderr, "Error reading LUA file %s: %s\n", luafile, lua_tostring(L, -1));
|
|
return EXIT_LUAFILE_HANDLING;
|
|
}
|
|
|
|
while ((bytesRead = getline(&line, &len, fp)) != -1) {
|
|
size_t formatLen;
|
|
const char *formatLine;
|
|
|
|
trim(line);
|
|
|
|
if (luaFilterCall(L, line, bytesRead)) {
|
|
continue;
|
|
}
|
|
|
|
formatLine = luaFormatCall(L, line, bytesRead, &formatLen);
|
|
printf("%s\n", formatLine);
|
|
}
|
|
if (errno) {
|
|
fprintf(stderr, "Error reading file %s: %s\n", logfile, strerror(errno));
|
|
return EXIT_LOGFILE_HANDLING;
|
|
}
|
|
|
|
free(line);
|
|
lua_close(L);
|
|
|
|
fclose(fp);
|
|
} |