音乐封面生成回调

from flask import Flask, request, jsonify
import requests
import os
from urllib.parse import urlparse

app = Flask(__name__)

@app.route('/suno-cover-callback', methods=['POST'])
def handle_callback():
    data = request.json
    
    code = data.get('code')
    msg = data.get('msg')
    callback_data = data.get('data', {})
    task_id = callback_data.get('taskId')
    images = callback_data.get('images')
    
    print(f"收到封面生成回调: {task_id}, 状态: {code}, 消息: {msg}")
    
    if code == 200:
        # 任务成功完成
        print("封面生成完成")
        
        if images:
            print("生成的封面图片:")
            for i, image_url in enumerate(images, 1):
                print(f"封面{i}: {image_url}")
            
            # 下载封面图片
            download_images(images, task_id)
            
    else:
        # 任务失败
        print(f"封面生成失败: {msg}")
        
        # 处理失败情况...
    
    # 返回200状态码确认收到回调
    return jsonify({'status': 'received'}), 200

def download_images(image_urls, task_id):
    """下载封面图片"""
    # 创建目录
    dir_path = f"covers/{task_id}"
    os.makedirs(dir_path, exist_ok=True)
    
    for i, url in enumerate(image_urls, 1):
        try:
            # 获取文件扩展名
            parsed_url = urlparse(url)
            file_ext = os.path.splitext(parsed_url.path)[1] or '.png'
            filename = os.path.join(dir_path, f"cover_{i}{file_ext}")
            
            # 下载文件
            response = requests.get(url, stream=True)
            if response.status_code == 200:
                with open(filename, 'wb') as f:
                    for chunk in response.iter_content(chunk_size=8192):
                        f.write(chunk)
                print(f"已保存封面: {filename}")
            else:
                print(f"下载失败 {url}: HTTP {response.status_code}")
                
        except Exception as e:
            print(f"下载失败 {url}: {e}")

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=3000)

/**
 * @param request 回调请求
 * @return ResponseEntity
 * @author zj
 * @description 处理Suno Cover生成回调
 * @date 2025/1/15
 **/
@PostMapping("/suno-cover-callback")
@Operation(summary = "Suno Cover生成回调", description = "接收Cover生成完成通知")
public ResponseEntity<Map<String, String>> handleCoverCallback(
    @RequestBody CoverCallbackRequest request) {
    
    log.info("收到封面生成回调: taskId={}, status={}, message={}", 
        request.getData().getTaskId(), request.getCode(), request.getMsg());
    
    try {
        if (request.getCode() == 200) {
            // 任务成功完成
            log.info("封面生成完成");
            List<String> images = request.getData().getImages();
            
            if (images != null && !images.isEmpty()) {
                log.info("生成的封面图片: {}", images);
                
                // 处理封面图片
                processCoverImages(request.getData().getTaskId(), images);
            }
            
        } else {
            // 任务失败
            log.error("封面生成失败: {}", request.getMsg());
            
            // 处理失败情况
            handleCoverGenerationFailure(request.getData().getTaskId(), 
                request.getCode(), request.getMsg());
        }
        
        // 更新任务状态
        coverTaskService.updateTaskStatus(request.getData().getTaskId(), 
            request.getCode(), request.getMsg(), request.getData().getImages());
            
    } catch (Exception e) {
        log.error("处理封面生成回调失败", e);
        return ResponseEntity.status(500)
            .body(Map.of("status", "error", "message", e.getMessage()));
    }
    
    // 返回200状态码确认收到回调
    return ResponseEntity.ok(Map.of("status", "received"));
}

/**
 * @param taskId 任务ID
 * @param imageUrls 图片URL列表
 * @author zj
 * @description 处理封面图片
 * @date 2025/1/15
 **/
private void processCoverImages(String taskId, List<String> imageUrls) {
    // 异步下载图片
    CompletableFuture.runAsync(() -> {
        try {
            downloadCoverImages(taskId, imageUrls);
        } catch (Exception e) {
            log.error("下载封面图片失败: taskId={}", taskId, e);
        }
    });
}

/**
 * @param taskId 任务ID
 * @param imageUrls 图片URL列表
 * @author zj
 * @description 下载封面图片
 * @date 2025/1/15
 **/
private void downloadCoverImages(String taskId, List<String> imageUrls) {
    String dirPath = "covers/" + taskId;
    File dir = new File(dirPath);
    if (!dir.exists()) {
        dir.mkdirs();
    }
    
    for (int i = 0; i < imageUrls.size(); i++) {
        String url = imageUrls.get(i);
        String filename = dirPath + "/cover_" + (i + 1) + ".png";
        
        try {
            downloadFile(url, filename);
            log.info("已保存封面: {}", filename);
        } catch (Exception e) {
            log.error("下载失败: {}", url, e);
        }
    }
}

@Data
public class CoverCallbackRequest {
    private Integer code;
    private String msg;
    private CoverCallbackData data;
}

@Data
public class CoverCallbackData {
    private String taskId;
    private List<String> images;
}

<?php
header('Content-Type: application/json');

// 获取POST数据
$input = file_get_contents('php://input');
$data = json_decode($input, true);

$code = $data['code'] ?? null;
$msg = $data['msg'] ?? '';
$callbackData = $data['data'] ?? [];
$taskId = $callbackData['taskId'] ?? '';
$images = $callbackData['images'] ?? null;

error_log("收到封面生成回调: $taskId, 状态: $code, 消息: $msg");

if ($code === 200) {
    // 任务成功完成
    error_log("封面生成完成");
    
    if ($images && is_array($images) && count($images) > 0) {
        error_log("生成的封面图片: " . implode(', ', $images));
        
        // 下载封面图片
        downloadCoverImages($taskId, $images);
    }
    
} else {
    // 任务失败
    error_log("封面生成失败: $msg");
    
    // 处理失败情况...
}

/**
 * 下载封面图片
 * @param string $taskId 任务ID
 * @param array $imageUrls 图片URL数组
 */
function downloadCoverImages($taskId, $imageUrls) {
    // 创建目录
    $dir = "covers/$taskId";
    if (!is_dir($dir)) {
        mkdir($dir, 0777, true);
    }
    
    foreach ($imageUrls as $index => $url) {
        $filename = $dir . "/cover_" . ($index + 1) . ".png";
        
        try {
            $imageContent = file_get_contents($url);
            if ($imageContent !== false) {
                file_put_contents($filename, $imageContent);
                error_log("已保存封面: $filename");
            } else {
                error_log("下载失败: $url");
            }
        } catch (Exception $e) {
            error_log("下载失败 $url: " . $e->getMessage());
        }
    }
}

// 返回200状态码确认收到回调
http_response_code(200);
echo json_encode(['status' => 'received']);
?>