mirror of
https://github.com/rustdesk/rustdesk.git
synced 2024-12-15 03:59:04 +08:00
107 lines
3.4 KiB
C++
107 lines
3.4 KiB
C++
#include <flutter/dart_project.h>
|
|
#include <flutter/flutter_view_controller.h>
|
|
#include <windows.h>
|
|
#include <tchar.h>
|
|
#include <iostream>
|
|
|
|
#include "flutter_window.h"
|
|
#include "utils.h"
|
|
// #include <bitsdojo_window_windows/bitsdojo_window_plugin.h>
|
|
|
|
#include <uni_links_desktop/uni_links_desktop_plugin.h>
|
|
|
|
typedef char** (*FUNC_RUSTDESK_CORE_MAIN)(int*);
|
|
typedef void (*FUNC_RUSTDESK_FREE_ARGS)( char**, int);
|
|
const char* uniLinksPrefix = "rustdesk://";
|
|
|
|
// auto bdw = bitsdojo_window_configure(BDW_CUSTOM_FRAME | BDW_HIDE_ON_STARTUP);
|
|
int APIENTRY wWinMain(_In_ HINSTANCE instance, _In_opt_ HINSTANCE prev,
|
|
_In_ wchar_t *command_line, _In_ int show_command)
|
|
{
|
|
HINSTANCE hInstance = LoadLibraryA("librustdesk.dll");
|
|
if (!hInstance)
|
|
{
|
|
std::cout << "Failed to load librustdesk.dll" << std::endl;
|
|
return EXIT_FAILURE;
|
|
}
|
|
FUNC_RUSTDESK_CORE_MAIN rustdesk_core_main =
|
|
(FUNC_RUSTDESK_CORE_MAIN)GetProcAddress(hInstance, "rustdesk_core_main");
|
|
if (!rustdesk_core_main)
|
|
{
|
|
std::cout << "Failed to get rustdesk_core_main" << std::endl;
|
|
return EXIT_FAILURE;
|
|
}
|
|
FUNC_RUSTDESK_FREE_ARGS free_c_args =
|
|
(FUNC_RUSTDESK_FREE_ARGS)GetProcAddress(hInstance, "free_c_args");
|
|
if (!free_c_args)
|
|
{
|
|
std::cout << "Failed to get free_c_args" << std::endl;
|
|
return EXIT_FAILURE;
|
|
}
|
|
std::vector<std::string> command_line_arguments =
|
|
GetCommandLineArguments();
|
|
|
|
int args_len = 0;
|
|
char** c_args = rustdesk_core_main(&args_len);
|
|
if (!c_args)
|
|
{
|
|
std::cout << "Rustdesk core returns false, exiting without launching Flutter app" << std::endl;
|
|
return EXIT_SUCCESS;
|
|
}
|
|
std::vector<std::string> rust_args(c_args, c_args + args_len);
|
|
free_c_args(c_args, args_len);
|
|
|
|
// uni links dispatch
|
|
// only do uni links when dispatch a rustdesk links
|
|
auto prefix = std::string(uniLinksPrefix);
|
|
if (!command_line_arguments.empty() && command_line_arguments.front().compare(0, prefix.size(), prefix.c_str()) == 0) {
|
|
HWND hwnd = ::FindWindow(_T("FLUTTER_RUNNER_WIN32_WINDOW"), _T("RustDesk"));
|
|
if (hwnd != NULL) {
|
|
DispatchToUniLinksDesktop(hwnd);
|
|
|
|
::ShowWindow(hwnd, SW_NORMAL);
|
|
::SetForegroundWindow(hwnd);
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|
|
// Attach to console when present (e.g., 'flutter run') or create a
|
|
// new console when running with a debugger.
|
|
if (!::AttachConsole(ATTACH_PARENT_PROCESS) && ::IsDebuggerPresent())
|
|
{
|
|
CreateAndAttachConsole();
|
|
}
|
|
|
|
// Initialize COM, so that it is available for use in the library and/or
|
|
// plugins.
|
|
::CoInitializeEx(nullptr, COINIT_APARTMENTTHREADED);
|
|
|
|
flutter::DartProject project(L"data");
|
|
// connection manager hide icon from taskbar
|
|
bool showOnTaskBar = true;
|
|
auto cmParam = std::string("--cm");
|
|
if (!command_line_arguments.empty() && command_line_arguments.front().compare(0, cmParam.size(), cmParam.c_str()) == 0) {
|
|
showOnTaskBar = false;
|
|
}
|
|
command_line_arguments.insert(command_line_arguments.end(), rust_args.begin(), rust_args.end());
|
|
project.set_dart_entrypoint_arguments(std::move(command_line_arguments));
|
|
|
|
FlutterWindow window(project);
|
|
Win32Window::Point origin(10, 10);
|
|
Win32Window::Size size(800, 600);
|
|
if (!window.CreateAndShow(L"RustDesk", origin, size, showOnTaskBar))
|
|
{
|
|
return EXIT_FAILURE;
|
|
}
|
|
window.SetQuitOnClose(true);
|
|
|
|
::MSG msg;
|
|
while (::GetMessage(&msg, nullptr, 0, 0))
|
|
{
|
|
::TranslateMessage(&msg);
|
|
::DispatchMessage(&msg);
|
|
}
|
|
|
|
::CoUninitialize();
|
|
return EXIT_SUCCESS;
|
|
}
|