luahtml模板引擎,用C为LUA写一个超迷你的模板引擎.

中午在做HTTP服务器&＃xff0c;内嵌了LUA引擎作为业务逻辑部分. 但考虑到LUA输出HTML的性能不高&＃xff0c;况且 MVC 模式开发网页已经习惯了&＃xff0c;何不用C给lua写个最简单的模板引擎呢&＃xff1f;说做就做, 项目时间很紧张&＃xff0c;所以必须确定目标&＃xff0c;想了会儿。定了几个目标&＃xff1a;

我要的是编译型的模板,要支持缓存。

不需要强大的模板逻辑方面的指令。

将HTML模板文件编译成LUA源代码即可。

LUA 还是比较容易扩展的&＃xff0c;除了那栈用起来比较变态之外&＃xff0c;经过了几个小时的奋斗&＃xff0c;C&＃43;LUA代码。算是大功告成了。

#include

#include

#include

#include

#include

#include

#include

#include "lib/automem.h"

#include "lua.h"

#include "lauxlib.h"

#include "lua-tpl.h"

/*

LUA 最简单的模板引擎.

*/

const char LUAFMT_TPL_META[]&＃61;"LUAFMT_TPL";

void lua_cfmt_createmetatable (lua_State *L);

void lua_register_class(lua_State * L,const luaL_Reg * methods,const char * name,lua_CFunction has_index);

#if defined(_WIN32) || defined(_WIN64)

#define open_open

#define close_close

#define read_read

#define stat( x , y )_stat( x , y )

#endif

static char * file_get_contents(const char * file,int * sz){

char * ret &＃61; NULL;

struct stat st;

int fd;

*sz &＃61; 0;

if ((fd &＃61; open(file, O_RDONLY|O_BINARY)) !&＃61; -1)

{

if (fstat(fd, &st) !&＃61; -1)

{

ret &＃61; (char *)malloc(st.st_size);

if(NULL !&＃61; ret){

while( *sz

*sz &＃43;&＃61; read(fd, ret &＃43; *sz, st.st_size - (*sz));

_lseek(fd, *sz, SEEK_SET);

}

}

close(fd);

}

}

return ret;

}

int file_put_contents(const char* file_name, automem_t* mem)

{

FILE * fp &＃61; fopen(file_name,"wb&＃43;");

if(NULL !&＃61; fp)

{

fwrite(mem->pdata, mem->size, 1, fp);

fclose(fp);

return 1;

}

return 0;

}

enum{

tpl_state_normal,

tpl_state_scode_1,

tpl_state_code,

tpl_state_ecode_1,

tpl_state_escape,

};

#define append_end_stringfield(a,b,c) \

automem_append_voidp( (a) , cmd, lcmd); \

automem_append_voidp( (a), (b) , (c)); \

automem_append_voidp( (a) , ecmd, lecmd); \

automem_append_voidp( (a), ")\n", 2);

static void lua_tpl_compile_local(lua_State * L, automem_t * mem, const char * buf,int lbuf){

int state &＃61; tpl_state_normal, i &＃61; 0;

const char * sbuf; char c;

size_t lcmd &＃61; sizeof("request.print([[") -1,

lecmd &＃61; sizeof("]])") -1,

lpre &＃61; 0;

const char * cmd &＃61; "request.print([[",

* ecmd &＃61; "]])",

*pre &＃61; NULL;

if(lua_isstring(L, 3))

cmd &＃61;luaL_checklstring(L, 3, &lcmd);

if(lua_isstring(L, 4))

ecmd &＃61;luaL_checklstring(L, 4, &lecmd);

if(lua_isstring(L, 5))

pre &＃61;luaL_checklstring(L, 5, &lpre);

if(NULL !&＃61; pre){

automem_append_voidp(mem, pre, lpre);

automem_append_byte(mem,&＃39;\n&＃39;);

}

sbuf &＃61; buf;

while(i

c &＃61; buf[i];

switch (state)

{

case tpl_state_normal:

switch(c){

case &＃39;{&＃39;:

state &＃61; tpl_state_scode_1;

break;

}

break;

case tpl_state_scode_1:

switch(c){

case &＃39;#&＃39;:

state &＃61;tpl_state_code;

append_end_stringfield(mem,sbuf, &buf[i] - sbuf-1);

sbuf &＃61; &buf[i&＃43;1];

break;

default:

state &＃61; tpl_state_normal;

break;

}

break;

case tpl_state_code:

switch(c){

case &＃39;#&＃39;:

state &＃61; tpl_state_ecode_1;

break;

}

case tpl_state_ecode_1:

switch(c){

case &＃39;}&＃39;:

automem_append_voidp(mem,sbuf, &buf[i] - sbuf-1);

automem_append_byte(mem,&＃39;\n&＃39;);

sbuf &＃61; &buf[i&＃43;1];

state &＃61; tpl_state_normal;

break;

default:

state&＃61;tpl_state_code;

}

default:

break;

}

i&＃43;&＃43;;

}

if(tpl_state_normal &＃61;&＃61; state){

append_end_stringfield(mem,sbuf, &buf[i] - sbuf);

}

}

/* 对模板文件进行编译.*/

static int lua_tpl_compile(lua_State * L)

