package com.dsideal.base.Util;

import cn.hutool.core.io.FileUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson.JSONArray;

import java.io.File;
import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpResponse;
import java.nio.charset.StandardCharsets;
import java.util.concurrent.CompletableFuture;
import java.util.stream.Stream;

public class CallDeepSeek {
    private static final String API_KEY = "sk-44ae895eeb614aa1a9c6460579e322f1"; // 请替换为您的API KEY
    private static final String API_URL = "https://api.deepseek.com/v1/chat/completions";

    /**
     * 通用的DeepSeek API调用方法（流式响应）
     *
     * @param prompt   用户提示词
     * @param listener 响应监听器
     */
    public static void callDeepSeekStream(String prompt, SSEListener listener) {
        callDeepSeekStream(prompt, listener, null, false);
    }

    /**
     * 通用的DeepSeek API调用方法（流式响应）
     *
     * @param prompt     用户提示词
     * @param listener   响应监听器
     * @param outputPath 输出文件路径（可选）
     * @param saveToFile 是否保存到文件
     */
    public static void callDeepSeekStream(String prompt, SSEListener listener, String outputPath, boolean saveToFile) {
        new Thread(() -> {
            StringBuilder fullResponse = new StringBuilder();
            try {
                System.out.println("开始调用DeepSeek API...");
                JSONObject jsonPayload = createRequestPayload(prompt);
                System.out.println("请求载荷: " + jsonPayload.toString());
                
                HttpClient client = createHttpClient();
                java.net.http.HttpRequest request = createHttpRequest(jsonPayload);
                System.out.println("发送HTTP请求到: " + API_URL);

                CompletableFuture<Void> future = client.sendAsync(request, HttpResponse.BodyHandlers.ofLines())
                        .thenAccept(response -> {
                            System.out.println("收到响应，状态码: " + response.statusCode());
                            handleStreamResponse(response, fullResponse, listener, outputPath, saveToFile);
                        }).exceptionally(e -> {
                            System.err.println("请求异常: " + e.getMessage());
                            listener.onError("请求或处理异常: " + e.getMessage());
                            e.printStackTrace();
                            return null;
                        });
                future.join();

            } catch (Exception e) {
                System.err.println("发生意外错误: " + e.getMessage());
                listener.onError("发生意外错误: " + e.getMessage());
                e.printStackTrace();
            }
        }).start();
    }


    /**
     * 创建HTTP请求载荷
     */
    private static JSONObject createRequestPayload(String prompt) {
        JSONObject jsonPayload = new JSONObject();
        jsonPayload.set("model", "deepseek-chat");
        JSONObject message = new JSONObject();
        message.set("role", "user");
        message.set("content", prompt);
        JSONArray messages = new JSONArray();
        messages.add(message);
        jsonPayload.set("messages", messages);
        jsonPayload.set("stream", true);
        return jsonPayload;
    }

    /**
     * 创建HTTP客户端
     */
    private static HttpClient createHttpClient() {
        return HttpClient.newBuilder()
                .version(HttpClient.Version.HTTP_1_1)
                .build();
    }

    /**
     * 创建HTTP请求
     */
    private static java.net.http.HttpRequest createHttpRequest(JSONObject jsonPayload) {
        return java.net.http.HttpRequest.newBuilder()
                .uri(URI.create(API_URL))
                .header("Content-Type", "application/json")
                .header("Authorization", "Bearer " + API_KEY)
                .header("Accept", "text/event-stream")
                .POST(java.net.http.HttpRequest.BodyPublishers.ofString(jsonPayload.toString(), StandardCharsets.UTF_8))
                .build();
    }

    /**
     * 处理流式响应
     */
    private static void handleStreamResponse(HttpResponse<Stream<String>> response,
                                             StringBuilder fullResponse,
                                             SSEListener listener,
                                             String outputPath,
                                             boolean saveToFile) {
        System.out.println("处理流式响应，状态码: " + response.statusCode());
        if (response.statusCode() == 200) {
            System.out.println("开始处理SSE数据流...");
            response.body().forEach(line -> {
                System.out.println("收到SSE行: " + line);
                if (line.startsWith("data:")) {
                    String data = line.substring(5).trim();
                    System.out.println("解析SSE数据: " + data);
                    if (!data.equals("[DONE]")) {
                        try {
                            JSONObject jsonData = JSONUtil.parseObj(data);
                            if (jsonData.containsKey("choices")) {
                                String content = jsonData.getJSONArray("choices")
                                        .getJSONObject(0)
                                        .getJSONObject("delta")
                                        .getStr("content", "");
                                if (content != null && !content.isEmpty()) {
                                    System.out.println("提取到内容: " + content);
                                    fullResponse.append(content);
                                    listener.onData(content);
                                }
                            }
                        } catch (Exception e) {
                            System.err.println("解析SSE JSON数据错误: " + data + " \nError: " + e.getMessage());
                            e.printStackTrace();
                        }
                    } else {
                        System.out.println("收到结束标记 [DONE]");
                    }
                }
            });

            // 流结束后的处理
            String responseContent = fullResponse.toString();
            System.out.println("流处理完成，总内容长度: " + responseContent.length());
            if (saveToFile && outputPath != null) {
                FileUtil.writeString(responseContent, new File(outputPath), "UTF-8");
                listener.onComplete("内容已成功保存到" + outputPath);
            } else {
                listener.onComplete(responseContent);
            }
        } else {
            System.err.println("API请求失败，状态码: " + response.statusCode());
            listener.onError("API请求失败: " + response.statusCode() + " Body: " + response.body().toString());
        }
    }


    public static void main(String[] args) {
        String prompt = "你好，你是谁?";

        // 使用通用方法
        callDeepSeekStream(prompt, new SSEListener() {
            @Override
            public void onData(String data) {
                System.out.print(data);
                System.out.flush();
            }

            @Override
            public void onComplete(String fullResponse) {
                System.out.println("\n\n完整回复: " + fullResponse);
            }

            @Override
            public void onError(String error) {
                System.err.println("错误: " + error);
            }
        });
    }

    public interface SSEListener {
        void onData(String data);

        void onComplete(String fullResponse);

        void onError(String error);
    }

    /**
     * 同步调用DeepSeek API（非流式响应）
     * 
     * @param prompt 用户提示词
     * @return DeepSeek的完整响应内容
     */
    public static String callDeepSeek(String prompt) {
        try {
            JSONObject jsonPayload = createRequestPayload(prompt);
            // 设置为非流式响应
            jsonPayload.set("stream", false);
            
            HttpClient client = createHttpClient();
            java.net.http.HttpRequest request = createHttpRequest(jsonPayload);
            
            HttpResponse<String> response = client.send(request, HttpResponse.BodyHandlers.ofString());
            
            if (response.statusCode() == 200) {
                JSONObject responseJson = JSONUtil.parseObj(response.body());
                if (responseJson.containsKey("choices")) {
                    return responseJson.getJSONArray("choices")
                            .getJSONObject(0)
                            .getJSONObject("message")
                            .getStr("content", "");
                }
            } else {
                System.err.println("DeepSeek API调用失败: " + response.statusCode() + " Body: " + response.body());
                return null;
            }
        } catch (Exception e) {
            System.err.println("调用DeepSeek API时出错: " + e.getMessage());
            e.printStackTrace();
            return null;
        }
        return null;
    }
}