@fermindi/pwn-cli 0.1.1 → 0.2.0

package/src/services/notification-service.js

@@ -1,342 +1,342 @@
-/**
- * PWN Notification Service
- *
- * Multi-channel notification system for task completion alerts.
- * Supports: ntfy.sh, desktop notifications, and extensible channels.
- */
-
-import { existsSync, readFileSync } from 'fs';
-import { join } from 'path';
-
-/**
- * @typedef {Object} NotificationConfig
- * @property {boolean} enabled - Global enable/disable
- * @property {string} defaultChannel - Default channel to use
- * @property {Object} channels - Channel-specific configurations
- */
-
-/**
- * @typedef {Object} NotificationPayload
- * @property {string} title - Notification title
- * @property {string} message - Notification body
- * @property {string} [priority] - Priority level (low, default, high, urgent)
- * @property {string[]} [tags] - Tags/emojis for the notification
- * @property {string} [url] - Click action URL
- */
-
-/**
- * Default configuration
- */
-const DEFAULT_CONFIG = {
-  enabled: true,
-  defaultChannel: 'desktop',
-  channels: {
-    ntfy: {
-      enabled: false,
-      server: 'https://ntfy.sh',
-      topic: '',
-      priority: 'default'
-    },
-    desktop: {
-      enabled: true
-    },
-    pushover: {
-      enabled: false,
-      userKey: '',
-      apiToken: ''
-    }
-  }
-};
-
-/**
- * Load notification configuration from workspace
- * @param {string} cwd - Working directory
- * @returns {NotificationConfig}
- */
-export function loadConfig(cwd = process.cwd()) {
-  const configPath = join(cwd, '.ai', 'config', 'notifications.json');
-
-  if (!existsSync(configPath)) {
-    return DEFAULT_CONFIG;
-  }
-
-  try {
-    const content = readFileSync(configPath, 'utf8');
-    const userConfig = JSON.parse(content);
-
-    // Merge with defaults
-    return {
-      ...DEFAULT_CONFIG,
-      ...userConfig,
-      channels: {
-        ...DEFAULT_CONFIG.channels,
-        ...userConfig.channels
-      }
-    };
-  } catch {
-    return DEFAULT_CONFIG;
-  }
-}
-
-/**
- * Send notification through configured channels
- * @param {NotificationPayload} payload - Notification data
- * @param {Object} options - Send options
- * @param {string} [options.channel] - Specific channel to use
- * @param {string} [options.cwd] - Working directory
- * @returns {Promise<{success: boolean, channel: string, error?: string}>}
- */
-export async function send(payload, options = {}) {
-  const { channel: specificChannel, cwd = process.cwd() } = options;
-  const config = loadConfig(cwd);
-
-  if (!config.enabled) {
-    return { success: false, channel: 'none', error: 'Notifications disabled' };
-  }
-
-  const channelName = specificChannel || config.defaultChannel;
-  const channelConfig = config.channels[channelName];
-
-  if (!channelConfig) {
-    return { success: false, channel: channelName, error: `Unknown channel: ${channelName}` };
-  }
-
-  if (!channelConfig.enabled) {
-    return { success: false, channel: channelName, error: `Channel ${channelName} is disabled` };
-  }
-
-  try {
-    switch (channelName) {
-      case 'ntfy':
-        return await sendNtfy(payload, channelConfig);
-      case 'desktop':
-        return await sendDesktop(payload, channelConfig);
-      case 'pushover':
-        return await sendPushover(payload, channelConfig);
-      default:
-        return { success: false, channel: channelName, error: `No handler for channel: ${channelName}` };
-    }
-  } catch (error) {
-    return { success: false, channel: channelName, error: error.message };
-  }
-}
-
-/**
- * Send notification via ntfy.sh
- * @param {NotificationPayload} payload
- * @param {Object} config
- */
-async function sendNtfy(payload, config) {
-  const { server = 'https://ntfy.sh', topic, priority = 'default' } = config;
-
-  if (!topic) {
-    return { success: false, channel: 'ntfy', error: 'ntfy topic not configured' };
-  }
-
-  const url = `${server}/${topic}`;
-
-  const headers = {
-    'Title': payload.title,
-    'Priority': payload.priority || priority,
-  };
-
-  if (payload.tags && payload.tags.length > 0) {
-    headers['Tags'] = payload.tags.join(',');
-  }
-
-  if (payload.url) {
-    headers['Click'] = payload.url;
-  }
-
-  const response = await fetch(url, {
-    method: 'POST',
-    headers,
-    body: payload.message
-  });
-
-  if (!response.ok) {
-    return { success: false, channel: 'ntfy', error: `HTTP ${response.status}` };
-  }
-
-  return { success: true, channel: 'ntfy' };
-}
-
-/**
- * Send desktop notification
- * Uses native OS notifications via child process
- * @param {NotificationPayload} payload
- * @param {Object} config
- */
-async function sendDesktop(payload, config) {
-  const { title, message } = payload;
-
-  // Use PowerShell on Windows for native toast notifications
-  if (process.platform === 'win32') {
-    const { exec } = await import('child_process');
-    const { promisify } = await import('util');
-    const execAsync = promisify(exec);
-
-    // Escape quotes in message
-    const escapedTitle = title.replace(/"/g, '`"');
-    const escapedMessage = message.replace(/"/g, '`"');
-
-    const script = `
-      [Windows.UI.Notifications.ToastNotificationManager, Windows.UI.Notifications, ContentType = WindowsRuntime] | Out-Null
-      [Windows.Data.Xml.Dom.XmlDocument, Windows.Data.Xml.Dom.XmlDocument, ContentType = WindowsRuntime] | Out-Null
-      $template = @"
-      <toast>
-        <visual>
-          <binding template="ToastText02">
-            <text id="1">${escapedTitle}</text>
-            <text id="2">${escapedMessage}</text>
-          </binding>
-        </visual>
-      </toast>
-"@
-      $xml = New-Object Windows.Data.Xml.Dom.XmlDocument
-      $xml.LoadXml($template)
-      $toast = [Windows.UI.Notifications.ToastNotification]::new($xml)
-      [Windows.UI.Notifications.ToastNotificationManager]::CreateToastNotifier("PWN").Show($toast)
-    `;
-
-    try {
-      await execAsync(`powershell -Command "${script.replace(/\n/g, ' ')}"`, {
-        windowsHide: true
-      });
-      return { success: true, channel: 'desktop' };
-    } catch (error) {
-      // Fallback to simple msg command
-      try {
-        await execAsync(`msg %username% "${escapedTitle}: ${escapedMessage}"`, {
-          windowsHide: true
-        });
-        return { success: true, channel: 'desktop' };
-      } catch {
-        return { success: false, channel: 'desktop', error: 'Desktop notifications not available' };
-      }
-    }
-  }
-
-  // macOS
-  if (process.platform === 'darwin') {
-    const { exec } = await import('child_process');
-    const { promisify } = await import('util');
-    const execAsync = promisify(exec);
-
-    const escapedTitle = title.replace(/"/g, '\\"');
-    const escapedMessage = message.replace(/"/g, '\\"');
-
-    try {
-      await execAsync(`osascript -e 'display notification "${escapedMessage}" with title "${escapedTitle}"'`);
-      return { success: true, channel: 'desktop' };
-    } catch (error) {
-      return { success: false, channel: 'desktop', error: error.message };
-    }
-  }
-
-  // Linux (notify-send)
-  if (process.platform === 'linux') {
-    const { exec } = await import('child_process');
-    const { promisify } = await import('util');
-    const execAsync = promisify(exec);
-
-    try {
-      await execAsync(`notify-send "${title}" "${message}"`);
-      return { success: true, channel: 'desktop' };
-    } catch (error) {
-      return { success: false, channel: 'desktop', error: 'notify-send not available' };
-    }
-  }
-
-  return { success: false, channel: 'desktop', error: `Unsupported platform: ${process.platform}` };
-}
-
-/**
- * Send notification via Pushover
- * @param {NotificationPayload} payload
- * @param {Object} config
- */
-async function sendPushover(payload, config) {
-  const { userKey, apiToken } = config;
-
-  if (!userKey || !apiToken) {
-    return { success: false, channel: 'pushover', error: 'Pushover credentials not configured' };
-  }
-
-  const priorityMap = {
-    'low': -1,
-    'default': 0,
-    'high': 1,
-    'urgent': 2
-  };
-
-  const body = new URLSearchParams({
-    token: apiToken,
-    user: userKey,
-    title: payload.title,
-    message: payload.message,
-    priority: priorityMap[payload.priority || 'default'] || 0
-  });
-
-  if (payload.url) {
-    body.append('url', payload.url);
-  }
-
-  const response = await fetch('https://api.pushover.net/1/messages.json', {
-    method: 'POST',
-    body
-  });
-
-  if (!response.ok) {
-    const data = await response.json().catch(() => ({}));
-    return { success: false, channel: 'pushover', error: data.errors?.join(', ') || `HTTP ${response.status}` };
-  }
-
-  return { success: true, channel: 'pushover' };
-}
-
-/**
- * Test notification configuration
- * @param {string} channel - Channel to test
- * @param {string} cwd - Working directory
- * @returns {Promise<{success: boolean, channel: string, error?: string}>}
- */
-export async function test(channel, cwd = process.cwd()) {
-  return send({
-    title: 'PWN Test Notification',
-    message: 'If you see this, notifications are working!',
-    tags: ['white_check_mark', 'robot']
-  }, { channel, cwd });
-}
-
-/**
- * Send task completion notification
- * @param {string} taskId - Task identifier
- * @param {string} taskName - Task description
- * @param {Object} options - Additional options
- */
-export async function notifyTaskComplete(taskId, taskName, options = {}) {
-  return send({
-    title: 'Task Completed',
-    message: `${taskId}: ${taskName}`,
-    tags: ['white_check_mark'],
-    priority: 'default',
-    ...options
-  }, options);
-}
-
-/**
- * Send error notification
- * @param {string} title - Error title
- * @param {string} message - Error details
- * @param {Object} options - Additional options
- */
-export async function notifyError(title, message, options = {}) {
-  return send({
-    title,
-    message,
-    tags: ['x', 'warning'],
-    priority: 'high',
-    ...options
-  }, options);
-}
+/**
+ * PWN Notification Service
+ *
+ * Multi-channel notification system for task completion alerts.
+ * Supports: ntfy.sh, desktop notifications, and extensible channels.
+ */
+
+import { existsSync, readFileSync } from 'fs';
+import { join } from 'path';
+
+/**
+ * @typedef {Object} NotificationConfig
+ * @property {boolean} enabled - Global enable/disable
+ * @property {string} defaultChannel - Default channel to use
+ * @property {Object} channels - Channel-specific configurations
+ */
+
+/**
+ * @typedef {Object} NotificationPayload
+ * @property {string} title - Notification title
+ * @property {string} message - Notification body
+ * @property {string} [priority] - Priority level (low, default, high, urgent)
+ * @property {string[]} [tags] - Tags/emojis for the notification
+ * @property {string} [url] - Click action URL
+ */
+
+/**
+ * Default configuration
+ */
+const DEFAULT_CONFIG = {
+  enabled: true,
+  defaultChannel: 'desktop',
+  channels: {
+    ntfy: {
+      enabled: false,
+      server: 'https://ntfy.sh',
+      topic: '',
+      priority: 'default'
+    },
+    desktop: {
+      enabled: true
+    },
+    pushover: {
+      enabled: false,
+      userKey: '',
+      apiToken: ''
+    }
+  }
+};
+
+/**
+ * Load notification configuration from workspace
+ * @param {string} cwd - Working directory
+ * @returns {NotificationConfig}
+ */
+export function loadConfig(cwd = process.cwd()) {
+  const configPath = join(cwd, '.ai', 'config', 'notifications.json');
+
+  if (!existsSync(configPath)) {
+    return DEFAULT_CONFIG;
+  }
+
+  try {
+    const content = readFileSync(configPath, 'utf8');
+    const userConfig = JSON.parse(content);
+
+    // Merge with defaults
+    return {
+      ...DEFAULT_CONFIG,
+      ...userConfig,
+      channels: {
+        ...DEFAULT_CONFIG.channels,
+        ...userConfig.channels
+      }
+    };
+  } catch {
+    return DEFAULT_CONFIG;
+  }
+}
+
+/**
+ * Send notification through configured channels
+ * @param {NotificationPayload} payload - Notification data
+ * @param {Object} options - Send options
+ * @param {string} [options.channel] - Specific channel to use
+ * @param {string} [options.cwd] - Working directory
+ * @returns {Promise<{success: boolean, channel: string, error?: string}>}
+ */
+export async function send(payload, options = {}) {
+  const { channel: specificChannel, cwd = process.cwd() } = options;
+  const config = loadConfig(cwd);
+
+  if (!config.enabled) {
+    return { success: false, channel: 'none', error: 'Notifications disabled' };
+  }
+
+  const channelName = specificChannel || config.defaultChannel;
+  const channelConfig = config.channels[channelName];
+
+  if (!channelConfig) {
+    return { success: false, channel: channelName, error: `Unknown channel: ${channelName}` };
+  }
+
+  if (!channelConfig.enabled) {
+    return { success: false, channel: channelName, error: `Channel ${channelName} is disabled` };
+  }
+
+  try {
+    switch (channelName) {
+      case 'ntfy':
+        return await sendNtfy(payload, channelConfig);
+      case 'desktop':
+        return await sendDesktop(payload, channelConfig);
+      case 'pushover':
+        return await sendPushover(payload, channelConfig);
+      default:
+        return { success: false, channel: channelName, error: `No handler for channel: ${channelName}` };
+    }
+  } catch (error) {
+    return { success: false, channel: channelName, error: error.message };
+  }
+}
+
+/**
+ * Send notification via ntfy.sh
+ * @param {NotificationPayload} payload
+ * @param {Object} config
+ */
+async function sendNtfy(payload, config) {
+  const { server = 'https://ntfy.sh', topic, priority = 'default' } = config;
+
+  if (!topic) {
+    return { success: false, channel: 'ntfy', error: 'ntfy topic not configured' };
+  }
+
+  const url = `${server}/${topic}`;
+
+  const headers = {
+    'Title': payload.title,
+    'Priority': payload.priority || priority,
+  };
+
+  if (payload.tags && payload.tags.length > 0) {
+    headers['Tags'] = payload.tags.join(',');
+  }
+
+  if (payload.url) {
+    headers['Click'] = payload.url;
+  }
+
+  const response = await fetch(url, {
+    method: 'POST',
+    headers,
+    body: payload.message
+  });
+
+  if (!response.ok) {
+    return { success: false, channel: 'ntfy', error: `HTTP ${response.status}` };
+  }
+
+  return { success: true, channel: 'ntfy' };
+}
+
+/**
+ * Send desktop notification
+ * Uses native OS notifications via child process
+ * @param {NotificationPayload} payload
+ * @param {Object} config
+ */
+async function sendDesktop(payload, config) {
+  const { title, message } = payload;
+
+  // Use PowerShell on Windows for native toast notifications
+  if (process.platform === 'win32') {
+    const { exec } = await import('child_process');
+    const { promisify } = await import('util');
+    const execAsync = promisify(exec);
+
+    // Escape quotes in message
+    const escapedTitle = title.replace(/"/g, '`"');
+    const escapedMessage = message.replace(/"/g, '`"');
+
+    const script = `
+      [Windows.UI.Notifications.ToastNotificationManager, Windows.UI.Notifications, ContentType = WindowsRuntime] | Out-Null
+      [Windows.Data.Xml.Dom.XmlDocument, Windows.Data.Xml.Dom.XmlDocument, ContentType = WindowsRuntime] | Out-Null
+      $template = @"
+      <toast>
+        <visual>
+          <binding template="ToastText02">
+            <text id="1">${escapedTitle}</text>
+            <text id="2">${escapedMessage}</text>
+          </binding>
+        </visual>
+      </toast>
+"@
+      $xml = New-Object Windows.Data.Xml.Dom.XmlDocument
+      $xml.LoadXml($template)
+      $toast = [Windows.UI.Notifications.ToastNotification]::new($xml)
+      [Windows.UI.Notifications.ToastNotificationManager]::CreateToastNotifier("PWN").Show($toast)
+    `;
+
+    try {
+      await execAsync(`powershell -Command "${script.replace(/\n/g, ' ')}"`, {
+        windowsHide: true
+      });
+      return { success: true, channel: 'desktop' };
+    } catch (error) {
+      // Fallback to simple msg command
+      try {
+        await execAsync(`msg %username% "${escapedTitle}: ${escapedMessage}"`, {
+          windowsHide: true
+        });
+        return { success: true, channel: 'desktop' };
+      } catch {
+        return { success: false, channel: 'desktop', error: 'Desktop notifications not available' };
+      }
+    }
+  }
+
+  // macOS
+  if (process.platform === 'darwin') {
+    const { exec } = await import('child_process');
+    const { promisify } = await import('util');
+    const execAsync = promisify(exec);
+
+    const escapedTitle = title.replace(/"/g, '\\"');
+    const escapedMessage = message.replace(/"/g, '\\"');
+
+    try {
+      await execAsync(`osascript -e 'display notification "${escapedMessage}" with title "${escapedTitle}"'`);
+      return { success: true, channel: 'desktop' };
+    } catch (error) {
+      return { success: false, channel: 'desktop', error: error.message };
+    }
+  }
+
+  // Linux (notify-send)
+  if (process.platform === 'linux') {
+    const { exec } = await import('child_process');
+    const { promisify } = await import('util');
+    const execAsync = promisify(exec);
+
+    try {
+      await execAsync(`notify-send "${title}" "${message}"`);
+      return { success: true, channel: 'desktop' };
+    } catch (error) {
+      return { success: false, channel: 'desktop', error: 'notify-send not available' };
+    }
+  }
+
+  return { success: false, channel: 'desktop', error: `Unsupported platform: ${process.platform}` };
+}
+
+/**
+ * Send notification via Pushover
+ * @param {NotificationPayload} payload
+ * @param {Object} config
+ */
+async function sendPushover(payload, config) {
+  const { userKey, apiToken } = config;
+
+  if (!userKey || !apiToken) {
+    return { success: false, channel: 'pushover', error: 'Pushover credentials not configured' };
+  }
+
+  const priorityMap = {
+    'low': -1,
+    'default': 0,
+    'high': 1,
+    'urgent': 2
+  };
+
+  const body = new URLSearchParams({
+    token: apiToken,
+    user: userKey,
+    title: payload.title,
+    message: payload.message,
+    priority: priorityMap[payload.priority || 'default'] || 0
+  });
+
+  if (payload.url) {
+    body.append('url', payload.url);
+  }
+
+  const response = await fetch('https://api.pushover.net/1/messages.json', {
+    method: 'POST',
+    body
+  });
+
+  if (!response.ok) {
+    const data = await response.json().catch(() => ({}));
+    return { success: false, channel: 'pushover', error: data.errors?.join(', ') || `HTTP ${response.status}` };
+  }
+
+  return { success: true, channel: 'pushover' };
+}
+
+/**
+ * Test notification configuration
+ * @param {string} channel - Channel to test
+ * @param {string} cwd - Working directory
+ * @returns {Promise<{success: boolean, channel: string, error?: string}>}
+ */
+export async function test(channel, cwd = process.cwd()) {
+  return send({
+    title: 'PWN Test Notification',
+    message: 'If you see this, notifications are working!',
+    tags: ['white_check_mark', 'robot']
+  }, { channel, cwd });
+}
+
+/**
+ * Send task completion notification
+ * @param {string} taskId - Task identifier
+ * @param {string} taskName - Task description
+ * @param {Object} options - Additional options
+ */
+export async function notifyTaskComplete(taskId, taskName, options = {}) {
+  return send({
+    title: 'Task Completed',
+    message: `${taskId}: ${taskName}`,
+    tags: ['white_check_mark'],
+    priority: 'default',
+    ...options
+  }, options);
+}
+
+/**
+ * Send error notification
+ * @param {string} title - Error title
+ * @param {string} message - Error details
+ * @param {Object} options - Additional options
+ */
+export async function notifyError(title, message, options = {}) {
+  return send({
+    title,
+    message,
+    tags: ['x', 'warning'],
+    priority: 'high',
+    ...options
+  }, options);
+}