mirror of
https://github.com/rizinorg/cutter.git
synced 2024-12-22 12:56:11 +00:00
86 lines
3.1 KiB
C++
86 lines
3.1 KiB
C++
|
|
#include <QJsonDocument>
|
|
#include <QJsonArray>
|
|
#include <QJsonObject>
|
|
#include <QDebug>
|
|
#include <QThread>
|
|
|
|
#include "JupyterConnection.h"
|
|
|
|
JupyterConnection::JupyterConnection(QObject *parent) : QObject(parent)
|
|
{
|
|
process = nullptr;
|
|
}
|
|
|
|
JupyterConnection::~JupyterConnection()
|
|
{
|
|
cmdServer->stop();
|
|
process->terminate();
|
|
}
|
|
|
|
const char *jupyterPyCode = "import sys\n"
|
|
"from notebook.notebookapp import *\n"
|
|
"\n"
|
|
"\n"
|
|
"class CutterNotebookApp(NotebookApp):\n"
|
|
" def start(self):\n"
|
|
" \"\"\" see NotebookApp.start() \"\"\"\n"
|
|
"\n"
|
|
" super(NotebookApp, self).start()\n"
|
|
"\n"
|
|
" self.write_server_info_file()\n"
|
|
"\n"
|
|
" if self.token and self._token_generated:\n"
|
|
" url = url_concat(self.connection_url, {'token': self.token})\n"
|
|
" sys.stdout.write(url + \"\\n\")\n"
|
|
" sys.stdout.flush()\n"
|
|
"\n"
|
|
" self.io_loop = ioloop.IOLoop.current()\n"
|
|
" if sys.platform.startswith('win'):\n"
|
|
" # add no-op to wake every 5s\n"
|
|
" # to handle signals that may be ignored by the inner loop\n"
|
|
" pc = ioloop.PeriodicCallback(lambda: None, 5000)\n"
|
|
" pc.start()\n"
|
|
" try:\n"
|
|
" self.io_loop.start()\n"
|
|
" except KeyboardInterrupt:\n"
|
|
" self.log.info(_(\"Interrupted...\"))\n"
|
|
" finally:\n"
|
|
" self.remove_server_info_file()\n"
|
|
" self.cleanup_kernels()\n"
|
|
"\n"
|
|
"\n"
|
|
"if __name__ == \"__main__\":\n"
|
|
" CutterNotebookApp.launch_instance()";
|
|
|
|
void JupyterConnection::start()
|
|
{
|
|
process = new QProcess(this);
|
|
connect(process, &QProcess::readyReadStandardError, this, &JupyterConnection::readStandardError);
|
|
connect(process, &QProcess::readyReadStandardOutput, this, &JupyterConnection::readStandardOutput);
|
|
connect(process, &QProcess::errorOccurred, this, [](QProcess::ProcessError error){ qWarning() << "Jupyter error occurred:" << error; });
|
|
process->start("python3", {"-c", jupyterPyCode});
|
|
|
|
QThread *cmdServerThread = new QThread(this);
|
|
cmdServer = new CommandServer();
|
|
cmdServer->moveToThread(cmdServerThread);
|
|
connect(cmdServer, &CommandServer::error, this, [](QString err){ qWarning() << "CmdServer error:" << err; });
|
|
connect(cmdServerThread, SIGNAL (started()), cmdServer, SLOT (process()));
|
|
connect(cmdServer, SIGNAL (finished()), cmdServerThread, SLOT (quit()));
|
|
connect(cmdServer, SIGNAL (finished()), cmdServer, SLOT (deleteLater()));
|
|
connect(cmdServerThread, SIGNAL (finished()), cmdServerThread, SLOT (deleteLater()));
|
|
cmdServerThread->start();
|
|
}
|
|
|
|
void JupyterConnection::readStandardError()
|
|
{
|
|
auto data = process->readAllStandardError();
|
|
printf("Jupyter stderr: %s\n", data.constData());
|
|
}
|
|
|
|
void JupyterConnection::readStandardOutput()
|
|
{
|
|
auto data = process->readAllStandardOutput();
|
|
printf("Jupyter stdout: %s\n", data.constData());
|
|
emit urlReceived(data);
|
|
} |