DifyApiConfigController.java 4.7 KB
package com.aigeo.ai.controller;

import com.aigeo.ai.entity.DifyApiConfig;
import com.aigeo.ai.service.DifyApiConfigService;
import com.aigeo.common.Result;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.List;

/**
 * AI配置控制器
 */
@Slf4j
@RestController
@RequestMapping("/api/ai-configs")
@Tag(name = "AI配置管理", description = "AI配置相关接口")
public class DifyApiConfigController {
    
    @Autowired
    private DifyApiConfigService difyApiConfigService;
    
    @PostMapping
    @Operation(summary = "创建AI配置", description = "创建新的AI配置")
    public Result<DifyApiConfig> createConfig(@RequestBody DifyApiConfig config) {
        try {
            DifyApiConfig savedConfig = difyApiConfigService.save(config);
            return Result.success("配置创建成功", savedConfig);
        } catch (Exception e) {
            log.error("创建AI配置失败", e);
            return Result.error("配置创建失败");
        }
    }
    
    @GetMapping("/{id}")
    @Operation(summary = "获取AI配置详情", description = "根据ID获取AI配置详情")
    public Result<DifyApiConfig> getConfigById(@PathVariable Integer id) {
        try {
            return difyApiConfigService.findById(id)
                    .map(config -> Result.success("查询成功", config))
                    .orElse(Result.error("配置不存在"));
        } catch (Exception e) {
            log.error("获取AI配置详情失败, id: {}", id, e);
            return Result.error("查询失败");
        }
    }
    
    @GetMapping
    @Operation(summary = "获取AI配置列表", description = "获取所有AI配置列表")
    public Result<List<DifyApiConfig>> getAllConfigs() {
        try {
            List<DifyApiConfig> configs = difyApiConfigService.findAll();
            return Result.success("查询成功", configs);
        } catch (Exception e) {
            log.error("获取AI配置列表失败", e);
            return Result.error("查询失败");
        }
    }
    
    @GetMapping("/company/{companyId}")
    @Operation(summary = "根据公司ID获取AI配置列表", description = "根据公司ID获取AI配置列表")
    public Result<List<DifyApiConfig>> getConfigsByCompanyId(@PathVariable Integer companyId) {
        try {
            List<DifyApiConfig> configs = difyApiConfigService.findByCompanyId(companyId);
            return Result.success("查询成功", configs);
        } catch (Exception e) {
            log.error("根据公司ID获取AI配置列表失败, companyId: {}", companyId, e);
            return Result.error("查询失败");
        }
    }
    
    @GetMapping("/company/{companyId}/active")
    @Operation(summary = "根据公司ID获取启用的AI配置列表", description = "根据公司ID获取启用的AI配置列表")
    public Result<List<DifyApiConfig>> getActiveConfigsByCompanyId(@PathVariable Integer companyId) {
        try {
            List<DifyApiConfig> configs = difyApiConfigService.findActiveByCompanyId(companyId);
            return Result.success("查询成功", configs);
        } catch (Exception e) {
            log.error("根据公司ID获取启用的AI配置列表失败, companyId: {}", companyId, e);
            return Result.error("查询失败");
        }
    }
    
    @PutMapping("/{id}")
    @Operation(summary = "更新AI配置", description = "更新AI配置信息")
    public Result<DifyApiConfig> updateConfig(@PathVariable Integer id, @RequestBody DifyApiConfig config) {
        try {
            if (!difyApiConfigService.findById(id).isPresent()) {
                return Result.error("配置不存在");
            }
            config.setId(id);
            DifyApiConfig updatedConfig = difyApiConfigService.save(config);
            return Result.success("配置更新成功", updatedConfig);
        } catch (Exception e) {
            log.error("更新AI配置失败, id: {}", id, e);
            return Result.error("配置更新失败");
        }
    }
    
    @DeleteMapping("/{id}")
    @Operation(summary = "删除AI配置", description = "删除指定ID的AI配置")
    public Result<String> deleteConfig(@PathVariable Integer id) {
        try {
            if (!difyApiConfigService.findById(id).isPresent()) {
                return Result.error("配置不存在");
            }
            difyApiConfigService.deleteById(id);
            return Result.success("配置删除成功");
        } catch (Exception e) {
            log.error("删除AI配置失败, id: {}", id, e);
            return Result.error("配置删除失败");
        }
    }
}