@dezhouzhang/archery-query-mcp 0.1.0

package/dist/services.d.ts

@@ -0,0 +1,39 @@
+import type { DescribeTableResponse, ExecuteQueryResponse, PrepareQueryResponse, QueryHistoryResponse } from "./responses.js";
+import { type QueryScopeInput, QueryScopeStore } from "./scope.js";
+import { PreparedQueryStore } from "./store.js";
+import type { ArcheryClient, PolicyConfig } from "./types.js";
+export type ServiceContext = {
+    readonly userSubject: string;
+    readonly policy: PolicyConfig;
+    readonly archeryClient: ArcheryClient;
+    readonly preparedQueries: PreparedQueryStore;
+    readonly scopes: QueryScopeStore;
+};
+export type PrepareQueryInput = QueryScopeInput & {
+    readonly bugContext: string;
+    readonly sql: string;
+    readonly limitNum: number;
+};
+export type ExecuteQueryInput = {
+    readonly prepareId: string;
+    readonly confirmationToken: string | null;
+    readonly maxRows: number;
+};
+export type DescribeTableInput = QueryScopeInput & {
+    readonly schemaName: string;
+    readonly tableName: string;
+};
+export type QueryHistoryInput = QueryScopeInput & {
+    readonly search: string;
+    readonly limit: number;
+};
+export declare function createServiceContext(options: {
+    readonly userSubject: string;
+    readonly policy: PolicyConfig;
+    readonly archeryClient: ArcheryClient;
+}): ServiceContext;
+export declare function prepareQuery(context: ServiceContext, input: PrepareQueryInput): Promise<PrepareQueryResponse>;
+export declare function executeQuery(context: ServiceContext, input: ExecuteQueryInput): Promise<ExecuteQueryResponse>;
+export declare function describeTable(context: ServiceContext, input: DescribeTableInput): Promise<DescribeTableResponse>;
+export declare function queryHistory(context: ServiceContext, input: QueryHistoryInput): Promise<QueryHistoryResponse>;
+//# sourceMappingURL=services.d.ts.map

package/dist/services.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"services.d.ts","sourceRoot":"","sources":["../src/services.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,qBAAqB,EACrB,oBAAoB,EACpB,oBAAoB,EACpB,oBAAoB,EACrB,MAAM,gBAAgB,CAAA;AACvB,OAAO,EAAE,KAAK,eAAe,EAAE,eAAe,EAAgB,MAAM,YAAY,CAAA;AAChF,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAA;AAC/C,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAA6B,MAAM,YAAY,CAAA;AAExF,MAAM,MAAM,cAAc,GAAG;IAC3B,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAA;IAC5B,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAA;IAC7B,QAAQ,CAAC,aAAa,EAAE,aAAa,CAAA;IACrC,QAAQ,CAAC,eAAe,EAAE,kBAAkB,CAAA;IAC5C,QAAQ,CAAC,MAAM,EAAE,eAAe,CAAA;CACjC,CAAA;AAED,MAAM,MAAM,iBAAiB,GAAG,eAAe,GAAG;IAChD,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAA;IAC3B,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAA;IACpB,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAA;CAC1B,CAAA;AAED,MAAM,MAAM,iBAAiB,GAAG;IAC9B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAA;IAC1B,QAAQ,CAAC,iBAAiB,EAAE,MAAM,GAAG,IAAI,CAAA;IACzC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAA;CACzB,CAAA;AAED,MAAM,MAAM,kBAAkB,GAAG,eAAe,GAAG;IACjD,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAA;IAC3B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAA;CAC3B,CAAA;AAED,MAAM,MAAM,iBAAiB,GAAG,eAAe,GAAG;IAChD,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAA;CACvB,CAAA;AAED,wBAAgB,oBAAoB,CAAC,OAAO,EAAE;IAC5C,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAA;IAC5B,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAA;IAC7B,QAAQ,CAAC,aAAa,EAAE,aAAa,CAAA;CACtC,GAAG,cAAc,CAQjB;AAED,wBAAsB,YAAY,CAChC,OAAO,EAAE,cAAc,EACvB,KAAK,EAAE,iBAAiB,GACvB,OAAO,CAAC,oBAAoB,CAAC,CAwD/B;AAED,wBAAsB,YAAY,CAChC,OAAO,EAAE,cAAc,EACvB,KAAK,EAAE,iBAAiB,GACvB,OAAO,CAAC,oBAAoB,CAAC,CAsC/B;AAED,wBAAsB,aAAa,CACjC,OAAO,EAAE,cAAc,EACvB,KAAK,EAAE,kBAAkB,GACxB,OAAO,CAAC,qBAAqB,CAAC,CAqBhC;AAED,wBAAsB,YAAY,CAChC,OAAO,EAAE,cAAc,EACvB,KAAK,EAAE,iBAAiB,GACvB,OAAO,CAAC,oBAAoB,CAAC,CAa/B"}

package/dist/services.js

