@tool-bridge/server 0.1.0 → 0.3.0

package/dist/mother_duck-CPGOFEXE.js

@@ -0,0 +1,322 @@
+import {
+  createProviderPlugin,
+  guardedFetch,
+  requireApiKey,
+  upstreamError
+} from "./chunk-CSJEZHQO.js";
+import "./chunk-K7FVSFUX.js";
+import {
+  TBError,
+  external_exports
+} from "./chunk-JLTYA2B3.js";
+import "./chunk-MLKGABMK.js";
+
+// ../plugins/src/mother_duck/api.ts
+var SERVICE = "mother_duck";
+var API_BASE = "https://api.motherduck.com";
+function text(value) {
+  if (typeof value !== "string") return void 0;
+  const trimmed = value.trim();
+  return trimmed === "" ? void 0 : trimmed;
+}
+function record(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value) ? value : void 0;
+}
+function compact(input) {
+  return Object.fromEntries(Object.entries(input).filter(([, value]) => value !== void 0));
+}
+function errorMessage(payload) {
+  const body = record(payload);
+  if (body === void 0) return void 0;
+  return text(body.message) ?? text(body.error) ?? text(body.code);
+}
+async function readPayload(response) {
+  const body = await response.text().catch(() => "");
+  if (body.trim() === "") return {};
+  try {
+    return JSON.parse(body);
+  } catch {
+    return { message: body };
+  }
+}
+function isEmpty(payload) {
+  const object = record(payload);
+  return object !== void 0 && Object.keys(object).length === 0;
+}
+async function request(ctx, path, input = {}) {
+  const apiKey = requireApiKey(ctx, SERVICE);
+  const headers = {
+    accept: "application/json",
+    authorization: `Bearer ${apiKey}`
+  };
+  if (input.body !== void 0) headers["content-type"] = "application/json";
+  let response;
+  try {
+    response = await guardedFetch(new URL(path, API_BASE).toString(), {
+      method: input.method ?? "GET",
+      headers,
+      ...input.body === void 0 ? {} : { body: JSON.stringify(input.body) }
+    });
+  } catch (error) {
+    throw upstreamError(
+      502,
+      error instanceof Error ? `MotherDuck \u8BF7\u6C42\u5931\u8D25: ${error.message}` : "MotherDuck \u8BF7\u6C42\u5931\u8D25"
+    );
+  }
+  const payload = await readPayload(response);
+  if (!response.ok) {
+    throw upstreamError(response.status, errorMessage(payload) ?? `MotherDuck \u8BF7\u6C42\u5931\u8D25,HTTP ${response.status}`);
+  }
+  if (input.allowEmpty === true && isEmpty(payload)) return {};
+  return payload;
+}
+function requireArrayProperty(payload, key) {
+  const value = record(payload)?.[key];
+  if (!Array.isArray(value)) {
+    throw new TBError("unavailable", `MotherDuck \u7684 ${key} \u54CD\u5E94\u975E\u6CD5`, { retryable: true });
+  }
+  return value;
+}
+function normalizeToken(value) {
+  const token = record(value);
+  if (token === void 0) {
+    throw new TBError("unavailable", "MotherDuck \u8FD4\u56DE\u7684 token \u4E0D\u662F\u5BF9\u8C61", { retryable: true });
+  }
+  return compact({
+    id: text(token.id),
+    name: text(token.name),
+    token: text(token.token),
+    expire_at: text(token.expire_at),
+    created_ts: text(token.created_ts),
+    read_only: typeof token.read_only === "boolean" ? token.read_only : void 0,
+    token_type: text(token.token_type),
+    raw: token
+  });
+}
+function responseUsername(payload, fallback) {
+  return text(record(payload)?.username) ?? fallback;
+}
+async function listActiveAccounts(_input, ctx) {
+  const payload = await request(ctx, "/v1/active_accounts");
+  return { accounts: requireArrayProperty(payload, "accounts") };
+}
+async function createUser(input, ctx) {
+  const username = input.username.trim();
+  const payload = await request(ctx, "/v1/users", { method: "POST", body: { username } });
+  return { username: responseUsername(payload, username) };
+}
+async function deleteUser(input, ctx) {
+  const username = input.username.trim();
+  const payload = await request(ctx, `/v1/users/${encodeURIComponent(username)}`, { method: "DELETE" });
+  return { username: responseUsername(payload, username) };
+}
+async function listTokens(input, ctx) {
+  const path = `/v1/users/${encodeURIComponent(input.username.trim())}/tokens`;
+  return { tokens: requireArrayProperty(await request(ctx, path), "tokens").map(normalizeToken) };
+}
+async function createToken(input, ctx) {
+  const path = `/v1/users/${encodeURIComponent(input.username.trim())}/tokens`;
+  const payload = await request(ctx, path, {
+    method: "POST",
+    body: compact({ name: input.name.trim(), ttl: input.ttl, token_type: input.token_type })
+  });
+  return { token: normalizeToken(payload) };
+}
+async function deleteToken(input, ctx) {
+  const username = encodeURIComponent(input.username.trim());
+  const tokenId = encodeURIComponent(input.token_id.trim());
+  await request(ctx, `/v1/users/${username}/tokens/${tokenId}`, { method: "DELETE", allowEmpty: true });
+  return { success: true };
+}
+async function getUserDucklingConfig(input, ctx) {
+  const path = `/v1/users/${encodeURIComponent(input.username.trim())}/instances`;
+  return { config: await request(ctx, path) };
+}
+async function setUserDucklingConfig(input, ctx) {
+  const path = `/v1/users/${encodeURIComponent(input.username.trim())}/instances`;
+  const payload = await request(ctx, path, { method: "PUT", body: { config: input.config } });
+  return { config: payload };
+}
+
+// ../plugins/src/mother_duck/schema.ts
+var listActiveAccountsInput = external_exports.strictObject({}).describe("No input is required to list active MotherDuck accounts.");
+var listActiveAccountsOutput = external_exports.strictObject({
+  accounts: external_exports.array(external_exports.strictObject({
+    username: external_exports.string().min(1).max(255).describe("The MotherDuck username within the organization.").optional(),
+    ducklings: external_exports.array(external_exports.strictObject({
+      id: external_exports.string().describe("The Duckling identifier, such as rw or rs.N.").optional(),
+      type: external_exports.enum(["read_write", "read_scaling"]).describe("The MotherDuck Duckling type.").optional(),
+      status: external_exports.string().min(1).describe("The MotherDuck Duckling status, such as active or cooldown.").optional()
+    }).describe("A MotherDuck Duckling attached to an active account.")).describe("The active Ducklings for the account.").optional()
+  }).describe("A MotherDuck active account.")).describe("The active accounts in the organization.")
+}).describe("The active MotherDuck accounts returned by the Admin API.");
+var createUserInput = external_exports.strictObject({
+  username: external_exports.string().min(1).max(255).describe("The MotherDuck username within the organization.")
+}).describe("Input for creating a MotherDuck user.");
+var createUserOutput = external_exports.strictObject({
+  username: external_exports.string().min(1).max(255).describe("The MotherDuck username within the organization.")
+}).describe("The created MotherDuck user.");
+var deleteUserInput = external_exports.strictObject({
+  username: external_exports.string().min(1).max(255).describe("The MotherDuck username within the organization.")
+}).describe("Input for deleting a MotherDuck user.");
+var deleteUserOutput = external_exports.strictObject({
+  username: external_exports.string().min(1).max(255).describe("The MotherDuck username within the organization.")
+}).describe("The deleted MotherDuck user.");
+var listTokensInput = external_exports.strictObject({
+  username: external_exports.string().min(1).max(255).describe("The MotherDuck username within the organization.")
+}).describe("Input for listing MotherDuck user tokens.");
+var listTokensOutput = external_exports.strictObject({
+  tokens: external_exports.array(external_exports.strictObject({
+    id: external_exports.string().describe("The token UUID.").optional(),
+    name: external_exports.string().describe("The token display name.").optional(),
+    token: external_exports.string().describe("The newly-created token secret when MotherDuck returns it.").optional(),
+    expire_at: external_exports.string().describe("The timestamp when the token expires.").optional(),
+    created_ts: external_exports.string().describe("The timestamp when the token was created.").optional(),
+    read_only: external_exports.boolean().describe("Whether the token is read-only.").optional(),
+    token_type: external_exports.enum(["read_write", "read_scaling"]).describe("The MotherDuck token type.").optional(),
+    raw: external_exports.looseObject({}).describe("The raw token object returned by MotherDuck.")
+  }).describe("A MotherDuck access token.")).describe("The user's MotherDuck access tokens.")
+}).describe("The MotherDuck access tokens for the user.");
+var createTokenInput = external_exports.strictObject({
+  username: external_exports.string().min(1).max(255).describe("The MotherDuck username within the organization."),
+  name: external_exports.string().min(1).max(255).describe("The token display name."),
+  ttl: external_exports.number().min(300).max(31536e3).describe("Token expiration in seconds.").optional(),
+  token_type: external_exports.enum(["read_write", "read_scaling"]).describe("The MotherDuck token type.").optional()
+}).describe("Input for creating a MotherDuck user token.");
+var createTokenOutput = external_exports.strictObject({
+  token: external_exports.strictObject({
+    id: external_exports.string().describe("The token UUID.").optional(),
+    name: external_exports.string().describe("The token display name.").optional(),
+    token: external_exports.string().describe("The newly-created token secret when MotherDuck returns it.").optional(),
+    expire_at: external_exports.string().describe("The timestamp when the token expires.").optional(),
+    created_ts: external_exports.string().describe("The timestamp when the token was created.").optional(),
+    read_only: external_exports.boolean().describe("Whether the token is read-only.").optional(),
+    token_type: external_exports.enum(["read_write", "read_scaling"]).describe("The MotherDuck token type.").optional(),
+    raw: external_exports.looseObject({}).describe("The raw token object returned by MotherDuck.")
+  }).describe("A MotherDuck access token.")
+}).describe("The newly-created MotherDuck token.");
+var deleteTokenInput = external_exports.strictObject({
+  username: external_exports.string().min(1).max(255).describe("The MotherDuck username within the organization."),
+  token_id: external_exports.string().min(1).describe("The MotherDuck access token identifier.")
+}).describe("Input for deleting a MotherDuck user token.");
+var deleteTokenOutput = external_exports.strictObject({
+  success: external_exports.boolean().describe("Whether MotherDuck accepted the token deletion request.")
+}).describe("The normalized MotherDuck token deletion result.");
+var getUserDucklingConfigInput = external_exports.strictObject({
+  username: external_exports.string().min(1).max(255).describe("The MotherDuck username within the organization.")
+}).describe("Input for retrieving a MotherDuck user's Duckling configuration.");
+var getUserDucklingConfigOutput = external_exports.strictObject({
+  config: external_exports.strictObject({
+    read_write: external_exports.strictObject({
+      instance_size: external_exports.enum(["pulse", "standard", "jumbo", "mega", "giga"]).describe("The MotherDuck instance size."),
+      cooldown_seconds: external_exports.int().min(60).max(86400).describe("Cooldown duration in seconds.").optional()
+    }).describe("MotherDuck read-write Duckling configuration.").optional(),
+    read_scaling: external_exports.strictObject({
+      instance_size: external_exports.enum(["pulse", "standard", "jumbo", "mega", "giga"]).describe("The MotherDuck instance size."),
+      flock_size: external_exports.number().min(0).max(64).describe("The number of read-scaling Ducklings."),
+      cooldown_seconds: external_exports.int().min(60).max(86400).describe("Cooldown duration in seconds.").optional()
+    }).describe("MotherDuck read-scaling Duckling configuration.").optional()
+  }).describe("MotherDuck Duckling configuration for a user.")
+}).describe("The MotherDuck Duckling configuration for the user.");
+var setUserDucklingConfigInput = external_exports.strictObject({
+  username: external_exports.string().min(1).max(255).describe("The MotherDuck username within the organization."),
+  config: external_exports.strictObject({
+    read_write: external_exports.strictObject({
+      instance_size: external_exports.enum(["pulse", "standard", "jumbo", "mega", "giga"]).describe("The MotherDuck instance size."),
+      cooldown_seconds: external_exports.int().min(60).max(86400).describe("Cooldown duration in seconds.").optional()
+    }).describe("MotherDuck read-write Duckling configuration.").optional(),
+    read_scaling: external_exports.strictObject({
+      instance_size: external_exports.enum(["pulse", "standard", "jumbo", "mega", "giga"]).describe("The MotherDuck instance size."),
+      flock_size: external_exports.number().min(0).max(64).describe("The number of read-scaling Ducklings."),
+      cooldown_seconds: external_exports.int().min(60).max(86400).describe("Cooldown duration in seconds.").optional()
+    }).describe("MotherDuck read-scaling Duckling configuration.").optional()
+  }).describe("MotherDuck Duckling configuration for a user.")
+}).describe("Input for setting a MotherDuck user's Duckling configuration.");
+var setUserDucklingConfigOutput = external_exports.strictObject({
+  config: external_exports.strictObject({
+    read_write: external_exports.strictObject({
+      instance_size: external_exports.enum(["pulse", "standard", "jumbo", "mega", "giga"]).describe("The MotherDuck instance size."),
+      cooldown_seconds: external_exports.int().min(60).max(86400).describe("Cooldown duration in seconds.").optional()
+    }).describe("MotherDuck read-write Duckling configuration.").optional(),
+    read_scaling: external_exports.strictObject({
+      instance_size: external_exports.enum(["pulse", "standard", "jumbo", "mega", "giga"]).describe("The MotherDuck instance size."),
+      flock_size: external_exports.number().min(0).max(64).describe("The number of read-scaling Ducklings."),
+      cooldown_seconds: external_exports.int().min(60).max(86400).describe("Cooldown duration in seconds.").optional()
+    }).describe("MotherDuck read-scaling Duckling configuration.").optional()
+  }).describe("MotherDuck Duckling configuration for a user.")
+}).describe("The updated MotherDuck Duckling configuration for the user.");
+var motherDuckActions = {
+  list_active_accounts: {
+    description: "List active MotherDuck accounts and their active Ducklings in the organization.",
+    effect: "read",
+    inputSchema: listActiveAccountsInput,
+    outputSchema: external_exports.toJSONSchema(listActiveAccountsOutput, { io: "output", unrepresentable: "any" })
+  },
+  create_user: {
+    description: "Create a MotherDuck member user in the organization.",
+    effect: "write",
+    inputSchema: createUserInput,
+    outputSchema: external_exports.toJSONSchema(createUserOutput, { io: "output", unrepresentable: "any" })
+  },
+  delete_user: {
+    description: "Permanently delete a MotherDuck user and all of their data.",
+    effect: "destructive",
+    inputSchema: deleteUserInput,
+    outputSchema: external_exports.toJSONSchema(deleteUserOutput, { io: "output", unrepresentable: "any" })
+  },
+  list_tokens: {
+    description: "List MotherDuck access tokens for a user.",
+    effect: "read",
+    inputSchema: listTokensInput,
+    outputSchema: external_exports.toJSONSchema(listTokensOutput, { io: "output", unrepresentable: "any" })
+  },
+  create_token: {
+    description: "Create a MotherDuck access token for a user.",
+    effect: "write",
+    inputSchema: createTokenInput,
+    outputSchema: external_exports.toJSONSchema(createTokenOutput, { io: "output", unrepresentable: "any" })
+  },
+  delete_token: {
+    description: "Invalidate a MotherDuck access token for a user.",
+    effect: "destructive",
+    inputSchema: deleteTokenInput,
+    outputSchema: external_exports.toJSONSchema(deleteTokenOutput, { io: "output", unrepresentable: "any" })
+  },
+  get_user_duckling_config: {
+    description: "Retrieve MotherDuck Duckling configuration for a user.",
+    effect: "read",
+    inputSchema: getUserDucklingConfigInput,
+    outputSchema: external_exports.toJSONSchema(getUserDucklingConfigOutput, { io: "output", unrepresentable: "any" })
+  },
+  set_user_duckling_config: {
+    description: "Set MotherDuck Duckling configuration for a user.",
+    effect: "write",
+    inputSchema: setUserDucklingConfigInput,
+    outputSchema: external_exports.toJSONSchema(setUserDucklingConfigOutput, { io: "output", unrepresentable: "any" })
+  }
+};
+
+// ../plugins/src/mother_duck/index.ts
+function createMotherDuckPlugin() {
+  return createProviderPlugin({
+    description: "MotherDuck",
+    actions: motherDuckActions,
+    // 上游 credentialValidators 就是打 /v1/active_accounts 试凭证,这里对应到同一个 action。
+    credentialProbe: "list_active_accounts",
+    handlers: {
+      list_active_accounts: listActiveAccounts,
+      create_user: createUser,
+      delete_user: deleteUser,
+      list_tokens: listTokens,
+      create_token: createToken,
+      delete_token: deleteToken,
+      get_user_duckling_config: getUserDucklingConfig,
+      set_user_duckling_config: setUserDucklingConfig
+    }
+  });
+}
+var mother_duck_default = createMotherDuckPlugin();
+export {
+  createMotherDuckPlugin,
+  mother_duck_default as default
+};