luajitos

dom.lua (6605B)

---

 -- dom.lua - Simple DOM Element structure
 
 local Element = {}
 Element.__index = Element
 
 function Element.new(tag, attrs)
     attrs = attrs or {}
     local self = setmetatable({}, Element)
 
     self.tag = tag or "div"
     self.id = attrs.id
     self.className = attrs.class or ""
     self.attributes = attrs
     self.children = {}
     self.parent = nil
     self.text = ""
 
     -- Layout (computed during render)
     self.x = 0
     self.y = 0
     self.width = 0
     self.height = 0
 
     return self
 end
 
 function Element:addChild(child)
     child.parent = self
     table.insert(self.children, child)
     return child
 end
 
 function Element:setText(text)
     self.text = text or ""
 end
 
 function Element:hasClass(name)
     return self.className:find(name, 1, true) ~= nil
 end
 
 function Element:getAttribute(name)
     return self.attributes[name]
 end
 
 function Element:findById(id)
     if self.id == id then
         return self
     end
     for _, child in ipairs(self.children) do
         local found = child:findById(id)
         if found then return found end
     end
     return nil
 end
 
 function Element:findByTag(tag)
     local results = {}
     if self.tag == tag then
         table.insert(results, self)
     end
     for _, child in ipairs(self.children) do
         local childResults = child:findByTag(tag)
         for _, elem in ipairs(childResults) do
             table.insert(results, elem)
         end
     end
     return results
 end
 
 function Element:findByClass(className)
     local results = {}
     if self:hasClass(className) then
         table.insert(results, self)
     end
     for _, child in ipairs(self.children) do
         local childResults = child:findByClass(className)
         for _, elem in ipairs(childResults) do
             table.insert(results, elem)
         end
     end
     return results
 end
 
 -- Find first element matching a CSS selector
 -- Supports: #id, .class, tagname, tag#id, tag.class, [attr], [attr=value]
 function Element:querySelector(selector)
     if not selector or selector == "" then
         return nil
     end
 
     -- Parse selector
     local tag, id, class, attrName, attrValue
 
     -- Check for #id
     local idMatch = selector:match("^#([%w%-_]+)$")
     if idMatch then
         return self:findById(idMatch)
     end
 
     -- Check for .class
     local classMatch = selector:match("^%.([%w%-_]+)$")
     if classMatch then
         local results = self:findByClass(classMatch)
         return results[1]
     end
 
     -- Check for [attr] or [attr=value]
     local attrOnly = selector:match("^%[([%w%-_]+)%]$")
     if attrOnly then
         attrName = attrOnly
     else
         local attrWithVal, attrVal = selector:match("^%[([%w%-_]+)=[\"']?([^\"'%]]+)[\"']?%]$")
         if attrWithVal then
             attrName = attrWithVal
             attrValue = attrVal
         end
     end
 
     -- Check for tag#id
     local tagId, idPart = selector:match("^([%w%-_]+)#([%w%-_]+)$")
     if tagId then
         tag = tagId:lower()
         id = idPart
     end
 
     -- Check for tag.class
     local tagClass, classPart = selector:match("^([%w%-_]+)%.([%w%-_]+)$")
     if tagClass then
         tag = tagClass:lower()
         class = classPart
     end
 
     -- Check for plain tag
     if not tag and not id and not class and not attrName then
         tag = selector:lower()
     end
 
     -- Search recursively
     return self:_querySelectorSearch(tag, id, class, attrName, attrValue)
 end
 
 function Element:_querySelectorSearch(tag, id, class, attrName, attrValue)
     -- Check if this element matches
     local matches = true
 
     if tag and self.tag ~= tag then
         matches = false
     end
     if id and self.id ~= id then
         matches = false
     end
     if class and not self:hasClass(class) then
         matches = false
     end
     if attrName then
         local attr = self.attributes[attrName]
         if attr == nil then
             matches = false
         elseif attrValue and tostring(attr) ~= attrValue then
             matches = false
         end
     end
 
     if matches and (tag or id or class or attrName) then
         return self
     end
 
     -- Search children
     for _, child in ipairs(self.children) do
         local found = child:_querySelectorSearch(tag, id, class, attrName, attrValue)
         if found then
             return found
         end
     end
 
     return nil
 end
 
 -- Find all elements matching a CSS selector
 function Element:querySelectorAll(selector)
     local results = {}
 
     if not selector or selector == "" then
         return results
     end
 
     -- Parse selector (same as querySelector)
     local tag, id, class, attrName, attrValue
 
     local idMatch = selector:match("^#([%w%-_]+)$")
     if idMatch then
         local elem = self:findById(idMatch)
         if elem then table.insert(results, elem) end
         return results
     end
 
     local classMatch = selector:match("^%.([%w%-_]+)$")
     if classMatch then
         return self:findByClass(classMatch)
     end
 
     local attrOnly = selector:match("^%[([%w%-_]+)%]$")
     if attrOnly then
         attrName = attrOnly
     else
         local attrWithVal, attrVal = selector:match("^%[([%w%-_]+)=[\"']?([^\"'%]]+)[\"']?%]$")
         if attrWithVal then
             attrName = attrWithVal
             attrValue = attrVal
         end
     end
 
     local tagId, idPart = selector:match("^([%w%-_]+)#([%w%-_]+)$")
     if tagId then
         tag = tagId:lower()
         id = idPart
     end
 
     local tagClass, classPart = selector:match("^([%w%-_]+)%.([%w%-_]+)$")
     if tagClass then
         tag = tagClass:lower()
         class = classPart
     end
 
     if not tag and not id and not class and not attrName then
         tag = selector:lower()
     end
 
     self:_querySelectorAllSearch(tag, id, class, attrName, attrValue, results)
     return results
 end
 
 function Element:_querySelectorAllSearch(tag, id, class, attrName, attrValue, results)
     local matches = true
 
     if tag and self.tag ~= tag then
         matches = false
     end
     if id and self.id ~= id then
         matches = false
     end
     if class and not self:hasClass(class) then
         matches = false
     end
     if attrName then
         local attr = self.attributes[attrName]
         if attr == nil then
             matches = false
         elseif attrValue and tostring(attr) ~= attrValue then
             matches = false
         end
     end
 
     if matches and (tag or id or class or attrName) then
         table.insert(results, self)
     end
 
     for _, child in ipairs(self.children) do
         child:_querySelectorAllSearch(tag, id, class, attrName, attrValue, results)
     end
 end
 
 return {
     Element = Element
 }