draftgo-cli 4.0.1 → 4.0.23

package/src/customServices.js

@@ -8,7 +8,11 @@ const AdmZip = require('adm-zip');
 const { loadProjectConfig } = require('./projectConfig');
 const { WorktreeError } = require('./worktree/errors');
 
-const FILES = ['service.go', 'go.mod', 'go.sum', 'service.json'];
+const FILES = ['service.go', 'service.json'];
+const MANAGED_MODULE_FILES = ['go.mod', 'go.sum', 'draftgo_local_main.go', '.draftgo-managed.json'];
+const MANAGED_SDK_DIRECTORY = '.draftgo-sdk';
+const SDK_BUNDLE_ROOT = path.resolve(__dirname, '..', 'resources', 'custom-service-sdk');
+const dependencyDirectivePattern = /^\s*\/\/\s*draftgo:require\s+([^\s@]+)@([^\s]+)\s*$/gm;
 function root(projectDir) { return path.join(projectDir, '.draftgo', 'worktree', 'custom-services'); }
 function baseRoot(projectDir) { return path.join(projectDir, '.draftgo', 'worktree', '.base', 'custom-services'); }
 function conflictRoot(projectDir, id) { return path.join(projectDir, '.draftgo', 'conflicts', 'custom-services', safe(id)); }
