package deepseek;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import okhttp3.*;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeUnit;

public class DeepSeekUtils {
    static Gson gson;

    public static Gson getGson() {
        if (gson != null) {
            return gson;
        }
        Gson gson = new GsonBuilder()
                .serializeNulls()
                .create();
        return gson;
    }

    /**
     * 配置参数，调用接口
     *
     * @param args
     * @throws IOException
     */
    public static void main(String[] args) throws IOException {
        String api = "sk-5bebad68c4c34e79b38bd621e371d3b0";
        int timeout = 120;
        long t1 = System.currentTimeMillis();

        OkHttpClient client = new OkHttpClient().newBuilder()
                .connectTimeout(timeout, TimeUnit.SECONDS)
                .readTimeout(timeout, TimeUnit.SECONDS)
                .writeTimeout(timeout, TimeUnit.SECONDS)
                .callTimeout(timeout, TimeUnit.SECONDS)
                .retryOnConnectionFailure(true)
                .connectionPool(new ConnectionPool(10, 5, TimeUnit.MINUTES))
                .build();
        MediaType mediaType = MediaType.parse("application/json");
        DpRequestBean dpRequestBean = new DpRequestBean();
        List<DpRequestBean.MessagesBean> messages = new ArrayList<>();
        DpRequestBean.MessagesBean messagesBean = new DpRequestBean.MessagesBean();
        messagesBean.setContent("介绍一下你");
        messagesBean.setRole("user");

        messages.add(messagesBean);
        dpRequestBean.setMessages(messages);
        dpRequestBean.setModel("deepseek-chat");

        DpRequestBean.ResponseFormatBean responseFormatBean = new DpRequestBean.ResponseFormatBean();
        responseFormatBean.setType("text");
        dpRequestBean.setResponse_format(responseFormatBean);
        dpRequestBean.setTool_choice("none");
        dpRequestBean.setTemperature(1);
        dpRequestBean.setTop_p(1);
        dpRequestBean.setMax_tokens(2048);
        dpRequestBean.setFrequency_penalty(0);
        dpRequestBean.setStream(true);
        dpRequestBean.setStop(null);
        dpRequestBean.setTools(null);
        dpRequestBean.setLogprobs(false);
        dpRequestBean.setTop_logprobs(null);
        dpRequestBean.setStream_options(null);
        dpRequestBean.setTop_logprobs(null);
        // 创建 GsonBuilder 并配置为不忽略 null 字段

        String gsonJson = getGson().toJson(dpRequestBean);
        System.out.println(gsonJson);
        System.out.println("开始请求--------------------------------------");
        RequestBody body = RequestBody.create(mediaType, gsonJson);
        Request request = new Request.Builder()
                .url("https://api.deepseek.com/chat/completions")
                .method("POST", body)
                .addHeader("Content-Type", "application/json")
                .addHeader("Accept", "application/json")
                .addHeader("Authorization", "Bearer " + api + "")
                .build();
        client.newCall(request).enqueue(new Callback() {
            @Override
            public void onFailure(Call call, IOException e) {
                System.err.println("请求失败: " + e.getMessage());
            }

            @Override
            public void onResponse(Call call, Response response) {
                if (!response.isSuccessful()) {
                    System.err.println("响应失败: " + response.code());
                    return;
                }
                long t2 = System.currentTimeMillis();
                System.out.println("耗时：" + (t2 - t1) + "ms");
                System.err.println("响应成功: " + response.code());
                StringBuffer stringBuffer = new StringBuffer();
                try {
                    ResponseBody responseBody = response.body();
                    if (responseBody == null) {
                        return;
                    }
                    // 处理流式响应
                    okio.BufferedSource source = responseBody.source();
                    while (!source.exhausted()) {
                        String line = source.readUtf8Line();
                        // 处理每一行响应数据
//                            System.out.println(line);
                        if (line == null) {
                            continue;
                        }
                        if (!line.contains("data:")) {
                            continue;
                        }
                        String[] sp = line.split("data:");
                        if (sp == null || sp.length <= 1) {
                            continue;
                        }
                        String json = sp[1];
                        if (json == null || json.isEmpty()) {
                            continue;
                        }
                        //解析 JSON 数据
                        DpResponseStreamBean streamBean = getGson().fromJson(json, DpResponseStreamBean.class);
                        stringBuffer.append(streamBean.getChoices().get(0).getDelta().getContent());
                        System.err.println(stringBuffer);
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                } finally {
                    System.err.println("----end-------");
                    long t3 = System.currentTimeMillis();
                    System.out.println("耗时：" + (t3 - t1) + "ms");
                }
            }
        });
    }

}