@yeaft/webchat-agent 0.1.224 → 0.1.226

package/conductor/ui-messages.js

@@ -0,0 +1,170 @@
+/**
+ * Conductor — UI Message Helpers (V5)
+ *
+ * Sends WebSocket messages to server/frontend.
+ * Maintains uiMessages persistence list on the conductor instance.
+ *
+ * No sessionId — Conductor is a singleton per Agent.
+ */
+import ctx from '../context.js';
+import { saveConductorMeta } from './persistence.js';
+
+/**
+ * Send conductor message to server (forwarded to Web frontend)
+ */
+export function sendConductorMessage(msg) {
+  if (ctx.sendToServer) {
+    ctx.sendToServer(msg);
+  }
+}
+
+/**
+ * Send Conductor Claude output to frontend
+ * outputType: 'text' | 'tool_use' | 'tool_result' | 'system' | 'task_created' | 'task_forwarded'
+ */
+export function sendConductorOutput(conductor, outputType, rawMessage, extra = {}) {
+  sendConductorMessage({
+    type: 'conductor_output',
+    outputType,
+    data: rawMessage,
+    ...extra
+  });
+
+  // Record trimmed UI messages
+  if (outputType === 'text') {
+    const content = rawMessage?.message?.content;
+    let text = '';
+    if (typeof content === 'string') {
+      text = content;
+    } else if (Array.isArray(content)) {
+      text = content.filter(b => b.type === 'text').map(b => b.text).join('');
+    }
+    if (!text) return;
+    let found = false;
+    for (let i = conductor.uiMessages.length - 1; i >= 0; i--) {
+      const msg = conductor.uiMessages[i];
+      if (msg.source === 'conductor' && msg.type === 'text' && msg._streaming) {
+        msg.content += text;
+        found = true;
+        break;
+      }
+    }
+    if (!found) {
+      conductor.uiMessages.push({
+        source: 'conductor', type: 'text', content: text,
+        _streaming: true, timestamp: Date.now()
+      });
+    }
+  } else if (outputType === 'system') {
+    const content = rawMessage?.message?.content;
+    let text = '';
+    if (typeof content === 'string') text = content;
+    else if (Array.isArray(content)) text = content.filter(b => b.type === 'text').map(b => b.text).join('');
+    if (!text) return;
+    conductor.uiMessages.push({
+      source: 'conductor', type: 'system', content: text, timestamp: Date.now()
+    });
+  } else if (outputType === 'task_created') {
+    conductor.uiMessages.push({
+      source: 'conductor', type: 'task_created',
+      taskId: extra.taskId, taskTitle: extra.taskTitle,
+      content: `Created task: ${extra.taskTitle}`,
+      timestamp: Date.now()
+    });
+  } else if (outputType === 'task_forwarded') {
+    conductor.uiMessages.push({
+      source: 'conductor', type: 'task_forwarded',
+      taskId: extra.taskId,
+      content: `Forwarded message to task: ${extra.taskId}`,
+      timestamp: Date.now()
+    });
+  } else if (outputType === 'tool_use') {
+    endConductorStreaming(conductor);
+    const content = rawMessage?.message?.content;
+    if (Array.isArray(content)) {
+      for (const block of content) {
+        if (block.type === 'tool_use') {
+          const input = block.input || {};
+          const trimmedInput = {};
+          if (input.file_path) trimmedInput.file_path = input.file_path;
+          if (input.command) trimmedInput.command = input.command.substring(0, 200);
+          if (input.pattern) trimmedInput.pattern = input.pattern;
+          conductor.uiMessages.push({
+            source: 'conductor', type: 'tool',
+            toolName: block.name, toolId: block.id,
+            toolInput: Object.keys(trimmedInput).length > 0 ? trimmedInput : null,
+            content: `${block.name} ${input.file_path || input.command?.substring(0, 60) || ''}`,
+            hasResult: false,
+            timestamp: Date.now()
+          });
+        }
+      }
+    }
+  } else if (outputType === 'tool_result') {
+    const toolId = rawMessage?.message?.tool_use_id;
+    if (toolId) {
+      for (let i = conductor.uiMessages.length - 1; i >= 0; i--) {
+        if (conductor.uiMessages[i].type === 'tool' && conductor.uiMessages[i].toolId === toolId) {
+          conductor.uiMessages[i].hasResult = true;
+          break;
+        }
+      }
+    }
+  }
+}
+
+/**
+ * End Conductor streaming message
+ */
+export function endConductorStreaming(conductor) {
+  for (let i = conductor.uiMessages.length - 1; i >= 0; i--) {
+    if (conductor.uiMessages[i].source === 'conductor' && conductor.uiMessages[i]._streaming) {
+      delete conductor.uiMessages[i]._streaming;
+      break;
+    }
+  }
+}
+
+/**
+ * Record user message to uiMessages
+ */
+export function recordUserMessage(conductor, content) {
+  conductor.uiMessages.push({
+    source: 'user', type: 'text', content,
+    timestamp: Date.now()
+  });
+}
+
+/**
+ * Send conductor status update to frontend
+ */
+export function sendStatusUpdate(conductor) {
+  const tasks = {};
+  for (const [taskId, t] of conductor.tasks) {
+    tasks[taskId] = {
+      taskId: t.taskId,
+      title: t.title,
+      workDir: t.workDir,
+      scenario: t.scenario,
+      status: t.status,
+      activeActors: t.activeActors || [],
+      currentStep: t.currentStep || '',
+      lastUpdate: t.lastUpdate || t.updatedAt
+    };
+  }
+
+  sendConductorMessage({
+    type: 'conductor_status',
+    status: conductor.status,
+    tasks,
+    costUsd: conductor.costUsd,
+    totalInputTokens: conductor.totalInputTokens,
+    totalOutputTokens: conductor.totalOutputTokens,
+    activeClaudes: conductor.activeClaudes || 0
+  });
+
+  // Async persist
+  saveConductorMeta(conductor).catch(e =>
+    console.warn('[Conductor] Failed to save meta:', e.message)
+  );
+}