@@ -18,9 +22,10 @@ function sha256(value) { return crypto.createHash('sha256').update(value).digest
 function relative(projectDir, file) { return path.relative(projectDir, file).replace(/\\/g, '/'); }
 function loadManifest(projectDir) {
   const file = manifestPath(projectDir);
-  if (!fs.existsSync(file)) return { schema_version: 1, entries: {} };
+  if (!fs.existsSync(file)) return { schema_version: 2, entries: {} };
   const value = JSON.parse(fs.readFileSync(file, 'utf8'));
-  if (!value || value.schema_version !== 1 || !value.entries || Array.isArray(value.entries)) throw new WorktreeError('INVALID_WORKTREE_MANIFEST', 'Invalid custom-service manifest.');
+  if (value && value.schema_version === 1) throw new WorktreeError('WORKTREE_SCHEMA_UNSUPPORTED', 'Custom-service worktree schema v1 is obsolete; create a new checkout with this CLI.');
+  if (!value || value.schema_version !== 2 || !value.entries || Array.isArray(value.entries)) throw new WorktreeError('INVALID_WORKTREE_MANIFEST', 'Invalid custom-service manifest.');
   return value;
 }
 function atomicWrite(file, value) {
@@ -84,13 +89,104 @@ function readFiles(directory) {
 }
 function createArchive(files) {
   const zip = new AdmZip();
-  for (const filename of FILES) zip.addFile(filename, files[filename]);
+  // Archive hashes are part of the checkout/commit concurrency contract. Keep
+  // ZIP metadata deterministic so the same two user files always produce the
+  // same content hash, independent of wall-clock time.
+  for (const filename of FILES) {
+    zip.addFile(filename, files[filename]);
+    const entry = zip.getEntry(filename);
+    if (entry?.header) entry.header.time = new Date(0);
+  }
   return zip.toBuffer();
 }
 function fileHashes(files) { return Object.fromEntries(FILES.map((filename) => [filename, sha256(files[filename])])); }
 function checkoutHash(files) { return sha256(createArchive(files)); }
+
+function sdkFingerprint(files) {
+  const hash = crypto.createHash('sha256');
+  for (const filename of ['go.mod', 'go.sum']) { hash.update(filename); if (files[filename]) hash.update(files[filename]); }
+  for (const filename of Object.keys(files).filter((item) => item.endsWith('.go')).sort()) { hash.update(filename); hash.update(files[filename]); }
+  return hash.digest('hex');
+}
+
+function loadSDKBundle() {
+  const manifestFile = path.join(SDK_BUNDLE_ROOT, 'manifest.json');
+  if (!fs.existsSync(manifestFile)) throw new WorktreeError('LOCAL_SDK_MISSING', 'draftgo-cli is missing its managed custom-service SDK; reinstall or upgrade the CLI.');
+  const manifest = JSON.parse(fs.readFileSync(manifestFile, 'utf8'));
+  if (!manifest || manifest.schema_version !== 1 || !/^[a-f0-9]{64}$/.test(String(manifest.fingerprint || '')) || !Array.isArray(manifest.files)) {
+    throw new WorktreeError('LOCAL_SDK_INVALID', 'draftgo-cli contains an invalid managed custom-service SDK manifest.');
+  }
+  const files = {};
+  for (const item of manifest.files) {
+    const filename = String(item?.path || '');
+    if (!/^(?:(?:[A-Za-z0-9._-]+\/)*[A-Za-z0-9._-]+\.go|go\.mod|go\.sum)$/.test(filename) || filename.includes('..')) {
+      throw new WorktreeError('LOCAL_SDK_INVALID', `Managed custom-service SDK contains an invalid path: ${filename || '<empty>'}.`);
+    }
+    const content = fs.readFileSync(path.join(SDK_BUNDLE_ROOT, filename));
+    if (sha256(content) !== item.sha256) throw new WorktreeError('LOCAL_SDK_INVALID', `Managed custom-service SDK hash mismatch: ${filename}.`);
+    files[filename] = content;
+  }
+  if (!files['go.mod'] || !Object.keys(files).some((filename) => filename.endsWith('.go'))) {
+    throw new WorktreeError('LOCAL_SDK_INVALID', 'Managed custom-service SDK is incomplete.');
+  }
+  if (sdkFingerprint(files) !== manifest.fingerprint) throw new WorktreeError('LOCAL_SDK_INVALID', 'Managed custom-service SDK fingerprint mismatch.');
+  return { manifest, files };
+}
+
+function dependencyRequirements(source) {
+  const byModule = new Map();
+  for (const match of String(source).matchAll(dependencyDirectivePattern)) {
+    const moduleName = match[1].trim(); const version = match[2].trim();
+    if (byModule.has(moduleName) && byModule.get(moduleName) !== version) {
+      throw new WorktreeError('INVALID_SERVICE_DEPENDENCY', `Dependency ${moduleName} declares conflicting versions.`);
+    }
+    byModule.set(moduleName, version);
+  }
+  return [...byModule.entries()].sort(([left], [right]) => left.localeCompare(right)).map(([module, version]) => ({ module, version }));
+}
+
+function sdkGoVersion(bundle) {
+  const match = bundle.files['go.mod'].toString('utf8').match(/^go\s+([^\s]+)\s*$/m);
+  if (!match) throw new WorktreeError('LOCAL_SDK_INVALID', 'Managed custom-service SDK go.mod has no Go version.');
+  return match[1];
+}
+
+function managedModule(source, bundle) {
+  const requirements = dependencyRequirements(source);
+  const lines = [
+    '// Code generated by draftgo-cli. DO NOT EDIT.',
+    '// Local development only; DraftGo uploads only service.go and service.json.',
+    'module draftgo.local/customservice', '', `go ${sdkGoVersion(bundle)}`, '', 'require (', '\tdraftgo/sdk v0.0.0',
+    ...requirements.map((item) => `\t${item.module} ${item.version}`),
+    ')', '', `replace draftgo/sdk => ./${MANAGED_SDK_DIRECTORY}`, '',
+  ];
+  return { content: `${lines.join('\n')}`, requirements };
+}
+
+function syncManagedDevelopmentFiles(directory) {
+  const bundle = loadSDKBundle();
+  const source = fs.readFileSync(path.join(directory, 'service.go'), 'utf8');
+  const moduleValue = managedModule(source, bundle); const moduleHash = sha256(moduleValue.content);
+  const markerFile = path.join(directory, '.draftgo-managed.json');
+  let prior = null;
+  try { prior = JSON.parse(fs.readFileSync(markerFile, 'utf8')); } catch { /* generated state is disposable */ }
+  const preserveGoSum = prior?.schema_version === 1 && prior.module_hash === moduleHash && fs.existsSync(path.join(directory, 'go.sum'));
+
+  const sdkDirectory = path.join(directory, MANAGED_SDK_DIRECTORY);
+  fs.rmSync(sdkDirectory, { recursive: true, force: true }); fs.mkdirSync(sdkDirectory, { recursive: true });
+  for (const [filename, content] of Object.entries(bundle.files)) atomicWrite(path.join(sdkDirectory, filename), content);
+  atomicWrite(path.join(directory, 'go.mod'), moduleValue.content);
+  if (!preserveGoSum) atomicWrite(path.join(directory, 'go.sum'), '');
+  atomicWrite(path.join(directory, 'draftgo_local_main.go'), `// Code generated by draftgo-cli. DO NOT EDIT.\n// Local compile entrypoint; DraftGo never uploads this file.\npackage main\n\nimport "draftgo/sdk"\n\nfunc main() {\n\tapp := sdk.NewApp()\n\tRegister(app)\n}\n`);
+  atomicWrite(markerFile, `${JSON.stringify({
+    schema_version: 1, owner: 'draftgo-cli', notice: 'Generated local-development files; do not edit or commit.',
+    sdk_fingerprint: bundle.manifest.fingerprint, module_hash: moduleHash, requirements: moduleValue.requirements,
+  }, null, 2)}\n`);
+  return { sdk_fingerprint: bundle.manifest.fingerprint, module_hash: moduleHash, requirements: moduleValue.requirements };
+}
 function remoteFileHashes(remote) {
-  return { 'service.go': remote.source_hash, 'go.mod': remote.go_mod_hash, 'go.sum': remote.go_sum_hash, 'service.json': remote.metadata_hash };
+  if (remote.checkout_schema_version != null && Number(remote.checkout_schema_version) !== 2) throw new WorktreeError('WORKTREE_SCHEMA_UNSUPPORTED', 'DraftGo custom-service checkout schema is not supported; upgrade DraftGo and create a new checkout.');
+  return { 'service.go': remote.source_hash, 'service.json': remote.metadata_hash };
 }
 function filesMatchRemote(files, remote) {
   const local = fileHashes(files); const expected = remoteFileHashes(remote);
@@ -129,6 +225,7 @@ function localState(entry) {
   return { local, base, local_hashes: localHashes, base_hashes: baseHashes, base_valid: baseValid,
     modified: FILES.some((filename) => localHashes[filename] !== baseHashes[filename]) };
 }
+
 async function checkout(projectDir, ids, options = {}) {
   const config = options.config || loadProjectConfig(projectDir); const manifest = loadManifest(projectDir);
   const targets = uniqueIDs(ids);
@@ -140,6 +237,7 @@ async function checkout(projectDir, ids, options = {}) {
     const archive = await download(config, remote, options); const files = archiveFiles(archive);
     const service = JSON.parse(files['service.json'].toString('utf8')); const localDir = old?.local_dir || serviceDir(projectDir, id, service.slug); const baseDir = serviceBaseDir(projectDir, id);
     writeFiles(localDir, files); writeFiles(baseDir, files);
+    syncManagedDevelopmentFiles(localDir);
     const entry = { server: config.server, resource_type: 'custom_services', resource_id: String(id), title: remote.title,
       slug: service.slug, local_dir: localDir, base_dir: baseDir, local_path: relative(projectDir, localDir), base_path: relative(projectDir, baseDir),
       content_type: remote.content_type, base_revision: remote.base_revision, base_etag: remote.etag,
@@ -171,13 +269,20 @@ function diffFile(baseFile, localFile, filename) {
 }
 function diff(projectDir, id) {
   const entry = getEntry(projectDir, id); if (!entry) throw new WorktreeError('RESOURCE_NOT_CHECKED_OUT', `Custom service ${id} is not checked out.`);
-  const state = localState(entry); const differences = FILES.map((filename) => diffFile(path.join(entry.base_dir, filename), path.join(entry.local_dir, filename), filename)).filter(Boolean);
-  return { changed: differences.length > 0, entry, files: FILES.map((filename) => ({ filename, changed: state.local_hashes[filename] !== state.base_hashes[filename] })), output: differences.join('\n') };
+  const state = localState(entry);
+  const files = FILES.map((filename) => {
+    const output = diffFile(path.join(entry.base_dir, filename), path.join(entry.local_dir, filename), filename);
+    return { filename, changed: state.local_hashes[filename] !== state.base_hashes[filename], output };
+  });
+  const result = { changed: files.some((file) => file.changed), entry,
+    files: files.map(({ filename, changed }) => ({ filename, changed })), output: files.map((file) => file.output).filter(Boolean).join('\n') };
+  Object.defineProperty(result, 'file_outputs', { value: files, enumerable: false });
+  return result;
 }
 async function writeConflict(projectDir, entry, remote, archive) {
   const directory = conflictRoot(projectDir, entry.resource_id); const baseDir = path.join(directory, 'base'); const localDir = path.join(directory, 'local'); const remoteDir = path.join(directory, 'remote');
   writeFiles(baseDir, readFiles(entry.base_dir)); writeFiles(localDir, readFiles(entry.local_dir)); writeFiles(remoteDir, archiveFiles(archive));
-  const record = { schema_version: 1, status: 'unresolved', code: 'RESOURCE_VERSION_CONFLICT', resource_type: 'custom_services', resource_id: entry.resource_id,
+  const record = { schema_version: 2, status: 'unresolved', code: 'RESOURCE_VERSION_CONFLICT', resource_type: 'custom_services', resource_id: entry.resource_id,
     base_path: relative(projectDir, baseDir), local_path: relative(projectDir, localDir), remote_path: relative(projectDir, remoteDir), worktree_local_path: entry.local_path,
     expected_revision: entry.base_revision, actual_revision: remote.base_revision, expected_hash: entry.base_hash, actual_hash: remote.content_hash,
     actual_etag: remote.etag, actual_updated_at: remote.updated_at, created_at: new Date().toISOString() };
@@ -209,6 +314,7 @@ async function commit(projectDir, ids, options = {}) {
   const preflight = await Promise.allSettled(targets.map(async (id) => {
     const entry = manifest.entries[entryKey(id)]; if (!entry) throw new WorktreeError('RESOURCE_NOT_CHECKED_OUT', `Custom service ${id} is not checked out.`);
     if (entry.server !== config.server) throw new WorktreeError('CHECKOUT_SERVER_MISMATCH', 'Checkout belongs to a different DraftGo server.');
+    syncManagedDevelopmentFiles(entry.local_dir);
     const state = localState(entry); if (!state.base_valid) throw new WorktreeError('BASE_HASH_MISMATCH', `Custom service ${id} base does not match its manifest.`);
     if (!state.modified) return { id, entry, state, unchanged: true };
     const current = await metadata(config, id, options);
@@ -240,7 +346,7 @@ async function commit(projectDir, ids, options = {}) {
       if ([409, 412].includes(error.status)) { const latest = await metadata(config, id, options); const remoteArchive = await download(config, latest, options); error.details = await writeConflict(projectDir, entry, latest, remoteArchive); }
       throw error;
     }
-    applyRemoteEntry(entry, committed, state.local); manifest.entries[entryKey(id)] = entry;
+    applyRemoteEntry(entry, committed, state.local); syncManagedDevelopmentFiles(entry.local_dir); manifest.entries[entryKey(id)] = entry;
     return { resource_type: 'custom_services', resource_id: String(id), status: 'committed', revision: committed.base_revision, hash: committed.content_hash };
   }));
   const results = uploaded.filter((item) => item.status === 'fulfilled').map((item) => item.value);
@@ -283,7 +389,7 @@ function syncCheckoutEntry(manifest, id, remote) {
   if (!entry) throw new WorktreeError('RESOURCE_NOT_CHECKED_OUT', `Custom service ${id} is not checked out.`);
   const state = localState(entry);
   if (state.modified) throw new WorktreeError('LOCAL_CHANGES_PRESENT', `Custom service ${id} changed locally during its remote lifecycle.`);
-  applyRemoteEntry(entry, remote, state.local); manifest.entries[entryKey(id)] = entry;
+  applyRemoteEntry(entry, remote, state.local); syncManagedDevelopmentFiles(entry.local_dir); manifest.entries[entryKey(id)] = entry;
   return entry;
 }
 async function validateCommitted(projectDir, id, config, manifest, options = {}) {
@@ -314,15 +420,29 @@ function parseHandler(handler) {
   return { kind: 'manual', name: value };
 }
 async function test(projectDir, id, input = {}, options = {}) {
-  const config = options.config || loadProjectConfig(projectDir); await commit(projectDir, [id], { ...options, config });
-  const manifest = loadManifest(projectDir); const validation = await validateCommitted(projectDir, id, config, manifest, options);
-  const entry = manifest.entries[entryKey(id)];
-  if (!validation.cached) saveManifest(projectDir, manifest);
-  const result = await jsonRequest(config, 'POST', `scripts/${encodeURIComponent(id)}/draft/execute`, {
+  const config = options.config || loadProjectConfig(projectDir);
+  const source = String(options.source || 'draft').toLowerCase();
+  if (!['auto', 'draft', 'published'].includes(source)) throw new Error('source must be auto, draft, or published.');
+  let manifest;
+  let entry;
+  let revision;
+  if (source === 'draft') {
+    await commit(projectDir, [id], { ...options, config });
+    manifest = loadManifest(projectDir);
+    const validation = await validateCommitted(projectDir, id, config, manifest, options);
+    entry = manifest.entries[entryKey(id)];
+    revision = entry.base_revision;
+    if (!validation.cached) saveManifest(projectDir, manifest);
+  } else {
+    manifest = loadManifest(projectDir);
+    entry = manifest.entries[entryKey(id)];
+  }
+  const result = await jsonRequest(config, 'POST', `scripts/${encodeURIComponent(id)}/trial`, {
+    source,
     input, selector: parseHandler(options.handler), headers: options.headers, user: options.user,
-    test_write: Boolean(options.testWrite), side_effect_policy: options.sideEffectPolicy || 'deny', revision: entry.base_revision,
+    test_write: Boolean(options.testWrite), side_effect_policy: options.sideEffectPolicy || 'deny', revision,
   }, options);
-  if (result.checkout) { syncCheckoutEntry(manifest, id, result.checkout); saveManifest(projectDir, manifest); }
+  if (result.checkout && entry) { syncCheckoutEntry(manifest, id, result.checkout); saveManifest(projectDir, manifest); }
   return result;
 }
 async function publish(projectDir, ids, options = {}) {
@@ -360,4 +480,4 @@ async function publish(projectDir, ids, options = {}) {
   return results;
 }
 
-module.exports = { FILES, checkout, commit, diff, validate, test, publish, inspectRemote, loadManifest, getEntry, metadata, archiveFiles, createArchive, serviceConflicts, showConflict, resolveConflict, parseHandler, validationStale };
+module.exports = { FILES, MANAGED_MODULE_FILES, MANAGED_SDK_DIRECTORY, checkout, commit, diff, validate, test, publish, inspectRemote, loadManifest, getEntry, metadata, archiveFiles, createArchive, serviceConflicts, showConflict, resolveConflict, parseHandler, validationStale, dependencyRequirements, managedModule, syncManagedDevelopmentFiles, loadSDKBundle, sdkFingerprint };

package/src/diffReport.js

@@ -0,0 +1,106 @@
+'use strict';
+
+const fs = require('fs');
+
+function lineStats(output) {
+  let additions = 0;
+  let deletions = 0;
+  let inHunk = false;
+  for (const line of String(output || '').split(/\r?\n/)) {
+    if (line.startsWith('@@ ')) {
+      inHunk = true;
+      continue;
+    }
+    if (line.startsWith('diff --git ')) {
+      inHunk = false;
+      continue;
+    }
+    if (!inHunk || line === '\') continue;
+    if (line.startsWith('+')) additions += 1;
+    else if (line.startsWith('-')) deletions += 1;
+  }
+  return { additions, deletions, lines_changed: additions + deletions };
+}
+
+function byteSize(file) {
+  return fs.statSync(file).size;
+}
+
+function fileStat(file) {
+  const base_bytes = byteSize(file.base_path);
+  const local_bytes = byteSize(file.local_path);
+  return {
+    path: file.path,
+    changed: Boolean(file.changed),
+    ...lineStats(file.output),
+    base_bytes,
+    local_bytes,
+    bytes_delta: local_bytes - base_bytes,
+  };
+}
+
+function resourceMetadata(entry) {
+  return {
+    resource_type: entry.resource_type,
+    resource_id: entry.resource_id,
+    title: entry.title || null,
+    route: entry.route || null,
+    slug: entry.slug || null,
+    local_path: entry.local_path || null,
+    base_path: entry.base_path || null,
+    base_version: entry.base_version ?? null,
+    base_revision: entry.base_revision ?? null,
+    base_etag: entry.base_etag ?? null,
+  };
+}
+
+function report(entry, files) {
+  const file_stats = files.map(fileStat);
+  const changed_files = file_stats.filter((file) => file.changed);
+  const additions = changed_files.reduce((total, file) => total + file.additions, 0);
+  const deletions = changed_files.reduce((total, file) => total + file.deletions, 0);
+  const base_bytes = file_stats.reduce((total, file) => total + file.base_bytes, 0);
+  const local_bytes = file_stats.reduce((total, file) => total + file.local_bytes, 0);
+  return {
+    resource: resourceMetadata(entry),
+    changed: changed_files.length > 0,
+    bytes: { base: base_bytes, local: local_bytes, delta: local_bytes - base_bytes },
+    stats: {
+      files_changed: changed_files.length,
+      additions,
+      deletions,
+      lines_changed: additions + deletions,
+      files: file_stats,
+    },
+  };
+}
+
+function resourceLabel(resource) {
+  return `${resource.resource_type} ${resource.resource_id}`;
+}
+
+function formatSummary(value) {
+  const { resource, bytes } = value;
+  const fields = [
+    `${resourceLabel(resource)}: ${value.changed ? 'changed' : 'no local changes'}`,
+    `title: ${resource.title || '-'}`,
+    `path: ${resource.local_path || '-'}`,
+    `base version: ${resource.base_version ?? '-'}`,
+    `base revision: ${resource.base_revision ?? '-'}`,
+    `bytes: ${bytes.base} -> ${bytes.local} (${bytes.delta >= 0 ? '+' : ''}${bytes.delta})`,
+  ];
+  if (resource.route) fields.splice(2, 0, `route: ${resource.route}`);
+  return `${fields.join('\n')}\n`;
+}
+
+function formatStat(value) {
+  const { stats } = value;
+  const files = stats.files.filter((file) => file.changed).map((file) => {
+    const marker = `${file.additions ? `${'+'.repeat(Math.min(file.additions, 20))}` : ''}${file.deletions ? `${'-'.repeat(Math.min(file.deletions, 20))}` : ''}` || '0';
+    return `${file.path} | ${file.lines_changed} ${marker}`;
+  });
+  const summary = `${stats.files_changed} file${stats.files_changed === 1 ? '' : 's'} changed, ${stats.additions} insertion${stats.additions === 1 ? '' : 's'}(+), ${stats.deletions} deletion${stats.deletions === 1 ? '' : 's'}(-)`;
+  return `${formatSummary(value)}${files.length ? `${files.join('\n')}\n` : ''}${summary}\n`;
+}
+
+module.exports = { lineStats, report, formatSummary, formatStat };

package/src/index.js

@@ -3,9 +3,11 @@
 const path = require('path');
 const { parse } = require('./cli');
 const { resolveCommand } = require('./commandRegistry');
+const { configureConsoleUtf8 } = require('./consoleEncoding');
 const log = require('./logger');
 
 async function run(argv) {
+  configureConsoleUtf8();
   const { command, positional, flags, errors } = parse(argv);
 
   if (errors.length) {

package/src/localRuntime/compose.js

@@ -44,7 +44,7 @@ function generate(projectDir, opts) {
   const env = readEnv(projectDir);
   const projectName = sanitizeProjectName(opts.projectName, 'draftgo');
   const database = opts.mysql.database;
-  const milvus = opts.milvus || { host: '127.0.0.1', port: 19530, username: '', password: '' };
+  const qdrant = opts.qdrant || { host: '127.0.0.1', port: 26333, apiKey: '' };
 
   env.APP_PORT = String(opts.appPort || 3000);
   env.SECRET_KEY = env.SECRET_KEY || randomHex(32);
@@ -53,14 +53,12 @@ function generate(projectDir, opts) {
   env.REDIS_PORT = String(opts.redis.port);
   env.REDIS_PASSWORD = opts.redis.password || '';
   env.REDIS_KEY_PREFIX = database;
-  env.MILVUS_ADDRESS = `${dockerHost(milvus.host)}:${milvus.port}`;
-  env.MILVUS_USERNAME = milvus.username || '';
-  env.MILVUS_PASSWORD = milvus.password || '';
-  env.MILVUS_DATABASE = 'default';
-  env.MILVUS_COLLECTION_PREFIX = database;
+  env.QDRANT_URL = `http://${dockerHost(qdrant.host)}:${qdrant.port}`;
+  env.QDRANT_API_KEY = qdrant.apiKey || '';
+  env.QDRANT_COLLECTION_PREFIX = database;
 
   const yaml = [
-    '# Generated by draftgo-cli. Shared MySQL, Redis, and Milvus stay on the host.',
+    '# Generated by draftgo-cli. Shared MySQL, Redis, and Qdrant stay on the host.',
     `name: ${projectName}`,
     'services:',
     '  app:',
@@ -80,23 +78,22 @@ function generate(projectDir, opts) {
     '      REDIS_PASSWORD: ${REDIS_PASSWORD}',
     '      REDIS_DB: "0"',
     '      REDIS_KEY_PREFIX: ${REDIS_KEY_PREFIX}',
-    '      MILVUS_ADDRESS: ${MILVUS_ADDRESS}',
-    '      MILVUS_USERNAME: ${MILVUS_USERNAME}',
-    '      MILVUS_PASSWORD: ${MILVUS_PASSWORD}',
-    '      MILVUS_DATABASE: ${MILVUS_DATABASE}',
-    '      MILVUS_COLLECTION_PREFIX: ${MILVUS_COLLECTION_PREFIX}',
-    '      LOCAL_UPLOAD_DIR: /app/backend/storage/uploads',
+    '      QDRANT_URL: ${QDRANT_URL}',
+    '      QDRANT_API_KEY: ${QDRANT_API_KEY}',
+    '      QDRANT_COLLECTION_PREFIX: ${QDRANT_COLLECTION_PREFIX}',
+    '      LOCAL_UPLOAD_DIR: /app/data/storage/uploads',
+    '      KB_STORAGE_DIR: /app/data/storage/kb',
+    '      DRAFTGO_GO_SERVICE_CACHE_DIR: /app/data/storage/custom-services/cache',
+    '      MCP_TEMP_DIR: /app/data/storage/mcp/tmp',
     '    volumes:',
-    '      - ./data/uploads:/app/backend/storage/uploads',
-    '      - ./data/logs:/app/backend/logs',
-    '      - ./data/db:/app/backend/db',
+    '      - ./data/storage:/app/data/storage',
     '    extra_hosts:',
     '      - "host.docker.internal:host-gateway"',
     '',
   ].join('\n');
   writeText(composePath(projectDir), yaml);
   writeEnv(projectDir, env);
-  for (const subdir of ['data/uploads', 'data/logs', 'data/db']) {
+  for (const subdir of ['data/storage', 'data/storage/uploads', 'data/storage/kb', 'data/storage/mcp/tmp', 'data/storage/custom-services/cache']) {
     try { fs.mkdirSync(path.join(root, subdir), { recursive: true }); } catch {}
   }
   return { dir: root, composeFile: composePath(projectDir), envFile: envPath(projectDir), projectName, env };

package/src/localRuntime/index.js

@@ -2,11 +2,11 @@
 
 const { spawnSync } = require('child_process');
 const log = require('../logger');
-const { ask, askRequired, askPassword, confirm } = require('../prompt');
+const { askRequired, askPassword, confirm } = require('../prompt');
 const { probePort, probeHttp, detectDocker } = require('./detect');
 const compose = require('./compose');
 const { ensureDatabase, testConnection, describeClient } = require('./mysqlClient');
-const { defaults, portOpen, probeRedis, probeMilvus, startService } = require('./services');
+const { defaults, portOpen, probeRedis, probeQdrant, startService } = require('./services');
 const { writeProjectConfig } = require('../projectConfig');
 const { appendGitignoreLine } = require('../fsx');
 
@@ -88,30 +88,29 @@ async function planRedis(docker) {
   return conn;
 }
 
-async function promptMilvusConnection() {
+async function promptQdrantConnection() {
   while (true) {
-    const host = await askRequired('  Milvus host', { default: '127.0.0.1' });
-    const port = Number(await askRequired('  Milvus port', { default: '19530' }));
-    const username = await ask('  Milvus username (empty for none)', { default: '' });
-    const password = await askPassword('  Milvus password (empty for none)', { default: '' });
-    const conn = { host, port, username, password };
-    if (await probeMilvus(conn)) return conn;
-    log.err('  Milvus health check failed. Check the address and credentials.');
+    const host = await askRequired('  Qdrant host', { default: '127.0.0.1' });
+    const port = Number(await askRequired('  Qdrant port', { default: '26333' }));
+    const apiKey = await askPassword('  Qdrant API key (empty for none)', { default: '' });
+    const conn = { host, port, apiKey };
+    if (await probeQdrant(conn)) return conn;
+    log.err('  Qdrant REST health check failed. Check the address and API key.');
   }
 }
 
-async function planMilvus(docker) {
-  log.step('Checking Milvus');
-  let conn = { ...defaults.milvus };
+async function planQdrant(docker) {
+  log.step('Checking Qdrant');
+  let conn = { ...defaults.qdrant };
   if (!await portOpen(conn.host, conn.port)) {
-    log.dim('  Milvus is not running locally; starting the shared local service.');
-    if (!startService(docker, 'milvus')) throw new Error('Unable to start shared Milvus.');
+    log.dim('  Qdrant is not running locally; starting the shared local service.');
+    if (!startService(docker, 'qdrant')) throw new Error('Unable to start shared Qdrant.');
   }
-  if (!await probeMilvus(conn)) {
-    log.warn('  Default unauthenticated Milvus health check failed.');
-    conn = await promptMilvusConnection();
+  if (!await probeQdrant(conn)) {
+    log.warn('  Default unauthenticated Qdrant REST health check failed.');
+    conn = await promptQdrantConnection();
   }
-  log.ok(`  Milvus ready: ${conn.host}:${conn.port}`);
+  log.ok(`  Qdrant ready: ${conn.host}:${conn.port}`);
   return conn;
 }
 
@@ -190,17 +189,17 @@ async function runWizard(projectDir, { yes = false, mcpCommands } = {}) {
   const projectName = await pickProjectName();
   const mysql = await planMysql(docker, projectName);
   const redis = await planRedis(docker);
-  const milvus = await planMilvus(docker);
+  const qdrant = await planQdrant(docker);
 
-  const out = compose.generate(projectDir, { projectName, appPort, mysql, redis, milvus });
+  const out = compose.generate(projectDir, { projectName, appPort, mysql, redis, qdrant });
   appendGitignoreLine(projectDir, '.draftgo/docker/');
   if (!composeUp(docker, out.dir) || !await waitForApp(appPort)) return 1;
 
   const url = `http://localhost:${appPort}`;
   let token = '';
   while (token.length < 10) {
-    token = String(await askPassword('Paste a DraftGo system access token (SAT)')).trim();
-    if (token.length < 10) log.err('  Token appears too short. Please try again.');
+    token = String(await askPassword('Paste a DraftGo user API Key')).trim();
+    if (token.length < 10) log.err('  API Key appears too short. Please try again.');
   }
   const cfgPath = writeProjectConfig(projectDir, url, token);
   await finishMcpSetup(projectDir, mcpCommands);

package/src/localRuntime/services.js

@@ -5,6 +5,8 @@
 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');
 
@@ -15,14 +17,14 @@ 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' },
   redis: { host: '127.0.0.1', port: 6379, password: '' },
-  milvus: { host: '127.0.0.1', port: 19530, username: '', password: '' },
+  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', 'redis'));
-  ensureDir(path.join(SERVICES_DIR, 'data', 'milvus'));
+  ensureDir(path.join(SERVICES_DIR, 'data', 'qdrant'));
   writeText(MYSQL_INIT_FILE, [
     "GRANT ALL PRIVILEGES ON *.* TO 'draftgo'@'%' WITH GRANT OPTION;",
     'FLUSH PRIVILEGES;',
@@ -66,27 +68,19 @@ function writeServiceFiles() {
     '      interval: 5s',
     '      timeout: 5s',
     '      retries: 20',
-    '  milvus:',
-    '    image: milvusdb/milvus:v2.6.0',
-    '    container_name: draftgo-local-milvus',
+    '  qdrant:',
+    '    image: qdrant/qdrant:v1.19.0',
+    '    container_name: draftgo-local-qdrant',
     '    restart: unless-stopped',
-    '    command: ["milvus", "run", "standalone"]',
-    '    security_opt:',
-    '      - seccomp:unconfined',
-    '    environment:',
-    '      DEPLOY_MODE: STANDALONE',
-    '      ETCD_USE_EMBED: "true"',
-    '      ETCD_DATA_DIR: /var/lib/milvus/etcd',
-    '      COMMON_STORAGETYPE: local',
     '    ports:',
-    '      - "127.0.0.1:19530:19530"',
+    '      - "127.0.0.1:26333:6333"',
     '    volumes:',
-    '      - ./data/milvus:/var/lib/milvus',
+    '      - ./data/qdrant:/qdrant/storage',
     '    healthcheck:',
-    '      test: ["CMD", "curl", "-f", "http://localhost:9091/healthz"]',
+    '      test: ["CMD-SHELL", "bash -c \':> /dev/tcp/127.0.0.1/6333\'"]',
     '      interval: 10s',
     '      timeout: 5s',
-    '      retries: 18',
+    '      retries: 12',
     '      start_period: 30s',
     '',
   ].join('\n'));
@@ -123,25 +117,22 @@ function probeRedis({ host, port, password = '' }) {
   });
 }
 
-function probeMilvus({ host, port, username = '', password = '' }) {
+function probeQdrant({ host, port, apiKey = '' }) {
   return new Promise((resolve) => {
-    const grpc = require('@grpc/grpc-js');
-    const client = new grpc.Client(`${host}:${port}`, grpc.credentials.createInsecure());
-    const metadata = new grpc.Metadata();
-    if (username || password) metadata.add('authorization', `Basic ${Buffer.from(`${username}:${password}`).toString('base64')}`);
-    client.makeUnaryRequest(
-      '/milvus.proto.milvus.MilvusService/CheckHealth',
-      (value) => value,
-      (value) => value,
-      Buffer.alloc(0),
-      metadata,
-      { deadline: Date.now() + 4000 },
-      (err, response) => {
-        client.close();
-        // CheckHealthResponse starts with field 1 (is_healthy) when the server is healthy.
-        resolve(!err && Buffer.isBuffer(response) && response.includes(0x01));
-      },
-    );
+    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));
   });
 }
 
@@ -158,6 +149,6 @@ module.exports = {
   defaults,
   portOpen,
   probeRedis,
-  probeMilvus,
+  probeQdrant,
   startService,
 };