draftgo-cli 1.0.4

package/src/localRuntime/services.js

@@ -0,0 +1,117 @@
+'use strict';
+
+// Shared local runtime dependencies live outside individual projects. Project Compose
+// files run DraftGo and isolated Redis, then connect back to MySQL and Qdrant.
+const net = require('net');
+const os = require('os');
+const path = require('path');
+const http = require('http');
+const https = require('https');
+const { spawnSync } = require('child_process');
+const { ensureDir, writeText } = require('../fsx');
+
+const SERVICES_DIR = path.join(os.homedir(), '.draftgo', 'local-services');
+const COMPOSE_FILE = path.join(SERVICES_DIR, 'docker-compose.yaml');
+const MYSQL_INIT_FILE = path.join(SERVICES_DIR, 'mysql-init.sql');
+
+const defaults = {
+  mysql: { host: '127.0.0.1', port: 3306, user: 'draftgo', password: 'draftgo' },
+  qdrant: { host: '127.0.0.1', port: 26333, apiKey: '' },
+};
+
+function writeServiceFiles() {
+  ensureDir(SERVICES_DIR);
+  ensureDir(path.join(SERVICES_DIR, 'data', 'mysql'));
+  ensureDir(path.join(SERVICES_DIR, 'data', 'qdrant'));
+  writeText(MYSQL_INIT_FILE, [
+    "GRANT ALL PRIVILEGES ON `draftgo\\_%`.* TO 'draftgo'@'%';",
+    'FLUSH PRIVILEGES;',
+    '',
+  ].join('\n'));
+  writeText(COMPOSE_FILE, [
+    'name: draftgo-local-services',
+    'services:',
+    '  mysql:',
+    '    image: mysql:8.0',
+    '    container_name: draftgo-local-mysql',
+    '    restart: unless-stopped',
+    '    command: ["--character-set-server=utf8mb4", "--collation-server=utf8mb4_unicode_ci"]',
+    '    environment:',
+    '      MYSQL_ROOT_PASSWORD: draftgo',
+    '      MYSQL_DATABASE: draftgo_bootstrap',
+    '      MYSQL_USER: draftgo',
+    '      MYSQL_PASSWORD: draftgo',
+    '    ports:',
+    '      - "127.0.0.1:3306:3306"',
+    '    volumes:',
+    '      - ./data/mysql:/var/lib/mysql',
+    '      - ./mysql-init.sql:/docker-entrypoint-initdb.d/01-grants.sql:ro',
+    '    healthcheck:',
+    '      test: ["CMD", "mysqladmin", "ping", "-h", "localhost", "-u", "root", "-pdraftgo"]',
+    '      interval: 5s',
+    '      timeout: 5s',
+    '      retries: 24',
+    '      start_period: 20s',
+    '  qdrant:',
+    '    image: qdrant/qdrant:v1.19.0',
+    '    container_name: draftgo-local-qdrant',
+    '    restart: unless-stopped',
+    '    ports:',
+    '      - "127.0.0.1:26333:6333"',
+    '    volumes:',
+    '      - ./data/qdrant:/qdrant/storage',
+    '    healthcheck:',
+    '      test: ["CMD-SHELL", "bash -c \':> /dev/tcp/127.0.0.1/6333\'"]',
+    '      interval: 10s',
+    '      timeout: 5s',
+    '      retries: 12',
+    '      start_period: 30s',
+    '',
+  ].join('\n'));
+}
+
+function portOpen(host, port, timeoutMs = 800) {
+  return new Promise((resolve) => {
+    const socket = net.createConnection({ host, port });
+    const finish = (ok) => { socket.destroy(); resolve(ok); };
+    socket.setTimeout(timeoutMs);
+    socket.once('connect', () => finish(true));
+    socket.once('timeout', () => finish(false));
+    socket.once('error', () => finish(false));
+  });
+}
+
+function probeQdrant({ host, port, apiKey = '' }) {
+  return new Promise((resolve) => {
+    const client = String(host || '').startsWith('https://') ? https : http;
+    const hostname = String(host || '').replace(/^https?:\/\//, '');
+    const request = client.get({
+      hostname,
+      port,
+      path: '/healthz',
+      headers: apiKey ? { 'api-key': apiKey } : undefined,
+      timeout: 4000,
+    }, (response) => {
+      response.resume();
+      resolve(response.statusCode >= 200 && response.statusCode < 300);
+    });
+    request.once('timeout', () => { request.destroy(); resolve(false); });
+    request.once('error', () => resolve(false));
+  });
+}
+
+function startService(docker, service) {
+  writeServiceFiles();
+  const result = spawnSync(docker.composeCmd, [
+    ...docker.composeArgs, '-f', COMPOSE_FILE, 'up', '-d', '--wait', service,
+  ], { cwd: SERVICES_DIR, stdio: 'inherit', shell: false });
+  return result.status === 0;
+}
+
+module.exports = {
+  SERVICES_DIR,
+  defaults,
+  portOpen,
+  probeQdrant,
+  startService,
+};

package/src/logger.js

@@ -0,0 +1,37 @@
+'use strict';
+
+// Minimal ANSI color helpers - no external deps.
+const isTTY = process.stdout && process.stdout.isTTY;
+const supportsColor = isTTY && !process.env.NO_COLOR;
+const isWindows = process.platform === 'win32';
+const wantsUnicode = process.env.DRAFTGO_UNICODE === '1';
+const useUnicode = wantsUnicode || !isWindows || process.env.TERM_PROGRAM === 'vscode' || process.env.WT_SESSION;
+const sym = useUnicode
+  ? { info: '\u2139', ok: '\u2713', err: '\u2717', step: '\u2192' }
+  : { info: 'i', ok: 'OK', err: 'x', step: '>' };
+
+const wrap = (code) => (s) => supportsColor ? `\x1b[${code}m${s}\x1b[0m` : String(s);
+
+const c = {
+  bold: wrap(1),
+  dim: wrap(2),
+  red: wrap(31),
+  green: wrap(32),
+  yellow: wrap(33),
+  blue: wrap(34),
+  magenta: wrap(35),
+  cyan: wrap(36),
+  gray: wrap(90),
+};
+
+function info(msg) { console.log(`${c.cyan(sym.info)} ${msg}`); }
+function ok(msg) { console.log(`${c.green(sym.ok)} ${msg}`); }
+function warn(msg) { console.warn(`${c.yellow('!')} ${msg}`); }
+function err(msg) { console.error(`${c.red(sym.err)} ${msg}`); }
+function step(msg) { console.log(`${c.blue(sym.step)} ${msg}`); }
+function title(msg) { console.log(`
+${c.bold(c.magenta(msg))}`); }
+function plain(msg) { console.log(msg); }
+function dim(msg) { console.log(c.dim(msg)); }
+
+module.exports = { c, info, ok, warn, err, step, title, plain, dim };