@@ -0,0 +1,171 @@
+import { randomUUID } from "node:crypto";
+import { sqlHash, validateSql } from "./policy.js";
+import { QueryScopeStore, resolveScope } from "./scope.js";
+import { PreparedQueryStore } from "./store.js";
+export function createServiceContext(options) {
+    return {
+        userSubject: options.userSubject,
+        policy: options.policy,
+        archeryClient: options.archeryClient,
+        preparedQueries: new PreparedQueryStore(),
+        scopes: new QueryScopeStore(),
+    };
+}
+export async function prepareQuery(context, input) {
+    const scopeResult = resolveScope(context.scopes, context.userSubject, input);
+    if (!scopeResult.ok) {
+        return {
+            prepareId: null,
+            normalizedSql: input.sql,
+            riskLevel: scopeResult.code,
+            requiresConfirmation: false,
+            policyFindings: [],
+            confirmationSummary: scopeResult.message,
+        };
+    }
+    const decision = validateSql(context.policy, scopeResult.scope.instanceName, scopeResult.scope.dbName, input.sql, input.limitNum);
+    const hash = sqlHash(decision.normalizedSql);
+    if (!decision.allowed) {
+        return {
+            prepareId: null,
+            normalizedSql: decision.normalizedSql,
+            riskLevel: "blocked",
+            requiresConfirmation: false,
+            policyFindings: decision.findings,
+            confirmationSummary: "Query blocked by MCP policy",
+        };
+    }
+    const prepareId = `prep_${randomUUID()}`;
+    const riskLevel = decision.riskLevel === "low" ? "low" : "medium";
+    const preparedQuery = {
+        id: prepareId,
+        requestUser: context.userSubject,
+        ...scopeResult.scope,
+        normalizedSql: decision.normalizedSql,
+        sqlHash: hash,
+        riskLevel,
+        expiresAt: new Date(Date.now() + 10 * 60 * 1000),
+        used: false,
+    };
+    context.preparedQueries.save(preparedQuery);
+    context.scopes.bind(context.userSubject, scopeResult.scope);
+    const requiresConfirmation = context.policy.requireConfirmation || riskLevel !== "low";
+    const baseSummary = `Review SQL hash ${hash} for ${scopeResult.scope.cloudName}/${scopeResult.scope.instanceName}/${scopeResult.scope.dbName}`;
+    return {
+        prepareId,
+        normalizedSql: decision.normalizedSql,
+        riskLevel,
+        requiresConfirmation,
+        policyFindings: decision.findings,
+        confirmationSummary: requiresConfirmation
+            ? `${baseSummary}; required confirmation token: CONFIRM:${prepareId}:${hash}`
+            : `${baseSummary}; low-risk query may execute without a confirmation token by policy`,
+    };
+}
+export async function executeQuery(context, input) {
+    const preparedQuery = context.preparedQueries.find(input.prepareId);
+    if (preparedQuery === null) {
+        return blockedExecution("Unknown prepare_id");
+    }
+    if (preparedQuery.used) {
+        return blockedExecution("Prepared query already used");
+    }
+    if (preparedQuery.expiresAt.getTime() < Date.now()) {
+        return blockedExecution("Prepared query expired");
+    }
+    if (requiresConfirmation(context.policy, preparedQuery)) {
+        const approval = verifyConfirmation(context.userSubject, preparedQuery, input.confirmationToken);
+        if (!approval.ok) {
+            return blockedExecution(approval.reason);
+        }
+    }
+    const effectiveMaxRows = input.maxRows <= 0
+        ? context.policy.defaultLimit
+        : Math.min(input.maxRows, context.policy.maxLimit);
+    const result = await context.archeryClient.executeQuery(preparedQuery, preparedQuery.normalizedSql, effectiveMaxRows);
+    context.preparedQueries.markUsed(input.prepareId);
+    return {
+        executionId: `exec_${randomUUID()}`,
+        status: result.status,
+        columns: result.columns,
+        rows: result.rows,
+        rowCount: result.rowCount,
+        durationMs: result.durationMs,
+        masked: result.masked,
+        archeryMessage: result.message,
+        archeryQueryLogId: result.archeryQueryLogId,
+    };
+}
+export async function describeTable(context, input) {
+    const scopeResult = resolveScope(context.scopes, context.userSubject, input);
+    if (!scopeResult.ok) {
+        return { columns: [], source: `${scopeResult.code}: ${scopeResult.message}` };
+    }
+    const blockReason = metadataPolicyBlock(context.policy, scopeResult.scope, input.tableName, "table");
+    if (blockReason !== null) {
+        return { columns: [], source: `blocked: ${blockReason}` };
+    }
+    const metadata = await context.archeryClient.describeTable(scopeResult.scope, input.schemaName, input.tableName);
+    context.scopes.bind(context.userSubject, scopeResult.scope);
+    return metadata;
+}
+export async function queryHistory(context, input) {
+    const scopeResult = resolveScope(context.scopes, context.userSubject, input);
+    if (!scopeResult.ok) {
+        return { matches: [], total: 0, truncated: false, status: scopeResult.code };
+    }
+    const blockReason = metadataPolicyBlock(context.policy, scopeResult.scope, input.search, "search");
+    if (blockReason !== null) {
+        return { matches: [], total: 0, truncated: false, status: "blocked" };
+    }
+    const limit = input.limit <= 0 ? 20 : Math.min(input.limit, 100);
+    const result = await context.archeryClient.queryHistory(scopeResult.scope, input.search, limit);
+    context.scopes.bind(context.userSubject, scopeResult.scope);
+    return { ...result, status: "ok" };
+}
+function requiresConfirmation(config, query) {
+    return config.requireConfirmation || query.riskLevel !== "low";
+}
+function verifyConfirmation(userSubject, query, token) {
+    if (token === null || token.trim().length === 0) {
+        return { ok: false, reason: "Confirmation token required" };
+    }
+    if (query.requestUser !== userSubject) {
+        return { ok: false, reason: "Confirmation token mismatch" };
+    }
+    const expectedToken = `CONFIRM:${query.id}:${query.sqlHash}`;
+    return token.trim() === expectedToken
+        ? { ok: true }
+        : { ok: false, reason: "Confirmation token mismatch" };
+}
+function blockedExecution(reason) {
+    return {
+        executionId: `exec_${randomUUID()}`,
+        status: "blocked",
+        columns: [],
+        rows: [],
+        rowCount: 0,
+        durationMs: 0,
+        masked: false,
+        archeryMessage: reason,
+        archeryQueryLogId: null,
+    };
+}
+function metadataPolicyBlock(config, scope, checkedText, checkedKind) {
+    if (config.allowedInstances.size > 0 && !config.allowedInstances.has(scope.instanceName)) {
+        return "Instance is outside the configured allowlist";
+    }
+    if (config.allowedDatabases.size > 0 && !config.allowedDatabases.has(scope.dbName)) {
+        return "Database is outside the configured allowlist";
+    }
+    const lowerText = checkedText.toLowerCase();
+    for (const blockedTable of config.blockedTables) {
+        if (blockedTable.trim().length > 0 && lowerText.includes(blockedTable.toLowerCase())) {
+            return checkedKind === "table"
+                ? "Table is blocked by policy"
+                : "Search references a blocked table";
+        }
+    }
+    return null;
+}
+//# sourceMappingURL=services.js.map

