qiangdanxitong/backend/src/main/java/com/designhub/service/ExternalAssetService.java

package com.designhub.service;

import com.designhub.dto.ExternalAssetDTO;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.*;
import org.springframework.stereotype.Service;
import org.springframework.web.client.RestTemplate;

/**
 * 第三方素材API服务
 * 用于查询外部素材系统的任务完成状态和下载链接
 */
@Slf4j
@Service
@RequiredArgsConstructor
public class ExternalAssetService {

    private final SystemConfigService systemConfigService;
    private final RestTemplate restTemplate;
    private final ObjectMapper objectMapper;

    /**
     * 查询任务素材
     * @param taskNo 任务编号
     * @return 素材信息，包含下载链接
     */
    public ExternalAssetDTO queryAssets(String taskNo) {
        String apiUrl = systemConfigService.getConfigValue("external_asset_api_url");
        String apiKey = systemConfigService.getConfigValue("external_asset_api_key");
        
        if (apiUrl == null || apiUrl.isEmpty()) {
            log.warn("第三方素材API地址未配置");
            ExternalAssetDTO result = new ExternalAssetDTO();
            result.setSuccess(false);
            ExternalAssetDTO.ErrorInfo error = new ExternalAssetDTO.ErrorInfo();
            error.setCode("CONFIG_ERROR");
            error.setMessage("第三方素材API未配置");
            result.setError(error);
            return result;
        }
        
        if (apiKey == null || apiKey.isEmpty()) {
            log.warn("第三方素材API Key未配置");
            ExternalAssetDTO result = new ExternalAssetDTO();
            result.setSuccess(false);
            ExternalAssetDTO.ErrorInfo error = new ExternalAssetDTO.ErrorInfo();
            error.setCode("CONFIG_ERROR");
            error.setMessage("第三方素材API Key未配置");
            result.setError(error);
            return result;
        }
        
        try {
            // 构建请求URL
            String url = apiUrl.endsWith("/") ? apiUrl : apiUrl + "/";
            url = url + "product-assets?taskNo=" + taskNo;
            
            log.info("查询第三方素材API: {}", url);
            
            // 构建请求头
            HttpHeaders headers = new HttpHeaders();
            headers.set("X-API-Key", apiKey);
            headers.setContentType(MediaType.APPLICATION_JSON);
            
            HttpEntity<String> entity = new HttpEntity<>(headers);
            
            // 发送请求
            ResponseEntity<String> response = restTemplate.exchange(
                url,
                HttpMethod.GET,
                entity,
                String.class
            );
            
            log.info("第三方素材API响应状态: {}", response.getStatusCode());
            log.debug("第三方素材API响应内容: {}", response.getBody());
            
            if (response.getStatusCode() == HttpStatus.OK && response.getBody() != null) {
                return objectMapper.readValue(response.getBody(), ExternalAssetDTO.class);
            } else {
                ExternalAssetDTO result = new ExternalAssetDTO();
                result.setSuccess(false);
                ExternalAssetDTO.ErrorInfo error = new ExternalAssetDTO.ErrorInfo();
                error.setCode("REQUEST_FAILED");
                error.setMessage("请求失败: " + response.getStatusCode());
                result.setError(error);
                return result;
            }
            
        } catch (Exception e) {
            log.error("调用第三方素材API失败", e);
            ExternalAssetDTO result = new ExternalAssetDTO();
            result.setSuccess(false);
            ExternalAssetDTO.ErrorInfo error = new ExternalAssetDTO.ErrorInfo();
            error.setCode("EXCEPTION");
            error.setMessage("调用API失败: " + e.getMessage());
            result.setError(error);
            return result;
        }
    }
    
    /**
     * 检查任务素材是否完成
     * @param taskNo 任务编号
     * @return 下载链接（如果完成），否则返回null
     */
    public String checkTaskCompletion(String taskNo) {
        ExternalAssetDTO result = queryAssets(taskNo);
        
        if (result.isSuccess() && result.getData() != null) {
            ExternalAssetDTO.DownloadUrls urls = result.getData().getDownloadUrls();
            if (urls != null && urls.getAll() != null && !urls.getAll().isEmpty()) {
                log.info("任务 {} 素材已完成，下载链接: {}", taskNo, urls.getAll());
                return urls.getAll();
            }
        }
        
        log.info("任务 {} 素材未完成或查询失败", taskNo);
        return null;
    }
    
    /**
     * 测试API连接
     * @return 测试结果
     */
    public boolean testConnection() {
        String apiUrl = systemConfigService.getConfigValue("external_asset_api_url");
        String apiKey = systemConfigService.getConfigValue("external_asset_api_key");
        
        if (apiUrl == null || apiUrl.isEmpty() || apiKey == null || apiKey.isEmpty()) {
            return false;
        }
        
        try {
            // 测试连接（使用一个测试任务号）
            String url = apiUrl.endsWith("/") ? apiUrl : apiUrl + "/";
            url = url + "product-assets?taskNo=TEST000000000000";
            
            HttpHeaders headers = new HttpHeaders();
            headers.set("X-API-Key", apiKey);
            
            HttpEntity<String> entity = new HttpEntity<>(headers);
            
            ResponseEntity<String> response = restTemplate.exchange(
                url,
                HttpMethod.GET,
                entity,
                String.class
            );
            
            // 只要能收到响应（无论是404还是200），说明连接正常
            return response.getStatusCode().is2xxSuccessful() || 
                   response.getStatusCode() == HttpStatus.NOT_FOUND;
                   
        } catch (Exception e) {
            log.error("测试第三方素材API连接失败", e);
            return false;
        }
    }
}