mirror of
https://gitee.com/mateos/mateclaw.git
synced 2026-09-14 03:33:43 +08:00
113 lines
3.9 KiB
Java
113 lines
3.9 KiB
Java
package vip.mate.tool.controller;
|
||
|
||
import io.swagger.v3.oas.annotations.Operation;
|
||
import io.swagger.v3.oas.annotations.tags.Tag;
|
||
import lombok.RequiredArgsConstructor;
|
||
import org.springframework.web.bind.annotation.*;
|
||
import vip.mate.common.result.R;
|
||
import vip.mate.tool.disclosure.DisclosureTier;
|
||
import vip.mate.tool.disclosure.ToolDisclosureService;
|
||
import vip.mate.tool.model.AvailableToolDTO;
|
||
import vip.mate.tool.model.ToolEntity;
|
||
import vip.mate.tool.service.AvailableToolService;
|
||
import vip.mate.tool.service.ToolService;
|
||
import vip.mate.workspace.core.annotation.RequireWorkspaceRole;
|
||
|
||
import java.util.List;
|
||
import java.util.Map;
|
||
|
||
/**
|
||
* 工具管理接口
|
||
*
|
||
* @author MateClaw Team
|
||
*/
|
||
@Tag(name = "工具管理")
|
||
@RestController
|
||
@RequestMapping("/api/v1/tools")
|
||
@RequiredArgsConstructor
|
||
public class ToolController {
|
||
|
||
private final ToolService toolService;
|
||
private final AvailableToolService availableToolService;
|
||
private final ToolDisclosureService toolDisclosureService;
|
||
|
||
@Operation(summary = "获取工具列表")
|
||
@GetMapping
|
||
@RequireWorkspaceRole("member")
|
||
public R<List<ToolEntity>> list() {
|
||
return R.ok(toolService.listTools());
|
||
}
|
||
|
||
@Operation(summary = "获取已启用工具列表")
|
||
@GetMapping("/enabled")
|
||
@RequireWorkspaceRole("member")
|
||
public R<List<ToolEntity>> listEnabled() {
|
||
return R.ok(toolService.listEnabledTools());
|
||
}
|
||
|
||
@Operation(summary = "获取员工可绑定的全部原子工具（含 MCP）")
|
||
@GetMapping("/available")
|
||
@RequireWorkspaceRole("member")
|
||
public R<List<AvailableToolDTO>> listAvailable() {
|
||
return R.ok(availableToolService.listAvailable());
|
||
}
|
||
|
||
@Operation(summary = "获取工具详情")
|
||
@GetMapping("/{id}")
|
||
@RequireWorkspaceRole("admin")
|
||
public R<ToolEntity> get(@PathVariable Long id) {
|
||
return R.ok(toolService.getTool(id));
|
||
}
|
||
|
||
@Operation(summary = "创建工具（MCP）")
|
||
@PostMapping
|
||
@RequireWorkspaceRole("admin")
|
||
public R<ToolEntity> create(@RequestBody ToolEntity tool) {
|
||
return R.ok(toolService.createTool(tool));
|
||
}
|
||
|
||
@Operation(summary = "更新工具")
|
||
@PutMapping("/{id}")
|
||
@RequireWorkspaceRole("admin")
|
||
public R<ToolEntity> update(@PathVariable Long id, @RequestBody ToolEntity tool) {
|
||
tool.setId(id);
|
||
return R.ok(toolService.updateTool(tool));
|
||
}
|
||
|
||
@Operation(summary = "删除工具")
|
||
@DeleteMapping("/{id}")
|
||
@RequireWorkspaceRole("admin")
|
||
public R<Void> delete(@PathVariable Long id) {
|
||
toolService.deleteTool(id);
|
||
return R.ok();
|
||
}
|
||
|
||
@Operation(summary = "启用/禁用工具")
|
||
@PutMapping("/{id}/toggle")
|
||
@RequireWorkspaceRole("admin")
|
||
public R<ToolEntity> toggle(@PathVariable Long id, @RequestParam boolean enabled) {
|
||
return R.ok(toolService.toggleTool(id, enabled));
|
||
}
|
||
|
||
@Operation(summary = "设置工具披露分级（core / extension）")
|
||
@PutMapping("/{id}/disclosure-tier")
|
||
@RequireWorkspaceRole("admin")
|
||
public R<ToolEntity> setDisclosureTier(@PathVariable Long id, @RequestBody Map<String, String> body) {
|
||
String tier = body == null ? null : body.get("tier");
|
||
if (!DisclosureTier.isValidToken(tier)) {
|
||
return R.fail(400, "tier must be 'core' or 'extension'");
|
||
}
|
||
ToolEntity tool = toolService.getTool(id);
|
||
String type = tool.getToolType();
|
||
// Only builtin / channel atomic tools are tiered on the row itself; MCP /
|
||
// ACP / skill tools are tiered at their owning source.
|
||
if (!"builtin".equals(type) && !"channel".equals(type)) {
|
||
return R.fail(409, "This tool's tier is decided by its owning server/endpoint/skill. "
|
||
+ "Modify it there instead.");
|
||
}
|
||
ToolEntity updated = toolService.setDisclosureTier(id, tier);
|
||
toolDisclosureService.invalidate();
|
||
return R.ok(updated);
|
||
}
|
||
}
|