package/dist/services.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"services.js","sourceRoot":"","sources":["../src/services.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AACxC,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAA;AAOlD,OAAO,EAAwB,eAAe,EAAE,YAAY,EAAE,MAAM,YAAY,CAAA;AAChF,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAA;AAiC/C,MAAM,UAAU,oBAAoB,CAAC,OAIpC;IACC,OAAO;QACL,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,aAAa,EAAE,OAAO,CAAC,aAAa;QACpC,eAAe,EAAE,IAAI,kBAAkB,EAAE;QACzC,MAAM,EAAE,IAAI,eAAe,EAAE;KAC9B,CAAA;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAAuB,EACvB,KAAwB;IAExB,MAAM,WAAW,GAAG,YAAY,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC,CAAA;IAC5E,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC;QACpB,OAAO;YACL,SAAS,EAAE,IAAI;YACf,aAAa,EAAE,KAAK,CAAC,GAAG;YACxB,SAAS,EAAE,WAAW,CAAC,IAAI;YAC3B,oBAAoB,EAAE,KAAK;YAC3B,cAAc,EAAE,EAAE;YAClB,mBAAmB,EAAE,WAAW,CAAC,OAAO;SACzC,CAAA;IACH,CAAC;IACD,MAAM,QAAQ,GAAG,WAAW,CAC1B,OAAO,CAAC,MAAM,EACd,WAAW,CAAC,KAAK,CAAC,YAAY,EAC9B,WAAW,CAAC,KAAK,CAAC,MAAM,EACxB,KAAK,CAAC,GAAG,EACT,KAAK,CAAC,QAAQ,CACf,CAAA;IACD,MAAM,IAAI,GAAG,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAA;IAC5C,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;QACtB,OAAO;YACL,SAAS,EAAE,IAAI;YACf,aAAa,EAAE,QAAQ,CAAC,aAAa;YACrC,SAAS,EAAE,SAAS;YACpB,oBAAoB,EAAE,KAAK;YAC3B,cAAc,EAAE,QAAQ,CAAC,QAAQ;YACjC,mBAAmB,EAAE,6BAA6B;SACnD,CAAA;IACH,CAAC;IACD,MAAM,SAAS,GAAG,QAAQ,UAAU,EAAE,EAAE,CAAA;IACxC,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAA;IACjE,MAAM,aAAa,GAAkB;QACnC,EAAE,EAAE,SAAS;QACb,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,GAAG,WAAW,CAAC,KAAK;QACpB,aAAa,EAAE,QAAQ,CAAC,aAAa;QACrC,OAAO,EAAE,IAAI;QACb,SAAS;QACT,SAAS,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QAChD,IAAI,EAAE,KAAK;KACZ,CAAA;IACD,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;IAC3C,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,WAAW,CAAC,KAAK,CAAC,CAAA;IAC3D,MAAM,oBAAoB,GAAG,OAAO,CAAC,MAAM,CAAC,mBAAmB,IAAI,SAAS,KAAK,KAAK,CAAA;IACtF,MAAM,WAAW,GAAG,mBAAmB,IAAI,QAAQ,WAAW,CAAC,KAAK,CAAC,SAAS,IAAI,WAAW,CAAC,KAAK,CAAC,YAAY,IAAI,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,CAAA;IAC9I,OAAO;QACL,SAAS;QACT,aAAa,EAAE,QAAQ,CAAC,aAAa;QACrC,SAAS;QACT,oBAAoB;QACpB,cAAc,EAAE,QAAQ,CAAC,QAAQ;QACjC,mBAAmB,EAAE,oBAAoB;YACvC,CAAC,CAAC,GAAG,WAAW,0CAA0C,SAAS,IAAI,IAAI,EAAE;YAC7E,CAAC,CAAC,GAAG,WAAW,qEAAqE;KACxF,CAAA;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAAuB,EACvB,KAAwB;IAExB,MAAM,aAAa,GAAG,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAA;IACnE,IAAI,aAAa,KAAK,IAAI,EAAE,CAAC;QAC3B,OAAO,gBAAgB,CAAC,oBAAoB,CAAC,CAAA;IAC/C,CAAC;IACD,IAAI,aAAa,CAAC,IAAI,EAAE,CAAC;QACvB,OAAO,gBAAgB,CAAC,6BAA6B,CAAC,CAAA;IACxD,CAAC;IACD,IAAI,aAAa,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;QACnD,OAAO,gBAAgB,CAAC,wBAAwB,CAAC,CAAA;IACnD,CAAC;IACD,IAAI,oBAAoB,CAAC,OAAO,CAAC,MAAM,EAAE,aAAa,CAAC,EAAE,CAAC;QACxD,MAAM,QAAQ,GAAG,kBAAkB,CAAC,OAAO,CAAC,WAAW,EAAE,aAAa,EAAE,KAAK,CAAC,iBAAiB,CAAC,CAAA;QAChG,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,OAAO,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;QAC1C,CAAC;IACH,CAAC;IACD,MAAM,gBAAgB,GACpB,KAAK,CAAC,OAAO,IAAI,CAAC;QAChB,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,YAAY;QAC7B,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;IACtD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,YAAY,CACrD,aAAa,EACb,aAAa,CAAC,aAAa,EAC3B,gBAAgB,CACjB,CAAA;IACD,OAAO,CAAC,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,CAAA;IACjD,OAAO;QACL,WAAW,EAAE,QAAQ,UAAU,EAAE,EAAE;QACnC,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,cAAc,EAAE,MAAM,CAAC,OAAO;QAC9B,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;KAC5C,CAAA;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,OAAuB,EACvB,KAAyB;IAEzB,MAAM,WAAW,GAAG,YAAY,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC,CAAA;IAC5E,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC;QACpB,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,WAAW,CAAC,IAAI,KAAK,WAAW,CAAC,OAAO,EAAE,EAAE,CAAA;IAC/E,CAAC;IACD,MAAM,WAAW,GAAG,mBAAmB,CACrC,OAAO,CAAC,MAAM,EACd,WAAW,CAAC,KAAK,EACjB,KAAK,CAAC,SAAS,EACf,OAAO,CACR,CAAA;IACD,IAAI,WAAW,KAAK,IAAI,EAAE,CAAC;QACzB,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,YAAY,WAAW,EAAE,EAAE,CAAA;IAC3D,CAAC;IACD,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,aAAa,CACxD,WAAW,CAAC,KAAK,EACjB,KAAK,CAAC,UAAU,EAChB,KAAK,CAAC,SAAS,CAChB,CAAA;IACD,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,WAAW,CAAC,KAAK,CAAC,CAAA;IAC3D,OAAO,QAAQ,CAAA;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAAuB,EACvB,KAAwB;IAExB,MAAM,WAAW,GAAG,YAAY,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC,CAAA;IAC5E,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC;QACpB,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,CAAC,IAAI,EAAE,CAAA;IAC9E,CAAC;IACD,MAAM,WAAW,GAAG,mBAAmB,CAAC,OAAO,CAAC,MAAM,EAAE,WAAW,CAAC,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;IAClG,IAAI,WAAW,KAAK,IAAI,EAAE,CAAC;QACzB,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,CAAA;IACvE,CAAC;IACD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAA;IAChE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,YAAY,CAAC,WAAW,CAAC,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;IAC/F,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,WAAW,CAAC,KAAK,CAAC,CAAA;IAC3D,OAAO,EAAE,GAAG,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,CAAA;AACpC,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAoB,EAAE,KAAoB;IACtE,OAAO,MAAM,CAAC,mBAAmB,IAAI,KAAK,CAAC,SAAS,KAAK,KAAK,CAAA;AAChE,CAAC;AAED,SAAS,kBAAkB,CACzB,WAAmB,EACnB,KAAoB,EACpB,KAAoB;IAEpB,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAChD,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,6BAA6B,EAAE,CAAA;IAC7D,CAAC;IACD,IAAI,KAAK,CAAC,WAAW,KAAK,WAAW,EAAE,CAAC;QACtC,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,6BAA6B,EAAE,CAAA;IAC7D,CAAC;IACD,MAAM,aAAa,GAAG,WAAW,KAAK,CAAC,EAAE,IAAI,KAAK,CAAC,OAAO,EAAE,CAAA;IAC5D,OAAO,KAAK,CAAC,IAAI,EAAE,KAAK,aAAa;QACnC,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE;QACd,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,6BAA6B,EAAE,CAAA;AAC1D,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAc;IACtC,OAAO;QACL,WAAW,EAAE,QAAQ,UAAU,EAAE,EAAE;QACnC,MAAM,EAAE,SAAS;QACjB,OAAO,EAAE,EAAE;QACX,IAAI,EAAE,EAAE;QACR,QAAQ,EAAE,CAAC;QACX,UAAU,EAAE,CAAC;QACb,MAAM,EAAE,KAAK;QACb,cAAc,EAAE,MAAM;QACtB,iBAAiB,EAAE,IAAI;KACxB,CAAA;AACH,CAAC;AAED,SAAS,mBAAmB,CAC1B,MAAoB,EACpB,KAAiB,EACjB,WAAmB,EACnB,WAA+B;IAE/B,IAAI,MAAM,CAAC,gBAAgB,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,CAAC;QACzF,OAAO,8CAA8C,CAAA;IACvD,CAAC;IACD,IAAI,MAAM,CAAC,gBAAgB,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;QACnF,OAAO,8CAA8C,CAAA;IACvD,CAAC;IACD,MAAM,SAAS,GAAG,WAAW,CAAC,WAAW,EAAE,CAAA;IAC3C,KAAK,MAAM,YAAY,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;QAChD,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;YACrF,OAAO,WAAW,KAAK,OAAO;gBAC5B,CAAC,CAAC,4BAA4B;gBAC9B,CAAC,CAAC,mCAAmC,CAAA;QACzC,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAA;AACb,CAAC"}

