z-clean 0.3.1 → 0.3.3

package/README.ko.md

@@ -55,17 +55,36 @@ AI 코딩 도구는 작업 중 많은 임시 런타임을 만듭니다.
 
 ## 설치
 
+먼저 설치나 정리 없이 읽기 전용 audit을 실행하세요:
+
 ```bash
-npm install -g z-clean
-zclean init
+npx --yes z-clean audit
 ```
 
-설치 없이 1회 실행:
+정리 전에 후보를 검토하세요:
 
 ```bash
 npx --yes z-clean
+npx --yes z-clean report
+```
+
+계속 사용하려면 전역 설치하세요:
+
+```bash
+npm install --global z-clean --foreground-scripts
+zclean report
+```
+
+`--foreground-scripts`를 사용하면 zclean 설치 워드마크가 표시됩니다. npm 7 이상은 lifecycle 출력을 기본적으로 숨기므로 `npm install -g z-clean`도 정상 설치되지만 완료 화면은 보이지 않을 수 있습니다.
+
+자동 실행은 선택 사항입니다:
+
+```bash
+zclean init
 ```
 
+`zclean init`은 Claude Code session hook과 OS scheduler를 설치합니다. dry-run 결과를 확인한 후에만 실행하세요. 상주 daemon을 설치하지는 않습니다.
+
 ## 자주 쓰는 명령
 
 ```bash
@@ -76,12 +95,15 @@ zclean report --json    # 자동화용 JSON report
 zclean cache            # workspace cache dry-run scan
 zclean cache --yes      # 지원되는 cache directory 삭제
 zclean cache --json     # cache 후보 JSON 출력
+zclean --pattern=my-agent-worker  # 이번 scan에 literal orphan pattern 추가
 zclean history --json   # 정리 이력 JSON
 zclean protect list     # 보호 목록 확인
 zclean protect add mcp-server-keep
 zclean doctor --json    # 설치/탐지/스케줄러 상태 점검
 ```
 
+패턴을 계속 사용하려면 `~/.zclean/config.json`의 `customPatterns` 배열에 문자열을 추가하세요. 패턴은 정규식이 아닌 대소문자 무시 literal 문자열이며 3–80자의 출력 가능한 문자만 허용됩니다. `node`, `node /`, `ode` 같은 일반 runtime 이름과 부분 문자열은 거부되고, orphan이면서 `maxAge`(기본 24시간)를 넘은 process만 후보가 됩니다. 잘못되거나 0인 기간은 안전한 기본값 24시간으로 돌아갑니다. whitelist, PID 재검증, dry-run, `--yes` 안전장치는 그대로 유지됩니다.
+
 다른 workspace cache를 확인하려면:
 
 ```bash

package/README.md

@@ -15,7 +15,7 @@ AI coding runtime hygiene
 [![Node.js](https://img.shields.io/badge/node-%3E%3D18-339933?style=flat-square&logo=node.js&logoColor=white)](#)
 [![Platform](https://img.shields.io/badge/platform-macOS%20%7C%20Linux%20%7C%20Windows-lightgrey?style=flat-square)](#platform-status)
 [![Zero Dependencies](https://img.shields.io/badge/dependencies-0-blue?style=flat-square)](#)
-[![Tests](https://img.shields.io/badge/tests-101%20passing-brightgreen?style=flat-square)](#)
+[![Tests](https://img.shields.io/badge/tests-142%20passing-brightgreen?style=flat-square)](#)
 [![GitHub stars](https://img.shields.io/github/stars/TheStack-ai/zclean?style=flat-square)](https://github.com/TheStack-ai/zclean)
 [![Mentioned in Awesome Claude Code Toolkit](https://awesome.re/mentioned-badge.svg)](https://github.com/rohitg00/awesome-claude-code-toolkit)
 
@@ -51,24 +51,28 @@ The goal is narrower: **AI coding cleanup and developer workspace hygiene**.
 
 ## Quick Start
 
+Start with a read-only audit. This does not install hooks, schedule jobs, or clean anything:
+
 ```bash
