FreeKill/src/main.cpp

// SPDX-License-Identifier: GPL-3.0-or-later

#include "client.h"
#include "util.h"
using namespace fkShell;

#include "packman.h"
#ifndef Q_OS_WASM
#include "server.h"
#else
#include <emscripten.h>
#endif

#if defined(Q_OS_LINUX) && !defined(Q_OS_ANDROID)
#include "shell.h"
#endif

#if defined(Q_OS_WIN32)
#include "applink.c"
#endif

#ifndef FK_SERVER_ONLY
#include <QFileDialog>
#include <QScreen>
#include <QSplashScreen>
#ifndef Q_OS_ANDROID
#include <QQuickStyle>
#endif
#include "qmlbackend.h"
#endif

#if defined(Q_OS_ANDROID) || defined(Q_OS_WASM)
static bool copyPath(const QString &srcFilePath, const QString &tgtFilePath) {
  QFileInfo srcFileInfo(srcFilePath);
  if (srcFileInfo.isDir()) {
    QDir targetDir(tgtFilePath);
    if (!targetDir.exists()) {
      targetDir.cdUp();
      if (!targetDir.mkdir(QFileInfo(tgtFilePath).fileName()))
        return false;
    }
    QDir sourceDir(srcFilePath);
    QStringList fileNames =
        sourceDir.entryList(QDir::Files | QDir::Dirs | QDir::NoDotAndDotDot |
                            QDir::Hidden | QDir::System);
    foreach (const QString &fileName, fileNames) {
      const QString newSrcFilePath = srcFilePath + QLatin1Char('/') + fileName;
      const QString newTgtFilePath = tgtFilePath + QLatin1Char('/') + fileName;
      if (!copyPath(newSrcFilePath, newTgtFilePath))
        return false;
    }
  } else {
    QFile::remove(tgtFilePath);
    if (!QFile::copy(srcFilePath, tgtFilePath))
      return false;
  }
  return true;
}
#endif

static void installFkAssets(const QString &src, const QString &dest) {
  QFile f(dest + "/fk_ver");
  if (f.exists() && f.open(QIODevice::ReadOnly)) {
    auto ver = f.readLine().simplified();
    if (ver == FK_VERSION) {
      return;
    }
  }
#ifdef Q_OS_ANDROID
  copyPath(src, dest);
#elif defined(Q_OS_LINUX)
  system(QString("cp -r %1 %2/..").arg(src).arg(dest).toUtf8());
#endif
}

#if defined(Q_OS_LINUX) && !defined(Q_OS_ANDROID)
#include <stdlib.h>
#include <unistd.h>
static void prepareForLinux() {
  // 如果用户执行的是 /usr/bin/FreeKill，那么这意味着 freekill 是被包管理器安装
  // 的，所以我们就需要把资源文件都复制到 ~/.local 中，并且切换当前目录
  // TODO: AppImage
  char buf[256] = {0};
  int len = readlink("/proc/self/exe", buf, 256);
  const char *home = getenv("HOME");
  if (!strcmp(buf, "/usr/bin/FreeKill")) {
    system("mkdir -p ~/.local/share/FreeKill");
    installFkAssets("/usr/share/FreeKill", QString("%1/.local/share/FreeKill").arg(home));
    chdir(home);
    chdir(".local/share/FreeKill");
  } else if (!strcmp(buf, "/usr/local/bin/FreeKill")) {
    system("mkdir -p ~/.local/share/FreeKill");
    installFkAssets("/usr/local/share/FreeKill", QString("%1/.local/share/FreeKill").arg(home));
    chdir(home);
    chdir(".local/share/FreeKill");
  }
}
#endif

static FILE *info_log = nullptr;
static FILE *err_log = nullptr;

