@yg-toolkit/datapull 0.1.0

package/dist/exporters/sqlserver.js

@@ -0,0 +1,126 @@
+import { fileURLToPath } from "node:url";
+import { join } from "node:path";
+import { mkdtemp, readFile, rm } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { DataPullError } from "../core/errors.js";
+import { runProcess } from "../process/runner.js";
+export class SqlServerExporter {
+    async listDatabases(connection) {
+        const result = await this.sqlcmd(connection, "master", "SET NOCOUNT ON; SELECT name FROM sys.databases WHERE state = 0 AND HAS_DBACCESS(name) = 1 ORDER BY name;");
+        return result
+            .split(/\r?\n/u)
+            .map((value) => value.trim())
+            .filter(Boolean);
+    }
+    async test(connection, database) {
+        const value = await this.sqlcmd(connection, database ?? "master", "SET NOCOUNT ON; SELECT CAST(SERVERPROPERTY('ProductVersion') AS nvarchar(128));");
+        const serverVersion = value.split(/\r?\n/u).map((line) => line.trim()).find(Boolean);
+        if (serverVersion === undefined) {
+            throw new DataPullError("DATABASE_CLIENT_FAILED", "SQL Server 未返回服务端版本。", 1);
+        }
+        return { serverVersion };
+    }
+    async exportObjects(connection, database, objectTypes) {
+        await this.test(connection, database);
+        const temporary = await mkdtemp(join(tmpdir(), "datapull-sqlserver-"));
+        try {
+            const script = fileURLToPath(new URL("../../resources/export-sqlserver.ps1", import.meta.url));
+            await runProcess("pwsh", ["-NoProfile", "-NonInteractive", "-File", script, "-OutputDirectory", temporary], {
+                env: {
+                    ...process.env,
+                    DATAPULL_SQLSERVER_HOST: required(connection.host, "host"),
+                    DATAPULL_SQLSERVER_PORT: String(connection.port ?? 1433),
+                    DATAPULL_SQLSERVER_DATABASE: database,
+                    DATAPULL_SQLSERVER_AUTH: connection.authMode,
+                    DATAPULL_SQLSERVER_USER: connection.username ?? "",
+                    DATAPULL_SQLSERVER_PASSWORD: connection.password ?? "",
+                    DATAPULL_SQLSERVER_TYPES: JSON.stringify(objectTypes),
+                    DATAPULL_SQLSERVER_ENCRYPT: "true",
+                    DATAPULL_SQLSERVER_TRUST_CERT: connection.tls?.trustServerCertificate === true ? "true" : "false",
+                },
+                timeoutMs: sqlServerExportTimeoutMs(process.env.DATAPULL_SQLSERVER_EXPORT_TIMEOUT_MS),
+                secrets: [connection.password ?? "", connection.secretUrl ?? ""],
+            });
+            const records = JSON.parse(await readFile(join(temporary, "objects.json"), "utf8"));
+            const objects = [];
+            for (const record of records) {
+                const ddl = (await readFile(join(temporary, record.file), "utf8"))
+                    .replaceAll("\r\n", "\n")
+                    .trim();
+                if (ddl.length === 0) {
+                    throw new DataPullError("DATABASE_OBJECT_READ_FAILED", `SQL Server 对象 ${record.identity} 的 DDL 为空。`, 1);
+                }
+                objects.push({
+                    type: record.objectType,
+                    ...(record.schema === undefined || record.schema === null
+                        ? {}
+                        : { schema: record.schema }),
+                    name: record.name,
+                    identity: record.identity,
+                    ddl: `${ddl}\n`,
+                });
+            }
+            return objects;
+        }
+        finally {
+            await rm(temporary, { force: true, recursive: true });
+        }
+    }
+    async sqlcmd(connection, database, query) {
+        const args = [
+            "-S",
+            `${required(connection.host, "host")},${connection.port ?? 1433}`,
+            "-d",
+            database,
+            "-N",
+            "-b",
+            "-h",
+            "-1",
+            "-W",
+            "-Q",
+            query,
+        ];
+        args.push(...sqlcmdTrustArguments(connection));
+        if (connection.authMode === "integrated")
+            args.push("-E");
+        else
+            args.push("-U", connection.username ?? "");
+        try {
+            const result = await runProcess("sqlcmd", args, {
+                env: {
+                    ...process.env,
+                    ...(connection.password === undefined
+                        ? {}
+                        : { SQLCMDPASSWORD: connection.password }),
+                },
+                timeoutMs: 120_000,
+                secrets: [connection.password ?? "", connection.secretUrl ?? ""],
+            });
+            return result.stdout;
+        }
+        catch (error) {
+            if (error instanceof DataPullError &&
+                /certificate|证书|certificate chain|unable to verify/iu.test(JSON.stringify(error.details ?? {}))) {
+                throw new DataPullError("SQLSERVER_TLS_CERTIFICATE_UNTRUSTED", "SQL Server 证书链无法验证。请安装可信 CA 链或修复服务器证书。", 1);
+            }
+            throw error;
+        }
+    }
+}
+export function sqlcmdTrustArguments(connection) {
+    return connection.tls?.trustServerCertificate === true ? ["-C"] : [];
+}
+export function sqlServerExportTimeoutMs(value) {
+    if (value === undefined || value.trim().length === 0)
+        return 900_000;
+    const timeoutMs = Number(value);
+    if (Number.isSafeInteger(timeoutMs) && timeoutMs > 0)
+        return timeoutMs;
+    throw new DataPullError("CONFIG_INVALID", "DATAPULL_SQLSERVER_EXPORT_TIMEOUT_MS 必须是正整数毫秒值。", 3);
+}
+function required(value, label) {
+    if (value !== undefined && value.length > 0)
+        return value;
+    throw new DataPullError("CONFIG_INVALID", `SQL Server 连接缺少 ${label}。`, 3);
+}
+//# sourceMappingURL=sqlserver.js.map

