fix deeplinks

ui/desktop/forge.config.ts

@@ -13,6 +13,13 @@ let cfg = {
     rfc3161TimeStampServer: 'http://timestamp.digicert.com',
     signWithParams: '/fd sha256 /tr http://timestamp.digicert.com /td sha256'
   },
+  // Protocol registration
+  protocols: [
+    {
+      name: "GooseProtocol",
+      schemes: ["goose"]
+    }
+  ],
   // macOS specific configuration
   osxSign: {
     entitlements: 'entitlements.plist',
@@ -100,4 +107,4 @@ module.exports = {
       [FuseV1Options.OnlyLoadAppFromAsar]: true,
     }),
   ],
-};
+};

ui/desktop/src/main.ts

@@ -14,6 +14,7 @@ import {
 } from 'electron';
 import started from 'electron-squirrel-startup';
 import path from 'node:path';
+import { handleSquirrelEvent } from './setup-events';
 import { startGoosed } from './goosed';
 import { getBinaryPath } from './utils/binaryPath';
 import { loadShellEnv } from './utils/loadEnv';
@@ -26,25 +27,112 @@ import {
   saveSettings,
   updateEnvironmentVariables,
 } from './utils/settings';
-const { exec } = require('child_process');
+import * as crypto from 'crypto';
+import * as electron from 'electron';
+import { exec as execCallback } from 'child_process';
+import { promisify } from 'util';
+
+const exec = promisify(execCallback);
+
+// Handle Squirrel events for Windows installer
+if (process.platform === 'win32') {
+  console.log('Windows detected, command line args:', process.argv);
+
+  if (handleSquirrelEvent()) {
+    // squirrel event handled and app will exit in 1000ms, so don't do anything else
+    process.exit(0);
+  }
+
+  // Handle the protocol on Windows during first launch
+  if (process.argv.length >= 2) {
+    const url = process.argv[1];
+    console.log('Checking URL from command line:', url);
+    if (url.startsWith('goose://')) {
+      console.log('Found goose:// URL in command line args');
+      app.emit('open-url', { preventDefault: () => {} }, url);
+    }
+  }
+}
+
+// Ensure single instance lock
+const gotTheLock = app.requestSingleInstanceLock();
+
+if (!gotTheLock) {
+  app.quit();
+} else {
+  app.on('second-instance', (event, commandLine, _workingDirectory) => {
+    // Someone tried to run a second instance
+    console.log('Second instance detected with args:', commandLine);
+
+    // Get existing window or create new one
+    const existingWindows = BrowserWindow.getAllWindows();
+    if (existingWindows.length > 0) {
+      const window = existingWindows[0];
+      if (window.isMinimized()) window.restore();
+      window.focus();
+
+      if (process.platform === 'win32') {
+        // Protocol handling for Windows
+        const url = commandLine[commandLine.length - 1];
+        console.log('Checking last arg for protocol:', url);
+        if (url.startsWith('goose://')) {
+          console.log('Found goose:// URL in second instance');
+          // Send the URL to the window
+          if (!window.webContents.isLoading()) {
+            window.webContents.send('add-extension', url);
+          } else {
+            window.webContents.once('did-finish-load', () => {
+              window.webContents.send('add-extension', url);
+            });
+          }
+        }
+      }
+    }
+  });
+}
 
 // Handle creating/removing shortcuts on Windows when installing/uninstalling.
 if (started) app.quit();
 
+// Register protocol handler
+if (process.platform === 'win32') {
+  const success = app.setAsDefaultProtocolClient('goose', process.execPath, ['--']);
+  console.log('Registering protocol handler for Windows:', success ? 'success' : 'failed');
+} else {
+  const success = app.setAsDefaultProtocolClient('goose');
+  console.log('Registering protocol handler:', success ? 'success' : 'failed');
+}
+
+// Log if we're the default protocol handler
+console.log('Is default protocol handler:', app.isDefaultProtocolClient('goose'));
+
 // Triggered when the user opens "goose://..." links
 app.on('open-url', async (event, url) => {
   event.preventDefault();
   console.log('open-url:', url);
 
-  const recentDirs = loadRecentDirs();
-  const openDir = recentDirs.length > 0 ? recentDirs[0] : null;
+  // Get existing window or create new one
+  let targetWindow: BrowserWindow;
+  const existingWindows = BrowserWindow.getAllWindows();
 
-  // Create the new Chat window
-  const newWindow = await createChat(app, undefined, openDir);
+  if (existingWindows.length > 0) {
+    targetWindow = existingWindows[0];
+    if (targetWindow.isMinimized()) targetWindow.restore();
+    targetWindow.focus();
+  } else {
+    const recentDirs = loadRecentDirs();
+    const openDir = recentDirs.length > 0 ? recentDirs[0] : null;
+    targetWindow = await createChat(app, undefined, openDir);
+  }
 
-  newWindow.webContents.once('did-finish-load', () => {
-    newWindow.webContents.send('add-extension', url);
-  });
+  // Wait for window to be ready before sending the extension URL
+  if (!targetWindow.webContents.isLoading()) {
+    targetWindow.webContents.send('add-extension', url);
+  } else {
+    targetWindow.webContents.once('did-finish-load', () => {
+      targetWindow.webContents.send('add-extension', url);
+    });
+  }
 });
 
 declare var MAIN_WINDOW_VITE_DEV_SERVER_URL: string;
