refactor(tauri): update script handling in execute function

- Modified the `execute` function to wrap string scripts in an IIFE, ensuring they are callable as statement expressions.
- Enhanced comments for clarity on the handling of both string and function scripts, improving understanding of the execution flow.

Author: goosewobbler

packages/tauri-service/src/service.ts

@@ -319,22 +319,22 @@ export default class TauriWorkerService {
       ...args: InnerArguments
     ): Promise<ReturnValue> {
       // For functions: use .toString() - produces valid JS function source
-      // For strings: pass as-is (wrapper template wraps in parentheses to make callable)
-      const scriptString = typeof script === 'function' ? script.toString() : script;
+      // For strings:
+      //   - embedded: pass as-is (WebDriver handles execution)
+      //   - non-embedded: wrap in async IIFE to make statement expressions callable
+      const scriptString =
+        typeof script === 'function' ? script.toString() : isEmbedded ? script : `(async () => ${script})()`;
 
       if (isEmbedded) {
         // For embedded WebDriver: skip console wrapper as console forwarding
         // is handled by tauri-plugin-webdriver.
         return originalExecute(scriptString, ...args) as Promise<ReturnValue>;
       }
 
-      // For tauri-driver: use sync execute with console wrapper
-      // Note: scriptString is passed as-is - wrap in IIFE to make both strings and functions callable
-      // For string scripts: "return x" -> "(() => return x)()" - wraps statement as expression
-      // For function scripts: "(a,b) => a+b" -> "((a,b) => a+b)()" - works as IIFE
+      // For non-embedded (tauri-driver/official): use sync execute with console wrapper
       const wrappedScript = `
             ${CONSOLE_WRAPPER_SCRIPT}
-            return ((${scriptString})()).apply(null, arguments);
+            return (${scriptString}).apply(null, arguments);
           `;
 
       return originalExecute(wrappedScript, ...args) as Promise<ReturnValue>;

packages/tauri-service/test/service.spec.ts

@@ -91,6 +91,52 @@ describe('TauriWorkerService', () => {
 
       expect(firstExecute).toBe(secondExecute);
     });
+
+    it('should wrap string scripts in IIFE for non-embedded providers', () => {
+      const mockExecute = vi.fn().mockResolvedValue(undefined);
+      const mockBrowser = createMockBrowser({ execute: mockExecute });
+      const service = new TauriWorkerService({ driverProvider: 'official' }, { 'wdio:tauriServiceOptions': {} });
+
+      (service as any).patchBrowserExecute(mockBrowser);
+      mockBrowser.execute('return document.title');
+
+      expect(mockExecute).toHaveBeenCalledWith(expect.stringContaining('(async () => return document.title)()'));
+    });
+
+    it('should pass function scripts as-is for non-embedded providers', () => {
+      const mockExecute = vi.fn().mockResolvedValue(undefined);
+      const mockBrowser = createMockBrowser({ execute: mockExecute });
+      const service = new TauriWorkerService({ driverProvider: 'official' }, { 'wdio:tauriServiceOptions': {} });
+
+      (service as any).patchBrowserExecute(mockBrowser);
+      const testFn = (a: number, b: number) => a + b;
+      mockBrowser.execute(testFn as any, 1, 2);
+
+      expect(mockExecute).toHaveBeenCalledWith(expect.stringContaining('(a, b) => a + b'), 1, 2);
+    });
+
+    it('should pass string scripts as-is for embedded provider', () => {
+      const mockExecute = vi.fn().mockResolvedValue(undefined);
+      const mockBrowser = createMockBrowser({ execute: mockExecute });
+      const service = new TauriWorkerService({ driverProvider: 'embedded' }, { 'wdio:tauriServiceOptions': {} });
+
+      (service as any).patchBrowserExecute(mockBrowser);
+      mockBrowser.execute('return document.title');
+
+      expect(mockExecute).toHaveBeenCalledWith('return document.title');
+    });
+
+    it('should pass function scripts as-is for embedded provider', () => {
+      const mockExecute = vi.fn().mockResolvedValue(undefined);
+      const mockBrowser = createMockBrowser({ execute: mockExecute });
+      const service = new TauriWorkerService({ driverProvider: 'embedded' }, { 'wdio:tauriServiceOptions': {} });
+
+      (service as any).patchBrowserExecute(mockBrowser);
+      const testFn = (a: number, b: number) => a + b;
+      mockBrowser.execute(testFn as any, 1, 2);
+
+      expect(mockExecute).toHaveBeenCalledWith(expect.stringContaining('(a, b) => a + b'), 1, 2);
+    });
   });
 
   describe('before()', () => {