Add GUI creation if no command line arguments are provided

gui.c

@@ -0,0 +1,211 @@
+#include <windows.h>
+#include <winuser.h>
+#include <assert.h>
+#include <stdio.h>
+
+/**
+ * The entire code could be better written, but at least it works.
+ *
+ * This will make installation and removal of the bridge WAY easier.
+ */
+
+LPTSTR GetErrorMessage();
+void print(char const *fmt, ...);
+void InstallService(int ServiceStartType, LPCSTR Path);
+void RemoveService();
+
+VOID HandleStartButton(BOOL Silent)
+{
+	SC_HANDLE hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
+	if (hSCManager == NULL)
+	{
+		print("OpenSCManager failed: %s\n", GetErrorMessage());
+		return;
+	}
+
+	SC_HANDLE schService = OpenService(hSCManager, "rpc-bridge",
+									   SERVICE_QUERY_CONFIG | SERVICE_CHANGE_CONFIG | SERVICE_START);
+	if (schService == NULL)
+	{
+		print("OpenService failed: %s\n", GetErrorMessage());
+		return;
+	}
+
+	DWORD dwBytesNeeded;
+	QueryServiceConfig(schService, NULL, 0, &dwBytesNeeded);
+	LPQUERY_SERVICE_CONFIG lpqsc = (LPQUERY_SERVICE_CONFIG)LocalAlloc(LPTR, dwBytesNeeded);
+	if (lpqsc == NULL)
+	{
+		print("LocalAlloc failed: %s\n", GetErrorMessage());
+		return;
+	}
+
+	if (!QueryServiceConfig(schService, lpqsc, dwBytesNeeded, &dwBytesNeeded))
+	{
+		print("QueryServiceConfig failed: %s\n", GetErrorMessage());
+		return;
+	}
+
+	if (StartService(schService, 0, NULL) == FALSE)
+	{
+		if (GetLastError() == ERROR_SERVICE_ALREADY_RUNNING)
+			return;
+		print("StartService failed: %s\n", GetErrorMessage());
+	}
+
+	LocalFree(lpqsc);
+	CloseServiceHandle(schService);
+	CloseServiceHandle(hSCManager);
+	if (Silent == FALSE)
+		MessageBox(NULL, "Bridge service started successfully", "Info", MB_OK);
+}
+
+VOID HandleInstallButton()
+{
+	char filename[MAX_PATH];
+	GetModuleFileName(NULL, filename, MAX_PATH);
+	CopyFile(filename, "C:\\windows\\bridge.exe", FALSE);
+	InstallService(SERVICE_AUTO_START, "C:\\windows\\bridge.exe --service");
+	MessageBox(NULL, "Bridge installed successfully", "Info", MB_OK);
+	HandleStartButton(TRUE);
+	ExitProcess(0);
+}
+
+VOID HandleRemoveButton()
+{
+	RemoveService();
+	MessageBox(NULL, "Bridge removed successfully", "Info", MB_OK);
+	ExitProcess(0);
+}
+
+LRESULT CALLBACK WndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
+{
+	switch (msg)
+	{
+	case WM_COMMAND:
+	{
+		switch (LOWORD(wParam))
+		{
+		case 1:
+			HandleStartButton(FALSE);
+			break;
+		case 2:
+			HandleInstallButton();
+			break;
+		case 3:
+			HandleRemoveButton();
+			break;
+		default:
+			break;
+		}
+		break;
+	}
+	case WM_CLOSE:
+		DestroyWindow(hwnd);
+		break;
+	case WM_DESTROY:
+		PostQuitMessage(0);
+		ExitProcess(0);
+		break;
+	default:
+		return DefWindowProc(hwnd, msg, wParam, lParam);
+	}
+	return 0;
+}
+
+int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,
+				   LPSTR lpCmdLine, int nCmdShow)
+{
+	SC_HANDLE hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
+	SC_HANDLE schService = OpenService(hSCManager, "rpc-bridge", SERVICE_START | SERVICE_QUERY_STATUS);
+
+	INT btnStartStyle = WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON;
+	INT btnRemoveStyle = WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON;
+	INT btnInstallStyle = WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON;
+
+	if (schService != NULL)
+	{
+		btnInstallStyle |= WS_DISABLED;
+
+		SERVICE_STATUS_PROCESS ssStatus;
+		DWORD dwBytesNeeded;
+		assert(QueryServiceStatusEx(schService, SC_STATUS_PROCESS_INFO, (LPBYTE)&ssStatus,
+									sizeof(SERVICE_STATUS_PROCESS), &dwBytesNeeded));
+
+		if (ssStatus.dwCurrentState == SERVICE_RUNNING ||
+			ssStatus.dwCurrentState == SERVICE_START_PENDING)
+			btnStartStyle |= WS_DISABLED;
+	}
+	else
+	{
+		btnStartStyle |= WS_DISABLED;
+		btnRemoveStyle |= WS_DISABLED;
+	}
+
+	CloseServiceHandle(schService);
+	CloseServiceHandle(hSCManager);
+
+	const char szClassName[] = "BridgeWindowClass";
+
+	WNDCLASSEX wc;
+	wc.cbSize = sizeof(WNDCLASSEX);
+	wc.style = 0;
+	wc.lpfnWndProc = WndProc;
+	wc.cbClsExtra = 0;
+	wc.cbWndExtra = 0;
+	wc.hInstance = hInstance;
+	wc.hIcon = LoadIcon(NULL, IDI_WINLOGO);
+	wc.hCursor = LoadCursor(NULL, IDC_ARROW);
+	wc.hbrBackground = (HBRUSH)(COLOR_WINDOW + 1);
+	wc.lpszMenuName = NULL;
+	wc.lpszClassName = szClassName;
+	wc.hIconSm = LoadIcon(NULL, IDI_WINLOGO);
+
+	assert(RegisterClassEx(&wc));
+
+	HWND hwnd = CreateWindowEx(WS_EX_CLIENTEDGE,
+							   szClassName,
+							   "Discord RPC Bridge",
+							   WS_OVERLAPPEDWINDOW & ~WS_THICKFRAME,
+							   (GetSystemMetrics(SM_CXSCREEN) - 400) / 2,
+							   (GetSystemMetrics(SM_CYSCREEN) - 150) / 2,
+							   400, 150,
+							   NULL, NULL, hInstance, NULL);
+
+	CreateWindow("STATIC", "Do you want to start, install or remove the bridge?",
+				 WS_CHILD | WS_VISIBLE | SS_CENTER,
+				 0, 0, 400, 50,
+				 hwnd, NULL, hInstance, NULL);
+
+	CreateWindow("BUTTON", "Start",
+				 btnStartStyle,
+				 50, 50, 100, 30,
+				 hwnd, (HMENU)1, hInstance, NULL);
+
+	CreateWindow("BUTTON", "Install",
+				 btnInstallStyle,
+				 150, 50, 100, 30,
+				 hwnd, (HMENU)2, hInstance, NULL);
+
+	CreateWindow("BUTTON", "Remove",
+				 btnRemoveStyle,
+				 250, 50, 100, 30,
+				 hwnd, (HMENU)3, hInstance, NULL);
+
+	ShowWindow(hwnd, nCmdShow);
+	UpdateWindow(hwnd);
+
+	MSG msg;
+	while (GetMessage(&msg, NULL, 0, 0) > 0)
+	{
+		TranslateMessage(&msg);
+		DispatchMessage(&msg);
+	}
+	return msg.wParam;
+}
+
+void CreateGUI()
+{
+	ShowWindow(GetConsoleWindow(), SW_MINIMIZE);
+	ExitProcess(WinMain(GetModuleHandle(NULL), NULL, GetCommandLine(), SW_SHOWNORMAL));
+}

