From c2db8b356c230ff532ac17ffadcfb9f37404be1b Mon Sep 17 00:00:00 2001
From: Cats Juice <wp1298554944@hotmail.com>
Date: Thu, 21 Dec 2023 15:44:21 +0800
Subject: [PATCH] feat(electron): handle onboarding-window moved between
 screens (#5345)

---
 .../components/affine/onboarding/style.css.ts |  3 +-
 .../frontend/electron/src/main/onboarding.ts  | 30 +++++++++++++++++--
 2 files changed, 29 insertions(+), 4 deletions(-)

diff --git a/packages/frontend/core/src/components/affine/onboarding/style.css.ts b/packages/frontend/core/src/components/affine/onboarding/style.css.ts
index 2f8ba3b561..3ae2d0294d 100644
--- a/packages/frontend/core/src/components/affine/onboarding/style.css.ts
+++ b/packages/frontend/core/src/components/affine/onboarding/style.css.ts
@@ -100,8 +100,7 @@ export const onboarding = style([
         animation: `${fadeIn} 0.8s linear`,
         // content: 'unset',
         background:
-          // 'linear-gradient(180deg, rgba(0,0,0,0.8) 0%, rgba(0,0,0,0) 99.58%)',
-          'linear-gradient(180deg, rgba(0,0,0,0.9) 0%, rgba(0,0,0,0) 80%)',
+          'linear-gradient(180deg, rgba(0,0,0,0.8) 0%, rgba(0,0,0,0) 99.58%)',
       },
       '&[data-is-window="true"][data-is-desktop="true"]::after': {
         opacity: 0,
diff --git a/packages/frontend/electron/src/main/onboarding.ts b/packages/frontend/electron/src/main/onboarding.ts
index 714bf19028..57b3adb625 100644
--- a/packages/frontend/electron/src/main/onboarding.ts
+++ b/packages/frontend/electron/src/main/onboarding.ts
@@ -1,10 +1,16 @@
-import { BrowserWindow, screen } from 'electron';
+import { BrowserWindow, type Display, screen } from 'electron';
 import { join } from 'path';
 
+import { isMacOS } from '../shared/utils';
 import { mainWindowOrigin } from './constants';
 // import { getExposedMeta } from './exposed';
 import { logger } from './logger';
 
+const getScreenSize = (display: Display) => {
+  const { width, height } = isMacOS() ? display.bounds : display.workArea;
+  return { width, height };
+};
+
 // todo: not all window need all of the exposed meta
 const getWindowAdditionalArguments = async () => {
   const { getExposedMeta } = await import('./exposed');
@@ -15,11 +21,24 @@ const getWindowAdditionalArguments = async () => {
   ];
 };
 
+function fullscreenAndCenter(browserWindow: BrowserWindow) {
+  const position = browserWindow.getPosition();
+  const size = browserWindow.getSize();
+  const currentDisplay = screen.getDisplayNearestPoint({
+    x: position[0] + size[0] / 2,
+    y: position[1] + size[1] / 2,
+  });
+  if (!currentDisplay) return;
+  const { width, height } = getScreenSize(currentDisplay);
+  browserWindow.setSize(width, height);
+  browserWindow.center();
+}
+
 async function createOnboardingWindow(additionalArguments: string[]) {
   logger.info('creating onboarding window');
 
   // get user's screen size
-  const { width, height } = screen.getPrimaryDisplay().workAreaSize;
+  const { width, height } = getScreenSize(screen.getPrimaryDisplay());
 
   const browserWindow = new BrowserWindow({
     width,
@@ -36,6 +55,7 @@ async function createOnboardingWindow(additionalArguments: string[]) {
     // skipTaskbar: true,
     transparent: true,
     hasShadow: false,
+    roundedCorners: false,
     webPreferences: {
       webgl: true,
       preload: join(__dirname, './preload.js'),
@@ -56,12 +76,18 @@ async function createOnboardingWindow(additionalArguments: string[]) {
   browserWindow.on('ready-to-show', () => {
     // forcing zoom factor to 1 to avoid onboarding display issues
     browserWindow.webContents.setZoomFactor(1);
+    fullscreenAndCenter(browserWindow);
     // TODO: add a timeout to avoid flickering, window is ready, but dom is not ready
     setTimeout(() => {
       browserWindow.show();
     }, 300);
   });
 
+  // When moved to another screen, resize to fit the screen
+  browserWindow.on('moved', () => {
+    fullscreenAndCenter(browserWindow);
+  });
+
   await browserWindow.loadURL(
     `${mainWindowOrigin}${mainWindowOrigin.endsWith('/') ? '' : '/'}onboarding`
   );