package/dist/store.d.ts

@@ -0,0 +1,8 @@
+import type { PreparedQuery } from "./types.js";
+export declare class PreparedQueryStore {
+    #private;
+    save(query: PreparedQuery): PreparedQuery;
+    find(prepareId: string): PreparedQuery | null;
+    markUsed(prepareId: string): void;
+}
+//# sourceMappingURL=store.d.ts.map

package/dist/store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"store.d.ts","sourceRoot":"","sources":["../src/store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAA;AAE/C,qBAAa,kBAAkB;;IAG7B,IAAI,CAAC,KAAK,EAAE,aAAa,GAAG,aAAa;IAKzC,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,aAAa,GAAG,IAAI;IAI7C,QAAQ,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;CAMlC"}

package/dist/store.js

@@ -0,0 +1,17 @@
+export class PreparedQueryStore {
+    #queries = new Map();
+    save(query) {
+        this.#queries.set(query.id, query);
+        return query;
+    }
+    find(prepareId) {
+        return this.#queries.get(prepareId) ?? null;
+    }
+    markUsed(prepareId) {
+        const query = this.#queries.get(prepareId);
+        if (query !== undefined) {
+            this.#queries.set(prepareId, { ...query, used: true });
+        }
+    }
+}
+//# sourceMappingURL=store.js.map