main.c

@@ -6,6 +6,7 @@
 FILE *g_logFile = NULL;
 BOOL RunningAsService = FALSE;
 
+void CreateGUI();
 void CreateBridge();
 void LaunchGame(int argc, char **argv);
 void ServiceMain(int argc, char *argv[]);
@@ -261,17 +262,25 @@ void HandleArguments(int argc, char *argv[])
 	else if (strcmp(argv[1], "--help") == 0)
 	{
 		printf("Usage:\n");
-		printf("  %s [args]\n", argv[0]);
+		printf("  %s [args]\n\n", argv[0]);
-		printf("  --help       - Show this help\n");
+
-		printf("  --install    - Install service\n");
+		printf("Arguments:\n");
-		printf("        This will copy the binary to C:\\windows\\bridge.exe and register it as a service\n");
+		printf("  --help         Show this help\n\n");
-		printf("  --uninstall  - Uninstall service\n");
+
-		printf("        This will remove the service and delete C:\\windows\\bridge.exe\n");
+		printf("  --install      Install service\n");
-		printf("  --steam      - Reserved for Steam\n");
+		printf("        This will copy the binary to C:\\windows\\bridge.exe and register it as a service\n\n");
-		printf("        This will start the service and exit (used with bridge.sh)\n");
+		printf("  --uninstall    Uninstall service\n");
-		printf("  --no-service - Do not run as service\n");
+
-		printf("        (only for --steam)\n");
+		printf("        This will remove the service and delete C:\\windows\\bridge.exe\n\n");
-		printf("  --service    - Reserved for service\n");
+		printf("  --steam        Reserved for Steam\n");
+
+		printf("        This will start the service and exit (used with bridge.sh)\n\n");
+		printf("  --no-service   Do not run as service\n");
+
+		printf("        (only for --steam)\n\n");
+		printf("  --service      Reserved for service\n\n");
+
+		printf("Note: If no arguments are provided, the GUI will be shown instead\n");
 		ExitProcess(0);
 	}
 }
@@ -290,6 +299,8 @@ int main(int argc, char *argv[])
 
 	if (argc > 1)
 		HandleArguments(argc, argv);
+	else
+		CreateGUI();
 
 	CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)CreateBridge,
 				 NULL, 0, NULL);