OSIT-AE-App-Native-Electron/src/main/shell_handlers.ts

import { ipcMain, shell } from 'electron';
import { exec, execSync } from 'child_process';
import * as fs from 'fs';
import * as path from 'path';
import * as os from 'os';
import { expandPath } from './file_utils';

export function registerShellHandlers() {
  ipcMain.handle('native:open-folder', async (event, folderPath: string) => {
    const cleanPath = expandPath(folderPath);
    const error = await shell.openPath(cleanPath);
    return { success: !error, error };
  });

  ipcMain.handle('native:run-cmd', async (event, { cmd, timeout = 30000 }) => {
    const cleanCmd = expandPath(cmd);
    return new Promise((resolve) => {
      exec(cleanCmd, { timeout }, (error, stdout, stderr) => {
        resolve({ success: !error, stdout: stdout.trim(), stderr: stderr.trim(), error: error ? error.message : null });
      });
    });
  });

  ipcMain.handle('native:run-cmd-sync', async (event, { cmd }) => {
    const cleanCmd = expandPath(cmd);
    try {
      const stdout = execSync(cleanCmd).toString();
      return { success: true, stdout: stdout.trim() };
    } catch (error: any) {
      return { success: false, error: error.message, stderr: error.stderr?.toString() };
    }
  });

  ipcMain.handle('native:run-osascript', async (event, script: string) => {
    if (os.platform() !== 'darwin') return { success: false, error: 'AppleScript is only available on macOS' };

    // HARDENED: Write script to a temp .scpt file rather than passing inline via -e.
    // The old -e approach (`osascript -e "..."`) has two fatal flaws:
    //   1. It breaks on multi-line scripts.
    //   2. It breaks on paths containing spaces or special characters (quotes, parens, etc.)
    // Writing to a file sidesteps both — no shell escaping needed at all.
    // The .scpt file is deleted immediately after execution (success or failure).
    // Worst case on crash: a stale .scpt in /tmp, cleared on next OS reboot.
    //
    // LEGACY (removed): const cmd = `osascript -e "${script.replace(/"/g, '\\"')}"`;

    const tmp_script_path = path.join(os.tmpdir(), `ae_osa_${Date.now()}.scpt`);
    return new Promise((resolve) => {
      try {
        fs.writeFileSync(tmp_script_path, script.trim());
      } catch (e: any) {
        resolve({ success: false, error: `Failed to write AppleScript temp file: ${e.message}` });
        return;
      }
      exec(`osascript "${tmp_script_path}"`, (error, stdout, stderr) => {
        try { fs.unlinkSync(tmp_script_path); } catch {}
        resolve({ success: !error, stdout: stdout.trim(), stderr: stderr.trim(), error: error ? error.message : null });
      });
    });
  });

  ipcMain.handle('native:kill-processes', async (event, { process_name_li = [] }) => {
    console.log(`Native: Killing processes -> `, process_name_li);
    const results = [];
    for (const name of process_name_li) {
      const cmd = os.platform() === 'win32'
        ? `taskkill /F /IM ${name} /T`
        : `pkill -f ${name}`;
      try {
        execSync(cmd);
        results.push({ name, success: true });
      } catch (e: any) {
        results.push({ name, success: false, error: e.message });
      }
    }
    return { success: true, results };
  });

  ipcMain.handle('native:open-local-file-v2', async (event, filePath: string) => {
    const cleanPath = expandPath(filePath);
    const error = await shell.openPath(cleanPath);
    return { success: !error, error };
  });

  ipcMain.handle('native:launch-presentation', async (event, { path: rawPath, app: appType = 'default' }) => {
    const cleanedPath = expandPath(rawPath);
    console.log(`Native: Launching Presentation -> ${cleanedPath} (App: ${appType})`);

    if (os.platform() === 'linux') {
      const cmd = `libreoffice --impress "${cleanedPath}"`;
      return new Promise((resolve) => {
        exec(cmd, (err, stdout, stderr) => {
          if (err) resolve({ success: false, error: err.message });
          else resolve({ success: true, stdout, stderr });
        });
      });
    }

    if (os.platform() === 'darwin') {
      let script = '';
      if (appType === 'keynote') {
        script = `
tell application "Keynote"
  activate
  open (POSIX file "${cleanedPath}")
  delay 1
  start (front document)
end tell
        `.trim();
      } else if (appType === 'powerpoint') {
        script = `
tell application "Microsoft PowerPoint"
  activate
  open (POSIX file "${cleanedPath}")
  delay 1
  run slide show of active presentation
end tell
        `.trim();
      }

      if (script) {
        const tmp_script_path = path.join(os.tmpdir(), `ae_launch_${Date.now()}.scpt`);
        return new Promise((resolve) => {
          try {
            fs.writeFileSync(tmp_script_path, script);
          } catch (e: any) {
            resolve({ success: false, error: `Failed to write AppleScript temp file: ${e.message}` });
            return;
          }
          exec(`osascript "${tmp_script_path}"`, (err) => {
            try { fs.unlinkSync(tmp_script_path); } catch {}
            if (err) resolve({ success: false, error: err.message });
            else resolve({ success: true });
          });
        });
      }
    }

    const error = await shell.openPath(cleanedPath);
    return { success: !error, error };
  });

  ipcMain.handle('native:control-presentation', async (event, { app, action }) => {
    if (os.platform() !== 'darwin') return { success: false, error: 'Presentation control is only available on macOS' };

    let script = '';
    if (app === 'powerpoint') {
      switch (action) {
        case 'next': script = 'tell application "Microsoft PowerPoint" to next slide of slide show view of active presentation'; break;
        case 'prev': script = 'tell application "Microsoft PowerPoint" to previous slide of slide show view of active presentation'; break;
        case 'start': script = 'tell application "Microsoft PowerPoint" to run slide show of active presentation'; break;
        case 'stop': script = 'tell application "Microsoft PowerPoint" to stop slide show of active presentation'; break;
      }
    } else if (app === 'keynote') {
      switch (action) {
        case 'next': script = 'tell application "Keynote" to show next'; break;
        case 'prev': script = 'tell application "Keynote" to show previous'; break;
        case 'start': script = 'tell application "Keynote" to start (front document)'; break;
        case 'stop': script = 'tell application "Keynote" to stop'; break;
      }
    }

    if (!script) return { success: false, error: `Unsupported app or action: ${app}/${action}` };

    return new Promise((resolve) => {
      exec(`osascript -e "${script.replace(/"/g, '\\"')}"`, (error, stdout, stderr) => {
        resolve({ success: !error, stdout: stdout.trim(), stderr: stderr.trim(), error: error ? error.message : null });
      });
    });
  });

  ipcMain.handle('native:list-tools', async () => {
    return [
      {
        name: 'open_folder',
        description: 'Opens a directory in the OS file explorer (Finder/Files/Explorer).',
        params: { path: 'string' }
      },
      {
        name: 'run_cmd',
        description: 'Executes an asynchronous shell command with a timeout.',
        params: { cmd: 'string', timeout: 'number (optional)' }
      },
      {
        name: 'run_cmd_sync',
        description: 'Executes a synchronous shell command.',
        params: { cmd: 'string' }
      },
      {
        name: 'run_osascript',
        description: 'Executes a raw AppleScript string (macOS only).',
        params: { script: 'string' }
      },
      {
        name: 'kill_processes',
        description: 'Forcefully terminates processes by name.',
        params: { process_name_li: 'string[]' }
      },
      {
        name: 'open_local_file_v2',
        description: 'Opens a local file using the default OS handler.',
        params: { filePath: 'string' }
      },
      {
        name: 'launch_presentation',
        description: 'Phase 5: Specialized launcher for PowerPoint, Keynote, and LibreOffice with auto-focus.',
        params: { path: 'string', app: 'default|powerpoint|keynote' }
      },
      {
        name: 'control_presentation',
        description: 'Phase 5: Remote navigation for active slideshows.',
        params: { app: 'powerpoint|keynote', action: 'next|prev|start|stop' }
      },
      {
        name: 'check_cache',
        description: 'Checks if a file exists in the local organized cache.',
        params: { cache_root: 'string', hash: 'string', hash_prefix_length: 'number' }
      },
      {
        name: 'download_to_cache',
        description: 'Downloads a file from the API directly into the native cache.',
        params: { url: 'string', cache_root: 'string', hash: 'string', api_key: 'string', account_id: 'string' }
      },
      {
        name: 'launch_from_cache',
        description: 'Atomic operation: Copies file from cache to temp with original name and launches via specialized handler.',
        params: { cache_root: 'string', hash: 'string', temp_root: 'string', filename: 'string' }
      },
      {
        name: 'get_device_info',
        description: 'Returns hardware and OS metadata (CPUs, RAM, IP addresses, Hostname).',
        params: {}
      }
    ];
  });
}