deskflow/src/lib/gui/ipc/DaemonIpcClient.cpp

/*
 * Deskflow -- mouse and keyboard sharing utility
 * SPDX-FileCopyrightText: (C) 2025 Symless Ltd.
 * SPDX-License-Identifier: GPL-2.0-only WITH LicenseRef-OpenSSL-Exception
 */

#include "DaemonIpcClient.h"

#include "common/constants.h"

#include <QDebug>
#include <QLocalSocket>
#include <QObject>
#include <QString>

namespace deskflow::gui::ipc {

const auto kTimeout = 1000;

DaemonIpcClient::DaemonIpcClient(QObject *parent) : QObject(parent), m_socket{new QLocalSocket(this)}
{
}

bool DaemonIpcClient::connectToServer()
{
  qInfo("connecting to daemon ipc");

  m_socket->connectToServer(kDaemonIpcName);
  if (!m_socket->waitForConnected(kTimeout)) {
    qWarning() << "ipc client failed to connect to server:" << kDaemonIpcName;
    return false;
  }

  if (!sendMessage("hello", "hello", false)) {
    qWarning() << "ipc client failed to send hello";
    return false;
  }

  connect(m_socket, &QLocalSocket::disconnected, this, &DaemonIpcClient::handleDisconnected);
  connect(m_socket, &QLocalSocket::errorOccurred, this, &DaemonIpcClient::handleErrorOccurred);

  m_connected = true;
  qInfo() << "ipc client connected to server:" << kDaemonIpcName;
  return true;
}

void DaemonIpcClient::handleDisconnected()
{
  qInfo() << "ipc client disconnected from server";
  m_connected = false;
}

void DaemonIpcClient::handleErrorOccurred()
{
  qCritical() << "ipc client error:" << m_socket->errorString();
  m_connected = false;
}

bool DaemonIpcClient::keepAlive()
{
  if (!isConnected() && !connectToServer()) {
    qWarning() << "ipc client keep alive failed to connect";
    return false;
  }

  if (!sendMessage("noop")) {
    qWarning() << "ipc client keep alive ping failed";
    m_connected = false;
    return false;
  }

  return true;
}

bool DaemonIpcClient::sendLogLevel(const QString &logLevel)
{
  if (!keepAlive())
    return false;

  sendMessage("logLevel=" + logLevel);
  return true;
}

bool DaemonIpcClient::sendCommand(const QString &command, ElevateMode elevateMode)
{
  if (!keepAlive())
    return false;

  sendMessage("elevate=" + (elevateMode == ElevateMode::kAlways ? QStringLiteral("yes") : QStringLiteral("no")));
  sendMessage("command=" + command);
  sendMessage("restart");
  return true;
}

bool DaemonIpcClient::sendMessage(const QString &message, const QString &expectAck, const bool expectConnected)
{
  if (expectConnected && !m_connected) {
    qWarning() << "cannot send command, ipc not connected";
    return false;
  }

  QByteArray messageData = message.toUtf8() + "\n";
  m_socket->write(messageData);
  if (!m_socket->waitForBytesWritten(kTimeout)) {
    qWarning() << "ipc client failed to write command";
    return false;
  }

  if (!m_socket->waitForReadyRead(kTimeout)) {
    qWarning() << "ipc client failed to read response";
    return false;
  }

  QByteArray response = m_socket->readAll();
  if (response.isEmpty()) {
    qWarning() << "ipc client got empty response";
    return false;
  }

  QString responseData = QString::fromUtf8(response);
  if (responseData.isEmpty()) {
    qWarning() << "ipc client failed to convert response to string";
    return false;
  }

  if (responseData != expectAck + "\n") {
    qWarning() << "ipc client got unexpected response: " << responseData;
    return false;
  }

  qDebug() << "ipc client sent message: " << messageData;
  return true;
}

} // namespace deskflow::gui::ipc