X-Git-Url: https://gitweb.ps.run/autorec/blobdiff_plain/e00ad6704b8538b1a553960531af7b50d367fe73..0c40371e1f68c7e7020867b25c85521d947b3f47:/src/main.cpp

diff --git a/src/main.cpp b/src/main.cpp
index 7366104..5e6eba3 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -8,6 +8,10 @@
 #define LAY_IMPLEMENTATION
 #include "layout.h"
 
+#include <fstream>
+#include <vector>
+#include <string>
+
 #include <Psapi.h>
 #include <stdbool.h>
 #include <stdio.h>
@@ -15,16 +19,53 @@
 #include <windows.h>
 #include <shlwapi.h>
 
+bool recording = false;
+HANDLE process = NULL;
+HWND hwnd;
+HINSTANCE hInstance;
+HICON icon_white, icon_green, icon_red;
+
+std::vector<std::string> gameExes;
+
+
+NOTIFYICONDATAA niData = { 0 };
+const UINT ICON_MSG = WM_APP+1;
+void
+ShowNotificationIcon()
+{
+  Shell_NotifyIconA(NIM_ADD, &niData);
+  Shell_NotifyIconA(NIM_SETVERSION, &niData);
+}
+
+void
+HideNotificationIcon()
+{
+  Shell_NotifyIconA(NIM_DELETE, &niData);
+}
+
+void changeIcon(HWND hwnd, HICON icon)
+{
+  niData.hIcon = icon;
+  if (! IsWindowVisible(hwnd))
+    Shell_NotifyIconA(NIM_MODIFY, &niData);
+  SendMessage(hwnd, WM_SETICON, 0, (LPARAM)icon);
+  SendMessage(hwnd, WM_SETICON, 1, (LPARAM)icon);
+}
+
+
+
 void
 startRecording()
 {
   ws::sendRequest("StartRecord");
+  changeIcon(hwnd, icon_green);
 }
 
 void
 stopRecording()
 {
   ws::sendRequest("StopRecord");
+  changeIcon(hwnd, icon_red);
 }
 
 bool
@@ -77,26 +118,143 @@ checkForegroundProcess(std::string exeName)
   HWND fgHwnd = GetForegroundWindow();
   HANDLE fgHandle = getHwndProcess(fgHwnd);
 
-  char filename[1024];
-  int len = GetModuleFileNameExA(fgHandle, NULL, filename, 1024);
+  char filename[MAX_PATH];
+  int len = GetModuleFileNameExA(fgHandle, NULL, filename, MAX_PATH);
+
+  if (strcmp(filename, exeName.c_str()) == 0)
+    return true;
+
   PathStripPathA(filename);
 
   return strcmp(filename, exeName.c_str()) == 0;
 }
 