package/dist/exporters/sqlserver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sqlserver.js","sourceRoot":"","sources":["../../src/exporters/sqlserver.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAC;AACzD,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAYlD,MAAM,OAAO,iBAAiB;IAC5B,KAAK,CAAC,aAAa,CAAC,UAA8B;QAChD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAC9B,UAAU,EACV,QAAQ,EACR,0GAA0G,CAC3G,CAAC;QACF,OAAO,MAAM;aACV,KAAK,CAAC,QAAQ,CAAC;aACf,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;aAC5B,MAAM,CAAC,OAAO,CAAC,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,IAAI,CACR,UAA8B,EAC9B,QAAiB;QAEjB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,MAAM,CAC7B,UAAU,EACV,QAAQ,IAAI,QAAQ,EACpB,iFAAiF,CAClF,CAAC;QACF,MAAM,aAAa,GAAG,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrF,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;YAChC,MAAM,IAAI,aAAa,CAAC,wBAAwB,EAAE,sBAAsB,EAAE,CAAC,CAAC,CAAC;QAC/E,CAAC;QACD,OAAO,EAAE,aAAa,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,UAA8B,EAC9B,QAAgB,EAChB,WAA8B;QAE9B,MAAM,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACtC,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,qBAAqB,CAAC,CAAC,CAAC;QACvE,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,aAAa,CAC1B,IAAI,GAAG,CAAC,sCAAsC,EAAE,OAAO,IAAI,CAAC,GAAG,CAAC,CACjE,CAAC;YACF,MAAM,UAAU,CACd,MAAM,EACN,CAAC,YAAY,EAAE,iBAAiB,EAAE,OAAO,EAAE,MAAM,EAAE,kBAAkB,EAAE,SAAS,CAAC,EACjF;gBACE,GAAG,EAAE;oBACH,GAAG,OAAO,CAAC,GAAG;oBACd,uBAAuB,EAAE,QAAQ,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC;oBAC1D,uBAAuB,EAAE,MAAM,CAAC,UAAU,CAAC,IAAI,IAAI,IAAI,CAAC;oBACxD,2BAA2B,EAAE,QAAQ;oBACrC,uBAAuB,EAAE,UAAU,CAAC,QAAQ;oBAC5C,uBAAuB,EAAE,UAAU,CAAC,QAAQ,IAAI,EAAE;oBAClD,2BAA2B,EAAE,UAAU,CAAC,QAAQ,IAAI,EAAE;oBACtD,wBAAwB,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;oBACrD,0BAA0B,EAAE,MAAM;oBAClC,6BAA6B,EAC3B,UAAU,CAAC,GAAG,EAAE,sBAAsB,KAAK,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO;iBACrE;gBACD,SAAS,EAAE,wBAAwB,CAAC,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC;gBACrF,OAAO,EAAE,CAAC,UAAU,CAAC,QAAQ,IAAI,EAAE,EAAE,UAAU,CAAC,SAAS,IAAI,EAAE,CAAC;aACjE,CACF,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CACxB,MAAM,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC,EAAE,MAAM,CAAC,CACzC,CAAC;YACjB,MAAM,OAAO,GAAqB,EAAE,CAAC;YACrC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC7B,MAAM,GAAG,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC;qBAC/D,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC;qBACxB,IAAI,EAAE,CAAC;gBACV,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACrB,MAAM,IAAI,aAAa,CACrB,6BAA6B,EAC7B,iBAAiB,MAAM,CAAC,QAAQ,YAAY,EAC5C,CAAC,CACF,CAAC;gBACJ,CAAC;gBACD,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,MAAM,CAAC,UAAU;oBACvB,GAAG,CAAC,MAAM,CAAC,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,KAAK,IAAI;wBACvD,CAAC,CAAC,EAAE;wBACJ,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;oBAC9B,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,GAAG,EAAE,GAAG,GAAG,IAAI;iBAChB,CAAC,CAAC;YACL,CAAC;YACD,OAAO,OAAO,CAAC;QACjB,CAAC;gBAAS,CAAC;YACT,MAAM,EAAE,CAAC,SAAS,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,MAAM,CAClB,UAA8B,EAC9B,QAAgB,EAChB,KAAa;QAEb,MAAM,IAAI,GAAG;YACX,IAAI;YACJ,GAAG,QAAQ,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,UAAU,CAAC,IAAI,IAAI,IAAI,EAAE;YACjE,IAAI;YACJ,QAAQ;YACR,IAAI;YACJ,IAAI;YACJ,IAAI;YACJ,IAAI;YACJ,IAAI;YACJ,IAAI;YACJ,KAAK;SACN,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,GAAG,oBAAoB,CAAC,UAAU,CAAC,CAAC,CAAC;QAC/C,IAAI,UAAU,CAAC,QAAQ,KAAK,YAAY;YAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;;YACrD,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;QAChD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,QAAQ,EAAE,IAAI,EAAE;gBAC9C,GAAG,EAAE;oBACH,GAAG,OAAO,CAAC,GAAG;oBACd,GAAG,CAAC,UAAU,CAAC,QAAQ,KAAK,SAAS;wBACnC,CAAC,CAAC,EAAE;wBACJ,CAAC,CAAC,EAAE,cAAc,EAAE,UAAU,CAAC,QAAQ,EAAE,CAAC;iBAC7C;gBACD,SAAS,EAAE,OAAO;gBAClB,OAAO,EAAE,CAAC,UAAU,CAAC,QAAQ,IAAI,EAAE,EAAE,UAAU,CAAC,SAAS,IAAI,EAAE,CAAC;aACjE,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IACE,KAAK,YAAY,aAAa;gBAC9B,qDAAqD,CAAC,IAAI,CACxD,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC,CACpC,EACD,CAAC;gBACD,MAAM,IAAI,aAAa,CACrB,qCAAqC,EACrC,wCAAwC,EACxC,CAAC,CACF,CAAC;YACJ,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF;AAED,MAAM,UAAU,oBAAoB,CAClC,UAA2C;IAE3C,OAAO,UAAU,CAAC,GAAG,EAAE,sBAAsB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AACvE,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,KAAyB;IAChE,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC;IACrE,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAChC,IAAI,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,IAAI,SAAS,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IACvE,MAAM,IAAI,aAAa,CACrB,gBAAgB,EAChB,iDAAiD,EACjD,CAAC,CACF,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CAAC,KAAyB,EAAE,KAAa;IACxD,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,KAAK,CAAC;IAC1D,MAAM,IAAI,aAAa,CAAC,gBAAgB,EAAE,mBAAmB,KAAK,GAAG,EAAE,CAAC,CAAC,CAAC;AAC5E,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+export {};
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":""}