package/conductor/worktree.js

@@ -0,0 +1,182 @@
+/**
+ * Conductor — Worktree Management (V5)
+ *
+ * V5 strategy: each task gets ONE worktree at creation time.
+ * Path: {workDir}/.conductor/task-N/worktree/
+ *
+ * This prevents file conflicts when multiple tasks work on the same project.
+ * - Read-write actors (coding/testing/review) → cwd = worktree
+ * - Read-only actors (planning/discussion/design) → cwd = {workDir}
+ */
+import { promises as fs } from 'fs';
+import { join, resolve } from 'path';
+import { execFile as execFileCb } from 'child_process';
+import { promisify } from 'util';
+
+const execFile = promisify(execFileCb);
+
+/**
+ * Create a worktree for a task.
+ * Called at task creation time.
+ *
+ * @param {string} workDir - Project root directory (git repo root)
+ * @param {string} taskId - Task ID (e.g. 'task-m1234-abcd')
+ * @param {string} taskDir - Full task directory path ({workDir}/.conductor/{taskId})
+ * @param {object} [options]
+ * @param {string} [options.baseBranch='HEAD'] - Base ref for the worktree
+ * @returns {Promise<string>} worktree path
+ */
+export async function createTaskWorktree(workDir, taskId, taskDir, options = {}) {
+  const { baseBranch = 'HEAD' } = options;
+
+  const wtDir = join(taskDir, 'worktree');
+  const branch = `conductor/${taskId}`;
+
+  // Check if worktree already exists
+  const exists = await directoryExists(wtDir);
+  if (exists) {
+    const known = await isKnownWorktree(workDir, wtDir);
+    if (known) {
+      console.log(`[Conductor/Worktree] Already exists: ${wtDir}`);
+      return wtDir;
+    }
+    // Orphaned directory, clean up and recreate
+    console.warn(`[Conductor/Worktree] Orphaned dir, removing: ${wtDir}`);
+    await fs.rm(wtDir, { recursive: true, force: true }).catch(() => {});
+  }
+
+  // Check if this is a git repo
+  const isGitRepo = await directoryExists(join(workDir, '.git'));
+  if (!isGitRepo) {
+    // Not a git repo — just create a regular directory (no worktree isolation possible)
+    console.warn(`[Conductor/Worktree] ${workDir} is not a git repo, creating plain directory`);
+    await fs.mkdir(wtDir, { recursive: true });
+    return wtDir;
+  }
+
+  try {
+    // Create branch (ignore if already exists)
+    try {
+      await execFile('git', ['branch', branch, baseBranch], { cwd: workDir });
+    } catch {
+      // Branch already exists, ignore
+    }
+
+    // Create worktree
+    await execFile('git', ['worktree', 'add', wtDir, branch], { cwd: workDir });
+    console.log(`[Conductor/Worktree] Created: ${wtDir} on branch ${branch}`);
+    return wtDir;
+  } catch (e) {
+    console.error(`[Conductor/Worktree] Failed to create worktree for ${taskId}:`, e.message);
+    // Fallback: create plain directory so task can still function
+    await fs.mkdir(wtDir, { recursive: true }).catch(() => {});
+    return wtDir;
+  }
+}
+
+/**
+ * Clean up a task's worktree
+ *
+ * @param {string} workDir - Project root directory
+ * @param {string} taskId - Task ID
+ * @param {string} taskDir - Full task directory path
+ */
+export async function cleanupTaskWorktree(workDir, taskId, taskDir) {
+  const wtDir = join(taskDir, 'worktree');
+  const branch = `conductor/${taskId}`;
+
+  if (!(await directoryExists(wtDir))) return;
+
+  const isGitRepo = await directoryExists(join(workDir, '.git'));
+  if (!isGitRepo) {
+    // Plain directory, just remove
+    await fs.rm(wtDir, { recursive: true, force: true }).catch(() => {});
+    return;
+  }
+
+  try {
+    await execFile('git', ['worktree', 'remove', wtDir, '--force'], { cwd: workDir });
+    console.log(`[Conductor/Worktree] Removed: ${wtDir}`);
+  } catch (e) {
+    console.warn(`[Conductor/Worktree] Failed to remove ${wtDir}:`, e.message);
+  }
+
+  try {
+    await execFile('git', ['branch', '-D', branch], { cwd: workDir });
+    console.log(`[Conductor/Worktree] Deleted branch: ${branch}`);
+  } catch (e) {
+    console.warn(`[Conductor/Worktree] Failed to delete branch ${branch}:`, e.message);
+  }
+}
+
+/**
+ * Clean up all conductor worktrees under a project
+ *
+ * @param {string} workDir - Project root directory
+ */
+export async function cleanupAllConductorWorktrees(workDir) {
+  const conductorDir = join(workDir, '.conductor');
+
+  if (!(await directoryExists(conductorDir))) return;
+
+  try {
+    const entries = await fs.readdir(conductorDir);
+    for (const entry of entries) {
+      if (!entry.startsWith('task-')) continue;
+      const taskDir = join(conductorDir, entry);
+      const stat = await fs.stat(taskDir).catch(() => null);
+      if (stat?.isDirectory()) {
+        await cleanupTaskWorktree(workDir, entry, taskDir);
+      }
+    }
+  } catch (e) {
+    console.error(`[Conductor/Worktree] Failed to cleanup all:`, e.message);
+  }
+}
+
+/**
+ * Get the worktree path for a task (for cwd routing)
+ */
+export function getTaskWorktreePath(taskDir) {
+  return join(taskDir, 'worktree');
+}
+
+/**
+ * Determine actor cwd based on specialty:
+ * - Read-write actors → worktree
+ * - Read-only actors → workDir (main project directory)
+ */
+export function getActorCwd(workDir, taskDir, specialty) {
+  const readWriteSpecialties = new Set(['coding', 'testing', 'review']);
+  if (readWriteSpecialties.has(specialty)) {
+    return getTaskWorktreePath(taskDir);
+  }
+  return workDir;
+}
+
+// =====================================================================
+// Helpers
+// =====================================================================
+
+async function directoryExists(dir) {
+  try {
+    await fs.access(dir);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+async function isKnownWorktree(projectDir, wtDir) {
+  const normalizedWtDir = resolve(wtDir);
+  try {
+    const { stdout } = await execFile('git', ['worktree', 'list', '--porcelain'], { cwd: projectDir });
+    for (const line of stdout.split('\n')) {
+      if (line.startsWith('worktree ')) {
+        const knownPath = resolve(line.slice('worktree '.length).trim());
+        if (knownPath === normalizedWtDir) return true;
+      }
+    }
+  } catch { /* ignore */ }
+  return false;
+}

package/conductor.js

@@ -1,46 +1,35 @@
 /**
- * Conductor Mode — 自适应项目执行引擎
+ * Conductor Mode — V5 Entry Point
  *
- * V2 的核心入口。与 crew.js 并存，不替代。
- *
- * 本文件聚合并重新导出 conductor 子模块的公共 API，
- * 供 message-router.js 消费。
- *
- * hideConductorSession / handleLoadConductorHistory 需要跨模块依赖，
- * 在这里注入，避免 persistence.js ↔ session.js 循环引用。
+ * Aggregates and re-exports conductor sub-module public API.
+ * Singleton model: one Conductor per Agent.
  */
 
-// Session 核心
+// Session core (singleton)
 export {
-  conductorSessions,
-  createConductorSession,
-  handleListConductorSessions,
-  resumeConductorSession,
+  getConductor,
+  initConductor,
   handleConductorUserInput,
-  handleUpdateWorkDir,
-  handleUpdateConductorSession,
-  stopConductorSession,
-  clearConductorSession
+  stopConductor,
+  clearConductor
 } from './conductor/session.js';
 
-// 持久化（原始导出）
-export { loadConductorIndex } from './conductor/persistence.js';
-
-// 依赖注入包装：hideConductorSession
-import { conductorSessions } from './conductor/session.js';
-import { sendConductorMessage } from './conductor/ui-messages.js';
-import {
-  hideConductorSession as _hideConductorSession,
-  handleLoadConductorHistory as _handleLoadConductorHistory
+// Persistence
+export {
+  handleLoadConductorHistory,
+  loadState as loadConductorState,
+  ensureConductorHome
 } from './conductor/persistence.js';
 
-export async function hideConductorSession(sessionId) {
-  return _hideConductorSession(sessionId, conductorSessions);
-}
+// UI messages
+import { sendConductorMessage } from './conductor/ui-messages.js';
+import { getConductor } from './conductor/session.js';
+import { handleLoadConductorHistory as _handleLoadConductorHistory } from './conductor/persistence.js';
 
-export async function handleLoadConductorHistory(msg) {
-  return _handleLoadConductorHistory(msg, conductorSessions, sendConductorMessage);
+// Dependency injection wrapper for history loading
+export async function handleConductorHistory(msg) {
+  return _handleLoadConductorHistory(msg, getConductor(), sendConductorMessage);
 }
 
-// 信号量
+// Semaphore
 export { globalSemaphore } from './conductor/semaphore.js';

package/connection/message-router.js

@@ -21,11 +21,9 @@ import {
   handleLoadCrewHistory, handleCheckCrewContext
 } from '../crew.js';
 import {
-  createConductorSession, handleListConductorSessions,
-  resumeConductorSession, handleConductorUserInput,
-  handleUpdateWorkDir, handleUpdateConductorSession,
-  stopConductorSession, clearConductorSession,
-  hideConductorSession, handleLoadConductorHistory
+  initConductor, handleConductorUserInput,
+  stopConductor, clearConductor,
+  handleConductorHistory
 } from '../conductor.js';
 import { sendToServer, flushMessageBuffer } from './buffer.js';
 import { handleRestartAgent, handleUpgradeAgent } from './upgrade.js';
@@ -260,44 +258,24 @@ export async function handleMessage(msg) {
       break;
 
     // Conductor (V2) messages
-    case 'create_conductor_session':
-      await createConductorSession(msg);
-      break;
-
-    case 'list_conductor_sessions':
-      await handleListConductorSessions(msg);
-      break;
-
-    case 'resume_conductor_session':
-      await resumeConductorSession(msg);
+    case 'open_conductor':
+      await initConductor(msg);
       break;
 
     case 'conductor_user_input':
       await handleConductorUserInput(msg);
       break;
 
-    case 'conductor_update_workdir':
-      await handleUpdateWorkDir(msg);
-      break;
-
-    case 'update_conductor_session':
-      await handleUpdateConductorSession(msg);
-      break;
-
-    case 'stop_conductor_session':
-      await stopConductorSession(msg.sessionId);
-      break;
-
-    case 'clear_conductor_session':
-      await clearConductorSession(msg.sessionId);
+    case 'stop_conductor':
+      await stopConductor();
       break;
 
-    case 'delete_conductor_session':
-      await hideConductorSession(msg.sessionId);
+    case 'clear_conductor':
+      await clearConductor();
       break;
 
     case 'conductor_load_history':
-      await handleLoadConductorHistory(msg);
+      await handleConductorHistory(msg);
       break;
 
     // Port proxy

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yeaft/webchat-agent",
-  "version": "0.1.224",
+  "version": "0.1.226",
   "description": "Remote agent for Yeaft WebChat — connects worker machines to the central server",
   "main": "index.js",
   "type": "module",
@@ -32,13 +32,7 @@
     "url": "https://github.com/yeaft/claude-web-chat/issues"
   },
   "files": [
-    "*.js",
-    "connection/",
-    "crew/",
-    "sdk/",
-    "scripts/",
-    "service/",
-    "workbench/"
+    "**/*.js"
   ],
   "license": "MIT",
   "author": "Yeaft",

package/scripts/agent-tray.ps1

@@ -1,134 +0,0 @@
-# Claude Agent 托盘管理器
-# 右键托盘图标可以查看日志、重启、停止等
-
-Add-Type -AssemblyName System.Windows.Forms
-Add-Type -AssemblyName System.Drawing
-
-$PM2AppName = "yeaft-agent"
-
-# 创建托盘图标
-$notifyIcon = New-Object System.Windows.Forms.NotifyIcon
-$notifyIcon.Text = "Yeaft Agent"
-$notifyIcon.Visible = $true
-
-# 创建 Claude 风格图标（橙色圆形 + 白色 C）
-function Create-ClaudeIcon {
-    $size = 32
-    $bitmap = New-Object System.Drawing.Bitmap($size, $size)
-    $graphics = [System.Drawing.Graphics]::FromImage($bitmap)
-    $graphics.SmoothingMode = [System.Drawing.Drawing2D.SmoothingMode]::AntiAlias
-    $graphics.TextRenderingHint = [System.Drawing.Text.TextRenderingHint]::AntiAlias
-
-    # 橙色背景圆形 (Claude 的品牌色)
-    $orangeBrush = New-Object System.Drawing.SolidBrush([System.Drawing.Color]::FromArgb(255, 204, 102, 51))
-    $graphics.FillEllipse($orangeBrush, 1, 1, $size - 2, $size - 2)
-
-    # 白色 "C" 字母
-    $whiteBrush = New-Object System.Drawing.SolidBrush([System.Drawing.Color]::White)
-    $font = New-Object System.Drawing.Font("Arial", 16, [System.Drawing.FontStyle]::Bold)
-    # 测量文字大小并精确居中
-    $textSize = $graphics.MeasureString("C", $font)
-    $x = ($size - $textSize.Width) / 2
-    $y = ($size - $textSize.Height) / 2
-    $graphics.DrawString("C", $font, $whiteBrush, $x, $y)
-
-    $graphics.Dispose()
-
-    # 转换为图标
-    $hIcon = $bitmap.GetHicon()
-    $icon = [System.Drawing.Icon]::FromHandle($hIcon)
-    return $icon
-}
-
-$notifyIcon.Icon = Create-ClaudeIcon
-
-# 创建右键菜单
-$contextMenu = New-Object System.Windows.Forms.ContextMenuStrip
-
-# 查看状态
-$menuStatus = New-Object System.Windows.Forms.ToolStripMenuItem
-$menuStatus.Text = "View Status"
-$menuStatus.Add_Click({
-    Start-Process "powershell" -ArgumentList "-NoExit -Command `"pm2 status`""
-})
-$contextMenu.Items.Add($menuStatus)
-
-# 查看日志
-$menuLogs = New-Object System.Windows.Forms.ToolStripMenuItem
-$menuLogs.Text = "View Logs"
-$menuLogs.Add_Click({
-    Start-Process "powershell" -ArgumentList "-NoExit -Command `"pm2 logs $PM2AppName --lines 100`""
-})
-$contextMenu.Items.Add($menuLogs)
-
-# 分隔线
-$contextMenu.Items.Add((New-Object System.Windows.Forms.ToolStripSeparator))
-
-# 重启
-$menuRestart = New-Object System.Windows.Forms.ToolStripMenuItem
-$menuRestart.Text = "Restart Agent"
-$menuRestart.Add_Click({
-    pm2 restart $PM2AppName
-    $notifyIcon.ShowBalloonTip(2000, "Yeaft Agent", "Agent restarted", [System.Windows.Forms.ToolTipIcon]::Info)
-})
-$contextMenu.Items.Add($menuRestart)
-
-# 停止
-$menuStop = New-Object System.Windows.Forms.ToolStripMenuItem
-$menuStop.Text = "Stop Agent"
-$menuStop.Add_Click({
-    pm2 stop $PM2AppName
-    $notifyIcon.ShowBalloonTip(2000, "Yeaft Agent", "Agent stopped", [System.Windows.Forms.ToolTipIcon]::Warning)
-})
-$contextMenu.Items.Add($menuStop)
-
-# 启动
-$menuStart = New-Object System.Windows.Forms.ToolStripMenuItem
-$menuStart.Text = "Start Agent"
-$menuStart.Add_Click({
-    pm2 start $PM2AppName
-    $notifyIcon.ShowBalloonTip(2000, "Yeaft Agent", "Agent started", [System.Windows.Forms.ToolTipIcon]::Info)
-})
-$contextMenu.Items.Add($menuStart)
-
-# 分隔线
-$contextMenu.Items.Add((New-Object System.Windows.Forms.ToolStripSeparator))
-
-# 打开日志文件夹
-$menuOpenLogs = New-Object System.Windows.Forms.ToolStripMenuItem
-$menuOpenLogs.Text = "Open Logs Folder"
-$menuOpenLogs.Add_Click({
-    $logDir = Join-Path $env:APPDATA "yeaft-agent\logs"
-    if (-not (Test-Path $logDir)) { New-Item -ItemType Directory -Path $logDir -Force | Out-Null }
-    Start-Process "explorer" -ArgumentList $logDir
-})
-$contextMenu.Items.Add($menuOpenLogs)
-
-# 分隔线
-$contextMenu.Items.Add((New-Object System.Windows.Forms.ToolStripSeparator))
-
-# 退出（停止 agent + 关闭托盘）
-$menuExit = New-Object System.Windows.Forms.ToolStripMenuItem
-$menuExit.Text = "Exit"
-$menuExit.Add_Click({
-    pm2 stop $PM2AppName 2>$null
-    $notifyIcon.ShowBalloonTip(1000, "Yeaft Agent", "Agent stopped", [System.Windows.Forms.ToolTipIcon]::Info)
-    Start-Sleep -Milliseconds 500
-    $notifyIcon.Visible = $false
-    $notifyIcon.Dispose()
-    [System.Windows.Forms.Application]::Exit()
-})
-$contextMenu.Items.Add($menuExit)
-
-$notifyIcon.ContextMenuStrip = $contextMenu
-
-# 双击打开日志
-$notifyIcon.Add_DoubleClick({
-    Start-Process "powershell" -ArgumentList "-NoExit -Command `"pm2 logs $PM2AppName`""
-})
-
-# 显示启动提示
-$notifyIcon.ShowBalloonTip(2000, "Yeaft Agent", "Tray manager started. Right-click for options.", [System.Windows.Forms.ToolTipIcon]::Info)
-
-# 保持运行
-[System.Windows.Forms.Application]::Run()

package/scripts/create-startup-shortcut.ps1

@@ -1,18 +0,0 @@
-# 创建开机自启动快捷方式
-$AgentDir = Split-Path -Parent $MyInvocation.MyCommand.Path
-$StartupFolder = [Environment]::GetFolderPath('Startup')
-$ShortcutPath = Join-Path $StartupFolder "Claude Agent.lnk"
-
-$WshShell = New-Object -ComObject WScript.Shell
-$Shortcut = $WshShell.CreateShortcut($ShortcutPath)
-$Shortcut.TargetPath = "powershell.exe"
-$Shortcut.Arguments = "-WindowStyle Hidden -ExecutionPolicy Bypass -File `"$AgentDir\start-with-tray.ps1`""
-$Shortcut.WorkingDirectory = $AgentDir
-$Shortcut.Description = "Claude Agent with Tray"
-$Shortcut.WindowStyle = 7  # Minimized
-$Shortcut.Save()
-
-Write-Host "Startup shortcut created at:" -ForegroundColor Green
-Write-Host "  $ShortcutPath" -ForegroundColor Cyan
-Write-Host ""
-Write-Host "The agent will now start automatically when you log in." -ForegroundColor Green