void fkMsgHandler(QtMsgType type, const QMessageLogContext &context,
                  const QString &msg) {
  auto date = QDate::currentDate();

  FILE *file;
  switch (type) {
  case QtDebugMsg:
  case QtInfoMsg:
    file = info_log;
    break;
  case QtWarningMsg:
  case QtCriticalMsg:
  case QtFatalMsg:
    file = err_log;
    break;
  }

  fprintf(stderr, "\r%02d/%02d ", date.month(), date.day());
  fprintf(stderr, "%s ",
          QTime::currentTime().toString("hh:mm:ss").toLatin1().constData());
  fprintf(file, "\r%02d/%02d ", date.month(), date.day());
  fprintf(file, "%s ",
          QTime::currentTime().toString("hh:mm:ss").toLatin1().constData());

  auto localMsg = msg.toUtf8();
  auto threadName = QThread::currentThread()->objectName().toLatin1();

  switch (type) {
  case QtDebugMsg:
    fprintf(stderr, "%s[D] %s\n", threadName.constData(),
            localMsg.constData());
    fprintf(file, "%s[D] %s\n", threadName.constData(),
            localMsg.constData());
    break;
  case QtInfoMsg:
    fprintf(stderr, "%s[%s] %s\n", threadName.constData(),
            Color("I", Green).toUtf8().constData(), localMsg.constData());
    fprintf(file, "%s[%s] %s\n", threadName.constData(),
            "I", localMsg.constData());
    break;
  case QtWarningMsg:
    fprintf(stderr, "%s[%s] %s\n", threadName.constData(),
            Color("W", Yellow, Bold).toUtf8().constData(),
            localMsg.constData());
    fprintf(file, "%s[%s] %s\n", threadName.constData(),
            "W", localMsg.constData());
    break;
  case QtCriticalMsg:
    fprintf(stderr, "%s[%s] %s\n", threadName.constData(),
            Color("C", Red, Bold).toUtf8().constData(), localMsg.constData());
    fprintf(file, "%s[%s] %s\n", threadName.constData(),
            "C", localMsg.constData());
#ifndef FK_SERVER_ONLY
    if (Backend != nullptr) {
      Backend->notifyUI(
          "ErrorDialog",
          QString("⛔ %1/Error occured!\n  %2").arg(threadName).arg(localMsg));
    }
#endif
    break;
  case QtFatalMsg:
    fprintf(stderr, "%s[%s] %s\n", threadName.constData(),
            Color("E", Red, Bold).toUtf8().constData(), localMsg.constData());
    fprintf(file, "%s[%s] %s\n", threadName.constData(),
            "E", localMsg.constData());
    break;
  }
}

