bs9 1.0.0 → 1.3.0

package/src/macos/launchd.ts

@@ -0,0 +1,402 @@
+#!/usr/bin/env bun
+
+import { execSync } from "node:child_process";
+import { existsSync, writeFileSync, mkdirSync } from "node:fs";
+import { join, dirname } from "node:path";
+import { homedir } from "node:os";
+
+interface LaunchdServiceConfig {
+  label: string;
+  programArguments: string[];
+  workingDirectory: string;
+  environmentVariables: Record<string, string>;
+  runAtLoad: boolean;
+  keepAlive: boolean;
+  standardOutPath?: string;
+  standardErrorPath?: string;
+  startInterval?: number;
+}
+
+interface LaunchdServiceStatus {
+  label: string;
+  pid?: number;
+  status: 'running' | 'stopped' | 'loaded' | 'unloaded';
+  lastExitStatus?: number;
+  exitTime?: Date;
+}
+
+class LaunchdServiceManager {
+  private launchAgentsDir: string;
+  private configPath: string;
+  
+  constructor() {
+    this.launchAgentsDir = join(homedir(), 'Library', 'LaunchAgents');
+    this.configPath = join(homedir(), '.bs9', 'launchd-services.json');
+    this.ensureDirectories();
+  }
+  
+  private ensureDirectories(): void {
+    if (!existsSync(this.launchAgentsDir)) {
+      mkdirSync(this.launchAgentsDir, { recursive: true });
+    }
+    
+    const configDir = dirname(this.configPath);
+    if (!existsSync(configDir)) {
+      mkdirSync(configDir, { recursive: true });
+    }
+  }
+  
+  private loadConfigs(): Record<string, LaunchdServiceConfig> {
+    try {
+      if (existsSync(this.configPath)) {
+        const content = require('fs').readFileSync(this.configPath, 'utf-8');
+        return JSON.parse(content);
+      }
+    } catch (error) {
+      console.warn('Failed to load launchd configs:', error);
+    }
+    return {};
+  }
+  
+  private saveConfigs(configs: Record<string, LaunchdServiceConfig>): void {
+    try {
+      writeFileSync(this.configPath, JSON.stringify(configs, null, 2));
+    } catch (error) {
+      console.error('Failed to save launchd configs:', error);
+    }
+  }
+  
+  private generatePlist(config: LaunchdServiceConfig): string {
+    const plistContent = {
+      Label: config.label,
+      ProgramArguments: config.programArguments,
+      WorkingDirectory: config.workingDirectory,
+      EnvironmentVariables: config.environmentVariables,
+      RunAtLoad: config.runAtLoad,
+      KeepAlive: config.keepAlive,
+      StandardOutPath: config.standardOutPath,
+      StandardErrorPath: config.standardErrorPath,
+      StartInterval: config.startInterval
+    };
+    
+    // Remove undefined values
+    Object.keys(plistContent).forEach(key => {
+      if ((plistContent as any)[key] === undefined) {
+        delete (plistContent as any)[key];
+      }
+    });
+    
+    return `<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+${Object.entries(plistContent).map(([key, value]) => {
+  if (typeof value === 'boolean') {
+    return `    <key>${key}</key>\n    <${value ? 'true' : 'false'}/>`;
+  } else if (typeof value === 'string') {
+    return `    <key>${key}</key>\n    <string>${value}</string>`;
+  } else if (Array.isArray(value)) {
+    return `    <key>${key}</key>\n    <array>\n${value.map(item => `        <string>${item}</string>`).join('\n')}\n    </array>`;
+  } else if (typeof value === 'object' && value !== null) {
+    return `    <key>${key}</key>\n    <dict>\n${Object.entries(value).map(([k, v]) => `        <key>${k}</key>\n        <string>${v}</string>`).join('\n')}\n    </dict>`;
+  }
+  return '';
+}).join('\n')}
+</dict>
+</plist>`;
+  }
+  
+  async createService(config: LaunchdServiceConfig): Promise<void> {
+    const configs = this.loadConfigs();
+    configs[config.label] = config;
+    this.saveConfigs(configs);
+    
+    // Generate plist file
+    const plistPath = join(this.launchAgentsDir, `${config.label}.plist`);
+    writeFileSync(plistPath, this.generatePlist(config));
+    
+    try {
+      // Load the service
+      execSync(`launchctl load "${plistPath}"`, { stdio: 'inherit' });
+      console.log(`✅ Launchd service '${config.label}' created and loaded successfully`);
+    } catch (error) {
+      console.error(`❌ Failed to create launchd service: ${error}`);
+      throw error;
+    }
+  }
+  
+  async startService(label: string): Promise<void> {
+    try {
+      execSync(`launchctl start "${label}"`, { stdio: 'inherit' });
+      console.log(`✅ Launchd service '${label}' started successfully`);
+    } catch (error) {
+      console.error(`❌ Failed to start launchd service: ${error}`);
+      throw error;
+    }
+  }
+  
+  async stopService(label: string): Promise<void> {
+    try {
+      execSync(`launchctl stop "${label}"`, { stdio: 'inherit' });
+      console.log(`✅ Launchd service '${label}' stopped successfully`);
+    } catch (error) {
+      console.error(`❌ Failed to stop launchd service: ${error}`);
+      throw error;
+    }
+  }
+  
+  async unloadService(label: string): Promise<void> {
+    const plistPath = join(this.launchAgentsDir, `${label}.plist`);
+    
+    try {
+      // Stop service first
+      try {
+        await this.stopService(label);
+      } catch {
+        // Service might not be running
+      }
+      
+      // Unload service
+      execSync(`launchctl unload "${plistPath}"`, { stdio: 'inherit' });
+      
+      // Remove plist file
+      require('fs').unlinkSync(plistPath);
+      
+      // Remove from config
+      const configs = this.loadConfigs();
+      delete configs[label];
+      this.saveConfigs(configs);
+      
+      console.log(`✅ Launchd service '${label}' unloaded and deleted successfully`);
+    } catch (error) {
+      console.error(`❌ Failed to unload launchd service: ${error}`);
+      throw error;
+    }
+  }
+  
+  async getServiceStatus(label: string): Promise<LaunchdServiceStatus | null> {
+    try {
+      const output = execSync(`launchctl list "${label}"`, { encoding: 'utf-8' });
+      
+      const lines = output.split('\n');
+      const dataLine = lines.find(line => line.includes(label));
+      
+      if (dataLine) {
+        const parts = dataLine.trim().split(/\s+/);
+        const status: LaunchdServiceStatus = {
+          label: label,
+          status: 'loaded'
+        };
+        
+        if (parts[0] !== '-') {
+          status.pid = parseInt(parts[0]);
+          status.status = 'running';
+        }
+        
+        if (parts[1] !== '-') {
+          status.lastExitStatus = parseInt(parts[1]);
+        }
+        
+        return status;
+      }
+    } catch (error) {
+      // Service might not exist
+    }
+    
+    return null;
+  }
+  
+  async listServices(): Promise<LaunchdServiceStatus[]> {
+    try {
+      const configs = this.loadConfigs();
+      const services: LaunchdServiceStatus[] = [];
+      
+      for (const label of Object.keys(configs)) {
+        const status = await this.getServiceStatus(label);
+        if (status) {
+          services.push(status);
+        }
+      }
+      
+      return services;
+    } catch (error) {
+      console.error('Failed to list launchd services:', error);
+      return [];
+    }
+  }
+  
+  async enableAutoStart(label: string): Promise<void> {
+    const configs = this.loadConfigs();
+    const config = configs[label];
+    
+    if (!config) {
+      throw new Error(`Service '${label}' not found`);
+    }
+    
+    config.runAtLoad = true;
+    config.keepAlive = true;
+    this.saveConfigs(configs);
+    
+    // Update plist file
+    const plistPath = join(this.launchAgentsDir, `${label}.plist`);
+    writeFileSync(plistPath, this.generatePlist(config));
+    
+    // Reload service
+    try {
+      execSync(`launchctl unload "${plistPath}"`, { stdio: 'inherit' });
+      execSync(`launchctl load "${plistPath}"`, { stdio: 'inherit' });
+      console.log(`✅ Launchd service '${label}' set to auto-start`);
+    } catch (error) {
+      console.error(`❌ Failed to configure auto-start: ${error}`);
+      throw error;
+    }
+  }
+  
+  async disableAutoStart(label: string): Promise<void> {
+    const configs = this.loadConfigs();
+    const config = configs[label];
+    
+    if (!config) {
+      throw new Error(`Service '${label}' not found`);
+    }
+    
+    config.runAtLoad = false;
+    config.keepAlive = false;
+    this.saveConfigs(configs);
+    
+    // Update plist file
+    const plistPath = join(this.launchAgentsDir, `${label}.plist`);
+    writeFileSync(plistPath, this.generatePlist(config));
+    
+    // Reload service
+    try {
+      execSync(`launchctl unload "${plistPath}"`, { stdio: 'inherit' });
+      execSync(`launchctl load "${plistPath}"`, { stdio: 'inherit' });
+      console.log(`✅ Launchd service '${label}' set to manual start`);
+    } catch (error) {
+      console.error(`❌ Failed to configure auto-start: ${error}`);
+      throw error;
+    }
+  }
+}
+
+export async function launchdCommand(action: string, options: any): Promise<void> {
+  console.log('🍎 BS9 macOS Launchd Service Management');
+  console.log('='.repeat(80));
+  
+  const manager = new LaunchdServiceManager();
+  
+  try {
+    switch (action) {
+      case 'create':
+        if (!options.name || !options.file) {
+          console.error('❌ --name and --file are required for create action');
+          process.exit(1);
+        }
+        
+        const config: LaunchdServiceConfig = {
+          label: options.name,
+          programArguments: [options.file, ...(options.args || [])],
+          workingDirectory: options.workingDir || process.cwd(),
+          environmentVariables: options.env ? JSON.parse(options.env) : {},
+          runAtLoad: options.autoStart !== false,
+          keepAlive: options.keepAlive !== false,
+          standardOutPath: options.logOut || join(homedir(), '.bs9', 'logs', `${options.name}.out.log`),
+          standardErrorPath: options.logErr || join(homedir(), '.bs9', 'logs', `${options.name}.err.log`)
+        };
+        
+        await manager.createService(config);
+        break;
+        
+      case 'start':
+        if (!options.name) {
+          console.error('❌ --name is required for start action');
+          process.exit(1);
+        }
+        await manager.startService(options.name);
+        break;
+        
+      case 'stop':
+        if (!options.name) {
+          console.error('❌ --name is required for stop action');
+          process.exit(1);
+        }
+        await manager.stopService(options.name);
+        break;
+        
+      case 'restart':
+        if (!options.name) {
+          console.error('❌ --name is required for restart action');
+          process.exit(1);
+        }
+        await manager.stopService(options.name);
+        await manager.startService(options.name);
+        break;
+        
+      case 'unload':
+        if (!options.name) {
+          console.error('❌ --name is required for unload action');
+          process.exit(1);
+        }
+        await manager.unloadService(options.name);
+        break;
+        
+      case 'status':
+        if (options.name) {
+          const status = await manager.getServiceStatus(options.name);
+          if (status) {
+            console.log(`📊 Service Status: ${status.label}`);
+            console.log(`   Status: ${status.status}`);
+            if (status.pid) console.log(`   PID: ${status.pid}`);
+            if (status.lastExitStatus !== undefined) console.log(`   Last Exit Status: ${status.lastExitStatus}`);
+          } else {
+            console.log(`❌ Service '${options.name}' not found`);
+          }
+        } else {
+          const services = await manager.listServices();
+          console.log('📋 BS9 macOS Services:');
+          console.log('-'.repeat(80));
+          console.log('LABEL'.padEnd(30) + 'STATUS'.padEnd(15) + 'PID'.padEnd(10) + 'EXIT STATUS');
+          console.log('-'.repeat(80));
+          
+          for (const service of services) {
+            console.log(
+              service.label.padEnd(30) +
+              service.status.padEnd(15) +
+              (service.pid?.toString() || '-').padEnd(10) +
+              (service.lastExitStatus?.toString() || '-')
+            );
+          }
+          
+          if (services.length === 0) {
+            console.log('No BS9 macOS services found.');
+          }
+        }
+        break;
+        
+      case 'enable':
+        if (!options.name) {
+          console.error('❌ --name is required for enable action');
+          process.exit(1);
+        }
+        await manager.enableAutoStart(options.name);
+        break;
+        
+      case 'disable':
+        if (!options.name) {
+          console.error('❌ --name is required for disable action');
+          process.exit(1);
+        }
+        await manager.disableAutoStart(options.name);
+        break;
+        
+      default:
+        console.error(`❌ Unknown action: ${action}`);
+        console.log('Available actions: create, start, stop, restart, unload, status, enable, disable');
+        process.exit(1);
+    }
+  } catch (error) {
+    console.error(`❌ Failed to ${action} macOS service: ${error}`);
+    process.exit(1);
+  }
+}