use zig to build

[autorec] / src / main.cpp

diff --git a/src/main.cpp b/src/main.cpp
index 5056acf844a0b702d4de597004f95c7909942eda..f7a1438538f435df29984ba1f7088a17485edc03 100644 (file)
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -5,11 +5,13 @@
 #include "win.h"\r
 #include "ws.h"\r
 \r
 #include "win.h"\r
 #include "ws.h"\r
 \r

-#include "../res/resource.h"\r
-\r

 #define LAY_IMPLEMENTATION\r
 #include "layout.h"\r
 \r
 #define LAY_IMPLEMENTATION\r
 #include "layout.h"\r
 \r

+#include <fstream>\r
+#include <vector>\r
+#include <string>\r
+\r

 #include <Psapi.h>\r
 #include <stdbool.h>\r
 #include <stdio.h>\r
 #include <Psapi.h>\r
 #include <stdbool.h>\r
 #include <stdio.h>\r


@@ -17,16 +19,53 @@
 #include <windows.h>\r
 #include <shlwapi.h>\r
 \r
 #include <windows.h>\r
 #include <shlwapi.h>\r
 \r

+bool recording = false;\r
+HANDLE process = NULL;\r
+HWND hwnd;\r
+HINSTANCE hInstance;\r
+HICON icon_white, icon_green, icon_red;\r
+\r
+std::vector<std::string> gameExes;\r
+\r
+\r
+NOTIFYICONDATAA niData = { 0 };\r
+const UINT ICON_MSG = WM_APP+1;\r
+void\r
+ShowNotificationIcon()\r
+{\r
+  Shell_NotifyIconA(NIM_ADD, &niData);\r
+  Shell_NotifyIconA(NIM_SETVERSION, &niData);\r
+}\r
+\r
+void\r
+HideNotificationIcon()\r
+{\r
+  Shell_NotifyIconA(NIM_DELETE, &niData);\r
+}\r
+\r
+void changeIcon(HWND hwnd, HICON icon)\r
+{\r
+  niData.hIcon = icon;\r
+  if (! IsWindowVisible(hwnd))\r
+    Shell_NotifyIconA(NIM_MODIFY, &niData);\r
+  SendMessage(hwnd, WM_SETICON, 0, (LPARAM)icon);\r
+  SendMessage(hwnd, WM_SETICON, 1, (LPARAM)icon);\r
+}\r
+\r
+\r
+\r

 void\r
 startRecording()\r
 {\r
   ws::sendRequest("StartRecord");\r
 void\r
 startRecording()\r
 {\r
   ws::sendRequest("StartRecord");\r

+  changeIcon(hwnd, icon_green);\r

 }\r
 \r
 void\r
 stopRecording()\r
 {\r
   ws::sendRequest("StopRecord");\r
 }\r
 \r
 void\r
 stopRecording()\r
 {\r
   ws::sendRequest("StopRecord");\r

+  changeIcon(hwnd, icon_red);\r

 }\r
 \r
 bool\r
 }\r
 \r
 bool\r


@@ -79,70 +118,175 @@ checkForegroundProcess(std::string exeName)
   HWND fgHwnd = GetForegroundWindow();\r
   HANDLE fgHandle = getHwndProcess(fgHwnd);\r
 \r
   HWND fgHwnd = GetForegroundWindow();\r
   HANDLE fgHandle = getHwndProcess(fgHwnd);\r
 \r

-  char filename[1024];\r
-  int len = GetModuleFileNameExA(fgHandle, NULL, filename, 1024);\r
+  char filename[MAX_PATH];\r
+  int len = GetModuleFileNameExA(fgHandle, NULL, filename, MAX_PATH);\r
+\r
+  if (strcmp(filename, exeName.c_str()) == 0)\r
+    return true;\r
+\r

   PathStripPathA(filename);\r
 \r
   return strcmp(filename, exeName.c_str()) == 0;\r
 }\r
 \r
   PathStripPathA(filename);\r
 \r
   return strcmp(filename, exeName.c_str()) == 0;\r
 }\r
 \r

