package cc.mrbird.febs.ai.controller.knowledge;

import cc.mrbird.febs.ai.entity.AiCompany;
import cc.mrbird.febs.ai.entity.AiKnowledgeFile;
import cc.mrbird.febs.ai.service.AiCompanyService;
import cc.mrbird.febs.ai.service.AiKnowledgeFileService;
import cc.mrbird.febs.common.annotation.ControllerEndpoint;
import cc.mrbird.febs.common.controller.BaseController;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.entity.QueryRequest;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;

import javax.validation.Valid;
import java.util.Map;

/**
 * @author Administrator
 */
@Slf4j
@Validated
@RestController
@RequiredArgsConstructor
@RequestMapping(value = "/admin/aiKnowledgeFile")
public class AiKnowledgeController extends BaseController {

    private final AiKnowledgeFileService aiKnowledgeFileService;

    @GetMapping("list")
    public FebsResponse list(AiKnowledgeFile dto, QueryRequest request) {
        String companyId = getCurrentUserCompanyId();
        dto.setCompanyId(companyId);
        Map<String, Object> data = getDataTable(aiKnowledgeFileService.listInPage(dto, request));
        return new FebsResponse().success().data(data);
    }

    @PostMapping("add")
    @ControllerEndpoint(operation = "新增", exceptionMessage = "操作失败")
    public FebsResponse add(@RequestBody @Valid AiKnowledgeFile dto) {

        String companyId = getCurrentUserCompanyId();
        dto.setCompanyId(companyId);
        return aiKnowledgeFileService.add(dto);
    }

    @PostMapping("update")
    @ControllerEndpoint(operation = "更新", exceptionMessage = "操作失败")
    public FebsResponse update(@RequestBody @Valid AiKnowledgeFile dto) {

        return aiKnowledgeFileService.update(dto);
    }
}