package/dist/index.js

@@ -0,0 +1,71 @@
+#!/usr/bin/env node
+import { CommanderError } from "commander";
+import { buildProgram } from "./cli/program.js";
+import { asDataPullError, DataPullError } from "./core/errors.js";
+import { Output } from "./core/output.js";
+import { assertSupportedNode } from "./core/runtime.js";
+const argv = process.argv.slice(2);
+const json = argv.includes("--json");
+if (argv.includes("--no-color"))
+    process.env.NO_COLOR = "1";
+const program = buildProgram();
+program.configureOutput({
+    writeOut: (value) => {
+        if (!json)
+            process.stdout.write(value);
+    },
+    writeErr: (value) => {
+        if (!json)
+            process.stderr.write(value);
+    },
+});
+try {
+    assertSupportedNode();
+    await program.parseAsync(process.argv);
+}
+catch (error) {
+    if (error instanceof CommanderError && error.exitCode === 0)
+        process.exitCode = 0;
+    else {
+        const normalized = normalizeCliError(error);
+        new Output({ command: commandName(argv), json }).failure(normalized, errorContext(argv, normalized));
+        process.exitCode = normalized.exitCode;
+    }
+}
+function normalizeCliError(error) {
+    if (error instanceof CommanderError) {
+        return new DataPullError(error.code === "commander.help" ? "INVALID_ARGUMENT" : "INVALID_ARGUMENT", error.message, 2);
+    }
+    return asDataPullError(error);
+}
+function commandName(args) {
+    const positional = args.filter((value, index) => {
+        if (value.startsWith("-"))
+            return false;
+        const previous = args[index - 1];
+        return previous === undefined || previous.startsWith("-") === false;
+    });
+    return positional.slice(0, 3).join(" ") || "datapull";
+}
+function errorContext(args, error) {
+    const details = error.details ?? {};
+    const context = {};
+    if (error.code === "OPERATION_CANCELLED")
+        context.outputFilesChanged = false;
+    for (const key of ["actionPlan", "installation", "manualRequired", "outputFilesChanged", "targets"]) {
+        if (details[key] !== undefined)
+            context[key] = details[key];
+    }
+    const connection = optionValue(args, "--connection");
+    const database = optionValue(args, "--database");
+    if (connection !== undefined)
+        context.connectionAlias = connection;
+    if (database !== undefined)
+        context.database = database;
+    return context;
+}
+function optionValue(args, option) {
+    const index = args.indexOf(option);
+    return index >= 0 ? args[index + 1] : undefined;
+}
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAClE,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAC1C,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAExD,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;AACnC,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AACrC,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC;IAAE,OAAO,CAAC,GAAG,CAAC,QAAQ,GAAG,GAAG,CAAC;AAC5D,MAAM,OAAO,GAAG,YAAY,EAAE,CAAC;AAC/B,OAAO,CAAC,eAAe,CAAC;IACtB,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE;QAClB,IAAI,CAAC,IAAI;YAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACzC,CAAC;IACD,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE;QAClB,IAAI,CAAC,IAAI;YAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACzC,CAAC;CACF,CAAC,CAAC;AAEH,IAAI,CAAC;IACH,mBAAmB,EAAE,CAAC;IACtB,MAAM,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;AACzC,CAAC;AAAC,OAAO,KAAK,EAAE,CAAC;IACf,IAAI,KAAK,YAAY,cAAc,IAAI,KAAK,CAAC,QAAQ,KAAK,CAAC;QAAE,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;SAC7E,CAAC;QACJ,MAAM,UAAU,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAC5C,IAAI,MAAM,CAAC,EAAE,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CACtD,UAAU,EACV,YAAY,CAAC,IAAI,EAAE,UAAU,CAAC,CAC/B,CAAC;QACF,OAAO,CAAC,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC;IACzC,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAc;IACvC,IAAI,KAAK,YAAY,cAAc,EAAE,CAAC;QACpC,OAAO,IAAI,aAAa,CACtB,KAAK,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,EACzE,KAAK,CAAC,OAAO,EACb,CAAC,CACF,CAAC;IACJ,CAAC;IACD,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;AAChC,CAAC;AAED,SAAS,WAAW,CAAC,IAAc;IACjC,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;QAC9C,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,OAAO,KAAK,CAAC;QACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QACjC,OAAO,QAAQ,KAAK,SAAS,IAAI,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC;IACtE,CAAC,CAAC,CAAC;IACH,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,UAAU,CAAC;AACxD,CAAC;AAED,SAAS,YAAY,CAAC,IAAc,EAAE,KAAoB;IACxD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC;IACpC,MAAM,OAAO,GAA4B,EAAE,CAAC;IAC5C,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB;QAAE,OAAO,CAAC,kBAAkB,GAAG,KAAK,CAAC;IAC7E,KAAK,MAAM,GAAG,IAAI,CAAC,YAAY,EAAE,cAAc,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,SAAS,CAAU,EAAE,CAAC;QAC7G,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,SAAS;YAAE,OAAO,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;IAC9D,CAAC;IACD,MAAM,UAAU,GAAG,WAAW,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;IACrD,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;IACjD,IAAI,UAAU,KAAK,SAAS;QAAE,OAAO,CAAC,eAAe,GAAG,UAAU,CAAC;IACnE,IAAI,QAAQ,KAAK,SAAS;QAAE,OAAO,CAAC,QAAQ,GAAG,QAAQ,CAAC;IACxD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,WAAW,CAAC,IAAc,EAAE,MAAc;IACjD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACnC,OAAO,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAClD,CAAC"}