-bool recording = false;\r
-HANDLE process = NULL;\r
+void ReadGameExes()\r
+{\r
+  std::ifstream ifs("games.txt");\r
+\r
+  gameExes.clear();\r
+\r
+  while (ifs) {\r
+    std::string str;\r
+    std::getline(ifs, str);\r
+    if (! str.empty())\r
+      gameExes.push_back(str);\r
+  }\r
+\r
+  ifs.close();\r
+}\r
+\r
+void WriteGameExes()\r
+{\r
+  std::ofstream ofs("games.txt", std::ios::trunc);\r
+\r
+  for (const auto &exe : gameExes)\r
+  {\r
+    ofs.write(exe.c_str(), exe.size());\r
+    ofs.write("\n", 1);\r
+  }\r
+\r
+  ofs.close();\r
+}\r
+\r
+PROCESS_INFORMATION StartOBS()\r
+{\r
+  PROCESS_INFORMATION pi;\r
+  STARTUPINFOA sui;\r
+  GetStartupInfoA(&sui);\r
+  CreateProcessA(nullptr, "C:\\Program Files\\obs-studio\\bin\\64bit\\obs64.exe --disable-shutdown-check",\r
+    nullptr, nullptr, false, NORMAL_PRIORITY_CLASS, nullptr, "C:\\Program Files\\obs-studio\\bin\\64bit", &sui, &pi);\r
+  return pi;\r
+}\r
+\r
+void StopOBS(PROCESS_INFORMATION &pi)\r
+{\r
+  EnumWindows([](HWND hwnd, LPARAM lParam) -> BOOL {\r
+    PROCESS_INFORMATION *pi = (PROCESS_INFORMATION*)lParam;\r
+\r
+    DWORD pid, tid;\r
+    tid = GetWindowThreadProcessId(hwnd, &pid);\r
+    if (tid == pi->dwThreadId && pid == pi->dwProcessId) {\r
+\r
+      char title[1024]; GetWindowTextA(hwnd, title, 1024);\r
+\r
+      if (strncmp(title, "OBS", 3) == 0) {\r
+        PostMessage(hwnd, WM_CLOSE, 0, 0);\r
+        return FALSE; \r
+      }\r
+    }\r
+    return TRUE;\r
+  }, (LPARAM)&pi);\r
+}\r
+\r
+/*\r
+TODO:\r
+  - Disconnect while recording\r
+*/\r
+\r
+\r

 \r
 int WINAPI\r
 WinMain(HINSTANCE hInstance,\r
         HINSTANCE hPrevInstance,\r
         LPSTR lpCmdLine,\r
         int nCmdShow)\r
 \r
 int WINAPI\r
 WinMain(HINSTANCE hInstance,\r
         HINSTANCE hPrevInstance,\r
         LPSTR lpCmdLine,\r
         int nCmdShow)\r