-npm install -g z-clean
-zclean init
+npx --yes z-clean audit
 ```
 
-One-off dry run without installation:
+Review the candidates before cleaning:
 
 ```bash
 npx --yes z-clean
+npx --yes z-clean report
 ```
 
-Review before cleaning:
+Install the CLI when you are ready to keep using it:
 
 ```bash
+npm install --global z-clean --foreground-scripts
 zclean report
-zclean report --json
 ```
 
+`--foreground-scripts` displays zclean's install wordmark. npm 7+ hides lifecycle output by default, so plain `npm install -g z-clean` still installs correctly but may not show the branded completion screen.
+
 Clean only when you decide:
 
 ```bash
@@ -76,6 +80,14 @@ zclean --yes
 zclean cache --yes
 ```
 
+Optional automation:
+
+```bash
+zclean init
+```
+
+`zclean init` installs the Claude Code session hook and the native OS scheduler. Run it only after reviewing the dry-run output. It does not install a persistent daemon.
+
 ## Why developers use it
 
 AI coding tools spawn many short-lived runtimes while they work:
@@ -101,6 +113,7 @@ When an agent crashes, a terminal closes, or a session ends abruptly, those chil
 | `zclean cache` | Dry-run scan for supported workspace cache directories |
 | `zclean cache --yes` | Remove supported workspace cache directories |
 | `zclean cache --json` | Machine-readable cache hygiene report |
+| `zclean --pattern=my-agent-worker` | Add one literal orphan-process pattern for this scan |
 | `zclean history` | Recent cleanup history |
 | `zclean history --json` | Sanitized cleanup history and cumulative stats |
 | `zclean protect list` | Show protected process patterns |
@@ -249,7 +262,8 @@ zclean doctor --json
   "sigterm_timeout": 10,
   "dryRunDefault": true,
   "logRetention": "30d",
-  "customAiDirs": []
+  "customAiDirs": [],
+  "customPatterns": []
 }
 ```
 
@@ -263,6 +277,9 @@ zclean doctor --json
 | `dryRunDefault` | `true` | Manual scans stay dry-run by default; cleanup still requires `--yes` |
 | `logRetention` | `"30d"` | Cleanup history retention |
 | `customAiDirs` | `[]` | Additional AI tool directories to detect, such as `[".mytool"]` |
+| `customPatterns` | `[]` | Case-insensitive literal command fragments for additional orphan processes |
+
+Custom patterns are deliberately restricted: values must contain 3–80 printable characters, generic runtime names and fragments such as `node`, `node /`, or `ode` are rejected, and values are treated as literal text rather than regular expressions. Candidates must be orphaned and older than `maxAge` (24 hours by default); invalid or zero durations fall back to that safe default. Whitelists, PID identity verification, batch limits, dry-run behavior, and the explicit `--yes` cleanup gate still apply.
 
 ## Supported tools
 

package/README.zh.md

@@ -55,17 +55,36 @@ AI 编程工具在工作时会启动很多临时运行时：
 
 ## 安装
 
+先运行只读 audit；它不会安装 hook、创建定时任务或执行清理：
+
 ```bash
-npm install -g z-clean
-zclean init
+npx --yes z-clean audit
 ```
 
-无需安装的一次性运行：
+清理前先查看候选项：
 
 ```bash
 npx --yes z-clean
+npx --yes z-clean report
+```
+
+准备长期使用时再全局安装：
+
+```bash
+npm install --global z-clean --foreground-scripts
+zclean report
+```
+
+`--foreground-scripts` 会显示 zclean 安装字标。npm 7 及以上默认隐藏 lifecycle 输出，因此普通的 `npm install -g z-clean` 仍会正常安装，但可能不显示品牌完成画面。
+
+自动运行是可选项：
+
+```bash
+zclean init
 ```
 
