package com.iailab.module.ai.api.template;

import com.iailab.framework.common.util.object.BeanUtils;
import com.iailab.module.ai.api.questionTemplate.QuestionTemplateApi;
import com.iailab.module.ai.api.questionTemplate.dto.AiModelDTO;
import com.iailab.module.ai.api.questionTemplate.dto.QuestionTemplateDTO;
import com.iailab.module.ai.controller.admin.questiontemplate.vo.QuestionTemplateReqVO;
import com.iailab.module.ai.service.model.AiModelService;
import com.iailab.module.ai.service.questiontemplate.QuestionTemplateService;
import jakarta.annotation.Resource;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.RestController;

import java.util.List;

/**
 * @description:
 * @author: dongyukun
 * @date: 2025/6/4 14:50
 **/
@RestController // 提供 RESTful API 接口，给 Feign 调用
@Validated
public class QuestionTemplateApiImpl implements QuestionTemplateApi {

    @Resource
    private QuestionTemplateService questionTemplateService;

    @Resource
    private AiModelService aiModelService;

    @Override
    public List<AiModelDTO> queryModelList() {
        List<AiModelDTO> list = BeanUtils.toBean(aiModelService.getModelListByStatusAndType(0, 1, null), AiModelDTO.class);
        list.forEach(item->{
            item.setChildren(
                    BeanUtils.toBean(questionTemplateService.getQuestionTemplateList(item.getId()), QuestionTemplateDTO.class));
        });
        return list;
    }

    @Override
    public List<QuestionTemplateDTO> queryTemplates(QuestionTemplateDTO reqDTO) {
        QuestionTemplateReqVO ReqVO = BeanUtils.toBean(reqDTO, QuestionTemplateReqVO.class);
        return BeanUtils.toBean(questionTemplateService.getQuestionTemplates(ReqVO), QuestionTemplateDTO.class);
    }
}