+//int main(int argc, char **argv)\r

 {\r
 {\r

+  hInstance = GetModuleHandle(0);\r

 \r
 \r

-  HWND window = win::Window("Title", "MyWindowClass", hInstance);\r
-\r
-  lay_id row1 = lay_item(&win::_::ctx);\r
-  lay_insert(&win::_::ctx, win::_::root, row1);\r
-  lay_set_size_xy(&win::_::ctx, row1, 0, 25);\r
-  lay_set_behave(&win::_::ctx, row1, LAY_LEFT);\r
-  lay_set_contain(&win::_::ctx, row1, LAY_ROW);\r
-  lay_set_margins_ltrb(&win::_::ctx, row1, 5, 5, 5, 5);\r
-  lay_id row2 = lay_item(&win::_::ctx);\r
-  lay_insert(&win::_::ctx, win::_::root, row2);\r
-  lay_set_size_xy(&win::_::ctx, row2, 0, 0);\r
-  lay_set_behave(&win::_::ctx, row2, LAY_FILL);\r
-  lay_set_contain(&win::_::ctx, row2, LAY_ROW);\r
-  lay_set_margins_ltrb(&win::_::ctx, row2, 5, 5, 5, 5);\r
-  lay_id col1 = lay_item(&win::_::ctx);\r
-  lay_set_size_xy(&win::_::ctx, col1, 80, 0);\r
-  lay_set_behave(&win::_::ctx, col1, LAY_VCENTER);\r
-  lay_set_contain(&win::_::ctx, col1, LAY_COLUMN);\r
-  lay_set_margins_ltrb(&win::_::ctx, col1, 5, 0, 5, 0);\r
-\r
-  HWND cbWindowTitle = win::CheckBox(window, "Window Title", row1, 100, 25, 0, 0);\r
-  HWND cbFullscreenWindow = win::CheckBox(window, "Any Fullscreen Application", row1, 200, 25, 0, 0);\r
-\r
-  HWND btnConnect = win::Button(window, "Connect", row1, 100, 25, 0, 0);\r
-  win::Callback(btnConnect, BN_CLICKED, [&]() {\r
-    ws::connect("ws://127.0.0.1:4444");\r
-  });\r
+  icon_white = LoadIconA(hInstance, MAKEINTRESOURCEA(IDI_ICON_WHITE));\r
+  icon_green = LoadIconA(hInstance, MAKEINTRESOURCEA(IDI_ICON_GREEN));\r
+  icon_red = LoadIconA(hInstance, MAKEINTRESOURCEA(IDI_ICON_RED));\r
+\r
+  win::Window window("Title", "MyWindowClass", hInstance);\r
+  hwnd = window.hwnd;\r
+\r
+  niData.cbSize = sizeof(niData);\r
+  niData.uID = 12345;\r
+  niData.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;\r
+  niData.hIcon = icon_white;\r
+  niData.hWnd = window.hwnd;\r
+  niData.uCallbackMessage = ICON_MSG;\r
+  niData.uVersion = NOTIFYICON_VERSION_4;\r

 \r
 \r

-  win::Callback(cbWindowTitle, BN_CLICKED, [&]() {\r
-    SendMessageA(cbWindowTitle, BM_SETCHECK, SendMessageA(cbWindowTitle, BM_GETCHECK, 0, 0) ? BST_UNCHECKED : BST_CHECKED, 0);\r
+  PROCESS_INFORMATION pi_obs = StartOBS();\r
+\r
+  window.handlers[WM_SIZE].push_back([](HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {\r
+    if (wParam == SIZE_MINIMIZED) {\r
+      ShowNotificationIcon();\r
+      ShowWindow(hwnd, false);\r
+    }\r
+  });\r
+  window.handlers[WM_DESTROY].push_back([](HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {\r
+    HideNotificationIcon();\r
+  });  \r
+  window.handlers[ICON_MSG].push_back([](HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {\r
+    if (LOWORD(lParam) == NIN_SELECT) {\r
+      HideNotificationIcon();\r
+      ShowWindow(hwnd, true);\r
+      SetForegroundWindow(hwnd);\r
+      SetActiveWindow(hwnd);\r
+    }\r

   });\r
   });\r

-  win::Callback(cbFullscreenWindow, BN_CLICKED, [&]() {\r
-    SendMessageA(cbFullscreenWindow, BM_SETCHECK, SendMessageA(cbFullscreenWindow, BM_GETCHECK, 0, 0) ? BST_UNCHECKED : BST_CHECKED, 0);\r
+  window.handlers[WM_GETMINMAXINFO].push_back([](HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {\r
+    MINMAXINFO *mmInfo = (MINMAXINFO*)lParam;\r
+    mmInfo->ptMinTrackSize.x = 400;\r
+    mmInfo->ptMinTrackSize.y = 200;\r

   });\r
   });\r

+  window.handlers[WM_QUERYENDSESSION].push_back([&pi_obs](HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {\r
+    StopOBS(pi_obs);\r
+  });\r
+  window.handlers[WM_CLOSE].push_back([&pi_obs](HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {\r
+    StopOBS(pi_obs);\r
+  });\r
+\r
+  lay_context *ctx = &window.ctx;\r
+  lay_id root = window.lId;\r
+\r
+  lay_id row1 = win::createLayId(&window.ctx, window.lId, 0, 25, LAY_ROW, LAY_LEFT);\r
+  lay_set_margins_ltrb(ctx, row1, 5, 5, 5, 5);\r
+  lay_id row2 = win::createLayId(&window.ctx, window.lId, 0, 0, LAY_ROW, LAY_FILL);\r
+  lay_set_margins_ltrb(ctx, row2, 5, 5, 5, 5);\r
+\r
+  win::CheckBox cbWindowTitle(&window, "Window Title", row1, 100, 25, 0, 0);\r
+  win::CheckBox cbFullscreenWindow(&window, "Any Fullscreen Application", row1, 200, 25, 0, 0);\r
+\r
+  win::Button btnConnect(&window, "Connect", row1, 100, 25, 0, 0);\r
+  btnConnect.onClick([&]() {\r
+    if (! ws::isConnected)\r
+      ws::connect("ws://127.0.0.1:4455");\r
+  });\r
+\r
+  win::ListBox lstActiveProcesses(&window, row2, 0, 0, 0, LAY_FILL);\r
+  \r
+  lay_id col1 = win::createLayId(&window.ctx, row2, 80, 0, LAY_COLUMN, LAY_VCENTER);\r
+  lstActiveProcesses.addStyle(WS_VSCROLL);\r
+\r
+  lay_set_margins_ltrb(ctx, col1, 5, 0, 5, 0);\r

 \r
 \r

-  HWND lstActiveProcesses = win::ListView(window, row2, 0, 0, 0, LAY_FILL);\r
-  lay_insert(&win::_::ctx, row2, col1);\r
-  HWND lstMonitoredProcesses = win::ListBox(window, row2, 0, 0, 0, LAY_FILL);\r
-  win::AddStyle(lstActiveProcesses, WS_VSCROLL);\r
-  win::AddStyle(lstMonitoredProcesses, WS_VSCROLL);\r
-\r
-  HWND btnUpdateWindows = win::Button(window, "Update", col1, 85, 25, 0, 0);\r
-  HWND btnStartMonitoringName = win::Button(window, "Exe name >>", col1, 85, 25, 0, 0);\r
-  HWND btnStartMonitoringPath = win::Button(window, "Full path >>", col1, 85, 25, 0, 0);\r
-  HWND btnStopMonitoring = win::Button(window, "Remove", col1, 85, 25, 0, 0);\r
-  win::Callback(btnUpdateWindows, BN_CLICKED, [&]() {\r
-    win::ListClear(lstActiveProcesses);\r
+  win::ListBox lstMonitoredProcesses(&window, row2, 0, 0, 0, LAY_FILL);\r
+  lstMonitoredProcesses.addStyle(WS_VSCROLL);\r
+\r
+  ReadGameExes();\r
+  for (const auto &exe : gameExes)\r
+    lstMonitoredProcesses.addString(exe);\r
+\r
+  win::Button btnUpdateWindows(&window, "Update", col1, 85, 25, 0, 0);\r
+  win::Button btnStartMonitoringName(&window, "Exe name >>", col1, 85, 25, 0, 0);\r
+  win::Button btnStartMonitoringPath(&window, "Full path >>", col1, 85, 25, 0, 0);\r
+  win::Button btnStopMonitoring(&window, "Remove", col1, 85, 25, 0, 0);\r
+  btnUpdateWindows.onClick([&]() {\r
+    lstActiveProcesses.clear();\r

     for (HWND hwnd = GetTopWindow(NULL); hwnd != nullptr;\r
          hwnd = GetNextWindow(hwnd, GW_HWNDNEXT)) {\r
       if (!IsWindowVisible(hwnd))\r
     for (HWND hwnd = GetTopWindow(NULL); hwnd != nullptr;\r
          hwnd = GetNextWindow(hwnd, GW_HWNDNEXT)) {\r
       if (!IsWindowVisible(hwnd))\r


@@ -153,49 +297,79 @@ WinMain(HINSTANCE hInstance,
 \r
       char str[1024];\r
       if (GetModuleFileNameExA(getHwndProcess(hwnd), 0, str, 1024) != 0 &&\r
 \r
       char str[1024];\r
       if (GetModuleFileNameExA(getHwndProcess(hwnd), 0, str, 1024) != 0 &&\r

-          win::ListFindString(lstActiveProcesses, str) == LB_ERR) {\r
-        win::ListAddString(lstActiveProcesses, str);\r
+          lstActiveProcesses.findString(str) == LB_ERR) {\r
+        lstActiveProcesses.addString(str);\r

       }\r
     }\r
   });\r
       }\r
     }\r
   });\r

-  win::Callback(btnStartMonitoringName, BN_CLICKED, [&]() {\r
-    int sel = win::ListGetSelectedIndex(lstActiveProcesses);\r
+  btnStartMonitoringName.onClick([&]() {\r
+    int sel = lstActiveProcesses.getSelectedIndex();\r

     if (sel < 0) return;\r
 \r
     if (sel < 0) return;\r
 \r

-    std::string selStr = win::ListGetText(lstActiveProcesses, sel);\r
+    std::string selStr = lstActiveProcesses.getText(sel);\r

     \r
     \r

-    char *filename = new char[selStr.size()];\r
+    char *filename = new char[selStr.size()+1];\r

     std::memcpy(filename, selStr.c_str(), selStr.size());\r
     std::memcpy(filename, selStr.c_str(), selStr.size());\r

+    filename[selStr.size()] = '\0';\r

     PathStripPathA(filename);\r
 \r
     PathStripPathA(filename);\r
 \r

-    if (win::ListFindString(lstMonitoredProcesses, std::string(filename)) == LB_ERR)\r
-      win::ListAddString(lstMonitoredProcesses, std::string(filename));\r
+    std::string filenameStr(filename);\r
+\r
+    if (lstMonitoredProcesses.findString(filenameStr) == LB_ERR)\r
+    {\r
+      lstMonitoredProcesses.addString(filenameStr);\r
+      gameExes.push_back(filenameStr);\r
+      WriteGameExes();\r
+    }\r

 \r
     delete[] filename;\r
   });\r
 \r
     delete[] filename;\r
   });\r

-  win::Callback(btnStartMonitoringPath, BN_CLICKED, [&]() {\r
-    int sel = win::ListGetSelectedIndex(lstActiveProcesses);\r
+  btnStartMonitoringPath.onClick([&]() {\r
+    int sel = lstActiveProcesses.getSelectedIndex();\r

     if (sel < 0) return;\r
     if (sel < 0) return;\r

-    std::string selStr = win::ListGetText(lstActiveProcesses, sel);\r
-    if (win::ListFindString(lstMonitoredProcesses, selStr) == LB_ERR)\r
-    win::ListAddString(lstMonitoredProcesses, selStr);\r
+    std::string selStr = lstActiveProcesses.getText(sel);\r
+    if (lstMonitoredProcesses.findString(selStr) == LB_ERR)\r
+    {\r
+      lstMonitoredProcesses.addString(selStr);\r
+      gameExes.push_back(selStr);\r
+      WriteGameExes();  \r
+    }\r

   });\r
   });\r

-  win::Callback(btnStopMonitoring, BN_CLICKED, [&]() {\r
-    int sel = win::ListGetSelectedIndex(lstMonitoredProcesses);\r
+  btnStopMonitoring.onClick([&]() {\r
+    int sel = lstMonitoredProcesses.getSelectedIndex();\r

     if (sel < 0) return;\r
     if (sel < 0) return;\r

-    win::ListRemove(lstMonitoredProcesses, sel);\r
+    lstMonitoredProcesses.remove(sel);\r
+    gameExes.erase(gameExes.begin() + sel);\r
+    WriteGameExes();\r

   });\r
 \r
   });\r
 \r

-  win::ShowWindow(window);\r
-\r
+  window.show();\r
+  window.setDefaultFont();\r
+\r
+  ws::onConnect = [&]() {\r
+    changeIcon(window.hwnd, icon_red);\r
+    btnConnect.setActive(false);\r
+  };\r
+  ws::onClose = [&]() {\r
+    changeIcon(window.hwnd, icon_white);\r
+    btnConnect.setActive(true);\r
+  };\r
+  ws::onError = ws::onClose;\r

   ws::init();\r
 \r
   ws::init();\r
 \r

-  SetTimer(window, 10123, 100, [](HWND, UINT, UINT_PTR, DWORD) {\r
+  window.setTimer(1000, [&btnConnect]() {\r
+    if (! ws::isConnected)\r
+      ws::connect("ws://127.0.0.1:4455");\r
+  });\r
+\r
+  window.setTimer(100, []() {\r

     if (!recording) {\r
     if (!recording) {\r

-      if (checkForegroundProcess("League of Legends.exe")) {\r
-        recording = true;\r
-        process = getHwndProcess(GetForegroundWindow());\r
-        startRecording();\r
+      for (auto exe : gameExes) {\r
+        if (checkForegroundProcess(exe)) {\r
+          recording = true;\r
+          process = getHwndProcess(GetForegroundWindow());\r
+          startRecording();\r
+        } \r

       }\r
     } else {\r
       if (!checkProcessRunning(process)) {\r
       }\r
     } else {\r
       if (!checkProcessRunning(process)) {\r


@@ -206,7 +380,9 @@ WinMain(HINSTANCE hInstance,
     }\r
   });\r
 \r
     }\r
   });\r
 \r

-  while (win::UpdateWindow(window)) {\r
+  while (window.update()) {\r

     ws::update();\r
   }\r
     ws::update();\r
   }\r

+\r
+  return 0;\r

 }\r
 }\r