// FreeKill 的程序主入口。整个程序就是从这里开始执行的。
int main(int argc, char *argv[]) {
  // 初始化一下各种杂项信息
  QThread::currentThread()->setObjectName("Main");
  if (!info_log) {
    info_log = fopen("freekill.server.info.log", "w+");
    if (!info_log) {
      qFatal("Cannot open info.log");
    }
  }
  if (!err_log) {
    err_log = fopen("freekill.server.error.log", "w+");
    if (!err_log) {
      qFatal("Cannot open error.log");
    }
  }

  qInstallMessageHandler(fkMsgHandler);
  QCoreApplication *app;
  QCoreApplication::setApplicationName("FreeKill");
  QCoreApplication::setApplicationVersion(FK_VERSION);

  QLocale l = QLocale::system();
  auto localeName = l.name();

#if defined(Q_OS_LINUX) && !defined(Q_OS_ANDROID)
  prepareForLinux();
#endif

#ifndef FK_CLIENT_ONLY
  // 分析命令行，如果有 -s 或者 --server 就在命令行直接开服务器
  QCommandLineParser parser;
  parser.setApplicationDescription("FreeKill server");
  parser.addVersionOption();
  parser.addOption({{"s", "server"}, "start server at <port>", "port"});
  parser.addOption({{"h", "help"}, "display help information"});
  QStringList cliOptions;
  for (int i = 0; i < argc; i++)
    cliOptions << argv[i];

  parser.parse(cliOptions);
  if (parser.isSet("version")) {
    parser.showVersion();
    return 0;
  } else if (parser.isSet("help")) {
    parser.showHelp();
    return 0;
  }

  bool startServer = parser.isSet("server");
  ushort serverPort = 9527;

  if (startServer) {
    app = new QCoreApplication(argc, argv);
    QTranslator translator;
    Q_UNUSED(translator.load("zh_CN.qm"));
    QCoreApplication::installTranslator(&translator);

    bool ok = false;
    if (parser.value("server").toInt(&ok) && ok)
      serverPort = parser.value("server").toInt();

    Pacman = new PackMan;
    Server *server = new Server;
    if (!server->listen(QHostAddress::Any, serverPort)) {
      qFatal("cannot listen on port %d!\n", serverPort);
      app->exit(1);
    } else {
      qInfo("Server is listening on port %d", serverPort);
#if defined(Q_OS_LINUX) && !defined(Q_OS_ANDROID)
      // Linux 服务器的话可以启用一个 Shell 来操作服务器。
      auto shell = new Shell;
      shell->start();
#endif
    }
    return app->exec();
  }
#endif

#ifdef FK_SERVER_ONLY
  // 根本没编译 GUI 相关的功能，直接在此退出
  qFatal("This is server-only build and have no GUI support.\n\
      Please use ./FreeKill -s to start a server in command line.");
#else

#ifdef Q_OS_WASM
  EM_ASM (
    FS.mkdir('/assets');
    FS.mount(IDBFS, {}, '/assets');
    FS.chdir('/assets');
    FS.syncfs(true, function(err) {
    });
  );
  copyPath(":/", QDir::currentPath());
#endif

  app = new QApplication(argc, argv);
#ifdef DESKTOP_BUILD
  ((QApplication *)app)->setWindowIcon(QIcon("image/icon.png"));
#endif

#define SHOW_SPLASH_MSG(msg)                                                   \
  splash.showMessage(msg, Qt::AlignHCenter | Qt::AlignBottom);

#ifdef Q_OS_ANDROID
  // 安卓：先切换到我们安装程序的那个外部存储目录去
  QJniObject::callStaticMethod<void>("org/notify/FreeKill/Helper", "InitView",
                                     "()V");
  QDir::setCurrent(
      "/storage/emulated/0/Android/data/org.notify.FreeKill/files");

  // 然后显示欢迎界面，并在需要时复制资源素材等
  QScreen *screen = qobject_cast<QApplication *>(app)->primaryScreen();
  QRect screenGeometry = screen->geometry();
  int screenWidth = screenGeometry.width();
  int screenHeight = screenGeometry.height();
  QSplashScreen splash(QPixmap("assets:/res/image/splash.jpg")
                           .scaled(screenWidth, screenHeight));
  splash.showFullScreen();
  SHOW_SPLASH_MSG("Copying resources...");
  installFkAssets("assets:/res", QDir::currentPath());
#else
  // 不是安卓，那么直接启动欢迎界面，也就是不复制东西了
  QSplashScreen splash(QPixmap("image/splash.jpg"));
  splash.show();
#endif

  SHOW_SPLASH_MSG("Loading qml files...");
  QQmlApplicationEngine *engine = new QQmlApplicationEngine;
#ifndef Q_OS_ANDROID
  QQuickStyle::setStyle("Material");
#endif

  QTranslator translator;
  Q_UNUSED(translator.load("zh_CN.qm"));
  QCoreApplication::installTranslator(&translator);

  QmlBackend backend;
  backend.setEngine(engine);

  Pacman = new PackMan;

  // 向 Qml 中先定义几个全局变量
  engine->rootContext()->setContextProperty("FkVersion", FK_VERSION);
  engine->rootContext()->setContextProperty("Backend", &backend);
  engine->rootContext()->setContextProperty("ModBackend", nullptr);
  engine->rootContext()->setContextProperty("Pacman", Pacman);
  engine->rootContext()->setContextProperty("SysLocale", localeName);

#ifdef QT_DEBUG
  bool debugging = true;
#else
  bool debugging = false;
#endif
  engine->rootContext()->setContextProperty("Debugging", debugging);

  QString system;
#if defined(Q_OS_ANDROID)
  system = "Android";
#elif defined(Q_OS_WASM)
  system = "Web";
  engine->rootContext()->setContextProperty("ServerAddr", "127.0.0.1:9527");
#elif defined(Q_OS_WIN32)
  system = "Win";
  ::system("chcp 65001");
#elif defined(Q_OS_LINUX)
  system = "Linux";
#else
  system = "Other";
#endif
  engine->rootContext()->setContextProperty("OS", system);

  engine->rootContext()->setContextProperty(
      "AppPath", QUrl::fromLocalFile(QDir::currentPath()));

  engine->addImportPath(QDir::currentPath());

  // 加载完全局变量后，就再去加载 main.qml，此时UI界面正式显示
  engine->load("Fk/main.qml");

  // qml 报错了就直接退出吧
  if (engine->rootObjects().isEmpty())
    return -1;

  // 关闭欢迎界面，然后进入Qt主循环
  splash.close();
  int ret = app->exec();

  // 先删除 engine
  // 防止报一堆错 "TypeError: Cannot read property 'xxx' of null"
  delete engine;
  delete Pacman;

#ifdef Q_OS_WASM
  EM_ASM (
    FS.syncfs(function(err) {});
  );
#endif

  if (info_log) {
    fclose(info_log);
    info_log = nullptr;
  }
  if (err_log) {
    fclose(err_log);
    info_log = nullptr;
  }

  return ret;
#endif
}