54 lines
1.2 KiB
Lua
54 lines
1.2 KiB
Lua
-- Making Lua source code look pretty.
|
|
-- A simple scanner based prettifier, which scans comments for @{ref} and code
|
|
-- for known modules and functions.
|
|
-- A module reference to an example `test-fun.lua` would look like
|
|
-- `@{example:test-fun}`.
|
|
require 'pl'
|
|
local lexer = require 'ldoc.lexer'
|
|
local tnext = lexer.skipws
|
|
local prettify = {}
|
|
|
|
local escaped_chars = {
|
|
['&'] = '&',
|
|
['<'] = '<',
|
|
['>'] = '>',
|
|
}
|
|
local escape_pat = '[&<>]'
|
|
|
|
local function escape(str)
|
|
return (str:gsub(escape_pat,escaped_chars))
|
|
end
|
|
|
|
local function span(t,val)
|
|
return ('<span class="%s">%s</span>'):format(t,val)
|
|
end
|
|
|
|
local spans = {keyword=true,number=true,string=true,comment=true}
|
|
|
|
function prettify.lua (code)
|
|
local res = List()
|
|
res:append(header)
|
|
res:append '<pre>\n'
|
|
|
|
local tok = lexer.lua(code,{},{})
|
|
local t,val = tok()
|
|
if not t then return nil,"empty file" end
|
|
while t do
|
|
val = escape(val)
|
|
if spans[t] then
|
|
if t == 'comment' then -- may contain @{ref}
|
|
val = prettify.resolve_inline_references(val)
|
|
end
|
|
res:append(span(t,val))
|
|
else
|
|
res:append(val)
|
|
end
|
|
t,val = tok()
|
|
end
|
|
res:append(footer)
|
|
return res:join ()
|
|
end
|
|
|
|
return prettify
|
|
|