214 lines
6.4 KiB
Lua
214 lines
6.4 KiB
Lua
require("fwutils.webapi")
|
||
-- CREATE TABLE `fw_menu` (
|
||
-- `id` int NOT NULL AUTO_INCREMENT,
|
||
-- `role_id` int DEFAULT NULL,
|
||
-- `title` varchar(255) COLLATE utf8mb4_unicode_ci DEFAULT NULL,
|
||
-- `path` varchar(255) COLLATE utf8mb4_unicode_ci DEFAULT NULL,
|
||
-- `icon` varchar(255) COLLATE utf8mb4_unicode_ci DEFAULT NULL,
|
||
-- `sort` int DEFAULT NULL,
|
||
-- `parent_id` int DEFAULT NULL,
|
||
-- `create_time` datetime DEFAULT NULL,
|
||
-- `update_time` datetime DEFAULT NULL,
|
||
-- `delete_time` datetime DEFAULT NULL,
|
||
-- PRIMARY KEY (`id`)
|
||
-- ) ENGINE=InnoDB AUTO_INCREMENT=3 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;
|
||
local M = {}
|
||
|
||
function M.get_by_id(id,conn)
|
||
local select = conn:select()
|
||
select:table("fw_menu LEFT JOIN fw_role ON fw_menu.role_id = fw_role.id")
|
||
select:field({
|
||
"fw_menu.id",
|
||
"fw_menu.title",
|
||
"fw_menu.path",
|
||
"fw_menu.icon",
|
||
"fw_menu.sort",
|
||
"fw_menu.parent_id",
|
||
"fw_menu.create_time",
|
||
"fw_menu.update_time",
|
||
"fw_menu.delete_time",
|
||
"fw_role.title as role_title",
|
||
"fw_role.id as role_id",
|
||
})
|
||
select:where_i32("fw_menu.id","=",id)
|
||
select:where_expression("AND fw_menu.delete_time IS NULL")
|
||
select:limit(0,1)
|
||
local result = select:query()
|
||
if result:row_count() == 0 then
|
||
return nil
|
||
end
|
||
local d = result:table()[1]
|
||
return d
|
||
end
|
||
function M.add(data,conn)
|
||
local insert = conn:insert()
|
||
insert:table("fw_menu")
|
||
insert:set_str("title",data.title)
|
||
insert:set_str("path",data.path)
|
||
insert:set_i32("role_id",data.role_id)
|
||
if data.icon ~= nil then
|
||
insert:set_str("icon",data.icon)
|
||
end
|
||
if data.sort ~= nil then
|
||
insert:set_i32("sort",data.sort)
|
||
end
|
||
if data.parent_id ~= nil and data.parent_id ~= 0 then
|
||
insert:set_i32("parent_id",data.parent_id)
|
||
end
|
||
insert:set_not_ppst("create_time","NOW()")
|
||
local d = insert:exec()
|
||
return d == 1
|
||
end
|
||
function M.update(data,conn)
|
||
local update = conn:update()
|
||
update:table("fw_menu")
|
||
if data.title ~= nil then
|
||
update:set_str("title",data.title)
|
||
end
|
||
if data.path ~= nil then
|
||
update:set_str("path",data.path)
|
||
end
|
||
if data.role_id ~= nil then
|
||
update:set_i32("role_id",data.role_id)
|
||
end
|
||
if data.icon ~= nil then
|
||
update:set_str("icon",data.icon)
|
||
end
|
||
if data.sort ~= nil then
|
||
update:set_i32("sort",data.sort)
|
||
end
|
||
if data.parent_id ~= nil then
|
||
update:set_i32("parent_id",data.parent_id)
|
||
end
|
||
update:set("update_time=NOW()")
|
||
update:where_i32("id","=",data.id)
|
||
local d = update:exec()
|
||
return d == 1
|
||
end
|
||
function M.delete(id,conn)
|
||
local update = conn:update()
|
||
update:table("fw_menu")
|
||
update:set("delete_time=NOW()")
|
||
update:where_i32("id","=",id)
|
||
local d = update:exec()
|
||
return d == 1
|
||
end
|
||
function M.list(search,limit,conn)
|
||
if search.role_id == -1 then
|
||
return {
|
||
count = 0,
|
||
data = {}
|
||
}
|
||
end
|
||
return query_model_ex(conn,[=[
|
||
fw_menu LEFT JOIN fw_role ON fw_menu.role_id = fw_role.id
|
||
]=],{
|
||
"fw_menu.id",
|
||
"fw_menu.title",
|
||
"fw_menu.path",
|
||
"fw_menu.icon",
|
||
"fw_menu.sort",
|
||
"fw_menu.parent_id",
|
||
"fw_menu.create_time",
|
||
"fw_menu.update_time",
|
||
"fw_menu.delete_time",
|
||
"fw_role.title as role_title",
|
||
"fw_role.id as role_id",
|
||
},limit.start,limit.length,function(sel)
|
||
|
||
if search.role_id ~= nil and search.role_id ~= 0 then
|
||
sel:where_i32("fw_menu.role_id","=",search.role_id)
|
||
end
|
||
sel:where_expression("AND fw_menu.delete_time IS NULL")
|
||
end,function(sel_data)
|
||
sel_data:orderby("fw_menu.sort DESC")
|
||
end)
|
||
end
|
||
M.make_bytecode = function(role_id,conn)
|
||
-- 查询菜单表
|
||
local select = conn:select()
|
||
select:table("fw_menu")
|
||
select:where_expression("AND delete_time IS NULL")
|
||
if role_id ~= nil then
|
||
select:where_i32("role_id", "=", role_id)
|
||
end
|
||
local result = select:query()
|
||
|
||
local bc = {}
|
||
local items_by_id = {} -- 通过id快速查找菜单项：{id = {role_id, title, item, parent_id}}
|
||
local items_with_parent = {} -- 存储有父级的菜单项
|
||
|
||
-- 第一遍：读取所有菜单项并存储
|
||
while result:next() do
|
||
local id = result:get("id")
|
||
local role_id = tostring(result:get("role_id"))
|
||
local title = result:get("title")
|
||
local path = result:get("path")
|
||
local icon = result:get("icon")
|
||
local sort = result:get("sort")
|
||
local parent_id = result:get("parent_id")
|
||
|
||
if not title or title == "" then
|
||
goto continue
|
||
end
|
||
|
||
if bc[role_id] == nil then
|
||
bc[role_id] = {}
|
||
end
|
||
|
||
local item = {
|
||
path = path,
|
||
icon = icon,
|
||
sort = sort,
|
||
}
|
||
|
||
-- 存储所有菜单项信息
|
||
items_by_id[id] = {
|
||
role_id = role_id,
|
||
title = title,
|
||
item = item,
|
||
parent_id = parent_id
|
||
}
|
||
|
||
-- 如果parent_id为空，则作为顶层菜单项
|
||
if parent_id == nil or parent_id == 0 then
|
||
bc[role_id][title] = item
|
||
else
|
||
-- 有父级，记录下来稍后处理
|
||
table.insert(items_with_parent, {
|
||
id = id,
|
||
role_id = role_id,
|
||
title = title,
|
||
item = item,
|
||
parent_id = parent_id
|
||
})
|
||
end
|
||
|
||
::continue::
|
||
end
|
||
|
||
-- 第二遍：处理有父级的菜单项，构建children结构
|
||
for _, menu_item in ipairs(items_with_parent) do
|
||
local parent_info = items_by_id[menu_item.parent_id]
|
||
if parent_info then
|
||
local parent_item = parent_info.item
|
||
|
||
-- 如果父项还没有children表，创建它
|
||
if not parent_item.children then
|
||
parent_item.children = {}
|
||
end
|
||
|
||
-- 将子项添加到父项的children中
|
||
parent_item.children[menu_item.title] = menu_item.item
|
||
end
|
||
end
|
||
|
||
local code = "return " .. require("serpent").serialize(bc, {comment = false})
|
||
utils.save_file(fw.website_dir().."/"..(fwutils_config.path.luabytecode:gsub("%.", "/")).."/menu_bc.lua",code)
|
||
return true
|
||
end
|
||
M.bytecode = function(role_id)
|
||
local menu_bc = require(fwutils_config.path.luabytecode..".menu_bc")
|
||
return menu_bc[string.format("%d",role_id)]
|
||
end
|
||
return M |