lightman-agent 1.0.5 → 1.0.6

package/src/commands/kiosk.ts

@@ -1,113 +1,113 @@
-import { z } from 'zod';
-import type { CommandHandler, KioskStatus } from '../lib/types.js';
-import type { KioskManager } from '../services/kiosk.js';
-import type { MultiScreenKioskManager } from '../services/multiScreenKiosk.js';
-import type { Logger } from '../lib/logger.js';
-
-// --- Zod Schemas ---
-const HttpUrlSchema = z.string().url().refine(
-  (val) => {
-    try {
-      const parsed = new URL(val);
-      return parsed.protocol === 'http:' || parsed.protocol === 'https:';
-    } catch {
-      return false;
-    }
-  },
-  { message: 'Must be a valid http/https URL' }
-);
-
-const LaunchArgsSchema = z.object({
-  url: HttpUrlSchema.optional(),
-});
-
-const NavigateArgsSchema = z.object({
-  url: HttpUrlSchema,
-});
-
-export function registerKioskCommands(
-  register: (command: string, handler: CommandHandler) => void,
-  kioskManager: KioskManager,
-  logger: Logger
-): void {
-  register('kiosk:launch', async (args) => {
-    const parsed = LaunchArgsSchema.safeParse(args ?? {});
-    if (!parsed.success) {
-      throw new Error('kiosk:launch requires a valid http/https URL');
-    }
-    const { url } = parsed.data;
-    logger.info(`kiosk:launch${url ? ` → ${url}` : ' (default URL)'}`);
-    const status: KioskStatus = await kioskManager.launch(url);
-    return status as unknown as Record<string, unknown>;
-  });
-
-  register('kiosk:kill', async () => {
-    logger.info('kiosk:kill');
-    await kioskManager.kill();
-    return { killed: true };
-  });
-
-  register('kiosk:navigate', async (args) => {
-    const parsed = NavigateArgsSchema.safeParse(args ?? {});
-    if (!parsed.success) {
-      const hasUrl = typeof args?.url === 'string' && args.url.length > 0;
-      if (!hasUrl) {
-        throw new Error('kiosk:navigate requires args.url');
-      }
-      throw new Error('kiosk:navigate requires a valid http/https URL');
-    }
-    const { url } = parsed.data;
-    logger.info(`kiosk:navigate → ${url}`);
-    await kioskManager.navigate(url);
-    return { navigated: true, url };
-  });
-
-  register('kiosk:restart', async () => {
-    logger.info('kiosk:restart');
-    const status: KioskStatus = await kioskManager.restart();
-    return status as unknown as Record<string, unknown>;
-  });
-
-  register('kiosk:status', async () => {
-    const status: KioskStatus = kioskManager.getStatus();
-    return status as unknown as Record<string, unknown>;
-  });
-}
-
-/**
- * Register multi-screen kiosk commands.
- * These allow the server to manage individual screens on multi-display devices.
- */
-export function registerMultiScreenKioskCommands(
-  register: (command: string, handler: CommandHandler) => void,
-  multiKiosk: MultiScreenKioskManager,
-  getIdentity: () => { deviceId: string; apiKey: string },
-  logger: Logger
-): void {
-  register('kiosk:multi:status', async () => {
-    const status = multiKiosk.getStatus();
-    return status as unknown as Record<string, unknown>;
-  });
-
-  register('kiosk:multi:navigate', async (args) => {
-    const hardwareId = args?.hardwareId as string;
-    const url = args?.url as string;
-    if (!hardwareId) throw new Error('kiosk:multi:navigate requires args.hardwareId');
-    if (!url) throw new Error('kiosk:multi:navigate requires args.url');
-    logger.info(`kiosk:multi:navigate ${hardwareId} → ${url}`);
-    await multiKiosk.navigateScreen(hardwareId, url, getIdentity());
-    return { navigated: true, hardwareId, url };
-  });
-
-  register('kiosk:multi:restart', async () => {
-    logger.info('kiosk:multi:restart');
-    const status = await multiKiosk.restartAll(getIdentity());
-    return status as unknown as Record<string, unknown>;
-  });
-
-  register('kiosk:multi:kill', async () => {
-    logger.info('kiosk:multi:kill');
-    await multiKiosk.killAll();
-    return { killed: true };
-  });
-}
+import { z } from 'zod';
+import type { CommandHandler, KioskStatus } from '../lib/types.js';
+import type { KioskManager } from '../services/kiosk.js';
+import type { MultiScreenKioskManager } from '../services/multiScreenKiosk.js';
+import type { Logger } from '../lib/logger.js';
+
+// --- Zod Schemas ---
+const HttpUrlSchema = z.string().url().refine(
+  (val) => {
+    try {
+      const parsed = new URL(val);
+      return parsed.protocol === 'http:' || parsed.protocol === 'https:';
+    } catch {
+      return false;
+    }
+  },
+  { message: 'Must be a valid http/https URL' }
+);
+
+const LaunchArgsSchema = z.object({
+  url: HttpUrlSchema.optional(),
+});
+
+const NavigateArgsSchema = z.object({
+  url: HttpUrlSchema,
+});
+
+export function registerKioskCommands(
+  register: (command: string, handler: CommandHandler) => void,
+  kioskManager: KioskManager,
+  logger: Logger
+): void {
+  register('kiosk:launch', async (args) => {
+    const parsed = LaunchArgsSchema.safeParse(args ?? {});
+    if (!parsed.success) {
+      throw new Error('kiosk:launch requires a valid http/https URL');
+    }
+    const { url } = parsed.data;
+    logger.info(`kiosk:launch${url ? ` → ${url}` : ' (default URL)'}`);
+    const status: KioskStatus = await kioskManager.launch(url);
+    return status as unknown as Record<string, unknown>;
+  });
+
+  register('kiosk:kill', async () => {
+    logger.info('kiosk:kill');
+    await kioskManager.kill();
+    return { killed: true };
+  });
+
+  register('kiosk:navigate', async (args) => {
+    const parsed = NavigateArgsSchema.safeParse(args ?? {});
+    if (!parsed.success) {
+      const hasUrl = typeof args?.url === 'string' && args.url.length > 0;
+      if (!hasUrl) {
+        throw new Error('kiosk:navigate requires args.url');
+      }
+      throw new Error('kiosk:navigate requires a valid http/https URL');
+    }
+    const { url } = parsed.data;
+    logger.info(`kiosk:navigate → ${url}`);
+    await kioskManager.navigate(url);
+    return { navigated: true, url };
+  });
+
+  register('kiosk:restart', async () => {
+    logger.info('kiosk:restart');
+    const status: KioskStatus = await kioskManager.restart();
+    return status as unknown as Record<string, unknown>;
+  });
+
+  register('kiosk:status', async () => {
+    const status: KioskStatus = kioskManager.getStatus();
+    return status as unknown as Record<string, unknown>;
+  });
+}
+
+/**
+ * Register multi-screen kiosk commands.
+ * These allow the server to manage individual screens on multi-display devices.
+ */
+export function registerMultiScreenKioskCommands(
+  register: (command: string, handler: CommandHandler) => void,
+  multiKiosk: MultiScreenKioskManager,
+  getIdentity: () => { deviceId: string; apiKey: string },
+  logger: Logger
+): void {
+  register('kiosk:multi:status', async () => {
+    const status = multiKiosk.getStatus();
+    return status as unknown as Record<string, unknown>;
+  });
+
+  register('kiosk:multi:navigate', async (args) => {
+    const hardwareId = args?.hardwareId as string;
+    const url = args?.url as string;
+    if (!hardwareId) throw new Error('kiosk:multi:navigate requires args.hardwareId');
+    if (!url) throw new Error('kiosk:multi:navigate requires args.url');
+    logger.info(`kiosk:multi:navigate ${hardwareId} → ${url}`);
+    await multiKiosk.navigateScreen(hardwareId, url, getIdentity());
+    return { navigated: true, hardwareId, url };
+  });
+
+  register('kiosk:multi:restart', async () => {
+    logger.info('kiosk:multi:restart');
+    const status = await multiKiosk.restartAll(getIdentity());
+    return status as unknown as Record<string, unknown>;
+  });
+
+  register('kiosk:multi:kill', async () => {
+    logger.info('kiosk:multi:kill');
+    await multiKiosk.killAll();
+    return { killed: true };
+  });
+}

