gitweb.ps.run Git - autorec/blob - src/main.cpp

   1 #ifndef WIN32\r
   2   #define WIN32\r
   3 #endif\r
   4 #include <WinSock2.h>\r
   5 #include "win.h"\r
   6 #include "ws.h"\r
   7 \r
   8 #include "../res/resource.h"\r
   9 \r
  10 #define LAY_IMPLEMENTATION\r
  11 #include "layout.h"\r
  12 \r
  13 #include <Psapi.h>\r
  14 #include <stdbool.h>\r
  15 #include <stdio.h>\r
  16 #include <string.h>\r
  17 #include <windows.h>\r
  18 #include <shlwapi.h>\r
  19 \r
  20 void\r
  21 startRecording()\r
  22 {\r
  23   ws::sendRequest("StartRecord");\r
  24 }\r
  25 \r
  26 void\r
  27 stopRecording()\r
  28 {\r
  29   ws::sendRequest("StopRecord");\r
  30 }\r
  31 \r
  32 bool\r
  33 checkProcessRunning(HANDLE handle)\r
  34 {\r
  35   DWORD exit_code;\r
  36 \r
  37   if (handle != NULL && GetExitCodeProcess(handle, &exit_code)) {\r
  38     return exit_code == STILL_ACTIVE;\r
  39   }\r
  40 \r
  41   return false;\r
  42 }\r
  43 \r
  44 HANDLE\r
  45 getHwndProcess(HWND hwnd)\r
  46 {\r
  47   DWORD processId, threadId = GetWindowThreadProcessId(hwnd, &processId);\r
  48   return OpenProcess(PROCESS_QUERY_INFORMATION, false, processId);\r
  49 }\r
  50 \r
  51 // HWND getProcessHwnd(HANDLE handle) {}\r
  52 \r
  53 bool\r
  54 checkFullscreenWindow()\r
  55 {\r
  56   HWND desktopHwnd = GetDesktopWindow();\r
  57   HWND fgHwnd = GetForegroundWindow();\r
  58 \r
  59   if (fgHwnd != desktopHwnd && fgHwnd != GetShellWindow()) {\r
  60     RECT windowRect, desktopRect;\r
  61     // Get Window and Desktop size\r
  62     GetWindowRect(fgHwnd, &windowRect);\r
  63     GetWindowRect(desktopHwnd, &desktopRect);\r
  64 \r
  65     bool fullscreen = windowRect.bottom == desktopRect.bottom &&\r
  66                       windowRect.top == desktopRect.top &&\r
  67                       windowRect.left == desktopRect.left &&\r
  68                       windowRect.right == desktopRect.right;\r
  69 \r
  70     return fullscreen;\r
  71   }\r
  72 \r
  73   return false;\r
  74 }\r
  75 \r
  76 bool\r
  77 checkForegroundProcess(std::string exeName)\r
  78 {\r
  79   HWND fgHwnd = GetForegroundWindow();\r
  80   HANDLE fgHandle = getHwndProcess(fgHwnd);\r
  81 \r
  82   char filename[1024];\r
  83   int len = GetModuleFileNameExA(fgHandle, NULL, filename, 1024);\r
  84   PathStripPathA(filename);\r
  85 \r
  86   return strcmp(filename, exeName.c_str()) == 0;\r
  87 }\r
  88 \r
  89 bool recording = false;\r
  90 HANDLE process = NULL;\r
  91 \r
  92 int WINAPI\r
  93 WinMain(HINSTANCE hInstance,\r
  94         HINSTANCE hPrevInstance,\r
  95         LPSTR lpCmdLine,\r
  96         int nCmdShow)\r
  97 {\r
  98 \r
  99   HWND window = win::Window("Title", "MyWindowClass", hInstance);\r
 100 \r
 101   lay_id row1 = lay_item(&win::_::ctx);\r
 102   lay_insert(&win::_::ctx, win::_::root, row1);\r
 103   lay_set_size_xy(&win::_::ctx, row1, 0, 25);\r
 104   lay_set_behave(&win::_::ctx, row1, LAY_LEFT);\r
 105   lay_set_contain(&win::_::ctx, row1, LAY_ROW);\r
 106   lay_set_margins_ltrb(&win::_::ctx, row1, 5, 5, 5, 5);\r
 107   lay_id row2 = lay_item(&win::_::ctx);\r
 108   lay_insert(&win::_::ctx, win::_::root, row2);\r
 109   lay_set_size_xy(&win::_::ctx, row2, 0, 0);\r
 110   lay_set_behave(&win::_::ctx, row2, LAY_FILL);\r
 111   lay_set_contain(&win::_::ctx, row2, LAY_ROW);\r
 112   lay_set_margins_ltrb(&win::_::ctx, row2, 5, 5, 5, 5);\r
 113   lay_id col1 = lay_item(&win::_::ctx);\r
 114   lay_set_size_xy(&win::_::ctx, col1, 80, 0);\r
 115   lay_set_behave(&win::_::ctx, col1, LAY_VCENTER);\r
 116   lay_set_contain(&win::_::ctx, col1, LAY_COLUMN);\r
 117   lay_set_margins_ltrb(&win::_::ctx, col1, 5, 0, 5, 0);\r
 118 \r
 119   HWND cbWindowTitle = win::CheckBox(window, "Window Title", row1, 100, 25, 0, 0);\r
 120   HWND cbFullscreenWindow = win::CheckBox(window, "Any Fullscreen Application", row1, 200, 25, 0, 0);\r
 121 \r
 122   HWND btnConnect = win::Button(window, "Connect", row1, 100, 25, 0, 0);\r
 123   win::Callback(btnConnect, BN_CLICKED, [&]() {\r
 124     ws::connect("ws://127.0.0.1:4444");\r
 125   });\r
 126 \r
 127   win::Callback(cbWindowTitle, BN_CLICKED, [&]() {\r
 128     SendMessageA(cbWindowTitle, BM_SETCHECK, SendMessageA(cbWindowTitle, BM_GETCHECK, 0, 0) ? BST_UNCHECKED : BST_CHECKED, 0);\r
 129   });\r
 130   win::Callback(cbFullscreenWindow, BN_CLICKED, [&]() {\r
 131     SendMessageA(cbFullscreenWindow, BM_SETCHECK, SendMessageA(cbFullscreenWindow, BM_GETCHECK, 0, 0) ? BST_UNCHECKED : BST_CHECKED, 0);\r
 132   });\r
 133 \r
 134   HWND lstActiveProcesses = win::ListView(window, row2, 0, 0, 0, LAY_FILL);\r
 135   lay_insert(&win::_::ctx, row2, col1);\r
 136   HWND lstMonitoredProcesses = win::ListBox(window, row2, 0, 0, 0, LAY_FILL);\r
 137   win::AddStyle(lstActiveProcesses, WS_VSCROLL);\r
 138   win::AddStyle(lstMonitoredProcesses, WS_VSCROLL);\r
 139 \r
 140   HWND btnUpdateWindows = win::Button(window, "Update", col1, 85, 25, 0, 0);\r
 141   HWND btnStartMonitoringName = win::Button(window, "Exe name >>", col1, 85, 25, 0, 0);\r
 142   HWND btnStartMonitoringPath = win::Button(window, "Full path >>", col1, 85, 25, 0, 0);\r
 143   HWND btnStopMonitoring = win::Button(window, "Remove", col1, 85, 25, 0, 0);\r
 144   win::Callback(btnUpdateWindows, BN_CLICKED, [&]() {\r
 145     win::ListClear(lstActiveProcesses);\r
 146     for (HWND hwnd = GetTopWindow(NULL); hwnd != nullptr;\r
 147          hwnd = GetNextWindow(hwnd, GW_HWNDNEXT)) {\r
 148       if (!IsWindowVisible(hwnd))\r
 149         continue;\r
 150 \r
 151       RECT rect;\r
 152       GetWindowRect(hwnd, &rect);\r
 153 \r
 154       char str[1024];\r
 155       if (GetModuleFileNameExA(getHwndProcess(hwnd), 0, str, 1024) != 0 &&\r
 156           win::ListFindString(lstActiveProcesses, str) == LB_ERR) {\r
 157         win::ListAddString(lstActiveProcesses, str);\r
 158       }\r
 159     }\r
 160   });\r
 161   win::Callback(btnStartMonitoringName, BN_CLICKED, [&]() {\r
 162     int sel = win::ListGetSelectedIndex(lstActiveProcesses);\r
 163     if (sel < 0) return;\r
 164 \r
 165     std::string selStr = win::ListGetText(lstActiveProcesses, sel);\r
 166     \r
 167     char *filename = new char[selStr.size()];\r
 168     std::memcpy(filename, selStr.c_str(), selStr.size());\r
 169     PathStripPathA(filename);\r
 170 \r
 171     if (win::ListFindString(lstMonitoredProcesses, std::string(filename)) == LB_ERR)\r
 172       win::ListAddString(lstMonitoredProcesses, std::string(filename));\r
 173 \r
 174     delete[] filename;\r
 175   });\r
 176   win::Callback(btnStartMonitoringPath, BN_CLICKED, [&]() {\r
 177     int sel = win::ListGetSelectedIndex(lstActiveProcesses);\r
 178     if (sel < 0) return;\r
 179     std::string selStr = win::ListGetText(lstActiveProcesses, sel);\r
 180     if (win::ListFindString(lstMonitoredProcesses, selStr) == LB_ERR)\r
 181     win::ListAddString(lstMonitoredProcesses, selStr);\r
 182   });\r
 183   win::Callback(btnStopMonitoring, BN_CLICKED, [&]() {\r
 184     int sel = win::ListGetSelectedIndex(lstMonitoredProcesses);\r
 185     if (sel < 0) return;\r
 186     win::ListRemove(lstMonitoredProcesses, sel);\r
 187   });\r
 188 \r
 189   win::ShowWindow(window);\r
 190 \r
 191   ws::init();\r
 192 \r
 193   SetTimer(window, 10123, 100, [](HWND, UINT, UINT_PTR, DWORD) {\r
 194     if (!recording) {\r
 195       if (checkForegroundProcess("League of Legends.exe")) {\r
 196         recording = true;\r
 197         process = getHwndProcess(GetForegroundWindow());\r
 198         startRecording();\r
 199       }\r
 200     } else {\r
 201       if (!checkProcessRunning(process)) {\r
 202         recording = false;\r
 203         process = NULL;\r
 204         stopRecording();\r
 205       }\r
 206     }\r
 207   });\r
 208 \r
 209   while (win::UpdateWindow(window)) {\r
 210     ws::update();\r
 211   }\r
 212 }\r