116 lines
4.5 KiB
C++
116 lines
4.5 KiB
C++
#ifndef OLLIE9PCLIENT_H
|
|
#define OLLIE9PCLIENT_H
|
|
|
|
#include <QObject>
|
|
#include <QProcess>
|
|
#include <QVariantList>
|
|
#if QT_VERSION_MAJOR >= 6
|
|
#include <QQmlEngine>
|
|
#endif
|
|
|
|
// 9P-based client for ollie. Uses ollie-9p subprocess for transport.
|
|
// Streaming files (chat, statewait) use persistent QProcess instances.
|
|
class Ollie9pClient : public QObject
|
|
{
|
|
Q_OBJECT
|
|
#if QT_VERSION_MAJOR >= 6
|
|
QML_ELEMENT
|
|
QML_SINGLETON
|
|
#endif
|
|
|
|
Q_PROPERTY(QVariantList sessions READ sessions NOTIFY sessionsChanged)
|
|
Q_PROPERTY(QString activeSessionId READ activeSessionId WRITE setActiveSessionId NOTIFY activeSessionIdChanged)
|
|
Q_PROPERTY(QString activeAgentId READ activeAgentId NOTIFY activeAgentIdChanged)
|
|
Q_PROPERTY(QString activeState READ activeState NOTIFY activeStateChanged)
|
|
Q_PROPERTY(bool rootBackendsLoaded READ rootBackendsLoaded NOTIFY rootBackendsLoadedChanged)
|
|
Q_PROPERTY(QStringList availableBackends READ availableBackends NOTIFY availableBackendsChanged)
|
|
Q_PROPERTY(bool rootAgentsLoaded READ rootAgentsLoaded NOTIFY rootAgentsLoadedChanged)
|
|
Q_PROPERTY(QStringList availableAgents READ availableAgents NOTIFY availableAgentsChanged)
|
|
Q_PROPERTY(bool rootModelsLoaded READ rootModelsLoaded NOTIFY rootModelsLoadedChanged)
|
|
Q_PROPERTY(QString currentBackend READ currentBackend WRITE setCurrentBackend NOTIFY currentBackendChanged)
|
|
|
|
public:
|
|
explicit Ollie9pClient(QObject *parent = nullptr);
|
|
~Ollie9pClient();
|
|
|
|
QVariantList sessions() const { return m_sessions; }
|
|
QString activeSessionId() const { return m_activeSessionId; }
|
|
void setActiveSessionId(const QString &id);
|
|
QString activeAgentId() const { return m_agentId; }
|
|
QString activeState() const { return m_activeState; }
|
|
|
|
bool rootBackendsLoaded() const { return m_rootBackendsLoaded; }
|
|
QStringList availableBackends() const { return m_availableBackends; }
|
|
bool rootAgentsLoaded() const { return m_rootAgentsLoaded; }
|
|
QStringList availableAgents() const { return m_availableAgents; }
|
|
bool rootModelsLoaded() const { return m_rootModelsLoaded; }
|
|
QString currentBackend() const { return m_currentBackend; }
|
|
Q_INVOKABLE void setCurrentBackend(const QString &backend) {
|
|
if (m_currentBackend == backend) return;
|
|
m_currentBackend = backend;
|
|
emit currentBackendChanged();
|
|
}
|
|
|
|
Q_INVOKABLE void refreshSessions();
|
|
Q_INVOKABLE QString readLog();
|
|
Q_INVOKABLE QString readLogForSession(const QString &sessionId, const QString &agentId);
|
|
Q_INVOKABLE bool submit(const QString &prompt);
|
|
Q_INVOKABLE bool interrupt();
|
|
Q_INVOKABLE bool kill();
|
|
Q_INVOKABLE QString getConfig();
|
|
Q_INVOKABLE QStringList getAgents(const QString &sessionId);
|
|
Q_INVOKABLE void setActiveAgentId(const QString &agentId);
|
|
Q_INVOKABLE void switchAgent(const QString &sessionId, const QString &agentId);
|
|
Q_INVOKABLE void loadRootBackends();
|
|
Q_INVOKABLE void loadRootAgents();
|
|
Q_INVOKABLE QStringList getAvailableModels(const QString &backend) const;
|
|
Q_INVOKABLE bool createSession(const QString &cwd, const QString &name, const QString &backend, const QString &model, const QString &agent, const QString &remote);
|
|
|
|
signals:
|
|
void sessionsChanged();
|
|
void activeSessionIdChanged();
|
|
void activeAgentIdChanged();
|
|
void activeStateChanged();
|
|
void chatReceived(const QString &text);
|
|
void streamingDone(); // chat process exited (session killed)
|
|
void rootBackendsLoadedChanged();
|
|
void availableBackendsChanged();
|
|
void rootAgentsLoadedChanged();
|
|
void availableAgentsChanged();
|
|
void rootModelsLoadedChanged();
|
|
void currentBackendChanged();
|
|
|
|
private slots:
|
|
void onChatReadyRead();
|
|
void onStateReadyRead();
|
|
void onChatFinished(int exitCode, QProcess::ExitStatus status);
|
|
void onStateFinished(int exitCode, QProcess::ExitStatus status);
|
|
|
|
private:
|
|
void startChatStream();
|
|
void startStateStream();
|
|
void stopStreams();
|
|
QString agentPath() const;
|
|
QByteArray run9p(const QStringList &args);
|
|
void ensureRootDataLoaded();
|
|
|
|
QVariantList m_sessions;
|
|
QString m_activeSessionId;
|
|
QString m_activeState = "idle";
|
|
QString m_agentId;
|
|
|
|
// Root data cache
|
|
bool m_rootBackendsLoaded = false;
|
|
QStringList m_availableBackends;
|
|
bool m_rootAgentsLoaded = false;
|
|
QStringList m_availableAgents;
|
|
bool m_rootModelsLoaded = false;
|
|
QVariantMap m_rootModels;
|
|
QString m_currentBackend;
|
|
|
|
QProcess *m_chatProc = nullptr;
|
|
QProcess *m_stateProc = nullptr;
|
|
};
|
|
|
|
#endif // OLLIE9PCLIENT_H
|