package/dist/interactive/connections.d.ts

@@ -0,0 +1,16 @@
+import { ConnectionService } from "../connections/service.js";
+import type { AuthMode, ConnectionConfig, Engine } from "../types.js";
+export interface ConnectionInput {
+    alias?: string | undefined;
+    engine?: Engine | undefined;
+    authMode?: AuthMode | undefined;
+    host?: string | undefined;
+    port?: number | undefined;
+    username?: string | undefined;
+    credentialRef?: string | undefined;
+    urlRef?: string | undefined;
+    sslMode?: string | undefined;
+    trustServerCertificate?: boolean | undefined;
+}
+export declare function promptAndAddConnection(service: ConnectionService, initial?: ConnectionInput): Promise<ConnectionConfig>;
+//# sourceMappingURL=connections.d.ts.map

package/dist/interactive/connections.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"connections.d.ts","sourceRoot":"","sources":["../../src/interactive/connections.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAoB,MAAM,2BAA2B,CAAC;AAEhF,OAAO,KAAK,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAEtE,MAAM,WAAW,eAAe;IAC9B,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,MAAM,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B,QAAQ,CAAC,EAAE,QAAQ,GAAG,SAAS,CAAC;IAChC,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,aAAa,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACnC,MAAM,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B,sBAAsB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CAC9C;AAED,wBAAsB,sBAAsB,CAC1C,OAAO,EAAE,iBAAiB,EAC1B,OAAO,GAAE,eAAoB,GAC5B,OAAO,CAAC,gBAAgB,CAAC,CAsG3B"}

package/dist/interactive/connections.js

