textautogeneratetextplugin.cpp

/*
  SPDX-FileCopyrightText: 2025 Laurent Montel <montel@kde.org>
 
  SPDX-License-Identifier: GPL-2.0-or-later
*/
 
#include "textautogeneratetextplugin.h"
#include "core/textautogeneratechatmodel.h"
#include "core/textautogeneratehistorymodel.h"
#include "core/textautogeneratemanager.h"
#include "core/textautogeneratemessage.h"
#include "textautogeneratehistoryinfo.h"
#include "textautogeneratetextcore_debug.h"
 
#include <QDateTime>
#include <QUuid>
 
using namespace TextAutogenerateText;
 
class TextAutogenerateText::TextAutogenerateTextPluginPrivate
{
public:
    bool hasError = false;
    bool isReady = false;
};
 
TextAutogenerateTextPlugin::TextAutogenerateTextPlugin(QObject *parent)
    : QObject{parent}
    , d(new TextAutogenerateText::TextAutogenerateTextPluginPrivate)
{
}
 
TextAutogenerateTextPlugin::~TextAutogenerateTextPlugin() = default;
 
void TextAutogenerateTextPlugin::setHasError(bool error)
{
    d->hasError = error;
}
 
bool TextAutogenerateTextPlugin::ready() const
{
    return d->isReady;
}
 
void TextAutogenerateTextPlugin::setReady(bool newReady)
{
    d->isReady = newReady;
}
 
void TextAutogenerateTextPlugin::sendMessage(const QString &str)
{
    if (ready()) {
        // User Message
        TextAutoGenerateMessage msg;
        msg.setSender(TextAutoGenerateMessage::Sender::User);
        msg.setContent(str);
        msg.setDateTime(QDateTime::currentSecsSinceEpoch());
        msg.setUuid(QUuid::createUuid().toByteArray(QUuid::Id128));
 
        // LLM Message
        TextAutoGenerateMessage msgLlm;
        msgLlm.setInProgress(true);
        msgLlm.setSender(TextAutoGenerateMessage::Sender::LLM);
        msgLlm.setDateTime(QDateTime::currentSecsSinceEpoch());
        msgLlm.setUuid(QUuid::createUuid().toByteArray(QUuid::Id128));
 
        msg.setAnswerUuid(msgLlm.uuid());
        TextAutoGenerateHistoryInfo info;
        info.setReferenceUuid(msg.uuid());
        info.setSubject(msg.content());
        info.setDateTime(msg.dateTime());
        // TODO add more
        TextAutogenerateManager::self()->textAutoGenerateHistoryModel()->addInfo(std::move(info));
 
        TextAutogenerateManager::self()->textAutoGenerateChatModel()->addMessage(std::move(msg));
        TextAutogenerateManager::self()->textAutoGenerateChatModel()->addMessage(std::move(msgLlm));
        sendToLLM(str);
 
    } else {
        qCWarning(TEXTAUTOGENERATETEXT_CORE_LOG) << "Plugin is not valid:";
    }
}
 
#include "moc_textautogeneratetextplugin.cpp"