package/src/commands/maintenance.ts

@@ -1,106 +1,106 @@
-import { execSync } from 'child_process';
-import { readFileSync, writeFileSync } from 'fs';
-import { resolve } from 'path';
-import type { CommandHandler } from '../lib/types.js';
-import type { Logger } from '../lib/logger.js';
-import type { Watchdog } from '../services/watchdog.js';
-
-export function registerMaintenanceCommands(
-  register: (command: string, handler: CommandHandler) => void,
-  watchdog: Watchdog,
-  logger: Logger
-): void {
-  // maintenance:cleanup — Run disk cleanup immediately
-  register('maintenance:cleanup', async () => {
-    logger.info('Manual disk cleanup requested');
-    const result = await watchdog.runDiskCleanup();
-    return { ...result };
-  });
-
-  // maintenance:status — Get watchdog recovery stats
-  register('maintenance:status', async () => {
-    const stats = watchdog.getStats();
-    const cooldowns = watchdog.getCooldowns();
-    return { stats, cooldowns };
-  });
-
-  // maintenance:restore-desktop — Switch from shell replacement back to normal desktop
-  // After this command, machine needs a reboot to take effect.
-  register('maintenance:restore-desktop', async () => {
-    if (process.platform !== 'win32') {
-      return { restored: false, message: 'Only available on Windows' };
-    }
-
-    logger.warn('RESTORE DESKTOP: Switching Windows shell back to explorer.exe');
-
-    try {
-      // Restore HKLM shell to explorer.exe
-      execSync(
-        'reg add "HKLM\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon" /v Shell /d explorer.exe /f',
-        { stdio: 'ignore', timeout: 10_000 }
-      );
-      // Remove HKCU shell override
-      execSync(
-        'reg delete "HKCU\\Software\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon" /v Shell /f',
-        { stdio: 'ignore', timeout: 10_000 }
-      );
-    } catch {
-      // HKCU key may not exist, that's fine
-    }
-
-    // Also set shellMode: false in the config so agent resumes normal kiosk management
-    try {
-      const configPath = resolve(process.cwd(), 'agent.config.json');
-      const raw = readFileSync(configPath, 'utf-8');
-      const config = JSON.parse(raw);
-      if (config.kiosk) {
-        config.kiosk.shellMode = false;
-      }
-      writeFileSync(configPath, JSON.stringify(config, null, 2), 'utf-8');
-      logger.info('Config updated: shellMode set to false');
-    } catch (err) {
-      logger.error('Failed to update config:', err instanceof Error ? err.message : String(err));
-    }
-
-    logger.warn('Desktop restored. Machine needs a REBOOT to take effect.');
-    return { restored: true, message: 'Desktop restored. Reboot required.' };
-  });
-
-  // maintenance:enable-shell — Switch from normal desktop to shell replacement mode
-  // After this command, machine needs a reboot to take effect.
-  register('maintenance:enable-shell', async () => {
-    if (process.platform !== 'win32') {
-      return { enabled: false, message: 'Only available on Windows' };
-    }
-
-    const shellBat = resolve(process.cwd(), 'lightman-shell.bat');
-    logger.warn(`ENABLE SHELL: Switching Windows shell to ${shellBat}`);
-
-    try {
-      // Set shell to lightman-shell.bat
-      execSync(
-        `reg add "HKLM\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon" /v Shell /d "\\"${shellBat}\\"" /f`,
-        { stdio: 'ignore', timeout: 10_000 }
-      );
-    } catch (err) {
-      return { enabled: false, message: `Registry update failed: ${err instanceof Error ? err.message : String(err)}` };
-    }
-
-    // Update config to shellMode: true
-    try {
-      const configPath = resolve(process.cwd(), 'agent.config.json');
-      const raw = readFileSync(configPath, 'utf-8');
-      const config = JSON.parse(raw);
-      if (config.kiosk) {
-        config.kiosk.shellMode = true;
-      }
-      writeFileSync(configPath, JSON.stringify(config, null, 2), 'utf-8');
-      logger.info('Config updated: shellMode set to true');
-    } catch (err) {
-      logger.error('Failed to update config:', err instanceof Error ? err.message : String(err));
-    }
-
-    logger.warn('Shell mode enabled. Machine needs a REBOOT to take effect.');
-    return { enabled: true, message: 'Shell mode enabled. Reboot required.' };
-  });
-}
+import { execSync } from 'child_process';
+import { readFileSync, writeFileSync } from 'fs';
+import { resolve } from 'path';
+import type { CommandHandler } from '../lib/types.js';
+import type { Logger } from '../lib/logger.js';
+import type { Watchdog } from '../services/watchdog.js';
+
+export function registerMaintenanceCommands(
+  register: (command: string, handler: CommandHandler) => void,
+  watchdog: Watchdog,
+  logger: Logger
+): void {
+  // maintenance:cleanup — Run disk cleanup immediately
+  register('maintenance:cleanup', async () => {
+    logger.info('Manual disk cleanup requested');
+    const result = await watchdog.runDiskCleanup();
+    return { ...result };
+  });
+
+  // maintenance:status — Get watchdog recovery stats
+  register('maintenance:status', async () => {
+    const stats = watchdog.getStats();
+    const cooldowns = watchdog.getCooldowns();
+    return { stats, cooldowns };
+  });
+
+  // maintenance:restore-desktop — Switch from shell replacement back to normal desktop
+  // After this command, machine needs a reboot to take effect.
+  register('maintenance:restore-desktop', async () => {
+    if (process.platform !== 'win32') {
+      return { restored: false, message: 'Only available on Windows' };
+    }
+
+    logger.warn('RESTORE DESKTOP: Switching Windows shell back to explorer.exe');
+
+    try {
+      // Restore HKLM shell to explorer.exe
+      execSync(
+        'reg add "HKLM\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon" /v Shell /d explorer.exe /f',
+        { stdio: 'ignore', timeout: 10_000 }
+      );
+      // Remove HKCU shell override
+      execSync(
+        'reg delete "HKCU\\Software\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon" /v Shell /f',
+        { stdio: 'ignore', timeout: 10_000 }
+      );
+    } catch {
+      // HKCU key may not exist, that's fine
+    }
+
+    // Also set shellMode: false in the config so agent resumes normal kiosk management
+    try {
+      const configPath = resolve(process.cwd(), 'agent.config.json');
+      const raw = readFileSync(configPath, 'utf-8');
+      const config = JSON.parse(raw);
+      if (config.kiosk) {
+        config.kiosk.shellMode = false;
+      }
+      writeFileSync(configPath, JSON.stringify(config, null, 2), 'utf-8');
+      logger.info('Config updated: shellMode set to false');
+    } catch (err) {
+      logger.error('Failed to update config:', err instanceof Error ? err.message : String(err));
+    }
+
+    logger.warn('Desktop restored. Machine needs a REBOOT to take effect.');
+    return { restored: true, message: 'Desktop restored. Reboot required.' };
+  });
+
+  // maintenance:enable-shell — Switch from normal desktop to shell replacement mode
+  // After this command, machine needs a reboot to take effect.
+  register('maintenance:enable-shell', async () => {
+    if (process.platform !== 'win32') {
+      return { enabled: false, message: 'Only available on Windows' };
+    }
+
+    const shellBat = resolve(process.cwd(), 'lightman-shell.bat');
+    logger.warn(`ENABLE SHELL: Switching Windows shell to ${shellBat}`);
+
+    try {
+      // Set shell to lightman-shell.bat
+      execSync(
+        `reg add "HKLM\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon" /v Shell /d "\\"${shellBat}\\"" /f`,
+        { stdio: 'ignore', timeout: 10_000 }
+      );
+    } catch (err) {
+      return { enabled: false, message: `Registry update failed: ${err instanceof Error ? err.message : String(err)}` };
+    }
+
+    // Update config to shellMode: true
+    try {
+      const configPath = resolve(process.cwd(), 'agent.config.json');
+      const raw = readFileSync(configPath, 'utf-8');
+      const config = JSON.parse(raw);
+      if (config.kiosk) {
+        config.kiosk.shellMode = true;
+      }
+      writeFileSync(configPath, JSON.stringify(config, null, 2), 'utf-8');
+      logger.info('Config updated: shellMode set to true');
+    } catch (err) {
+      logger.error('Failed to update config:', err instanceof Error ? err.message : String(err));
+    }
+
+    logger.warn('Shell mode enabled. Machine needs a REBOOT to take effect.');
+    return { enabled: true, message: 'Shell mode enabled. Reboot required.' };
+  });
+}