package/dist/store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"store.js","sourceRoot":"","sources":["../src/store.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,kBAAkB;IACpB,QAAQ,GAAG,IAAI,GAAG,EAAyB,CAAA;IAEpD,IAAI,CAAC,KAAoB;QACvB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAA;QAClC,OAAO,KAAK,CAAA;IACd,CAAC;IAED,IAAI,CAAC,SAAiB;QACpB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,IAAI,CAAA;IAC7C,CAAC;IAED,QAAQ,CAAC,SAAiB;QACxB,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;QAC1C,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,GAAG,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;QACxD,CAAC;IACH,CAAC;CACF"}

package/dist/tools.d.ts

@@ -0,0 +1,65 @@
+import { z } from "zod";
+import { type ServiceContext } from "./services.js";
+import type { JsonValue } from "./types.js";
+export declare const PrepareQueryShape: {
+    bugContext: z.ZodString;
+    cloudName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    instanceName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    dbName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    sql: z.ZodString;
+    limitNum: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+};
+export declare const PrepareQuerySchema: z.ZodObject<{
+    bugContext: z.ZodString;
+    cloudName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    instanceName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    dbName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    sql: z.ZodString;
+    limitNum: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+}, z.core.$strip>;
+export declare const ExecuteQueryShape: {
+    prepareId: z.ZodString;
+    confirmationToken: z.ZodDefault<z.ZodNullable<z.ZodOptional<z.ZodString>>>;
+    maxRows: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+};
+export declare const ExecuteQuerySchema: z.ZodObject<{
+    prepareId: z.ZodString;
+    confirmationToken: z.ZodDefault<z.ZodNullable<z.ZodOptional<z.ZodString>>>;
+    maxRows: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+}, z.core.$strip>;
+export declare const DescribeTableShape: {
+    cloudName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    instanceName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    dbName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    schemaName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    tableName: z.ZodString;
+};
+export declare const DescribeTableSchema: z.ZodObject<{
+    cloudName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    instanceName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    dbName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    schemaName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    tableName: z.ZodString;
+}, z.core.$strip>;
+export declare const QueryHistoryShape: {
+    search: z.ZodString;
+    cloudName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    instanceName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    dbName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+};
+export declare const QueryHistorySchema: z.ZodObject<{
+    search: z.ZodString;
+    cloudName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    instanceName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    dbName: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+}, z.core.$strip>;
+export declare function callTool(context: ServiceContext, name: string, args: unknown): Promise<JsonValue>;
+export declare function toolDefinitions(): readonly JsonValue[];
+export declare class UnknownToolError extends Error {
+    readonly toolName: string;
+    readonly name = "UnknownToolError";
+    constructor(toolName: string);
+}
+//# sourceMappingURL=tools.d.ts.map

package/dist/tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../src/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAKL,KAAK,cAAc,EACpB,MAAM,eAAe,CAAA;AACtB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAA;AAK3C,eAAO,MAAM,iBAAiB;;;;;;;CAO7B,CAAA;AAED,eAAO,MAAM,kBAAkB;;;;;;;iBAA8B,CAAA;AAE7D,eAAO,MAAM,iBAAiB;;;;CAI7B,CAAA;AAED,eAAO,MAAM,kBAAkB;;;;iBAA8B,CAAA;AAE7D,eAAO,MAAM,kBAAkB;;;;;;CAM9B,CAAA;AAED,eAAO,MAAM,mBAAmB;;;;;;iBAA+B,CAAA;AAE/D,eAAO,MAAM,iBAAiB;;;;;;CAM7B,CAAA;AAED,eAAO,MAAM,kBAAkB;;;;;;iBAA8B,CAAA;AAE7D,wBAAsB,QAAQ,CAC5B,OAAO,EAAE,cAAc,EACvB,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,OAAO,GACZ,OAAO,CAAC,SAAS,CAAC,CAapB;AAED,wBAAgB,eAAe,IAAI,SAAS,SAAS,EAAE,CAmCtD;AAED,qBAAa,gBAAiB,SAAQ,KAAK;IAE7B,QAAQ,CAAC,QAAQ,EAAE,MAAM;IADrC,QAAQ,CAAC,IAAI,sBAAqB;gBACb,QAAQ,EAAE,MAAM;CAGtC"}

package/dist/tools.js

