// SPDX-FileCopyrightText: 2026 GXDE OS Team
// SPDX-License-Identifier: GPL-3.0-or-later

#pragma once

#include <QJsonArray>
#include <QNetworkAccessManager>
#include <QObject>
#include <QString>
#include <QVector>

struct ChatMessage
{
    QString role; // system / user / assistant
    QString content;
};

struct ToolCall
{
    QString id;
    QString name;
    QString arguments;
};

/**
 * Minimal client for the DeepSeek (OpenAI compatible) chat API.
 *
 * \a send streams (SSE), \a sendOnce performs a non streaming request used for
 * context compression and other housekeeping calls.
 */
class ChatClient : public QObject
{
    Q_OBJECT

public:
    explicit ChatClient(QObject *parent = nullptr);

    bool busy() const { return m_reply != nullptr; }
    void setTools(const QJsonArray &tools);
    void send(const QVector<ChatMessage> &messages, const QJsonArray &extraMessages = QJsonArray());
    void sendOnce(const QVector<ChatMessage> &messages);
    void cancel();

signals:
    void delta(const QString &text);
    void finished(const QString &text);
    void failed(const QString &error);
    void onceFinished(const QString &text);
    void onceFailed(const QString &error);
    void toolCallsReady(const QVector<ToolCall> &calls);

private:
    void handleReadyRead();
    void handleFinished();
    void handlePayload(const QByteArray &payload);

    QNetworkAccessManager m_manager;
    class QNetworkReply *m_reply = nullptr;
    class QNetworkReply *m_onceReply = nullptr;
    QByteArray m_buffer;
    QJsonArray m_tools;
    QVector<ToolCall> m_toolCalls;
    QString m_fullText;
    bool m_done = false;
};