package cc.mrbird.febs.ai.service.impl;

import cc.mrbird.febs.ai.entity.AiCompanyWorkflow;
import cc.mrbird.febs.ai.mapper.AiCompanyWorkflowMapper;
import cc.mrbird.febs.ai.service.AiCompanyWorkflowService;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

@Slf4j
@Service
@RequiredArgsConstructor
public class AiCompanyWorkflowServiceImpl extends ServiceImpl<AiCompanyWorkflowMapper, AiCompanyWorkflow> implements AiCompanyWorkflowService {

    private final AiCompanyWorkflowMapper aiCompanyWorkflowMapper;

    @Override
    public AiCompanyWorkflow getByTypeAndCompanyId(Integer type, String companyId) {
        LambdaQueryWrapper<AiCompanyWorkflow> query = Wrappers.lambdaQuery(AiCompanyWorkflow.class);
        query.eq(AiCompanyWorkflow::getType, type);
        if (StrUtil.isEmpty(companyId)){
            query.isNull(AiCompanyWorkflow::getCompanyId);
        }else{
            query.eq(AiCompanyWorkflow::getCompanyId, companyId);
        }
        query.last("limit 1");
        return aiCompanyWorkflowMapper.selectOne(query);
    }
}