X-Git-Url: https://gitweb.ps.run/autorec/blobdiff_plain/e90fc8a9809127820252a09dd7bfbbbc7a6f295d..4264345d06a81cb561047d0973c44a129db934aa:/src/main.cpp

diff --git a/src/main.cpp b/src/main.cpp
index 5056acf..d89c199 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -5,8 +5,6 @@
 #include "win.h"
 #include "ws.h"
 
-#include "../res/resource.h"
-
 #define LAY_IMPLEMENTATION
 #include "layout.h"
 
@@ -17,16 +15,52 @@
 #include <windows.h>
 #include <shlwapi.h>
 
+bool recording = false;
+HANDLE process = NULL;
+HWND hwnd;
+HINSTANCE hInstance;
+
+
+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, HINSTANCE hInstance, WORD id)
+{
+  HICON icon = LoadIcon(hInstance, MAKEINTRESOURCE(id));
+  HideNotificationIcon();
+  niData.hIcon = icon;
+  if (! IsWindowVisible(hwnd))
+    ShowNotificationIcon();
+  SendMessage(hwnd, WM_SETICON, 0, (LPARAM)icon);
+  SendMessage(hwnd, WM_SETICON, 1, (LPARAM)icon);
+}
+
+
+
 void
 startRecording()
 {
   ws::sendRequest("StartRecord");
+  changeIcon(hwnd, hInstance, IDI_ICON_GREEN);
 }
 
 void
 stopRecording()
 {
   ws::sendRequest("StopRecord");
+  changeIcon(hwnd, hInstance, IDI_ICON_RED);
 }
 
 bool
@@ -86,63 +120,86 @@ checkForegroundProcess(std::string exeName)
   return strcmp(filename, exeName.c_str()) == 0;
 }
 
-bool recording = false;
-HANDLE process = NULL;
-
 int WINAPI
 WinMain(HINSTANCE hInstance,
         HINSTANCE hPrevInstance,
         LPSTR lpCmdLine,
         int nCmdShow)
+//int main(int argc, char **argv)
 {
+  hInstance = GetModuleHandle(0);
+
+  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 = LoadIconA(hInstance, MAKEINTRESOURCEA(IDI_ICON_WHITE));
+  niData.hWnd = window.hwnd;
+  niData.uCallbackMessage = ICON_MSG;
+  niData.uVersion = NOTIFYICON_VERSION_4;
+
+  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);
+    }
+  });
 
-  HWND window = win::Window("Title", "MyWindowClass", hInstance);
-
-  lay_id row1 = lay_item(&win::_::ctx);
-  lay_insert(&win::_::ctx, win::_::root, row1);
-  lay_set_size_xy(&win::_::ctx, row1, 0, 25);
-  lay_set_behave(&win::_::ctx, row1, LAY_LEFT);
-  lay_set_contain(&win::_::ctx, row1, LAY_ROW);
-  lay_set_margins_ltrb(&win::_::ctx, row1, 5, 5, 5, 5);
-  lay_id row2 = lay_item(&win::_::ctx);
-  lay_insert(&win::_::ctx, win::_::root, row2);
-  lay_set_size_xy(&win::_::ctx, row2, 0, 0);
-  lay_set_behave(&win::_::ctx, row2, LAY_FILL);
-  lay_set_contain(&win::_::ctx, row2, LAY_ROW);
-  lay_set_margins_ltrb(&win::_::ctx, row2, 5, 5, 5, 5);
-  lay_id col1 = lay_item(&win::_::ctx);
-  lay_set_size_xy(&win::_::ctx, col1, 80, 0);
-  lay_set_behave(&win::_::ctx, col1, LAY_VCENTER);
-  lay_set_contain(&win::_::ctx, col1, LAY_COLUMN);
-  lay_set_margins_ltrb(&win::_::ctx, col1, 5, 0, 5, 0);
-
-  HWND cbWindowTitle = win::CheckBox(window, "Window Title", row1, 100, 25, 0, 0);
-  HWND cbFullscreenWindow = win::CheckBox(window, "Any Fullscreen Application", row1, 200, 25, 0, 0);
-
-  HWND btnConnect = win::Button(window, "Connect", row1, 100, 25, 0, 0);
-  win::Callback(btnConnect, BN_CLICKED, [&]() {
-    ws::connect("ws://127.0.0.1:4444");
+  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;
   });
 