@@ -0,0 +1,97 @@
+import { confirm, input, password, select } from "@inquirer/prompts";
+import { ConnectionService, createConnection } from "../connections/service.js";
+import { DataPullError } from "../core/errors.js";
+export async function promptAndAddConnection(service, initial = {}) {
+    const alias = initial.alias ?? (await input({ message: "连接别名：", required: true }));
+    if ((await service.list()).some((connection) => connection.alias.toLocaleLowerCase("en-US") === alias.toLocaleLowerCase("en-US"))) {
+        throw new DataPullError("CONNECTION_ALIAS_CONFLICT", `连接别名已存在：${alias}`, 2);
+    }
+    const engine = initial.engine ??
+        (await select({
+            message: "数据库类型：",
+            choices: [
+                { name: "MySQL", value: "mysql" },
+                { name: "PostgreSQL", value: "postgresql" },
+                { name: "SQL Server", value: "sqlserver" },
+            ],
+        }));
+    const authMode = initial.authMode ??
+        (await select({
+            message: "认证方式：",
+            choices: [
+                { name: "用户名和密码", value: "password" },
+                { name: "完整连接 URL", value: "url" },
+                ...(engine === "sqlserver" && process.platform === "win32"
+                    ? [{ name: "Windows 集成认证", value: "integrated" }]
+                    : []),
+            ],
+        }));
+    if (authMode === "integrated" && (engine !== "sqlserver" || process.platform !== "win32")) {
+        throw new DataPullError("INVALID_ARGUMENT", "Windows 集成认证只支持 Windows 上的 SQL Server。", 2);
+    }
+    let host = initial.host;
+    let port = initial.port;
+    let username = initial.username;
+    let credentialRef = initial.credentialRef;
+    let urlRef = initial.urlRef;
+    let pendingSecret;
+    if (authMode !== "url") {
+        host ??= await input({ message: "数据库主机：", required: true });
+        if (port === undefined) {
+            const defaultPort = engine === "mysql" ? 3306 : engine === "postgresql" ? 5432 : 1433;
+            const value = await input({ message: "端口：", default: String(defaultPort) });
+            port = Number(value);
+        }
+    }
+    if (authMode === "password") {
+        username ??= await input({ message: "用户名：", required: true });
+        credentialRef ??= await input({
+            message: "密码变量名：",
+            default: defaultReference(alias, "PASSWORD"),
+        });
+        pendingSecret = await password({ message: "数据库密码（输入不会回显）：", mask: "*" });
+    }
+    else if (authMode === "url") {
+        urlRef ??= await input({
+            message: "连接 URL 变量名：",
+            default: defaultReference(alias, "URL"),
+        });
+        pendingSecret = await password({ message: "完整连接 URL（输入不会回显）：", mask: "*" });
+    }
+    const connection = createConnection({
+        alias,
+        engine,
+        authMode,
+        ...(host === undefined ? {} : { host }),
+        ...(port === undefined ? {} : { port }),
+        ...(username === undefined ? {} : { username }),
+        ...(credentialRef === undefined ? {} : { credentialRef }),
+        ...(urlRef === undefined ? {} : { urlRef }),
+        ...(initial.sslMode === undefined ? {} : { sslMode: initial.sslMode }),
+        ...(initial.trustServerCertificate === undefined
+            ? {}
+            : { trustServerCertificate: initial.trustServerCertificate }),
+    });
+    if (connection.tls?.trustServerCertificate === true) {
+        process.stderr.write("警告：该连接将保持加密，但不会验证 SQL Server 的身份，可能受到中间人攻击。\n");
+    }
+    const approved = await confirm({
+        message: `确认登记连接 ${connection.alias}（${connection.engine} / ${connection.authMode}）？`,
+        default: true,
+    });
+    if (!approved)
+        throw new DataPullError("OPERATION_CANCELLED", "已取消登记连接。", 1);
+    const reference = authMode === "url" ? urlRef : credentialRef;
+    if (pendingSecret !== undefined && reference !== undefined) {
+        await service.store.setCredential(reference, pendingSecret);
+    }
+    return service.add(connection);
+}
+function defaultReference(alias, suffix) {
+    const stem = alias
+        .toUpperCase()
+        .replace(/[^A-Z0-9_]/gu, "_")
+        .replace(/^[^A-Z_]/u, "_$&");
+    return `DATAPULL_${stem}_${suffix}`;
+}
+//# sourceMappingURL=connections.js.map