-bool recording = false;
-HANDLE process = NULL;
+void ReadGameExes()
+{
+  std::ifstream ifs("games.txt");
+
+  gameExes.clear();
+
+  while (ifs) {
+    std::string str;
+    std::getline(ifs, str);
+    if (! str.empty())
+      gameExes.push_back(str);
+  }
 
-// int WINAPI
-// WinMain(HINSTANCE hInstance,
-//         HINSTANCE hPrevInstance,
-//         LPSTR lpCmdLine,
-//         int nCmdShow)
-int main(int argc, char **argv)
+  ifs.close();
+}
+
+void WriteGameExes()
 {
+  std::ofstream ofs("games.txt", std::ios::trunc);
 
-  //win::Window window("Title", "MyWindowClass", hInstance);
-  win::Window window("Title", "MyWindowClass", GetModuleHandle(0));
+  for (const auto &exe : gameExes)
+  {
+    ofs.write(exe.c_str(), exe.size());
+    ofs.write("\n", 1);
+  }
+
+  ofs.close();
+}
+
+PROCESS_INFORMATION StartOBS()
+{
+  // delete .sentinel folder to stop obs from displaying
+  // error message on startup
+  system("del %appdata%\\obs-studio\\.sentinel /Q");
+
+  PROCESS_INFORMATION pi;
+  STARTUPINFOA sui;
+  GetStartupInfoA(&sui);
+  CreateProcessA(nullptr, "C:\\Program Files\\obs-studio\\bin\\64bit\\obs64.exe",
+    nullptr, nullptr, false, NORMAL_PRIORITY_CLASS, nullptr, "C:\\Program Files\\obs-studio\\bin\\64bit", &sui, &pi);
+  return pi;
+}
+
+void StopOBS(PROCESS_INFORMATION &pi)
+{
+  EnumWindows([](HWND hwnd, LPARAM lParam) -> BOOL {
+    PROCESS_INFORMATION *pi = (PROCESS_INFORMATION*)lParam;
+
+    DWORD pid, tid;
+    tid = GetWindowThreadProcessId(hwnd, &pid);
+    if (tid == pi->dwThreadId && pid == pi->dwProcessId) {
+
+      char title[1024]; GetWindowTextA(hwnd, title, 1024);
+
+      if (strncmp(title, "OBS", 3) == 0) {
+        PostMessage(hwnd, WM_CLOSE, 0, 0);
+        return FALSE; 
+      }
+    }
+    return TRUE;
+  }, (LPARAM)&pi);
+}
+
+/*
+TODO:
+  - Disconnect while recording
+*/
+
+
+
+int WINAPI
+WinMain(HINSTANCE hInstance,
+        HINSTANCE hPrevInstance,
+        LPSTR lpCmdLine,
+        int nCmdShow)
+//int main(int argc, char **argv)
+{
+  hInstance = GetModuleHandle(0);
+
+  icon_white = LoadIconA(hInstance, MAKEINTRESOURCEA(IDI_ICON_WHITE));
+  icon_green = LoadIconA(hInstance, MAKEINTRESOURCEA(IDI_ICON_GREEN));
+  icon_red = LoadIconA(hInstance, MAKEINTRESOURCEA(IDI_ICON_RED));
+
+  win::Window window("Title", "MyWindowClass", hInstance);
+  hwnd = window.hwnd;
+
+  niData.cbSize = sizeof(niData);
+  niData.uID = 12345;
+  niData.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
+  niData.hIcon = icon_white;
+  niData.hWnd = window.hwnd;
+  niData.uCallbackMessage = ICON_MSG;
+  niData.uVersion = NOTIFYICON_VERSION_4;
+
+  PROCESS_INFORMATION pi_obs = StartOBS();
+
+  window.handlers[WM_SIZE].push_back([](HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {
+    if (wParam == SIZE_MINIMIZED) {
+      ShowNotificationIcon();
+      ShowWindow(hwnd, false);
+    }
+  });
+  window.handlers[WM_DESTROY].push_back([](HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {
+    HideNotificationIcon();
+  });  
+  window.handlers[ICON_MSG].push_back([](HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {
+    if (LOWORD(lParam) == NIN_SELECT) {
+      HideNotificationIcon();
+      ShowWindow(hwnd, true);
+      SetForegroundWindow(hwnd);
+      SetActiveWindow(hwnd);
+    }
+  });
+  window.handlers[WM_GETMINMAXINFO].push_back([](HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {
+    MINMAXINFO *mmInfo = (MINMAXINFO*)lParam;
+    mmInfo->ptMinTrackSize.x = 400;
+    mmInfo->ptMinTrackSize.y = 200;
+  });
+  window.handlers[WM_QUERYENDSESSION].push_back([&pi_obs](HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {
+    StopOBS(pi_obs);
+  });  window.handlers[WM_ENDSESSION].push_back([&pi_obs](HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {
+    StopOBS(pi_obs);
+  });
+  window.handlers[WM_CLOSE].push_back([&pi_obs](HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {
+    StopOBS(pi_obs);
+  });
 
   lay_context *ctx = &window.ctx;
   lay_id root = window.lId;
@@ -111,18 +269,24 @@ int main(int argc, char **argv)
 
   win::Button btnConnect(&window, "Connect", row1, 100, 25, 0, 0);
   btnConnect.onClick([&]() {
-    ws::connect("ws://127.0.0.1:4444");
+    if (! ws::isConnected)
+      ws::connect("ws://127.0.0.1:4455");
   });
 
   win::ListBox lstActiveProcesses(&window, row2, 0, 0, 0, LAY_FILL);
   
   lay_id col1 = win::createLayId(&window.ctx, row2, 80, 0, LAY_COLUMN, LAY_VCENTER);
+  lstActiveProcesses.addStyle(WS_VSCROLL);
+
   lay_set_margins_ltrb(ctx, col1, 5, 0, 5, 0);
 
   win::ListBox lstMonitoredProcesses(&window, row2, 0, 0, 0, LAY_FILL);
-  lstActiveProcesses.addStyle(WS_VSCROLL);
   lstMonitoredProcesses.addStyle(WS_VSCROLL);
 
+  ReadGameExes();
+  for (const auto &exe : gameExes)
+    lstMonitoredProcesses.addString(exe);
+
   win::Button btnUpdateWindows(&window, "Update", col1, 85, 25, 0, 0);
   win::Button btnStartMonitoringName(&window, "Exe name >>", col1, 85, 25, 0, 0);
   win::Button btnStartMonitoringPath(&window, "Full path >>", col1, 85, 25, 0, 0);
@@ -150,12 +314,19 @@ int main(int argc, char **argv)
 
     std::string selStr = lstActiveProcesses.getText(sel);
     
-    char *filename = new char[selStr.size()];
+    char *filename = new char[selStr.size()+1];
     std::memcpy(filename, selStr.c_str(), selStr.size());
+    filename[selStr.size()] = '\0';
     PathStripPathA(filename);
 
-    if (lstMonitoredProcesses.findString(std::string(filename)) == LB_ERR)
-      lstMonitoredProcesses.addString(std::string(filename));
+    std::string filenameStr(filename);
+
+    if (lstMonitoredProcesses.findString(filenameStr) == LB_ERR)
+    {
+      lstMonitoredProcesses.addString(filenameStr);
+      gameExes.push_back(filenameStr);
+      WriteGameExes();
+    }
 
     delete[] filename;
   });
@@ -164,24 +335,47 @@ int main(int argc, char **argv)
     if (sel < 0) return;
     std::string selStr = lstActiveProcesses.getText(sel);
     if (lstMonitoredProcesses.findString(selStr) == LB_ERR)
-    lstMonitoredProcesses.addString(selStr);
+    {
+      lstMonitoredProcesses.addString(selStr);
+      gameExes.push_back(selStr);
+      WriteGameExes();  
+    }
   });
   btnStopMonitoring.onClick([&]() {
     int sel = lstMonitoredProcesses.getSelectedIndex();
     if (sel < 0) return;
     lstMonitoredProcesses.remove(sel);
+    gameExes.erase(gameExes.begin() + sel);
+    WriteGameExes();
   });
 
   window.show();
-
+  window.setDefaultFont();
+
+  ws::onConnect = [&]() {
+    changeIcon(window.hwnd, icon_red);
+    btnConnect.setActive(false);
+  };
+  ws::onClose = [&]() {
+    changeIcon(window.hwnd, icon_white);
+    btnConnect.setActive(true);
+  };
+  ws::onError = ws::onClose;
   ws::init();
 
-  SetTimer(window.hwnd, 10123, 100, [](HWND, UINT, UINT_PTR, DWORD) {
+  window.setTimer(1000, [&btnConnect]() {
+    if (! ws::isConnected)
+      ws::connect("ws://127.0.0.1:4455");
+  });
+
+  window.setTimer(100, []() {
     if (!recording) {
-      if (checkForegroundProcess("League of Legends.exe")) {
-        recording = true;
-        process = getHwndProcess(GetForegroundWindow());
-        startRecording();
+      for (auto exe : gameExes) {
+        if (checkForegroundProcess(exe)) {
+          recording = true;
+          process = getHwndProcess(GetForegroundWindow());
+          startRecording();
+        } 
       }
     } else {
       if (!checkProcessRunning(process)) {
@@ -195,4 +389,6 @@ int main(int argc, char **argv)
   while (window.update()) {
     ws::update();
   }
+
+  return 0;
 }