antd-web-apps/info/controllers/IndexController.lua

BaseController:subclass(
    "IndexController",
    {
        registry = {},
        models = {"sections", "category"}
    }
)

local sectionsByCid = function(db, id)
    local cond = {exp = { ["and"] = {{["="] = {cid = id}}, {["= "] = {publish = 1}} }}, order = {start = "DESC"}}
    local data, a = db:find(cond)
    return data, a
end

function IndexController:index(...)
    local args = {...}
    -- now read all the data
    -- get all root sections as the toc
    local cond = {exp = {["="] = {pid = 0}}, order = {name = "ASC"}}
    local data, a = self.category:find(cond)
    local toc = {}
    if not data then
        return self:error("Cannot query the ToC")
    end
    -- find all children category of the toc
    for key, cat in pairs(data) do
        cat.name = cat.name:gsub("^%d+%.", "")
        table.insert(toc, {cat.name, cat.id})
        cond = {exp = {["="] = {pid = cat.id}}, order = {name = "ASC"}}
        local children, b = self.category:find(cond)
        if children and #children > 0 then
            for k, v in pairs(children) do
                v.sections = sectionsByCid(self.sections, v.id)
            end
            cat.children = children
        else
            cat.sections = sectionsByCid(self.sections, cat.id)
        end
    end
    self.template:set("data", data)
    self.template:set("toc", toc)
    return true
end

function IndexController:notoc(...)
    self.template:setView("index")
    return self:index(table.unpack({...}))
end

function IndexController:actionnotfound(...)
    return self:notoc(table.unpack({...}))
end

function IndexController:pdf(...)
    local tmp_file = WWW_ROOT.."/cv_exported.pdf"
    local cmd = "wkhtmltopdf "..HTTP_ROOT.."/"..self.registry.user.."/index/notoc "..tmp_file
    print(cmd)
    local r = os.execute(cmd)
    if r then
        local mime = std.mimeOf(tmp_file)
        std.header(mime)
        std.f(tmp_file)
        return false
    else
        return self:error("Sorry.Problem generate PDF file")
    end
end