mirror of
https://github.com/msva/lua-htmlparser.git
synced 2024-11-27 12:44:22 +00:00
58 lines
1.6 KiB
Lua
58 lines
1.6 KiB
Lua
|
local ElementNode = require("ElementNode")
|
||
|
|
local voidelements = require("voidelements")
|
||
|
|
|
||
|
|
local HtmlParser = {}
|
||
|
|
|
||
|
|
local function parse(text)
|
||
|
|
local root = ElementNode:new(text)
|
||
|
|
|
||
|
|
local node, descend, tpos, opentags = root, true, 1, {}
|
||
|
|
while true do
|
||
|
|
local openstart, name
|
||
|
|
openstart, tpos, name = string.find(root._text, "<(%w+)[^>]*>", tpos)
|
||
|
|
if not name then break end
|
||
|
|
local tag = ElementNode:new(name, node, descend, openstart, tpos)
|
||
|
|
node = tag
|
||
|
|
|
||
|
|
local tagst, apos = tag:gettext(), 1
|
||
|
|
while true do
|
||
|
|
local start, k, quote, v
|
||
|
|
start, apos, k, quote = string.find(tagst, "%s+([^%s=]+)=(['\"]?)", apos)
|
||
|
|
if not k then break end
|
||
|
|
local pattern = "=([^%s'\">]*)"
|
||
|
|
if quote ~= '' then
|
||
|
|
pattern = quote .. "([^" .. quote .. "]*)" .. quote
|
||
|
|
end
|
||
|
|
start, apos, v = string.find(tagst, pattern, apos)
|
||
|
|
tag:addattribute(k, v)
|
||
|
|
end
|
||
|
|
|
||
|
|
if voidelements[string.lower(tag.name)] then
|
||
|
|
descend = false
|
||
|
|
tag:close()
|
||
|
|
else
|
||
|
|
opentags[tag.name] = tag
|
||
|
|
end
|
||
|
|
|
||
|
|
local closeend = tpos
|
||
|
|
while true do
|
||
|
|
local closestart, closing, closename
|
||
|
|
closestart, closeend, closing, closename = string.find(root._text, "[^<]*<(/?)(%w+)", closeend)
|
||
|
|
closing = closing and closing ~= ''
|
||
|
|
if not closing then break end
|
||
|
|
tag = opentags[closename]
|
||
|
|
opentags[closename] = nil
|
||
|
|
closestart = string.find(root._text, "<", closestart)
|
||
|
|
tag:close(closestart, closeend + 1)
|
||
|
|
node = tag.parent
|
||
|
|
descend = true
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
return root
|
||
|
|
end
|
||
|
|
HtmlParser.parse = parse
|
||
|
|
|
||
|
|
return HtmlParser
|
||
|
|
|