UI for Win32

examples/desktop-server/win32/res.rc

@@ -0,0 +1 @@
+150 ICON DISCARDABLE "systray.ico"

examples/desktop-server/win32/winmain.c

@@ -0,0 +1,131 @@
+#include <shellapi.h>
+#include <shlobj.h>
+
+static const char *s_server_name = "Mongoose v" MG_VERSION;
+static HANDLE hThread;  // Serving thread
+static NOTIFYICONDATA TrayIcon;
+
+enum {
+  ID_QUIT = 100,  // Menu item IDs
+  ID_SEPARATOR,
+  ID_SET_DIR,
+  ID_ICON_MONGOOSE = 150,  // Icons for mongoose menu
+};
+
+static const char *get_my_url(void) {
+  static char buf[100];
+  char local_ip[50] = "", hostname[200];
+  struct hostent *he;
+  if (gethostname(hostname, sizeof(hostname)) == 0 &&
+      (he = gethostbyname(hostname)) != NULL) {
+    snprintf(local_ip, sizeof(local_ip), "%s",
+             inet_ntoa(*(struct in_addr *) he->h_addr_list[0]));
+  }
+  if (local_ip[0] == '\0') snprintf(local_ip, sizeof(local_ip), "127.0.0.1");
+  snprintf(buf, sizeof(buf), "http://%s:%d", local_ip,
+           mg_url_port(s_listening_address));
+  return buf;
+}
+
+static LRESULT CALLBACK WindowProc(HWND hWnd, UINT msg, WPARAM wParam,
+                                   LPARAM lParam) {
+  POINT pt;
+  HMENU hMenu;
+  switch (msg) {
+    case WM_COMMAND:
+      switch (LOWORD(wParam)) {
+        case ID_QUIT:
+          Shell_NotifyIcon(NIM_DELETE, &TrayIcon);
+          PostQuitMessage(0);
+          return 0;
+        case ID_SET_DIR: {
+          char path[PATH_MAX] = "";
+          BROWSEINFO bi;
+          memset(&bi, 0, sizeof(bi));
+          bi.hwndOwner = hWnd;
+          bi.lpszTitle = "Choose root directory:";
+          bi.ulFlags = BIF_RETURNONLYFSDIRS;
+          SHGetPathFromIDList(SHBrowseForFolder(&bi), path);
+          if (path[0] != '\0') {
+            FILE *fp = opencfg(__argv[0], "w");
+            s_root_dir = strdup(path);
+            if (fp != NULL) {
+              fprintf(fp, "-d %s\n", path);
+              fclose(fp);
+            }
+          }
+          break;
+        }
+      }
+      break;
+    case WM_USER:
+      switch (lParam) {
+        case WM_RBUTTONUP:
+        case WM_LBUTTONUP:
+        case WM_LBUTTONDBLCLK: {
+          char buf[1024];
+          hMenu = CreatePopupMenu();
+          AppendMenu(hMenu, MF_STRING | MF_GRAYED, ID_SEPARATOR, s_server_name);
+          AppendMenu(hMenu, MF_SEPARATOR, ID_SEPARATOR, "");
+          snprintf(buf, sizeof(buf), "Shared directory: %s", s_root_dir);
+          AppendMenu(hMenu, MF_STRING | MF_GRAYED, ID_SEPARATOR, buf);
+          snprintf(buf, sizeof(buf), "URL: %s", get_my_url());
+          AppendMenu(hMenu, MF_STRING | MF_GRAYED, ID_SEPARATOR, buf);
+          AppendMenu(hMenu, MF_SEPARATOR, ID_SEPARATOR, "");
+          AppendMenu(hMenu, MF_STRING, ID_SET_DIR,
+                     "Change shared directory ...");
+          AppendMenu(hMenu, MF_STRING, ID_QUIT, "Exit");
+          GetCursorPos(&pt);
+          SetForegroundWindow(hWnd);
+          TrackPopupMenu(hMenu, 0, pt.x, pt.y, 0, hWnd, NULL);
+          PostMessage(hWnd, WM_NULL, 0, 0);
+          DestroyMenu(hMenu);
+          break;
+        }
+      }
+      break;
+    case WM_CLOSE:
+      Shell_NotifyIcon(NIM_DELETE, &TrayIcon);
+      PostQuitMessage(0);
+      return 0;  // We've just sent our own quit message, with proper hwnd.
+    default:
+      break;
+  }
+  return DefWindowProc(hWnd, msg, wParam, lParam);
+}
+
+int WINAPI WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPSTR cmdline, int show) {
+  WNDCLASS cls;
+  HWND hWnd;
+  MSG msg = {0};
+
+  init_mongoose(__argc, __argv);
+  memset(&cls, 0, sizeof(cls));
+  cls.lpfnWndProc = (WNDPROC) WindowProc;
+  cls.hIcon = LoadIcon(NULL, IDI_APPLICATION);
+  cls.lpszClassName = s_server_name;
+  RegisterClass(&cls);
+  hWnd = CreateWindow(cls.lpszClassName, s_server_name, WS_OVERLAPPEDWINDOW, 0,
+                      0, 0, 0, NULL, NULL, NULL, NULL);
+  ShowWindow(hWnd, SW_HIDE);
+
+  TrayIcon.cbSize = sizeof(TrayIcon);
+  TrayIcon.uID = ID_ICON_MONGOOSE;
+  TrayIcon.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
+  TrayIcon.hIcon =
+      LoadIcon(GetModuleHandle(NULL), MAKEINTRESOURCE(ID_ICON_MONGOOSE));
+  TrayIcon.hWnd = hWnd;
+  snprintf(TrayIcon.szTip, sizeof(TrayIcon.szTip), "%s", s_server_name);
+  TrayIcon.uCallbackMessage = WM_USER;
+  Shell_NotifyIcon(NIM_ADD, &TrayIcon);
+
+  while (msg.message != WM_QUIT) {
+    mg_mgr_poll(&s_mgr, 1);
+    if (PeekMessage(&msg, hWnd, 0, 0, PM_REMOVE) > 0) {
+      TranslateMessage(&msg);
+      DispatchMessage(&msg);
+    }
+  }
+  mg_mgr_free(&s_mgr);
+  return msg.wParam;
+}