package tech.aiflowy.ai.controller;

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import com.mybatisflex.core.paginate.Page;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import tech.aiflowy.ai.entity.AiFirstMenu;
import tech.aiflowy.ai.entity.AiSecondMenu;
import tech.aiflowy.ai.service.AiFirstMenuService;
import tech.aiflowy.ai.service.AiSecondMenuService;
import tech.aiflowy.common.domain.Result;

import java.util.List;

@RestController
@RequestMapping("/api/v1/aiMenu")
public class AiMenuController {

    @Autowired
    private AiFirstMenuService aiFirstMenuService;

    /**
     * 创建一级菜单
     */
    @PostMapping("FirstMenu")
    public Result create(@RequestBody AiFirstMenu menu) {
        boolean success = aiFirstMenuService.save(menu);
        return Result.success(success);
    }

    /**
     * 删除一级菜单
     */
    @DeleteMapping("/FirstMenu/{id}")
    public Result delete(@PathVariable Integer id) {
        boolean success = aiFirstMenuService.removeById(id);
        return Result.success(success);
    }

    /**
     * 更新一级菜单
     */
    @PutMapping("FirstMenu")
    public Result updateFirstMenu(@RequestBody AiFirstMenu menu) {
        boolean success = aiFirstMenuService.updateById(menu);
        return Result.success(success);
    }

    /**
     * 根据ID查询一级菜单
     */
    @GetMapping("/FirstMenu/{id}")
    public Result getByIdFirstMenu(@PathVariable Integer id) {
        AiFirstMenu menu = aiFirstMenuService.getById(id);
        return Result.success(menu);
    }

    /**
     * 分页查询一级菜单
     */
    @GetMapping("/FirstMenu/page")
    public Result pageFirstMenu(
            @RequestParam(defaultValue = "1") Integer pageNumber,
            @RequestParam(defaultValue = "10") Integer pageSize) {
        Page<AiFirstMenu> page = new Page<>(pageNumber, pageSize);
        page = aiFirstMenuService.page(page);
        return Result.success(page);
    }

    /**
     * 获取所有一级菜单
     */
    @GetMapping("FirstMenu/list")
    public Result listFirstMenu() {
        List<AiFirstMenu> list = aiFirstMenuService.list();
        return Result.success(list);
    }
    @Autowired
    private AiSecondMenuService aiSecondMenuService;

    /**
     * 创建二级菜单
     */
    @PostMapping("SecondMenu")
    public Result createSecondMenu(@RequestBody AiSecondMenu menu) {
        boolean success = aiSecondMenuService.save(menu);
        return Result.success(success);
    }

    /**
     * 删除二级菜单
     */
    @DeleteMapping("SecondMenu/{id}")
    public Result deleteSecondMenu(@PathVariable Integer id) {
        boolean success = aiSecondMenuService.removeById(id);
        return Result.success(success);
    }

    /**
     * 更新二级菜单
     */
    @PutMapping("SecondMenu/update")
    public Result updateSecondMenu(@RequestBody AiSecondMenu menu) {
        boolean success = aiSecondMenuService.updateById(menu);
        return Result.success(success);
    }

    /**
     * 根据ID查询二级菜单
     */
    @GetMapping("SecondMenu/{id}")
    public Result getByIdSecondMenu(@PathVariable Integer id) {
        AiSecondMenu menu = aiSecondMenuService.getById(id);
        return Result.success(menu);
    }

    /**
     * 分页查询二级菜单
     */
    @GetMapping("SecondMenu/page")
    public Result pageSecondMenu(
            @RequestParam(defaultValue = "1") Integer pageNumber,
            @RequestParam(defaultValue = "10") Integer pageSize) {
        Page<AiSecondMenu> page = new Page<>(pageNumber, pageSize);
        page = aiSecondMenuService.page(page);
        return Result.success(page);
    }

    /**
     * 获取所有一级菜单
     */
    @GetMapping("SecondMenu/list")
    public Result listSecondMenu() {
        List<AiSecondMenu> list = aiSecondMenuService.list();
        return Result.success(list);
    }
}