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

import cc.mrbird.febs.ai.entity.AiProductPrompt;
import cc.mrbird.febs.ai.entity.AiProductQuestionLabel;
import cc.mrbird.febs.ai.service.AiProductPromptService;
import cc.mrbird.febs.ai.service.AiProductQuestionLabelService;
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.List;
import java.util.Map;

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


    private final AiProductQuestionLabelService aiProductQuestionLabelService;

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

    @PostMapping("add")
    @ControllerEndpoint(operation = "新增", exceptionMessage = "操作失败")
    public FebsResponse add(@RequestBody @Valid AiProductQuestionLabel dto) {
        String companyId = getCurrentUserCompanyId();
        dto.setCompanyId(companyId);
        return aiProductQuestionLabelService.add(dto);
    }

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

        return aiProductQuestionLabelService.update(dto);
    }

    @GetMapping("listAll")
    @ControllerEndpoint(operation = "获取所有标签", exceptionMessage = "获取标签失败")
    public FebsResponse listAll() {
        String companyId = getCurrentUserCompanyId();
        List<AiProductQuestionLabel> labels = aiProductQuestionLabelService.listAllByCompanyId(companyId);
        return new FebsResponse().success().data(labels);
    }
}