package/dist/interactive/connections.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connections.js","sourceRoot":"","sources":["../../src/interactive/connections.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AACrE,OAAO,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAChF,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAgBlD,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,OAA0B,EAC1B,OAAO,GAAoB,EAAE;IAE7B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACnF,IACE,CAAC,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CACzB,CAAC,UAAU,EAAE,EAAE,CACb,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CACnF,EACD,CAAC;QACD,MAAM,IAAI,aAAa,CACrB,2BAA2B,EAC3B,WAAW,KAAK,EAAE,EAClB,CAAC,CACF,CAAC;IACJ,CAAC;IACD,MAAM,MAAM,GACV,OAAO,CAAC,MAAM;QACd,CAAC,MAAM,MAAM,CAAS;YACpB,OAAO,EAAE,QAAQ;YACjB,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE;gBACjC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,YAAY,EAAE;gBAC3C,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,WAAW,EAAE;aAC3C;SACF,CAAC,CAAC,CAAC;IACN,MAAM,QAAQ,GACZ,OAAO,CAAC,QAAQ;QAChB,CAAC,MAAM,MAAM,CAAW;YACtB,OAAO,EAAE,OAAO;YAChB,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE;gBACrC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE;gBAClC,GAAG,CAAC,MAAM,KAAK,WAAW,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO;oBACxD,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,YAAqB,EAAE,CAAC;oBAC1D,CAAC,CAAC,EAAE,CAAC;aACR;SACF,CAAC,CAAC,CAAC;IACN,IAAI,QAAQ,KAAK,YAAY,IAAI,CAAC,MAAM,KAAK,WAAW,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,EAAE,CAAC;QAC1F,MAAM,IAAI,aAAa,CACrB,kBAAkB,EAClB,wCAAwC,EACxC,CAAC,CACF,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IACxB,IAAI,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IACxB,IAAI,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;IAChC,IAAI,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;IAC1C,IAAI,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAC5B,IAAI,aAAiC,CAAC;IACtC,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;QACvB,IAAI,KAAK,MAAM,KAAK,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,WAAW,GAAG,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,KAAK,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;YACtF,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YAC5E,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IACD,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;QAC5B,QAAQ,KAAK,MAAM,KAAK,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9D,aAAa,KAAK,MAAM,KAAK,CAAC;YAC5B,OAAO,EAAE,QAAQ;YACjB,OAAO,EAAE,gBAAgB,CAAC,KAAK,EAAE,UAAU,CAAC;SAC7C,CAAC,CAAC;QACH,aAAa,GAAG,MAAM,QAAQ,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3E,CAAC;SAAM,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;QAC9B,MAAM,KAAK,MAAM,KAAK,CAAC;YACrB,OAAO,EAAE,aAAa;YACtB,OAAO,EAAE,gBAAgB,CAAC,KAAK,EAAE,KAAK,CAAC;SACxC,CAAC,CAAC;QACH,aAAa,GAAG,MAAM,QAAQ,CAAC,EAAE,OAAO,EAAE,mBAAmB,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9E,CAAC;IAED,MAAM,UAAU,GAAG,gBAAgB,CAAC;QAClC,KAAK;QACL,MAAM;QACN,QAAQ;QACR,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC;QACvC,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC;QACvC,GAAG,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC;QAC/C,GAAG,CAAC,aAAa,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC;QACzD,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;QAC3C,GAAG,CAAC,OAAO,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;QACtE,GAAG,CAAC,OAAO,CAAC,sBAAsB,KAAK,SAAS;YAC9C,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,sBAAsB,EAAE,OAAO,CAAC,sBAAsB,EAAE,CAAC;KAChE,CAAC,CAAC;IACH,IAAI,UAAU,CAAC,GAAG,EAAE,sBAAsB,KAAK,IAAI,EAAE,CAAC;QACpD,OAAO,CAAC,MAAM,CAAC,KAAK,CAClB,+CAA+C,CAChD,CAAC;IACJ,CAAC;IACD,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC;QAC7B,OAAO,EAAE,UAAU,UAAU,CAAC,KAAK,IAAI,UAAU,CAAC,MAAM,MAAM,UAAU,CAAC,QAAQ,IAAI;QACrF,OAAO,EAAE,IAAI;KACd,CAAC,CAAC;IACH,IAAI,CAAC,QAAQ;QAAE,MAAM,IAAI,aAAa,CAAC,qBAAqB,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;IAC7E,MAAM,SAAS,GAAG,QAAQ,KAAK,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC;IAC9D,IAAI,aAAa,KAAK,SAAS,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;QAC3D,MAAM,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;IAC9D,CAAC;IACD,OAAO,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAa,EAAE,MAAc;IACrD,MAAM,IAAI,GAAG,KAAK;SACf,WAAW,EAAE;SACb,OAAO,CAAC,cAAc,EAAE,GAAG,CAAC;SAC5B,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;IAC/B,OAAO,YAAY,IAAI,IAAI,MAAM,EAAE,CAAC;AACtC,CAAC"}

package/dist/interactive/wizard.d.ts

@@ -0,0 +1,4 @@
+import { ConfigStore } from "../config/store.js";
+import { type PullExecutionResult } from "../pull/service.js";
+export declare function runWizard(store: ConfigStore): Promise<PullExecutionResult>;
+//# sourceMappingURL=wizard.d.ts.map

package/dist/interactive/wizard.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"wizard.d.ts","sourceRoot":"","sources":["../../src/interactive/wizard.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAIjD,OAAO,EAAe,KAAK,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAQ3E,wBAAsB,SAAS,CAAC,KAAK,EAAE,WAAW,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAyGhF"}

package/dist/interactive/wizard.js