-  win::Callback(cbWindowTitle, BN_CLICKED, [&]() {
-    SendMessageA(cbWindowTitle, BM_SETCHECK, SendMessageA(cbWindowTitle, BM_GETCHECK, 0, 0) ? BST_UNCHECKED : BST_CHECKED, 0);
+  lay_context *ctx = &window.ctx;
+  lay_id root = window.lId;
+
+  lay_id row1 = win::createLayId(&window.ctx, window.lId, 0, 25, LAY_ROW, LAY_LEFT);
+  lay_set_margins_ltrb(ctx, row1, 5, 5, 5, 5);
+  lay_id row2 = win::createLayId(&window.ctx, window.lId, 0, 0, LAY_ROW, LAY_FILL);
+  lay_set_margins_ltrb(ctx, row2, 5, 5, 5, 5);
+
+  win::CheckBox cbWindowTitle(&window, "Window Title", row1, 100, 25, 0, 0);
+  win::CheckBox cbFullscreenWindow(&window, "Any Fullscreen Application", row1, 200, 25, 0, 0);
+
+  win::Button btnConnect(&window, "Connect", row1, 100, 25, 0, 0);
+  btnConnect.onClick([&]() {
+    ws::connect("ws://127.0.0.1:4444");
   });
-  win::Callback(cbFullscreenWindow, BN_CLICKED, [&]() {
-    SendMessageA(cbFullscreenWindow, BM_SETCHECK, SendMessageA(cbFullscreenWindow, BM_GETCHECK, 0, 0) ? BST_UNCHECKED : BST_CHECKED, 0);
+  win::Button btnTest(&window, "Test", row1, 100, 25, 0, 0);
+  btnTest.onClick([&]() {
+    changeIcon(window.hwnd, hInstance, IDI_ICON_GREEN);
   });
 
-  HWND lstActiveProcesses = win::ListView(window, row2, 0, 0, 0, LAY_FILL);
-  lay_insert(&win::_::ctx, row2, col1);
-  HWND lstMonitoredProcesses = win::ListBox(window, row2, 0, 0, 0, LAY_FILL);
-  win::AddStyle(lstActiveProcesses, WS_VSCROLL);
-  win::AddStyle(lstMonitoredProcesses, WS_VSCROLL);
-
-  HWND btnUpdateWindows = win::Button(window, "Update", col1, 85, 25, 0, 0);
-  HWND btnStartMonitoringName = win::Button(window, "Exe name >>", col1, 85, 25, 0, 0);
-  HWND btnStartMonitoringPath = win::Button(window, "Full path >>", col1, 85, 25, 0, 0);
-  HWND btnStopMonitoring = win::Button(window, "Remove", col1, 85, 25, 0, 0);
-  win::Callback(btnUpdateWindows, BN_CLICKED, [&]() {
-    win::ListClear(lstActiveProcesses);
+  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);
+  lstMonitoredProcesses.addStyle(WS_VSCROLL);
+
+  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);
+  win::Button btnStopMonitoring(&window, "Remove", col1, 85, 25, 0, 0);
+  btnUpdateWindows.onClick([&]() {
+    lstActiveProcesses.clear();
     for (HWND hwnd = GetTopWindow(NULL); hwnd != nullptr;
          hwnd = GetNextWindow(hwnd, GW_HWNDNEXT)) {
       if (!IsWindowVisible(hwnd))
@@ -153,46 +210,50 @@ WinMain(HINSTANCE hInstance,
 
       char str[1024];
       if (GetModuleFileNameExA(getHwndProcess(hwnd), 0, str, 1024) != 0 &&
-          win::ListFindString(lstActiveProcesses, str) == LB_ERR) {
-        win::ListAddString(lstActiveProcesses, str);
+          lstActiveProcesses.findString(str) == LB_ERR) {
+        lstActiveProcesses.addString(str);
       }
     }
   });
-  win::Callback(btnStartMonitoringName, BN_CLICKED, [&]() {
-    int sel = win::ListGetSelectedIndex(lstActiveProcesses);
+  btnStartMonitoringName.onClick([&]() {
+    int sel = lstActiveProcesses.getSelectedIndex();
     if (sel < 0) return;
 
-    std::string selStr = win::ListGetText(lstActiveProcesses, sel);
+    std::string selStr = lstActiveProcesses.getText(sel);
     
     char *filename = new char[selStr.size()];
     std::memcpy(filename, selStr.c_str(), selStr.size());
     PathStripPathA(filename);
 
-    if (win::ListFindString(lstMonitoredProcesses, std::string(filename)) == LB_ERR)
-      win::ListAddString(lstMonitoredProcesses, std::string(filename));
+    if (lstMonitoredProcesses.findString(std::string(filename)) == LB_ERR)
+      lstMonitoredProcesses.addString(std::string(filename));
 
     delete[] filename;
   });
-  win::Callback(btnStartMonitoringPath, BN_CLICKED, [&]() {
-    int sel = win::ListGetSelectedIndex(lstActiveProcesses);
+  btnStartMonitoringPath.onClick([&]() {
+    int sel = lstActiveProcesses.getSelectedIndex();
     if (sel < 0) return;
-    std::string selStr = win::ListGetText(lstActiveProcesses, sel);
-    if (win::ListFindString(lstMonitoredProcesses, selStr) == LB_ERR)
-    win::ListAddString(lstMonitoredProcesses, selStr);
+    std::string selStr = lstActiveProcesses.getText(sel);
+    if (lstMonitoredProcesses.findString(selStr) == LB_ERR)
+    lstMonitoredProcesses.addString(selStr);
   });
-  win::Callback(btnStopMonitoring, BN_CLICKED, [&]() {
-    int sel = win::ListGetSelectedIndex(lstMonitoredProcesses);
+  btnStopMonitoring.onClick([&]() {
+    int sel = lstMonitoredProcesses.getSelectedIndex();
     if (sel < 0) return;
-    win::ListRemove(lstMonitoredProcesses, sel);
+    lstMonitoredProcesses.remove(sel);
   });
 
-  win::ShowWindow(window);
+  window.show();
+  window.setDefaultFont();
 
+  ws::onConnect = [&]() {
+    changeIcon(window.hwnd, hInstance, IDI_ICON_RED);
+  };
   ws::init();
 
-  SetTimer(window, 10123, 100, [](HWND, UINT, UINT_PTR, DWORD) {
+  SetTimer(window.hwnd, 10123, 100, [](HWND, UINT, UINT_PTR, DWORD) {
     if (!recording) {
-      if (checkForegroundProcess("League of Legends.exe")) {
+      if (checkForegroundProcess("notepad.exe")) {
         recording = true;
         process = getHwndProcess(GetForegroundWindow());
         startRecording();
@@ -206,7 +267,7 @@ WinMain(HINSTANCE hInstance,
     }
   });
 
-  while (win::UpdateWindow(window)) {
+  while (window.update()) {
     ws::update();
   }
 }