@@ -77,8 +165,7 @@ const getGooseProvider = () => {
 };
 
 const generateSecretKey = () => {
-  const crypto = require('crypto');
-  let key = crypto.randomBytes(32).toString('hex');
+  const key = crypto.randomBytes(32).toString('hex');
   process.env.GOOSE_SERVER__SECRET_KEY = key;
   return key;
 };
@@ -110,8 +197,7 @@ const createLauncher = () => {
   });
 
   // Center on screen
-  const { screen } = require('electron');
-  const primaryDisplay = screen.getPrimaryDisplay();
+  const primaryDisplay = electron.screen.getPrimaryDisplay();
   const { width, height } = primaryDisplay.workAreaSize;
   const windowBounds = launcherWindow.getBounds();
 
@@ -141,8 +227,6 @@ let windowCounter = 0;
 const windowMap = new Map<number, BrowserWindow>();
 
 const createChat = async (app, query?: string, dir?: string, version?: string) => {
-  const env = version ? { GOOSE_AGENT_VERSION: version } : {};
-
   // Apply current environment settings before creating chat
   updateEnvironmentVariables(envToggles);
 
@@ -186,8 +270,7 @@ const createChat = async (app, query?: string, dir?: string, version?: string) =
 
   // Load the index.html of the app.
   const queryParam = query ? `?initialQuery=${encodeURIComponent(query)}` : '';
-  const { screen } = require('electron');
-  const primaryDisplay = screen.getPrimaryDisplay();
+  const primaryDisplay = electron.screen.getPrimaryDisplay();
   const { width } = primaryDisplay.workAreaSize;
 
   // Increment window counter to track number of windows
@@ -349,7 +432,7 @@ ipcMain.handle('select-file-or-directory', async () => {
 
 ipcMain.handle('check-ollama', async () => {
   try {
-    return new Promise((resolve, reject) => {
+    return new Promise((resolve) => {
       // Run `ps` and filter for "ollama"
       exec('ps aux | grep -iw "[o]llama"', (error, stdout, stderr) => {
         if (error) {

ui/desktop/src/setup-events.ts

@@ -0,0 +1,71 @@
+import { app } from 'electron';
+import * as path from 'path';
+import { spawn as spawnProcess } from 'child_process';
+import * as fs from 'fs';
+
+export function handleSquirrelEvent(): boolean {
+  if (process.argv.length === 1) {
+    return false;
+  }
+
+  const appFolder = path.resolve(process.execPath, '..');
+  const rootAtomFolder = path.resolve(appFolder, '..');
+  const updateDotExe = path.resolve(path.join(rootAtomFolder, 'Update.exe'));
+  const exeName = path.basename(process.execPath);
+
+  const spawnUpdate = function (args: string[]) {
+    try {
+      return spawnProcess(updateDotExe, args, { detached: true });
+    } catch (error) {
+      console.error('Failed to spawn update process:', error);
+      return null;
+    }
+  };
+
+  const squirrelEvent = process.argv[1];
+  switch (squirrelEvent) {
+    case '--squirrel-install':
+    case '--squirrel-updated': {
+      // Register protocol handler
+      spawnUpdate(['--createShortcut', exeName]);
+
+      // Register protocol
+      const regCommand = `Windows Registry Editor Version 5.00
+
+[HKEY_CLASSES_ROOT\\goose]
+@="URL:Goose Protocol"
+"URL Protocol"=""
+
+[HKEY_CLASSES_ROOT\\goose\\DefaultIcon]
+@="\\"${process.execPath.replace(/\\/g, '\\\\')},1\\""
+
+[HKEY_CLASSES_ROOT\\goose\\shell]
+
+[HKEY_CLASSES_ROOT\\goose\\shell\\open]
+
+[HKEY_CLASSES_ROOT\\goose\\shell\\open\\command]
+@="\\"${process.execPath.replace(/\\/g, '\\\\')}\\" \\"%1\\""`;
+
+      fs.writeFileSync('goose-protocol.reg', regCommand);
+      spawnProcess('regedit.exe', ['/s', 'goose-protocol.reg']);
+
+      setTimeout(() => app.quit(), 1000);
+      return true;
+    }
+
+    case '--squirrel-uninstall': {
+      // Remove protocol handler
+      spawnUpdate(['--removeShortcut', exeName]);
+      setTimeout(() => app.quit(), 1000);
+      return true;
+    }
+
+    case '--squirrel-obsolete': {
+      app.quit();
+      return true;
+    }
+
+    default:
+      return false;
+  }
+}