9.0.4

gui/lib/src/util/os.dart

@@ -2,18 +2,399 @@ import 'dart:io';
 
 import 'package:fluent_ui/fluent_ui.dart';
 import 'package:flutter/scheduler.dart';
+import 'dart:collection';
+import 'dart:ffi';
+import 'package:ffi/ffi.dart';
+import 'package:win32/win32.dart';
 
 final RegExp _winBuildRegex = RegExp(r'(?<=\(Build )(.*)(?=\))');
 
-bool get isWin11 {
-  var result = _winBuildRegex.firstMatch(Platform.operatingSystemVersion)?.group(1);
-  if(result == null){
-    return false;
+int? get windowsBuild {
+  final result = _winBuildRegex.firstMatch(Platform.operatingSystemVersion)?.group(1);
+  if (result == null) {
+    return null;
   }
 
-  var intBuild = int.tryParse(result);
+  return int.tryParse(result);
+}
+
+bool get isWin11 {
+  final intBuild = windowsBuild;
   return intBuild != null && intBuild > 22000;
 }
 
-bool get isDarkMode
-    => SchedulerBinding.instance.platformDispatcher.platformBrightness.isDark;
+bool get isDarkMode =>
+    SchedulerBinding.instance.platformDispatcher.platformBrightness.isDark;
+
+class _ServiceProvider10 extends IUnknown {
+  static const String _CLSID = "{C2F03A33-21F5-47FA-B4BB-156362A2F239}";
+  static const String _IID = "{6D5140C1-7436-11CE-8034-00AA006009FA}";
+
+  _ServiceProvider10._internal(Pointer<COMObject> ptr) : super(ptr);
+
+  factory _ServiceProvider10.createInstance() =>
+      _ServiceProvider10._internal(COMObject.createFromID(_CLSID, _IID));
+
+  Pointer<COMObject> queryService(String classId, String instanceId) {
+    final result = calloc<COMObject>();
+    final code = (ptr.ref.vtable + 3)
+            .cast<
+                Pointer<
+                    NativeFunction<
+                        HRESULT Function(Pointer, Pointer<GUID>, Pointer<GUID>,
+                            Pointer<COMObject>)>>>()
+            .value
+            .asFunction<
+                int Function(Pointer, Pointer<GUID>, Pointer<GUID>,
+                    Pointer<COMObject>)>()(ptr.ref.lpVtbl,
+        GUIDFromString(classId), GUIDFromString(instanceId), result);
+    if (code != 0) {
+      free(result);
+      throw WindowsException(code);
+    }
+
+    return result;
+  }
+}
+
+class IVirtualDesktop extends IUnknown {
+  static const String _CLSID = "{3F07F4BE-B107-441A-AF0F-39D82529072C}";
+
+  IVirtualDesktop._internal(super.ptr);
+
+  String getName() {
+    final result = calloc<HSTRING>();
+    final code = (ptr.ref.vtable + 5)
+        .cast<
+            Pointer<
+                NativeFunction<HRESULT Function(Pointer, Pointer<HSTRING>)>>>()
+        .value
+        .asFunction<
+            int Function(Pointer, Pointer<HSTRING>)>()(ptr.ref.lpVtbl, result);
+    if (code != 0) {
+      free(result);
+      throw WindowsException(code);
+    }
+
+    return convertFromHString(result.value);
+  }
+}
+
+class IApplicationView extends IUnknown {
+  // static const String _CLSID = "{372E1D3B-38D3-42E4-A15B-8AB2B178F513}";
+
+  IApplicationView._internal(super.ptr);
+}
+
+class _IObjectArray extends IUnknown {
+  _IObjectArray(super.ptr);
+
+  int getCount() {
+    final result = calloc<Int32>();
+    final code = (ptr.ref.vtable + 3)
+        .cast<
+            Pointer<
+                NativeFunction<HRESULT Function(Pointer, Pointer<Int32>)>>>()
+        .value
+        .asFunction<
+            int Function(Pointer, Pointer<Int32>)>()(ptr.ref.lpVtbl, result);
+    if (code != 0) {
+      free(result);
+      throw WindowsException(code);
+    }
+
+    return result.value;
+  }
+
+  Pointer<COMObject> getAt(int index, String guid) {
+    final result = calloc<COMObject>();
+    final code = (ptr.ref.vtable + 4)
+            .cast<
+                Pointer<
+                    NativeFunction<
+                        HRESULT Function(Pointer, Int32 index, Pointer<GUID>,
+                            Pointer<COMObject>)>>>()
+            .value
+            .asFunction<
+                int Function(
+                    Pointer, int index, Pointer<GUID>, Pointer<COMObject>)>()(
+        ptr.ref.lpVtbl, index, GUIDFromString(guid), result);
+    if (code != 0) {
+      free(result);
+      throw WindowsException(code);
+    }
+
+    return result;
+  }
+}
+
+typedef _IObjectMapper<T> = T Function(Pointer<COMObject>);
+
+class _IObjectArrayList<T> extends ListBase<T> {
+  final _IObjectArray _array;
+  final String _guid;
+  final _IObjectMapper<T> _mapper;
+
+  _IObjectArrayList(
+      {required _IObjectArray array,
+      required String guid,
+      required _IObjectMapper<T> mapper})
+      : _array = array,
+        _guid = guid,
+        _mapper = mapper;
+
+  @override
+  int get length => _array.getCount();
+
+  @override
+  set length(int newLength) {
+    throw UnsupportedError("Immutable list");
+  }
+
+  @override
+  T operator [](int index) => _mapper(_array.getAt(index, _guid));
+
+  @override
+  void operator []=(int index, T value) {
+    throw UnsupportedError("Immutable list");
+  }
+}
+
+class _IVirtualDesktopManagerInternal extends IUnknown {
+  static const String _CLSID = "{C5E0CDCA-7B6E-41B2-9FC4-D93975CC467B}";
+  static const String _IID_WIN10 = "{F31574D6-B682-4CDC-BD56-1827860ABEC6}";
+  static const String _IID_WIN_21H2 = "{B2F925B9-5A0F-4D2E-9F4D-2B1507593C10}";
+  static const String _IID_WIN_23H2 = "{A3175F2D-239C-4BD2-8AA0-EEBA8B0B138E}";
+  static const String _IID_WIN_23H2_3085 = "{53F5CA0B-158F-4124-900C-057158060B27}";
+
+  _IVirtualDesktopManagerInternal._internal(super.ptr);
+
+  int getDesktopsCount() {
+    final result = calloc<Int32>();
+    final code = (ptr.ref.vtable + 3)
+        .cast<
+            Pointer<
+                NativeFunction<HRESULT Function(Pointer, Pointer<Int32>)>>>()
+        .value
+        .asFunction<
+            int Function(Pointer, Pointer<Int32>)>()(ptr.ref.lpVtbl, result);
+    if (code != 0) {
+      free(result);
+      throw WindowsException(code);
+    }
+
+    return result.value;
+  }
+
+  List<IVirtualDesktop> getDesktops() {
+    final result = calloc<COMObject>();
+    final code = (ptr.ref.vtable + 7)
+            .cast<
+                Pointer<
+                    NativeFunction<
+                        HRESULT Function(Pointer, Pointer<COMObject>)>>>()
+            .value
+            .asFunction<int Function(Pointer, Pointer<COMObject>)>()(
+        ptr.ref.lpVtbl, result);
+    if (code != 0) {
+      free(result);
+      throw WindowsException(code);
+    }
+
+    final array = _IObjectArray(result);
+    return _IObjectArrayList(
+        array: array,
+        guid: IVirtualDesktop._CLSID,
+        mapper: (comObject) => IVirtualDesktop._internal(comObject));
+  }
+
+  void moveWindowToDesktop(IApplicationView view, IVirtualDesktop desktop) {
+    final code = (ptr.ref.vtable + 4)
+            .cast<
+                Pointer<
+                    NativeFunction<
+                        Int32 Function(Pointer, COMObject, COMObject)>>>()
+            .value
+            .asFunction<int Function(Pointer, COMObject, COMObject)>()(
+        ptr.ref.lpVtbl, view.ptr.ref, desktop.ptr.ref);
+    if (code != 0) {
+      throw WindowsException(code);
+    }
+  }
+
+  IVirtualDesktop createDesktop() {
+    final result = calloc<COMObject>();
+    final code = (ptr.ref.vtable + 10)
+            .cast<
+                Pointer<
+                    NativeFunction<
+                        HRESULT Function(Pointer, Pointer<COMObject>)>>>()
+            .value
+            .asFunction<int Function(Pointer, Pointer<COMObject>)>()(
+        ptr.ref.lpVtbl, result);
+    if (code != 0) {
+      free(result);
+      throw WindowsException(code);
+    }
+
+    return IVirtualDesktop._internal(result);
+  }
+
+  void removeDesktop(IVirtualDesktop desktop, IVirtualDesktop fallback) {
+    final code = (ptr.ref.vtable + 12)
+            .cast<
+                Pointer<
+                    NativeFunction<
+                        HRESULT Function(Pointer, COMObject, COMObject)>>>()
+            .value
+            .asFunction<int Function(Pointer, COMObject, COMObject)>()(
+        ptr.ref.lpVtbl, desktop.ptr.ref, fallback.ptr.ref);
+    if (code != 0) {
+      throw WindowsException(code);
+    }
+  }
+
+  void setDesktopName(IVirtualDesktop desktop, String newName) {
+    final code =
+        (ptr.ref.vtable + 15)
+                .cast<
+                    Pointer<
+                        NativeFunction<
+                            HRESULT Function(Pointer, COMObject, Int8)>>>()
+                .value
+                .asFunction<int Function(Pointer, COMObject, int)>()(
+            ptr.ref.lpVtbl, desktop.ptr.ref, convertToHString(newName));
+    if (code != 0) {
+      throw WindowsException(code);
+    }
+  }
+}
+
+class _IApplicationViewCollection extends IUnknown {
+  static const String _CLSID = "{1841C6D7-4F9D-42C0-AF41-8747538F10E5}";
+  static const String _IID = "{1841C6D7-4F9D-42C0-AF41-8747538F10E5}";
+
+  _IApplicationViewCollection._internal(super.ptr);
+
+  IApplicationView getViewForHWnd(int HWnd) {
+    final result = calloc<COMObject>();
+    final code =
+        (ptr.ref.vtable + 6)
+                .cast<
+                    Pointer<
+                        NativeFunction<
+                            HRESULT Function(
+                                Pointer, IntPtr, Pointer<COMObject>)>>>()
+                .value
+                .asFunction<int Function(Pointer, int, Pointer<COMObject>)>()(
+            ptr.ref.lpVtbl, HWnd, result);
+    if (code != 0) {
+      free(result);
+      throw WindowsException(code);
+    }
+
+    return IApplicationView._internal(result);
+  }
+}
+
+final class _Process extends Struct {
+  @Uint32()
+  external int pid;
+
+  @Uint32()
+  external int HWnd;
+
+  static int _filter(int HWnd, int lParam) {
+    final structure = Pointer.fromAddress(lParam).cast<_Process>();
+    final pidPointer = calloc<Uint32>();
+    GetWindowThreadProcessId(HWnd, pidPointer);
+    final pid = pidPointer.value;
+    free(pidPointer);
+    if (pid != structure.ref.pid) {
+      return TRUE;
+    }
+
+    structure.ref.HWnd = HWnd;
+    return FALSE;
+  }
+
+  static int getHWndFromPid(int pid) {
+    final result = calloc<_Process>();
+    result.ref.pid = pid;
+    EnumWindows(
+        Pointer.fromFunction<EnumWindowsProc>(_filter, TRUE), result.address);
+    final HWnd = result.ref.HWnd;
+    calloc.free(result);
+    return HWnd;
+  }
+}
+
+class VirtualDesktopManager {
+  static VirtualDesktopManager? _instance;
+
+  final _IVirtualDesktopManagerInternal windowManager;
+  final _IApplicationViewCollection applicationViewCollection;
+
+  VirtualDesktopManager._internal(this.windowManager, this.applicationViewCollection);
+
+  factory VirtualDesktopManager.getInstance() {
+    if (_instance != null) {
+      return _instance!;
+    }
+
+    final hr = CoInitializeEx(
+        nullptr, COINIT_APARTMENTTHREADED | COINIT_DISABLE_OLE1DDE);
+    if (FAILED(hr)) {
+      throw WindowsException(hr);
+    }
+
+    final shell = _ServiceProvider10.createInstance();
+    final windowManager = _createWindowManager(shell);
+    final applicationViewCollection = _IApplicationViewCollection._internal(
+        shell.queryService(_IApplicationViewCollection._CLSID,
+            _IApplicationViewCollection._IID));
+    return _instance =
+        VirtualDesktopManager._internal(windowManager, applicationViewCollection);
+  }
+
+  static _IVirtualDesktopManagerInternal _createWindowManager(_ServiceProvider10 shell) {
+    final build = windowsBuild;
+    if(build == null || build < 19044) {
+      return _IVirtualDesktopManagerInternal._internal(
+          shell.queryService(_IVirtualDesktopManagerInternal._CLSID,
+              _IVirtualDesktopManagerInternal._IID_WIN10));
+    }else if(build >= 19044 && build < 22631) {
+      return _IVirtualDesktopManagerInternal._internal(
+          shell.queryService(_IVirtualDesktopManagerInternal._CLSID,
+              _IVirtualDesktopManagerInternal._IID_WIN_21H2));
+    }else if(build >= 22631 && build < 22631) {
+      return _IVirtualDesktopManagerInternal._internal(
+          shell.queryService(_IVirtualDesktopManagerInternal._CLSID,
+              _IVirtualDesktopManagerInternal._IID_WIN_23H2));
+    }else {
+      return _IVirtualDesktopManagerInternal._internal(
+          shell.queryService(_IVirtualDesktopManagerInternal._CLSID,
+              _IVirtualDesktopManagerInternal._IID_WIN_23H2_3085));
+    }
+  }
+
+  int getDesktopsCount() => windowManager.getDesktopsCount();
+
+  List<IVirtualDesktop> getDesktops() => windowManager.getDesktops();
+
+  void moveWindowToDesktop(int pid, IVirtualDesktop desktop) {
+    final HWnd = _Process.getHWndFromPid(pid);
+    final window = applicationViewCollection.getViewForHWnd(HWnd);
+    windowManager.moveWindowToDesktop(window, desktop);
+  }
+
+  IVirtualDesktop createDesktop() => windowManager.createDesktop();
+
+  void removeDesktop(IVirtualDesktop desktop, [IVirtualDesktop? fallback]) {
+    fallback ??= getDesktops().first;
+    return windowManager.removeDesktop(desktop, fallback);
+  }
+
+  void setDesktopName(IVirtualDesktop desktop, String newName) =>
+      windowManager.setDesktopName(desktop, newName);
+}