@@ -0,0 +1,98 @@
+import { z } from "zod";
+import { describeTable, executeQuery, prepareQuery, queryHistory, } from "./services.js";
+const OptionalString = z.string().optional().default("");
+const OptionalNumber = z.number().int().optional().default(0);
+export const PrepareQueryShape = {
+    bugContext: z.string(),
+    cloudName: OptionalString,
+    instanceName: OptionalString,
+    dbName: OptionalString,
+    sql: z.string(),
+    limitNum: OptionalNumber,
+};
+export const PrepareQuerySchema = z.object(PrepareQueryShape);
+export const ExecuteQueryShape = {
+    prepareId: z.string(),
+    confirmationToken: z.string().optional().nullable().default(null),
+    maxRows: z.number().int().optional().default(100),
+};
+export const ExecuteQuerySchema = z.object(ExecuteQueryShape);
+export const DescribeTableShape = {
+    cloudName: OptionalString,
+    instanceName: OptionalString,
+    dbName: OptionalString,
+    schemaName: OptionalString,
+    tableName: z.string(),
+};
+export const DescribeTableSchema = z.object(DescribeTableShape);
+export const QueryHistoryShape = {
+    search: z.string(),
+    cloudName: OptionalString,
+    instanceName: OptionalString,
+    dbName: OptionalString,
+    limit: z.number().int().optional().default(20),
+};
+export const QueryHistorySchema = z.object(QueryHistoryShape);
+export async function callTool(context, name, args) {
+    switch (name) {
+        case "archery_prepare_query":
+            return prepareQuery(context, PrepareQuerySchema.parse(args));
+        case "archery_execute_query":
+            return executeQuery(context, ExecuteQuerySchema.parse(args));
+        case "archery_describe_table":
+            return describeTable(context, DescribeTableSchema.parse(args));
+        case "archery_query_history":
+            return queryHistory(context, QueryHistorySchema.parse(args));
+        default:
+            throw new UnknownToolError(name);
+    }
+}
+export function toolDefinitions() {
+    return [
+        tool("archery_prepare_query", "Validate and prepare a read-only SQL query for Archery execution", ["bugContext", "sql"], {
+            bugContext: stringSchema("Bug context or reason for the query"),
+            cloudName: stringSchema("Archery cloud name: hangzhou, beijing, or huawei"),
+            instanceName: stringSchema("Archery instance name"),
+            dbName: stringSchema("Database name"),
+            sql: stringSchema("Candidate read-only SQL"),
+            limitNum: integerSchema("Requested row limit"),
+        }),
+        tool("archery_execute_query", "Execute a prepared query through Archery", ["prepareId"], {
+            prepareId: stringSchema("Prepared query ID"),
+            confirmationToken: stringSchema("Human confirmation token"),
+            maxRows: integerSchema("Maximum rows"),
+        }),
+        tool("archery_describe_table", "Read table metadata through Archery", ["tableName"], {
+            cloudName: stringSchema("Archery cloud name"),
+            instanceName: stringSchema("Archery instance name"),
+            dbName: stringSchema("Database name"),
+            schemaName: stringSchema("Schema name"),
+            tableName: stringSchema("Table name"),
+        }),
+        tool("archery_query_history", "Search prior query history summaries", ["search"], {
+            search: stringSchema("Search text or SQL fragment"),
+            cloudName: stringSchema("Archery cloud name"),
+            instanceName: stringSchema("Archery instance name"),
+            dbName: stringSchema("Database name"),
+            limit: integerSchema("Maximum matches"),
+        }),
+    ];
+}
+export class UnknownToolError extends Error {
+    toolName;
+    name = "UnknownToolError";
+    constructor(toolName) {
+        super(`Unknown tool: ${toolName}`);
+        this.toolName = toolName;
+    }
+}
+function tool(name, description, required, properties) {
+    return { name, description, inputSchema: { type: "object", properties, required } };
+}
+function stringSchema(description) {
+    return { type: "string", description };
+}
+function integerSchema(description) {
+    return { type: "integer", description };
+}
+//# sourceMappingURL=tools.js.map