{

int cache &＃61; 0,i &＃61; 0, lbuf &＃61; 0;

size_t lfile;

const char *cfile &＃61; NULL, * file&＃61; luaL_checklstring(L, 1,&lfile),* buf;

automem_t mem;

if(lua_isboolean(L, 2))

cache &＃61;lua_toboolean(L, 2);

if(0 !&＃61; cache){

struct stat st1,st2;

cfile&＃61;(char *)malloc(lfile&＃43;5);

memcpy((char *)cfile,file,lfile);

strcpy((char *)cfile&＃43;lfile,".tpl");

if((0 &＃61;&＃61; stat(file,&st1)) && (0 &＃61;&＃61; stat(cfile, &st2)))

{

if(st1.st_mtime <&＃61; st2.st_mtime)

{

if(NULL !&＃61; (buf &＃61; file_get_contents(cfile, &lbuf)))

{

free((void *)cfile);

lua_pushlstring(L,buf, lbuf);

goto lua_tpl_compile_final;

}

}

}

}

if(NULL !&＃61; (buf &＃61; file_get_contents(file, &lbuf)))

{

automem_init(&mem,lbuf &＃43; 1024);

lua_tpl_compile_local(L, &mem, buf, lbuf);

free((void*)buf);

lua_pushlstring(L,(char *)mem.pdata,mem.size);

if(0 !&＃61; cache && NULL !&＃61;cfile)

file_put_contents(cfile,&mem);

automem_uninit(&mem);

}

if(NULL !&＃61; cfile)

free((void *)cfile);

lua_tpl_compile_final:

return 1;

}

static luaL_Reg fmt_tpl_reg[] &＃61; {

{"compile",lua_tpl_compile},

{NULL,NULL}

};

int luaopen_cfmt_tpl(lua_State * L)

{

luaL_newlib(L, fmt_tpl_reg);

lua_cfmt_createmetatable(L);

return 1;

}

整个LUA模块扩展就1个函数 compile()。在LUA中的原型如下&＃xff1a;

string compile(filePath,cached, write, prefix,suffix,init)

功能: 将html模板编译为lua代码.

参数:

filePath: html源文件的路径.

cached: 是否需要缓存.

writer: 内容输出函数名.

prefix: 文件分界符前缀.

init:初始代码,用于做参数展开之类的工作.

当然,光有C的接口还不够&＃xff0c;为了使它变得简单易用&＃xff0c;还需要用LUA对其包装一下^_^, 包装代码如下:

function util.tpl(writer)

local t &＃61; require "cfmt.tpl"

local tpl &＃61; {}

local args &＃61; {}

local _prefix&＃61;&＃39;[&＃61;[&＃39;

local _suffix&＃61;&＃39;]&＃61;]&＃39;

-- 创建的时候指定 writer

if nil ~&＃61; writer then args[&＃39;_&＃39;]&＃61;writer end

function tpl:assign(name,value)

args[name]&＃61;value

end

function tpl:boundary(prefix, suffix) --修改字符串边界符

if nil ~&＃61; prefix then _prefix&＃61;prefix end

if nil ~&＃61; suffix then _suffix&＃61;suffix end

end

function tpl:display(tpl,cache,writer) -- tpl 模板文件位置, cache 是否需要缓存 writer 可选,如果创建对象的时候指定了的话.

local i&＃61;1

if nil~&＃61;writer then args[&＃39;_&＃39;]&＃61;writer end

local init &＃61; {&＃39;local args &＃61; ...&＃39;}

for key,val in pairs(args) do

init[#init&＃43;1]&＃61;&＃39;local &＃39;..key..&＃39;&＃61;args["&＃39;..key..&＃39;"]&＃39;

i&＃61;i&＃43;1

end

init &＃61; table.concat(init,&＃39;\n&＃39;)

local code &＃61;t.compile(tpl,cache,&＃39;_(&＃39;.._prefix, _suffix,init)

load(code)(args)

end

return tpl;

end

接下来用起来就简单多了,上测试代码.

---外部进来的数据在这里做检测

function login:request(r)

local tpl &＃61; (require "util").tpl(r.print)

local users &＃61;{

{[&＃39;ID&＃39;]&＃61;1,[&＃39;username&＃39;]&＃61;&＃39;bywayboy&＃39;,[&＃39;age&＃39;]&＃61;31},

{[&＃39;ID&＃39;]&＃61;1,[&＃39;username&＃39;]&＃61;&＃39;liigo&＃39;,[&＃39;age&＃39;]&＃61;31},

{[&＃39;ID&＃39;]&＃61;1,[&＃39;username&＃39;]&＃61;&＃39;sunwei&＃39;,[&＃39;age&＃39;]&＃61;8},

}

local b&＃61;{[&＃39;a&＃39;]&＃61;12}

tpl:assign(&＃39;users&＃39;,users);

tpl:assign(&＃39;title&＃39;,"测试模板变量.")

tpl:assign(&＃39;name&＃39;,"某某童鞋")

tpl:display("D:\\VC\\CmdChannel\\win32\\Debug\\test.html",true)

end

再来一个模板文件示例:

table{border:1px solid #CCC;}

{# for key,val in pairs(users) do#}

{#end#}

{#_(name)#}

该文件最终生成的缓存文件为:

local args &＃61; ...

local _&＃61;args["_"]

local name&＃61;args["name"]

local users&＃61;args["users"]

local title&＃61;args["title"]

_([&＃61;[

_(title)

_([&＃61;[--方121212法

table{border:1px solid #CCC;}

]&＃61;])

for key,val in pairs(users) do

_([&＃61;[

_(val[&＃39;ID&＃39;])

_([&＃61;[

_(val[&＃39;username&＃39;])

_([&＃61;[

_(val[&＃39;age&＃39;])

_([&＃61;[

]&＃61;])

end

_([&＃61;[

]&＃61;])

_(name)

_([&＃61;[

]&＃61;])

输出结果如图:

最终生成的HTML代码如下:

table{border:1px solid #CCC;}

某某童鞋

目前该项目已经开源:

https://github.com/bywayboy/lua-modules