aboutsummaryrefslogtreecommitdiff
path: root/src/InstanceManager.cpp
blob: ec7d6e118cf82b4ff0a583d6a51984a216965bb5 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#include <QLocalSocket>
#include <QtCore>
#include <cstdlib>
#include <qnamespace.h>

#include "InstanceManager.hpp"

void InstanceManager::initialize() {
  bool connected = try_server_connection();
  if (!connected) {
    server_init();
  }
}

void InstanceManager::server_init() {
  server = new QLocalServer(this);
  is_server_mode = true;

  connect(server, &QLocalServer::newConnection, this, &InstanceManager::handle_new_connection);

  QLocalServer::removeServer(socket_path);
  if (!server->listen(socket_path)) {
    qCritical() << "Unable to start server:" << server->serverError();
    exit(1); // TODO: Do something else?
  }
}

void InstanceManager::write_message(ServerMsg &msg) {
  if (socket == nullptr)
    return;

  msg.write_to_socket(socket);
  socket->waitForBytesWritten();
}

void InstanceManager::close() {
  socket->waitForBytesWritten();
  socket->close();
}

void InstanceManager::handle_new_connection() {
  if (!server->hasPendingConnections())
    return;

  qDebug() << "New connection";
  auto *client = server->nextPendingConnection();
  if (client == nullptr)
    return;

  connect(client, &QLocalSocket::disconnected, client, &QLocalSocket::deleteLater);

  client->waitForReadyRead();
  while (client->canReadLine()) {
    QString data = client->readLine();

    qDebug() << "RECV:" << data;
    if (data.startsWith("lua ")) {
      auto lua_expr = data.sliced(4);
      emit lua_eval_requested(lua_expr);
    } else {
      QRegularExpression whitespace("\\s+");
      auto urls = data.split(whitespace, Qt::SkipEmptyParts);
      emit urls_open_requested(urls);
    }
  }

  client->close();
}

bool InstanceManager::try_server_connection() {
  socket = new QLocalSocket(this);
  socket->connectToServer(socket_path);

  auto connected = socket->waitForConnected(200);
  if (connected)
    return true;

  socket->close();
  socket->deleteLater();
  socket = nullptr;
  return false;
}

InstanceManager::~InstanceManager() {
  if (socket) {
    socket->disconnectFromServer();
    socket->close();
    socket->deleteLater();
  }
  if (server) {
    server->disconnect();
    server->deleteLater();
  }
}