blob: 357be7ea7ccdccb47109a671eb9cb36556f85699 [file] [log] [blame]
local base = _G
-- symbolic name for tag index, this allows accessing the tag by var[xml.TAG]
xml.TAG = 0
-- sets or returns tag of a LuaXML object
function xml.tag(var,tag)
if base.type(var)~="table" then return end
if base.type(tag)=="nil" then
return var[xml.TAG]
end
var[xml.TAG] = tag
end
-- creates a new LuaXML object either by setting the metatable of an existing Lua table or by setting its tag
function xml.new(arg)
if base.type(arg)=="table" then
base.setmetatable(arg,{__index=xml, __tostring=xml.str})
return arg
end
local var={}
base.setmetatable(var,{__index=xml, __tostring=xml.str})
if base.type(arg)=="string" then var[xml.TAG]=arg end
return var
end
-- appends a new subordinate LuaXML object to an existing one, optionally sets tag
function xml.append(var,tag)
if base.type(var)~="table" then return end
local newVar = xml.new(tag)
var[#var+1] = newVar
return newVar
end
-- converts any Lua var into an XML string
function xml.str(var,indent,tagValue)
if base.type(var)=="nil" then return end
local indent = indent or 0
local indentStr=""
for i = 1,indent do indentStr=indentStr.." " end
local tableStr=""
if base.type(var)=="table" then
local tag = var[0] or tagValue or base.type(var)
local s = indentStr.."<"..tag
for k,v in base.pairs(var) do -- attributes
if base.type(k)=="string" then
if base.type(v)=="table" and k~="_M" then -- otherwise recursiveness imminent
tableStr = tableStr..xml.str(v,indent+1,k)
else
s = s.." "..k.."=\""..xml.encode(base.tostring(v)).."\""
end
end
end
if #var==0 and #tableStr==0 then
s = s.." />\n"
elseif #var==1 and base.type(var[1])~="table" and #tableStr==0 then -- single element
s = s..">"..xml.encode(base.tostring(var[1])).."</"..tag..">\n"
else
s = s..">\n"
for k,v in base.ipairs(var) do -- elements
if base.type(v)=="string" then
s = s..indentStr.." "..xml.encode(v).." \n"
else
s = s..xml.str(v,indent+1)
end
end
s=s..tableStr..indentStr.."</"..tag..">\n"
end
return s
else
local tag = base.type(var)
return indentStr.."<"..tag.."> "..xml.encode(base.tostring(var)).." </"..tag..">\n"
end
end
-- saves a Lua var as xml file
function xml.save(var,filename)
if not var then return end
if not filename or #filename==0 then return end
local file = base.io.open(filename,"w")
file:write("<?xml version=\"1.0\"?>\n<!-- file \"",filename, "\", generated by LuaXML -->\n\n")
file:write(xml.str(var))
base.io.close(file)
end
-- recursively parses a Lua table for a substatement fitting to the provided tag and attribute
function xml.find(var, tag, attributeKey,attributeValue)
-- check input:
if base.type(var)~="table" then return end
if base.type(tag)=="string" and #tag==0 then tag=nil end
if base.type(attributeKey)~="string" or #attributeKey==0 then attributeKey=nil end
if base.type(attributeValue)=="string" and #attributeValue==0 then attributeValue=nil end
-- compare this table:
if tag~=nil then
if var[0]==tag and ( attributeValue == nil or var[attributeKey]==attributeValue ) then
base.setmetatable(var,{__index=xml, __tostring=xml.str})
return var
end
else
if attributeValue == nil or var[attributeKey]==attributeValue then
base.setmetatable(var,{__index=xml, __tostring=xml.str})
return var
end
end
-- recursively parse subtags:
for k,v in base.ipairs(var) do
if base.type(v)=="table" then
local ret = xml.find(v, tag, attributeKey,attributeValue)
if ret ~= nil then return ret end
end
end
end