Implement Stream Dialog

2022-07-09 22:07:14 +02:00
parent 81317fd43d
commit 67087e8251
10 changed files with 394 additions and 140 deletions
--- a/src/discordpage.cpp
+++ b/src/discordpage.cpp
@@ -0,0 +1,119 @@
+#include "discordpage.h"
+#include "virtmic.h"
+
+#include <QApplication>
+#include <QDesktopServices>
+#include <QFile>
+#include <QTimer>
+#include <QWebChannel>
+#include <QWebEngineScript>
+#include <QWebEngineScriptCollection>
+#include <QWebEngineSettings>
+
+DiscordPage::DiscordPage(QWidget *parent) : QWebEnginePage(parent) {
+  setBackgroundColor(QColor("#202225"));
+  m_virtmicProcess.setProcessChannelMode(QProcess::ForwardedChannels);
+
+  connect(this, &QWebEnginePage::featurePermissionRequested, this,
+          &DiscordPage::featurePermissionRequested);
+
+  settings()->setAttribute(QWebEngineSettings::ScreenCaptureEnabled, true);
+  settings()->setAttribute(QWebEngineSettings::JavascriptCanOpenWindows, true);
+  settings()->setAttribute(QWebEngineSettings::AllowRunningInsecureContent,
+                           true);
+  settings()->setAttribute(
+      QWebEngineSettings::AllowWindowActivationFromJavaScript, true);
+  settings()->setAttribute(QWebEngineSettings::FullScreenSupportEnabled, true);
+  settings()->setAttribute(QWebEngineSettings::PlaybackRequiresUserGesture,
+                           false);
+
+  setUrl(QUrl("https://discord.com/app"));
+
+  injectScript(":/assets/userscript.js");
+
+  connect(&m_streamDialog, &StreamDialog::requestedStreamStart, this,
+          &DiscordPage::startStream);
+}
+
+void DiscordPage::injectScript(QString source) {
+  qDebug() << "[main   ] Injecting " << source;
+
+  QFile userscript(source);
+
+  if (!userscript.open(QIODevice::ReadOnly)) {
+    qFatal("Failed to load %s with error: %s", source.toLatin1().constData(),
+           userscript.errorString().toLatin1().constData());
+  } else {
+    QByteArray userscriptJs = userscript.readAll();
+
+    QWebEngineScript script;
+
+    script.setSourceCode(userscriptJs);
+    script.setName("userscript.js");
+    script.setWorldId(QWebEngineScript::MainWorld);
+    script.setInjectionPoint(QWebEngineScript::DocumentCreation);
+    script.setRunsOnSubFrames(false);
+
+    scripts().insert(script);
+  }
+}
+
+void DiscordPage::featurePermissionRequested(const QUrl &securityOrigin,
+                                             QWebEnginePage::Feature feature) {
+  // Allow every permission asked
+  setFeaturePermission(securityOrigin, feature,
+                       QWebEnginePage::PermissionGrantedByUser);
+}
+
+bool DiscordPage::acceptNavigationRequest(const QUrl &url,
+                                          QWebEnginePage::NavigationType type,
+                                          bool isMainFrame) {
+  qDebug() << url;
+  if (type == QWebEnginePage::NavigationTypeLinkClicked) {
+    QDesktopServices::openUrl(url);
+    return false;
+  }
+  return true;
+};
+
+void DiscordPage::stopVirtmic() {
+  if (m_virtmicProcess.state() == QProcess::Running) {
+    qDebug() << "[virtmic] Stopping Virtmic";
+    m_virtmicProcess.kill();
+  }
+}
+
+void DiscordPage::startVirtmic(QString target) {
+  if (target != "") {
+    qDebug() << "[virtmic] Starting Virtmic with target" << target;
+    m_virtmicProcess.start(QApplication::arguments()[0], {"--virtmic", target});
+  }
+}
+
+void DiscordPage::javaScriptConsoleMessage(
+    QWebEnginePage::JavaScriptConsoleMessageLevel level, const QString &message,
+    int lineNumber, const QString &sourceID) {
+  if (message == "!discord-screenaudio-start-stream") {
+    if (m_streamDialog.isHidden())
+      m_streamDialog.setHidden(false);
+    else
+      m_streamDialog.activateWindow();
+  } else if (message == "!discord-screenaudio-stream-stopped") {
+    stopVirtmic();
+  } else {
+    qDebug() << "[discord]" << message;
+  }
+}
+
+void DiscordPage::startStream(QString target, uint width, uint height,
+                              uint frameRate) {
+  stopVirtmic();
+  startVirtmic(target);
+  // Wait a bit for the virtmic to start
+  QTimer::singleShot(target == "" ? 0 : 200, [=]() {
+    runJavaScript(QString("window.discordScreenaudioStartStream(%1, %2, %3);")
+                      .arg(width)
+                      .arg(height)
+                      .arg(frameRate));
+  });
+}