@@ -0,0 +1,284 @@
+import { checkbox, confirm, input, password, select } from "@inquirer/prompts";
+import { Listr } from "listr2";
+import { ConnectionService } from "../connections/service.js";
+import { ConfigStore } from "../config/store.js";
+import { DataPullError, asDataPullError } from "../core/errors.js";
+import { exporterFor } from "../exporters/factory.js";
+import { commonObjectTypes, OBJECT_TYPES } from "../exporters/objects.js";
+import { PullService } from "../pull/service.js";
+import { SkillInstaller } from "../skills/installer.js";
+import { AGENTS, parseSkillTarget } from "../skills/targets.js";
+import { ToolManager } from "../tools/manager.js";
+import { discoverProjectRoot, validatePathSegment } from "../utils/path.js";
+import { promptAndAddConnection } from "./connections.js";
+export async function runWizard(store) {
+    const created = !(await store.exists());
+    if (created) {
+        process.stdout.write(`DataPull 将创建用户级配置：${store.paths.root}\n`);
+        const approved = await confirm({ message: "继续初始化？", default: true });
+        if (!approved)
+            throw new DataPullError("OPERATION_CANCELLED", "已取消初始化。", 1);
+    }
+    await store.initialize();
+    const connections = new ConnectionService(store);
+    await promptSkillSetup(store);
+    let connection = await chooseConnection(connections);
+    let trustServerCertificateOnce = false;
+    while (true) {
+        const validation = await validateConnectionWithRecovery(connections, connection);
+        if (validation !== undefined) {
+            connection = validation.connection;
+            trustServerCertificateOnce = validation.trustServerCertificateOnce;
+            break;
+        }
+        connection = await chooseConnection(connections);
+    }
+    const resolved = await connections.resolve(connection.alias, undefined, {
+        ...(trustServerCertificateOnce ? { trustServerCertificate: true } : {}),
+    });
+    const exporter = exporterFor(connection.engine);
+    let enumerated = [];
+    try {
+        enumerated = await exporter.listDatabases(resolved);
+    }
+    catch (error) {
+        const normalized = asDataPullError(error);
+        process.stderr.write(`无法枚举数据库 [${normalized.code}]，仍可手工输入。\n`);
+    }
+    const database = await chooseDatabase(connection, enumerated);
+    validatePathSegment(database, "数据库名");
+    await exporter.test(await connections.resolve(connection.alias, database, {
+        ...(trustServerCertificateOnce ? { trustServerCertificate: true } : {}),
+    }), database);
+    const common = new Set(commonObjectTypes(connection.engine));
+    const selectedTypes = await checkbox({
+        message: "选择要获取的数据库对象（默认全选）：",
+        required: true,
+        choices: OBJECT_TYPES[connection.engine].map((type) => ({
+            name: `${common.has(type) ? "常用" : "高级"} · ${type}`,
+            value: type,
+            checked: true,
+        })),
+    });
+    const projectRoot = await discoverProjectRoot();
+    process.stdout.write([
+        "\n本次操作：",
+        `  连接：${connection.alias}`,
+        `  数据库：${database}`,
+        `  对象类型：${selectedTypes.join("、")}`,
+        `  项目根：${projectRoot}`,
+        ...(connection.tls?.trustServerCertificate === true || trustServerCertificateOnce
+            ? ["  SQL Server TLS：连接已加密，但不验证服务器身份"]
+            : []),
+        "  覆盖规则：所选类型整体更新；未选类型保留；任一所选类型失败则均不更新。",
+    ].join("\n") + "\n");
+    const approved = await confirm({ message: "开始获取结构文件？", default: true });
+    if (!approved)
+        throw new DataPullError("OPERATION_CANCELLED", "已取消结构文件拉取。", 1);
+    const pull = new PullService(connections);
+    let result;
+    const tasks = new Listr([
+        {
+            title: "获取数据库结构文件",
+            task: async (_context, task) => {
+                result = await pull.execute({
+                    connectionAlias: connection.alias,
+                    database,
+                    include: selectedTypes,
+                    installMissing: true,
+                    confirmed: true,
+                    trustServerCertificate: trustServerCertificateOnce,
+                    projectRoot,
+                    onStage: (stage) => {
+                        task.output = stage;
+                    },
+                });
+            },
+        },
+    ], { concurrent: false });
+    await tasks.run();
+    if (result === undefined) {
+        throw new DataPullError("UNEXPECTED_ERROR", "拉取任务未返回结果。", 1);
+    }
+    const finalResult = result;
+    process.stdout.write(`\n获取成功：${Object.values(finalResult.objectCounts).reduce((sum, count) => sum + count, 0)} 个对象\n` +
+        `输出目录：${finalResult.outputPath}\n` +
+        `已更新：${finalResult.updatedTypes.join("、")}\n` +
+        `已保留：${finalResult.preservedTypes.join("、") || "无"}\n`);
+    return finalResult;
+}
+async function chooseConnection(service) {
+    const connections = await service.list();
+    const value = await select({
+        message: "选择登记连接：",
+        choices: [
+            ...connections.map((connection) => ({
+                name: `${connection.alias} · ${connection.engine}${connection.tls?.trustServerCertificate === true ? " · ⚠ 未验证服务器身份" : ""}`,
+                value: connection.alias,
+            })),
+            { name: "＋ 添加新数据库连接", value: "__add__" },
+        ],
+    });
+    return value === "__add__" ? promptAndAddConnection(service) : service.get(value);
+}
+async function validateConnectionWithRecovery(service, connection) {
+    const tools = new ToolManager();
+    let current = connection;
+    let trustServerCertificateOnce = false;
+    while (true) {
+        try {
+            const missing = await tools.missing(current.engine);
+            if (missing.length > 0) {
+                const plans = tools.plans(missing);
+                process.stdout.write("\n检测到缺少数据库工具：\n");
+                for (const plan of plans) {
+                    process.stdout.write(`- ${plan.tool}\n  来源：${plan.source}\n  官方说明：${plan.sourceUrl}\n  命令：${plan.command} ${plan.args.join(" ")}\n  权限：${plan.permission}\n  影响：${plan.downloadImpact}\n`);
+                }
+                const install = await confirm({ message: "确认自动安装以上工具？", default: false });
+                if (!install) {
+                    throw new DataPullError("DATABASE_TOOL_MISSING", "缺少数据库工具，请按以上命令手工安装后重试。", 4);
+                }
+                await tools.ensure(current.engine, true, true);
+            }
+            const resolved = await service.resolve(current.alias, undefined, {
+                ...(trustServerCertificateOnce ? { trustServerCertificate: true } : {}),
+            });
+            await exporterFor(current.engine).test(resolved);
+            process.stdout.write(`连接 ${current.alias} 校验通过。\n`);
+            return { connection: current, trustServerCertificateOnce };
+        }
+        catch (error) {
+            const normalized = asDataPullError(error);
+            process.stderr.write(`连接校验失败 [${normalized.code}]：${normalized.message}\n`);
+            if (current.engine === "sqlserver" &&
+                normalized.code === "SQLSERVER_TLS_CERTIFICATE_UNTRUSTED") {
+                const tlsAction = await select({
+                    message: "服务器证书不受信任。信任后连接仍加密，但无法验证服务器身份，可能受到中间人攻击：",
+                    choices: [
+                        { name: "仅本次信任并继续", value: "once" },
+                        { name: "保存到该连接并继续", value: "save" },
+                        { name: "返回连接选择", value: "back" },
+                        { name: "退出", value: "exit" },
+                    ],
+                });
+                if (tlsAction === "once") {
+                    trustServerCertificateOnce = true;
+                    continue;
+                }
+                if (tlsAction === "save") {
+                    const approved = await confirm({
+                        message: "确认保存该风险设置？可稍后使用 connection update 恢复证书验证。",
+                        default: false,
+                    });
+                    if (!approved)
+                        continue;
+                    current = await service.update(current.alias, {
+                        tls: { encrypt: true, trustServerCertificate: true },
+                    });
+                    trustServerCertificateOnce = false;
+                    continue;
+                }
+                if (tlsAction === "back")
+                    return undefined;
+                throw new DataPullError("OPERATION_CANCELLED", "已退出向导。", 1);
+            }
+            const action = await select({
+                message: "请选择恢复操作：",
+                choices: [
+                    { name: "重试", value: "retry" },
+                    { name: "更新凭证", value: "credential" },
+                    { name: "返回连接选择", value: "back" },
+                    { name: "退出", value: "exit" },
+                ],
+            });
+            if (action === "retry")
+                continue;
+            if (action === "back")
+                return undefined;
+            if (action === "exit")
+                throw new DataPullError("OPERATION_CANCELLED", "已退出向导。", 1);
+            const reference = current.authMode === "url" ? current.urlRef : current.credentialRef;
+            if (reference === undefined) {
+                process.stderr.write("该连接没有可更新的凭证引用。\n");
+                continue;
+            }
+            const value = await password({ message: `更新 ${reference}（输入不会回显）：`, mask: "*" });
+            await service.store.setCredential(reference, value);
+        }
+    }
+}
+async function chooseDatabase(connection, enumerated) {
+    const values = unique([
+        ...connection.favoriteDatabases,
+        ...connection.recentDatabases,
+        ...enumerated,
+    ]);
+    const manual = "__manual__";
+    const chosen = await select({
+        message: "选择目标数据库：",
+        choices: [
+            ...values.map((database) => ({
+                name: `${connection.favoriteDatabases.includes(database) ? "★ " : ""}${database}`,
+                value: database,
+            })),
+            { name: "手工输入数据库名", value: manual },
+        ],
+    });
+    return chosen === manual
+        ? input({ message: "目标数据库名：", required: true })
+        : chosen;
+}
+async function promptSkillSetup(store) {
+    const config = await store.read();
+    if (config.onboarding.skillPrompted)
+        return;
+    const install = await confirm({
+        message: "是否为 Agent 安装 DataPull Skill？稍后也可运行 datapull skill install。",
+        default: true,
+    });
+    if (install) {
+        const agents = await checkbox({
+            message: "选择 Agent 工具（不检测本机是否已安装）：",
+            required: true,
+            choices: AGENTS.map((agent) => ({ name: agentLabel(agent), value: agent })),
+        });
+        const projectRoot = await discoverProjectRoot();
+        const targets = [];
+        for (const agent of agents) {
+            const scopes = await checkbox({
+                message: `${agentLabel(agent)} 的安装位置：`,
+                required: true,
+                choices: [
+                    { name: "用户级", value: "user", checked: true },
+                    { name: "项目级", value: "project" },
+                ],
+            });
+            for (const scope of scopes)
+                targets.push(parseSkillTarget(`${agent}:${scope}`, projectRoot));
+        }
+        process.stdout.write(`\n将写入以下 Skill 目标：\n${targets.map((target) => `- ${target.agent}:${target.scope} → ${target.path}`).join("\n")}\n`);
+        if (await confirm({ message: "确认创建以上目录并安装 Skill？", default: true })) {
+            const installer = new SkillInstaller();
+            for (const target of targets) {
+                const status = await installer.status(target);
+                if (status.status === "missing")
+                    await installer.install(target, true);
+                else
+                    process.stdout.write(`跳过已有目标：${target.path}（${status.status}）\n`);
+            }
+        }
+    }
+    await store.write({ ...config, onboarding: { skillPrompted: true } });
+}
+function unique(values) {
+    return [...new Set(values)];
+}
+function agentLabel(agent) {
+    return {
+        codex: "Codex",
+        claude: "Claude Code",
+        cursor: "Cursor",
+        trae: "Trae IDE",
+    }[agent];
+}
+//# sourceMappingURL=wizard.js.map