package/dist/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../src/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EACL,aAAa,EACb,YAAY,EACZ,YAAY,EACZ,YAAY,GAEb,MAAM,eAAe,CAAA;AAGtB,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;AACxD,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAA;AAE7D,MAAM,CAAC,MAAM,iBAAiB,GAAG;IAC/B,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;IACtB,SAAS,EAAE,cAAc;IACzB,YAAY,EAAE,cAAc;IAC5B,MAAM,EAAE,cAAc;IACtB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;IACf,QAAQ,EAAE,cAAc;CACzB,CAAA;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAA;AAE7D,MAAM,CAAC,MAAM,iBAAiB,GAAG;IAC/B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,iBAAiB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;IACjE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC;CAClD,CAAA;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAA;AAE7D,MAAM,CAAC,MAAM,kBAAkB,GAAG;IAChC,SAAS,EAAE,cAAc;IACzB,YAAY,EAAE,cAAc;IAC5B,MAAM,EAAE,cAAc;IACtB,UAAU,EAAE,cAAc;IAC1B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAA;AAED,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAA;AAE/D,MAAM,CAAC,MAAM,iBAAiB,GAAG;IAC/B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;IAClB,SAAS,EAAE,cAAc;IACzB,YAAY,EAAE,cAAc;IAC5B,MAAM,EAAE,cAAc;IACtB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;CAC/C,CAAA;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAA;AAE7D,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,OAAuB,EACvB,IAAY,EACZ,IAAa;IAEb,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,uBAAuB;YAC1B,OAAO,YAAY,CAAC,OAAO,EAAE,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAA;QAC9D,KAAK,uBAAuB;YAC1B,OAAO,YAAY,CAAC,OAAO,EAAE,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAA;QAC9D,KAAK,wBAAwB;YAC3B,OAAO,aAAa,CAAC,OAAO,EAAE,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAA;QAChE,KAAK,uBAAuB;YAC1B,OAAO,YAAY,CAAC,OAAO,EAAE,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAA;QAC9D;YACE,MAAM,IAAI,gBAAgB,CAAC,IAAI,CAAC,CAAA;IACpC,CAAC;AACH,CAAC;AAED,MAAM,UAAU,eAAe;IAC7B,OAAO;QACL,IAAI,CACF,uBAAuB,EACvB,kEAAkE,EAClE,CAAC,YAAY,EAAE,KAAK,CAAC,EACrB;YACE,UAAU,EAAE,YAAY,CAAC,qCAAqC,CAAC;YAC/D,SAAS,EAAE,YAAY,CAAC,kDAAkD,CAAC;YAC3E,YAAY,EAAE,YAAY,CAAC,uBAAuB,CAAC;YACnD,MAAM,EAAE,YAAY,CAAC,eAAe,CAAC;YACrC,GAAG,EAAE,YAAY,CAAC,yBAAyB,CAAC;YAC5C,QAAQ,EAAE,aAAa,CAAC,qBAAqB,CAAC;SAC/C,CACF;QACD,IAAI,CAAC,uBAAuB,EAAE,0CAA0C,EAAE,CAAC,WAAW,CAAC,EAAE;YACvF,SAAS,EAAE,YAAY,CAAC,mBAAmB,CAAC;YAC5C,iBAAiB,EAAE,YAAY,CAAC,0BAA0B,CAAC;YAC3D,OAAO,EAAE,aAAa,CAAC,cAAc,CAAC;SACvC,CAAC;QACF,IAAI,CAAC,wBAAwB,EAAE,qCAAqC,EAAE,CAAC,WAAW,CAAC,EAAE;YACnF,SAAS,EAAE,YAAY,CAAC,oBAAoB,CAAC;YAC7C,YAAY,EAAE,YAAY,CAAC,uBAAuB,CAAC;YACnD,MAAM,EAAE,YAAY,CAAC,eAAe,CAAC;YACrC,UAAU,EAAE,YAAY,CAAC,aAAa,CAAC;YACvC,SAAS,EAAE,YAAY,CAAC,YAAY,CAAC;SACtC,CAAC;QACF,IAAI,CAAC,uBAAuB,EAAE,sCAAsC,EAAE,CAAC,QAAQ,CAAC,EAAE;YAChF,MAAM,EAAE,YAAY,CAAC,6BAA6B,CAAC;YACnD,SAAS,EAAE,YAAY,CAAC,oBAAoB,CAAC;YAC7C,YAAY,EAAE,YAAY,CAAC,uBAAuB,CAAC;YACnD,MAAM,EAAE,YAAY,CAAC,eAAe,CAAC;YACrC,KAAK,EAAE,aAAa,CAAC,iBAAiB,CAAC;SACxC,CAAC;KACH,CAAA;AACH,CAAC;AAED,MAAM,OAAO,gBAAiB,SAAQ,KAAK;IAEpB;IADZ,IAAI,GAAG,kBAAkB,CAAA;IAClC,YAAqB,QAAgB;QACnC,KAAK,CAAC,iBAAiB,QAAQ,EAAE,CAAC,CAAA;QADf,aAAQ,GAAR,QAAQ,CAAQ;IAErC,CAAC;CACF;AAED,SAAS,IAAI,CACX,IAAY,EACZ,WAAmB,EACnB,QAA2B,EAC3B,UAAqB;IAErB,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,CAAA;AACrF,CAAC;AAED,SAAS,YAAY,CAAC,WAAmB;IACvC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAA;AACxC,CAAC;AAED,SAAS,aAAa,CAAC,WAAmB;IACxC,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,CAAA;AACzC,CAAC"}

package/dist/types.d.ts