+`zclean init` 会安装 Claude Code session hook 和原生 OS scheduler。请先检查 dry-run 结果；它不会安装常驻 daemon。
+
 ## 常用命令
 
 ```bash
@@ -76,12 +95,15 @@ zclean report --json    # 自动化 JSON report
 zclean cache            # workspace cache dry-run scan
 zclean cache --yes      # 删除支持的 cache directory
 zclean cache --json     # 输出 cache 候选 JSON
+zclean --pattern=my-agent-worker  # 为本次扫描添加 literal orphan pattern
 zclean history --json   # 清理历史 JSON
 zclean protect list     # 查看保护列表
 zclean protect add mcp-server-keep
 zclean doctor --json    # 检查安装、调度器和进程枚举状态
 ```
 
+如需长期使用，请把字符串加入 `~/.zclean/config.json` 的 `customPatterns` 数组。pattern 是忽略大小写的 literal 文本，不是正则表达式；只允许 3–80 个可打印字符，并拒绝 `node`、`node /`、`ode` 等通用 runtime 名称或片段。只有 orphan 且超过 `maxAge`（默认 24 小时）的 process 才会成为候选项；无效或为零的时长会回退到安全的 24 小时默认值。whitelist、PID 再验证、dry-run 和显式 `--yes` 安全限制仍然有效。
+
 扫描另一个 workspace：
 
 ```bash

package/bin/zclean.js

@@ -6,21 +6,21 @@ const { scan, hasScanErrors } = require('../src/scanner');
 const { killZombies } = require('../src/killer');
 const {
   loadConfig,
-  saveConfig,
   readLogs,
   pruneLogs,
-  DEFAULT_CONFIG,
   appendLog,
   getCumulativeStats,
   getConfigFile,
 } = require('../src/config');
 const { reportDryRun, reportKill, reportStatus, reportLogs, reportConfig, c, bold } = require('../src/reporter');
-const { installHook, removeHook } = require('../src/installer/hook');
+const { removeHook } = require('../src/installer/hook');
 const { runHistory, runProtect } = require('../src/commands/trust');
-const { installScheduler, uninstallScheduler } = require('../src/commands/scheduler');
+const { uninstallScheduler } = require('../src/commands/scheduler');
 const { runDoctor } = require('../src/doctor');
 const { runAudit } = require('../src/audit');
 const { runCache } = require('../src/cache');
+const { getCustomPatternError, normalizeCustomPattern } = require('../src/detector/patterns');
+const { runInit } = require('../src/commands/init');
 
 // Platform-specific installers (lazy loaded)
 const platform = os.platform();
@@ -33,8 +33,10 @@ const positional = [];
 
 for (const arg of args) {
   if (arg.startsWith('--')) {
-    const [key, value] = arg.substring(2).split('=');
-    flags[key] = value !== undefined ? value : true;
+    const body = arg.substring(2);
+    const separator = body.indexOf('=');
+    const key = separator === -1 ? body : body.slice(0, separator);
+    flags[key] = separator === -1 ? true : body.slice(separator + 1);
   } else if (arg.startsWith('-') && arg.length === 2) {
     flags[arg.substring(1)] = true;
   } else {
@@ -64,7 +66,7 @@ async function main() {
 
   switch (command) {
     case 'init':
-      return cmdInit(config);
+      return runInit({ config, platform });
 
     case 'status':
       return cmdStatus(config);
@@ -114,17 +116,19 @@ async function main() {
  * Dry-run unless --yes is passed.
  */
 function cmdScan(config) {
+  const scanConfig = withPatternFlag(config);
+  if (!scanConfig) return;
   const sessionPid = parseSessionPidFlag();
   const force = flags.yes || flags.y;
 
   console.log(bold('\n  zclean') + c('gray', ' — scanning for zombie processes...\n'));
 
-  const zombies = scan(config, { sessionPid });
+  const zombies = scan(scanConfig, { sessionPid });
 
   if (hasScanErrors(zombies)) {
     reportDryRun(zombies);
     appendLog({ action: 'scan-failed', errors: zombies.errors.length });
-    pruneLogs(config);
+    pruneLogs(scanConfig);
     process.exitCode = 1;
     return;
   }
@@ -137,12 +141,12 @@ function cmdScan(config) {
       return;
     }
     appendLog({ action: 'scan', found: zombies.length });
-    const results = killZombies(zombies, config);
+    const results = killZombies(zombies, scanConfig);
     results.cumulative = getCumulativeStats();
     reportKill(results);
   } else {
     // Dry-run mode
-    if (config.dryRunDefault === false) {
+    if (scanConfig.dryRunDefault === false) {
       console.log(c('yellow', '  dryRunDefault=false is not an auto-kill switch; pass --yes to clean.\n'));
     }
     reportDryRun(zombies);
@@ -152,40 +156,16 @@ function cmdScan(config) {
   }
 
   // Prune old logs
-  pruneLogs(config);
-}
-
-/**
- * init: Install hooks + scheduler.
- */
-function cmdInit(config) {
-  console.log(bold('\n  zclean init') + c('gray', ' — installing hooks and scheduler...\n'));
-
-  // 1. Save default config if none exists
-  const existingConfig = loadConfig();
-  if (JSON.stringify(existingConfig) === JSON.stringify(DEFAULT_CONFIG)) {
-    saveConfig(DEFAULT_CONFIG);
-    console.log(c('green', '  Config created:') + ` ${getConfigFile()}`);
-  } else {
-    console.log(c('gray', '  Config exists:') + ` ${getConfigFile()}`);
-  }
-
-  // 2. Install Claude Code hook
-  const hookResult = installHook();
-  const hookIcon = hookResult.installed ? c('green', '  Hook:') : c('yellow', '  Hook:');
-  console.log(`${hookIcon} ${hookResult.message}`);
-
-  // 3. Install platform-specific scheduler
-  installScheduler(platform);
-
-  console.log();
+  pruneLogs(scanConfig);
 }
 
 /**
  * status: Show current zombies and last cleanup info.
  */
 function cmdStatus(config) {
-  const zombies = scan(config);
+  const scanConfig = withPatternFlag(config);
+  if (!scanConfig) return;
+  const zombies = scan(scanConfig);
   const logs = readLogs(100);
   reportStatus(zombies, logs);
   if (hasScanErrors(zombies)) process.exitCode = 1;
@@ -232,8 +212,10 @@ function cmdDoctor(config) {
 }
 
 function cmdAudit(config, commandName = 'audit') {
+  const scanConfig = withPatternFlag(config);
+  if (!scanConfig) return;
   const sessionPid = parseSessionPidFlag();
-  const report = runAudit(config, { json: Boolean(flags.json), sessionPid, commandName });
+  const report = runAudit(scanConfig, { json: Boolean(flags.json), sessionPid, commandName });
   if (!report.risk.enumerationComplete) process.exitCode = 1;
 }
 
@@ -257,6 +239,19 @@ function parseSessionPidFlag() {
   process.exit(1);
 }
 
+function withPatternFlag(config) {
+  if (flags.pattern === undefined) return config;
+  const error = getCustomPatternError(flags.pattern);
+  if (error) {
+    console.error(c('red', `  ${error}`));
+    process.exitCode = 1;
+    return null;
+  }
+  const literal = normalizeCustomPattern(flags.pattern);
+  const configured = Array.isArray(config.customPatterns) ? config.customPatterns : [];
+  return { ...config, customPatterns: [...configured, literal] };
+}
+
 // ─── Help ───────────────────────────────────────────────────────────────────
 
 function printHelp() {
@@ -285,6 +280,7 @@ function printHelp() {
   ${bold('Options:')}
     --yes, -y             Kill found zombies (default: dry-run)
     --session-pid=PID     Filter by parent session PID
+    --pattern=TEXT        Add a literal orphan-process pattern
     --path=DIR            Workspace path for zclean cache
     --json                Print machine-readable output for supported commands
     --version, -v         Show version

package/package.json

@@ -1,15 +1,16 @@
 {
   "name": "z-clean",
-  "version": "0.3.1",
+  "version": "0.3.3",
   "description": "AI coding runtime hygiene reports, zombie cleanup, and safe workspace cache cleanup",
   "bin": {
     "zclean": "bin/zclean.js"
   },
   "scripts": {
-    "test": "node --test test/patterns.test.js test/scanner.test.js test/process-tree.test.js test/killer.test.js test/config.test.js test/installer-fallback.test.js test/cli.test.js test/cli-history-protect.test.js test/cli-doctor.test.js test/audit.test.js test/cache.test.js",
+    "test": "node --test test/patterns.test.js test/scanner.test.js test/process-tree.test.js test/killer.test.js test/config.test.js test/installer-fallback.test.js test/install-experience.test.js test/reporter.test.js test/cli.test.js test/cli-history-protect.test.js test/cli-doctor.test.js test/audit.test.js test/cache.test.js",
     "syntax": "node -e \"const fs=require('node:fs');const path=require('node:path');const {execFileSync}=require('node:child_process');const files=['bin/zclean.js'];function walk(dir){if(!fs.existsSync(dir))return;for(const entry of fs.readdirSync(dir,{withFileTypes:true})){const file=path.join(dir,entry.name);if(entry.isDirectory())walk(file);else if(entry.name.endsWith('.js'))files.push(file);}}for(const dir of ['src','test','scripts'])walk(dir);for(const file of files)execFileSync(process.execPath,['-c',file],{stdio:'inherit'});\"",
     "smoke": "node scripts/smoke.js",
-    "pack:check": "npm pack --dry-run --json"
+    "pack:check": "npm pack --dry-run --json",
+    "postinstall": "node scripts/postinstall.js"
   },
   "keywords": [
     "claude-code",
@@ -40,6 +41,7 @@
   "files": [
     "bin/",
     "src/",
+    "scripts/postinstall.js",
     "assets/zclean-hero.png",
     "LICENSE",
     "README.md",

package/scripts/postinstall.js

@@ -0,0 +1,7 @@
+#!/usr/bin/env node
+'use strict';
+
+const { version } = require('../package.json');
+const { renderPostinstall } = require('../src/cli-brand');
+
+process.stdout.write(renderPostinstall({ version }));

package/src/cli-brand.js

@@ -0,0 +1,129 @@
+'use strict';
+
+const { c, bold } = require('./reporter');
+const { sanitizeTerminalText } = require('./terminal-text');
+
+const WORDMARK = [
+  '  ███████  ██████ ██      ███████  █████  ███    ██',
+  '       ██ ██      ██      ██      ██   ██ ████   ██',
+  '     ███  ██      ██      █████   ███████ ██ ██  ██',
+  '   ██     ██      ██      ██      ██   ██ ██  ██ ██',
+  '  ███████  ██████ ███████ ███████ ██   ██ ██   ████',
+];
+
+const RULE = '  --------------------------------------------------------';
+const FRAME_WIDTH = 55;
+
+function renderPostinstall({ version }) {
+  return [
+    '',
+    c('gray', '     .   +  .'),
+    c('gray', '  +    .     :'),
+    ...WORDMARK.map((line) => c('cyan', line)),
+    '',
+    `  ${bold('Z / CLEAN')}  ${c('gray', `v${version}`)}`,
+    c('gray', '  AI CODING RUNTIME HYGIENE'),
+    c('gray', RULE),
+    '',
+    `  ${c('green', 'INSTALLED')}   z-clean`,
+    `  ${c('gray', 'COMMAND')}     zclean`,
+    `  ${c('gray', 'SAFETY')}      dry-run by default`,
+    '',
+    c('gray', '  Continue setup'),
+    `    ${c('cyan', '$ zclean init')}`,
+    '',
+    c('gray', '  Inspect first. Clean only when you decide.'),
+    '',
+  ].join('\n');
+}
+
+function renderInit({ version, steps, warningCount }) {
+  const lines = [
+    '',
+    c('gray', '     .   +  .'),
+    c('gray', '  +    .     :'),
+    ...WORDMARK.map((line) => c('cyan', line)),
+    '',
+    `  ${bold('Z / CLEAN')}  ${c('gray', `v${version}`)}`,
+    c('gray', '  RUNTIME HYGIENE SYSTEM'),
+    '',
+    c('gray', `  +${'-- SETUP '.padEnd(FRAME_WIDTH, '-')}+`),
+    c('gray', `  |${''.padEnd(FRAME_WIDTH)}|`),
+  ];
+
+  for (const step of steps) {
+    lines.push(renderStep(step));
+  }
+
+  const ready = warningCount === 0;
+  lines.push(
+    c('gray', `  |${''.padEnd(FRAME_WIDTH)}|`),
+    c('gray', `  +${''.padEnd(FRAME_WIDTH, '-')}+`),
+    '',
+    ready
+      ? c('green', '  SYSTEM READY')
+      : c('yellow', `  SETUP COMPLETED WITH ${warningCount} WARNING${warningCount === 1 ? '' : 'S'}`),
+    '',
+    `  ${c('gray', 'Inspect runtime leftovers')}   ${c('cyan', 'zclean audit')}`,
+    `  ${c('gray', 'Verify this installation')}    ${c('cyan', 'zclean doctor')}`,
+    '',
+    c('gray', '  Cleanup remains locked until you pass --yes.'),
+    ''
+  );
+
+  return lines.join('\n');
+}
+
+function renderStep(step) {
+  const state = String(step.state).padEnd(10);
+  const label = String(step.label).padEnd(14);
+  const tone = step.state === 'WARNING' ? 'yellow' : step.state === 'EXISTS' ? 'gray' : 'green';
+  const statusPrefix = `  ${step.index}  ${label} `;
+  const statusSpacing = ''.padEnd(FRAME_WIDTH - statusPrefix.length - state.length);
+  const detail = padToWidth(`      ${fitDetail(step.detail)}`, FRAME_WIDTH);
+  return [
+    `  |${c('gray', statusPrefix)}${c(tone, state)}${statusSpacing}|`,
+    `  |${c('gray', detail)}|`,
+  ].join('\n');
+}
+
+function fitDetail(value) {
+  const text = sanitizeTerminalText(value).replace(/\s+/g, ' ').trim();
+  if (displayWidth(text) <= 45) return text;
+  let output = '';
+  let width = 0;
+  for (const char of text) {
+    const next = charWidth(char);
+    if (width + next > 42) break;
+    output += char;
+    width += next;
+  }
+  return `${output}...`;
+}
+
+function padToWidth(value, width) {
+  return `${value}${''.padEnd(Math.max(0, width - displayWidth(value)))}`;
+}
+
+function displayWidth(value) {
+  return [...value].reduce((width, char) => width + charWidth(char), 0);
+}
+
+function charWidth(char) {
+  if (/\p{Mark}/u.test(char)) return 0;
+  const code = char.codePointAt(0);
+  const wide = code >= 0x1100 && (
+    code <= 0x115f || code === 0x2329 || code === 0x232a ||
+    (code >= 0x2e80 && code <= 0xa4cf && code !== 0x303f) ||
+    (code >= 0xac00 && code <= 0xd7a3) ||
+    (code >= 0xf900 && code <= 0xfaff) ||
+    (code >= 0xfe10 && code <= 0xfe6f) ||
+    (code >= 0xff00 && code <= 0xff60) ||
+    (code >= 0xffe0 && code <= 0xffe6) ||
+    (code >= 0x1f300 && code <= 0x1f9ff) ||
+    (code >= 0x20000 && code <= 0x3fffd)
+  );
+  return wide ? 2 : 1;
+}
+
+module.exports = { renderPostinstall, renderInit };

package/src/commands/init.js

@@ -0,0 +1,50 @@
+'use strict';
+
+const fs = require('node:fs');
+const os = require('node:os');
+const { saveConfig, getConfigFile } = require('../config');
+const { installHook } = require('../installer/hook');
+const { installScheduler, getSchedulerState } = require('./scheduler');
+const { renderInit } = require('../cli-brand');
+const { version } = require('../../package.json');
+
+function runInit({ config, platform }) {
+  const configFile = getConfigFile();
+  const configExists = fs.existsSync(configFile);
+  if (!configExists) saveConfig(config);
+
+  const hookResult = installHook();
+  const schedulerResult = installScheduler(platform);
+  const schedulerState = getSchedulerState(schedulerResult);
+  const steps = [
+    {
+      index: '01',
+      label: 'CONFIG',
+      state: configExists ? 'EXISTS' : 'READY',
+      detail: configFile.replace(os.homedir(), '~'),
+    },
+    {
+      index: '02',
+      label: 'CLAUDE HOOK',
+      state: hookResult.installed ? 'INSTALLED' : 'WARNING',
+      detail: hookResult.installed ? 'SessionEnd cleanup hook registered' : hookResult.message,
+    },
+    {
+      index: '03',
+      label: 'SCHEDULER',
+      state: schedulerState,
+      detail: schedulerState === 'ACTIVE' ? schedulerLabel(platform) : schedulerResult.message,
+    },
+  ];
+  const warningCount = steps.filter((step) => step.state === 'WARNING').length;
+  console.log(renderInit({ version, steps, warningCount }));
+}
+
+function schedulerLabel(platform) {
+  if (platform === 'darwin') return 'Hourly launchd hygiene check';
+  if (platform === 'linux') return 'Hourly systemd hygiene timer';
+  if (platform === 'win32') return 'Hourly Task Scheduler check';
+  return 'Runtime hygiene scheduler';
+}
+
+module.exports = { runInit };

package/src/commands/scheduler.js

@@ -6,53 +6,57 @@ function installScheduler(platform) {
   switch (platform) {
     case 'darwin': {
       const { installLaunchd } = require('../installer/launchd');
-      const result = installLaunchd();
-      const icon = result.installed ? c('green', '  Scheduler:') : c('yellow', '  Scheduler:');
-      console.log(`${icon} ${result.message}`);
-      break;
+      return installLaunchd();
     }
     case 'linux': {
       const { installSystemd } = require('../installer/systemd');
-      const result = installSystemd();
-      const icon = result.installed ? c('green', '  Scheduler:') : c('yellow', '  Scheduler:');
-      console.log(`${icon} ${result.message}`);
-      break;
+      return installSystemd();
     }
     case 'win32': {
       const { installTaskScheduler } = require('../installer/taskscheduler');
-      const result = installTaskScheduler();
-      const icon = result.installed ? c('green', '  Scheduler:') : c('yellow', '  Scheduler:');
-      console.log(`${icon} ${result.message}`);
-      break;
+      return installTaskScheduler();
     }
     default:
-      console.log(c('yellow', `  Scheduler: Unsupported platform (${platform}). Install a cron job manually.`));
+      return {
+        installed: false,
+        message: `Unsupported platform (${platform}). Install a cron job manually.`,
+      };
   }
 }
 
-function uninstallScheduler(platform) {
+function uninstallScheduler(platform, options = {}) {
+  const log = options.log || console.log;
+  let result;
   switch (platform) {
     case 'darwin': {
       const { removeLaunchd } = require('../installer/launchd');
-      const result = removeLaunchd();
-      console.log(`  Scheduler: ${result.message}`);
+      result = (options.remove || removeLaunchd)();
+      log(`  Scheduler: ${result.message}`);
       break;
     }
     case 'linux': {
       const { removeSystemd } = require('../installer/systemd');
-      const result = removeSystemd();
-      console.log(`  Scheduler: ${result.message}`);
+      result = (options.remove || removeSystemd)();
+      log(`  Scheduler: ${result.message}`);
       break;
     }
     case 'win32': {
       const { removeTaskScheduler } = require('../installer/taskscheduler');
-      const result = removeTaskScheduler();
-      console.log(`  Scheduler: ${result.message}`);
+      result = (options.remove || removeTaskScheduler)();
+      log(`  Scheduler: ${result.message}`);
       break;
     }
-    default:
-      console.log(c('yellow', `  Scheduler: Remove manually for ${platform}.`));
+    default: {
+      result = { removed: false, failed: false, message: `Remove manually for ${platform}.` };
+      log(c('yellow', `  Scheduler: ${result.message}`));
+    }
   }
+  if (result.failed) process.exitCode = 1;
+  return result;
+}
+
+function getSchedulerState(result) {
+  return result.installed && result.active !== false ? 'ACTIVE' : 'WARNING';
 }
 
-module.exports = { installScheduler, uninstallScheduler };
+module.exports = { installScheduler, uninstallScheduler, getSchedulerState };