so far, in lua red

lua.rule

;orginal
;function: [ function  ws* funcbody  ws* ]
;funcbody: [ #"(" opt parlist #")"  ws*  block  ws* "end" ]


function: [ "function"  ws* name #"(" opt parlist #")"  ws
      varlist #"="  explist ws*  
      ws* functioncall ws* 
      ws* "do" ws* functioncall ws* "end" ws* 
      ws* "while" ws* exp ws* "do" ws* functioncall ws* "end" ws* 
      ws* "repeat" ws* functioncall ws* "until" ws* exp ws* 
      ws* "if" ws* exp ws* "then" ws* functioncall ws*  ; any ["elseif"  ws* exp ws* "then" ws* functioncall ws* ] ; opt ["else" ws* functioncall ws*] end ws*  
      ;ws* "for" ws* name ws* #"=" ws* exp ws* any [#"," ws* exp ws*] "do" ws* block ws* "end" ws* | 
      ;ws* "for" ws* namelist ws* "in" ws* explist ws* "do" ws* block ws* "end" ws* | 
      ;ws* "function" funcname funcbody ws* | 
      ;ws* "local" ws* "function" ws* name funcbody ws* | 
      ;ws* "local" namelist opt [#"=" explist]]
ws* "end" ws*]
;funcbody  ws* ]


With this input:

function f(n)
 x = 1
 g(1)
 do h(3) end
 while true do g(2) end
 repeat f(1) until true
 if true then h(4) 

end