@@ -0,0 +1,78 @@
+export type JsonPrimitive = string | number | boolean | null;
+export type JsonValue = JsonPrimitive | readonly JsonValue[] | {
+    readonly [key: string]: JsonValue;
+};
+export type JsonObject = {
+    readonly [key: string]: JsonValue;
+};
+export type PolicyFinding = {
+    readonly code: string;
+    readonly message: string;
+};
+export type PolicyDecision = {
+    readonly allowed: boolean;
+    readonly normalizedSql: string;
+    readonly effectiveLimit: number;
+    readonly riskLevel: "low" | "medium" | "blocked";
+    readonly findings: readonly PolicyFinding[];
+};
+export type QueryScope = {
+    readonly cloudName: CloudName;
+    readonly instanceName: string;
+    readonly dbName: string;
+};
+export type CloudName = "hangzhou" | "beijing" | "huawei";
+export type PolicyConfig = {
+    readonly defaultLimit: number;
+    readonly maxLimit: number;
+    readonly requireConfirmation: boolean;
+    readonly allowedInstances: ReadonlySet<string>;
+    readonly allowedDatabases: ReadonlySet<string>;
+    readonly blockedTables: ReadonlySet<string>;
+};
+export type PreparedQuery = QueryScope & {
+    readonly id: string;
+    readonly requestUser: string;
+    readonly normalizedSql: string;
+    readonly sqlHash: string;
+    readonly riskLevel: "low" | "medium";
+    readonly expiresAt: Date;
+    readonly used: boolean;
+};
+export type QueryResult = {
+    readonly status: string;
+    readonly columns: readonly string[];
+    readonly rows: readonly JsonObject[];
+    readonly rowCount: number;
+    readonly durationMs: number;
+    readonly masked: boolean;
+    readonly message: string;
+    readonly archeryQueryLogId: string | null;
+};
+export type TableColumn = {
+    readonly name: string;
+    readonly type: string;
+    readonly sensitive: boolean;
+};
+export type TableMetadata = {
+    readonly columns: readonly TableColumn[];
+    readonly source: string;
+};
+export type HistoryMatch = {
+    readonly id: string;
+    readonly sqlPreview: string;
+    readonly instanceName: string;
+    readonly dbName: string;
+    readonly createdAt: string;
+};
+export type HistoryResult = {
+    readonly matches: readonly HistoryMatch[];
+    readonly total: number;
+    readonly truncated: boolean;
+};
+export interface ArcheryClient {
+    executeQuery(scope: QueryScope, sql: string, maxRows: number): Promise<QueryResult>;
+    describeTable(scope: QueryScope, schemaName: string, tableName: string): Promise<TableMetadata>;
+    queryHistory(scope: QueryScope, search: string, limit: number): Promise<HistoryResult>;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,aAAa,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,IAAI,CAAA;AAC5D,MAAM,MAAM,SAAS,GAAG,aAAa,GAAG,SAAS,SAAS,EAAE,GAAG;IAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,GAAG,SAAS,CAAA;CAAE,CAAA;AACpG,MAAM,MAAM,UAAU,GAAG;IAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,GAAG,SAAS,CAAA;CAAE,CAAA;AAE9D,MAAM,MAAM,aAAa,GAAG;IAC1B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAA;IACrB,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAA;CACzB,CAAA;AAED,MAAM,MAAM,cAAc,GAAG;IAC3B,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAA;IACzB,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAA;IAC9B,QAAQ,CAAC,cAAc,EAAE,MAAM,CAAA;IAC/B,QAAQ,CAAC,SAAS,EAAE,KAAK,GAAG,QAAQ,GAAG,SAAS,CAAA;IAChD,QAAQ,CAAC,QAAQ,EAAE,SAAS,aAAa,EAAE,CAAA;CAC5C,CAAA;AAED,MAAM,MAAM,UAAU,GAAG;IACvB,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAA;IAC7B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAA;IAC7B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;CACxB,CAAA;AAED,MAAM,MAAM,SAAS,GAAG,UAAU,GAAG,SAAS,GAAG,QAAQ,CAAA;AAEzD,MAAM,MAAM,YAAY,GAAG;IACzB,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAA;IAC7B,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAA;IACzB,QAAQ,CAAC,mBAAmB,EAAE,OAAO,CAAA;IACrC,QAAQ,CAAC,gBAAgB,EAAE,WAAW,CAAC,MAAM,CAAC,CAAA;IAC9C,QAAQ,CAAC,gBAAgB,EAAE,WAAW,CAAC,MAAM,CAAC,CAAA;IAC9C,QAAQ,CAAC,aAAa,EAAE,WAAW,CAAC,MAAM,CAAC,CAAA;CAC5C,CAAA;AAED,MAAM,MAAM,aAAa,GAAG,UAAU,GAAG;IACvC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAA;IACnB,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAA;IAC5B,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAA;IAC9B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAA;IACxB,QAAQ,CAAC,SAAS,EAAE,KAAK,GAAG,QAAQ,CAAA;IACpC,QAAQ,CAAC,SAAS,EAAE,IAAI,CAAA;IACxB,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAA;CACvB,CAAA;AAED,MAAM,MAAM,WAAW,GAAG;IACxB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,OAAO,EAAE,SAAS,MAAM,EAAE,CAAA;IACnC,QAAQ,CAAC,IAAI,EAAE,SAAS,UAAU,EAAE,CAAA;IACpC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAA;IACzB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAA;IAC3B,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAA;IACxB,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAA;IACxB,QAAQ,CAAC,iBAAiB,EAAE,MAAM,GAAG,IAAI,CAAA;CAC1C,CAAA;AAED,MAAM,MAAM,WAAW,GAAG;IACxB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAA;IACrB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAA;IACrB,QAAQ,CAAC,SAAS,EAAE,OAAO,CAAA;CAC5B,CAAA;AAED,MAAM,MAAM,aAAa,GAAG;IAC1B,QAAQ,CAAC,OAAO,EAAE,SAAS,WAAW,EAAE,CAAA;IACxC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;CACxB,CAAA;AAED,MAAM,MAAM,YAAY,GAAG;IACzB,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAA;IACnB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAA;IAC3B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAA;IAC7B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAA;CAC3B,CAAA;AAED,MAAM,MAAM,aAAa,GAAG;IAC1B,QAAQ,CAAC,OAAO,EAAE,SAAS,YAAY,EAAE,CAAA;IACzC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAA;IACtB,QAAQ,CAAC,SAAS,EAAE,OAAO,CAAA;CAC5B,CAAA;AAED,MAAM,WAAW,aAAa;IAC5B,YAAY,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAAA;IACnF,aAAa,CAAC,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAA;IAC/F,YAAY,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAA;CACvF"}

package/dist/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

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

package/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "@dezhouzhang/archery-query-mcp",
+  "version": "0.1.0",
+  "description": "MCP server for querying Archery safely via stdio or HTTP",
+  "type": "module",
+  "bin": {
+    "archery-query-mcp": "dist/cli.js"
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "typecheck": "tsc -p tsconfig.json --noEmit",
+    "test": "vitest run",
+    "start:stdio": "node dist/cli.js --transport stdio",
+    "start:http": "node dist/cli.js --transport http --port 8080"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.29.0",
+    "@hono/node-server": "^1.19.8",
+    "commander": "^14.0.2",
+    "hono": "^4.10.7",
+    "ky": "^1.14.0",
+    "zod": "^4.2.1"
+  },
+  "devDependencies": {
+    "@biomejs/biome": "^2.3.8",
+    "@types/node": "^25.0.3",
+    "typescript": "^5.9.3",
+    "vitest": "^4.0.16"
+  },
+  "engines": {
+    "node": ">=20.0.0"
+  }
+}