@confluentinc/mcp-confluent 1.0.1 → 1.0.3

package/dist/cli.js

@@ -1,7 +1,70 @@
-import { Command } from "@commander-js/extra-typings";
+import { Command, CommanderError, Option } from "@commander-js/extra-typings";
+import { ToolName } from "./confluent/tools/tool-name.js";
+import { logger } from "./logger.js";
+import { TransportType } from "./mcp/transports/types.js";
 import * as dotenv from "dotenv";
 import fs from "fs";
 import path from "path";
+import { getProperties } from "properties-file";
+import pkg from "../package.json" with { type: "json" };
+/**
+ * Get the package version from package.json
+ * @returns Package version string
+ */
+export function getPackageVersion() {
+    return pkg.version;
+}
+function parseTransportList(value) {
+    // Split, trim, and filter out empty strings
+    const types = value
+        .split(",")
+        .map((type) => type.trim())
+        .filter(Boolean);
+    // Validate each transport type
+    const validTypes = new Set(Object.values(TransportType));
+    const invalidTypes = types.filter((type) => !validTypes.has(type));
+    if (invalidTypes.length > 0) {
+        throw new Error(`Invalid transport type(s): ${invalidTypes.join(", ")}. Valid options: ${Array.from(validTypes).join(", ")}`);
+    }
+    // Deduplicate using Set
+    return Array.from(new Set(types));
+}
+/**
+ * Parses a comma-separated list of tool names and returns an array of valid tool names.
+ * Trims whitespace from each tool name.
+ * Exits the process with an error if any invalid tool names are provided.
+ *
+ * @param value - Comma-separated list of tool names
+ * @returns Array of valid tool names
+ */
+function parseToolList(value) {
+    return (value
+        .split(",")
+        .map((t) => t.trim())
+        // Filter out any empty strings that may result from extra commas or whitespace
+        .filter(Boolean));
+}
+/**
+ * Reads a file and returns an array of non-empty, non-comment lines.
+ * Lines starting with '#' are treated as comments and ignored.
+ * Trims whitespace from each line.
+ * Exits the process with an error if the file does not exist.
+ *
+ * @param filePath - Path to the file to read
+ * @returns Array of valid lines from the file
+ */
+function readFileLines(filePath) {
+    const absPath = path.resolve(filePath);
+    if (!fs.existsSync(absPath)) {
+        throw new Error(`Tool list file not found: ${absPath}`);
+    }
+    const lines = fs
+        .readFileSync(absPath, "utf-8")
+        .split("\n")
+        .map((line) => line.trim())
+        .filter((line) => line && !line.startsWith("#"));
+    return lines;
+}
 /**
  * Parse command line arguments with strong typing
  * @returns Parsed CLI options
@@ -10,44 +73,154 @@ export function parseCliArgs() {
     const program = new Command()
         .name("mcp-confluent")
         .description("Confluent MCP Server - Model Context Protocol implementation for Confluent Cloud")
-        .version(process.env.npm_package_version ?? "dev")
+        .version(getPackageVersion())
         .option("-e, --env-file <path>", "Load environment variables from file")
-        .action((options) => {
-        if (options.envFile) {
-            loadEnvironmentVariables(options);
-        }
-    })
+        .option("-k, --kafka-config-file <file>", "Path to a properties file for configuring kafka clients")
+        .addOption(new Option("-t, --transport <types>", "Transport types (comma-separated list)")
+        .choices(Object.values(TransportType))
+        .default(TransportType.STDIO)
+        .argParser((value) => parseTransportList(value)))
+        .option("--allow-tools <tools>", "Comma-separated list of tool names to allow. If provided, takes precedence over --allow-tools-file. Allow-list is applied before block-list.")
+        .option("--block-tools <tools>", "Comma-separated list of tool names to block. If provided, takes precedence over --block-tools-file. Block-list is applied after allow-list.")
+        .option("--allow-tools-file <file>", "File with tool names to allow (one per line). Used only if --allow-tools is not provided. Allow-list is applied before block-list.")
+        .option("--block-tools-file <file>", "File with tool names to block (one per line). Used only if --block-tools is not provided. Block-list is applied after allow-list.")
+        .option("--list-tools", "Print the final set of enabled tool names (with descriptions) after allow/block filtering and exit. Does not start the server.")
+        .option("--disable-confluent-cloud-tools", "Disable all tools that require Confluent Cloud REST APIs (cloud-only tools).")
         .allowExcessArguments(false)
         .exitOverride();
     try {
-        // Parse arguments and get options (no need for generic type parameter with extra-typings)
-        return program.parse().opts();
+        const opts = program.parse().opts();
+        if (opts.envFile) {
+            loadEnvironmentVariables(opts.envFile);
+        }
+        // Precedence: CLI > file > undefined
+        let allowTools = undefined;
+        let blockTools = undefined;
+        if (opts.allowTools) {
+            allowTools = parseToolList(opts.allowTools);
+        }
+        else if (opts.allowToolsFile) {
+            allowTools = readFileLines(opts.allowToolsFile);
+        }
+        if (opts.blockTools) {
+            blockTools = parseToolList(opts.blockTools);
+        }
+        else if (opts.blockToolsFile) {
+            blockTools = readFileLines(opts.blockToolsFile);
+        }
+        let kafkaConfig = {};
+        if (opts.kafkaConfigFile) {
+            kafkaConfig = parsePropertiesFile(opts.kafkaConfigFile);
+        }
+        return {
+            envFile: opts.envFile,
+            transports: Array.isArray(opts.transport)
+                ? opts.transport
+                : [opts.transport],
+            allowTools,
+            blockTools,
+            listTools: !!opts.listTools,
+            disableConfluentCloudTools: !!opts.disableConfluentCloudTools,
+            kafkaConfig: kafkaConfig,
+        };
     }
-    catch {
-        // This block is reached when --help or --version is called
-        // as these will throw an error due to exitOverride()
-        process.exit(0);
+    catch (error) {
+        if (error instanceof CommanderError &&
+            (error.code === "commander.helpDisplayed" ||
+                error.code === "commander.version")) {
+            process.exit(0);
+        }
+        if (error instanceof Error) {
+            logger.error({
+                error,
+                errorString: error.toString(),
+                errorMessage: error.message,
+            }, "Error parsing CLI options");
+        }
+        else {
+            logger.error({ error: String(error) }, "Error parsing CLI options");
+        }
+        process.exit(1);
+    }
+}
+/**
+ * Load environment variables from file
+ * @param envFile Path to the environment file
+ */
+export function loadEnvironmentVariables(envFile) {
+    const envPath = path.resolve(envFile);
+    // Check if file exists
+    if (!fs.existsSync(envPath)) {
+        throw new Error(`Environment file not found: ${envPath}`);
+    }
+    // Load environment variables from file
+    const result = dotenv.config({ path: envPath });
+    if (result.error) {
+        throw new Error(`Error loading environment variables: ${result.error}`);
     }
+    logger.info(`Loaded environment variables from ${envPath}`);
 }
 /**
- * Load environment variables from file if specified in options
- * @param options CLI options containing envFile path
+ * Filters and returns a sorted list of enabled ToolNames based on CLI allow/block options.
+ *
+ * This function determines which tools should be enabled for the server by applying
+ * the following logic:
+ *   1. If an allow list (`cliOptions.allowTools`) is provided and non-empty, only those
+ *      tool names present in the allow list (and valid) will be enabled. Any invalid
+ *      tool names in the allow list are ignored and a warning is logged.
+ *   2. If a block list (`cliOptions.blockTools`) is provided and non-empty, any tool
+ *      names present in the block list (and valid) will be removed from the enabled
+ *      set. Any invalid tool names in the block list are ignored and a warning is logged.
+ *   3. If neither allow nor block lists are provided, all available tools are enabled.
+ *
+ * The returned list is always sorted alphabetically.
+ *
+ * @param cliOptions - The parsed CLI options containing allow/block tool lists.
+ * @returns An alphabetically sorted array of enabled ToolNames.
  */
-export function loadEnvironmentVariables(options) {
-    if (options.envFile) {
-        const envPath = path.resolve(options.envFile);
-        // Check if file exists
-        if (!fs.existsSync(envPath)) {
-            console.error(`Environment file not found: ${envPath}`);
-            return;
-        }
-        // Load environment variables from file
-        const result = dotenv.config({ path: envPath });
-        if (result.error) {
-            console.error(`Error loading environment variables: ${result.error.message}`);
-            return;
-        }
-        console.error(`Loaded environment variables from ${envPath}`);
+export function getFilteredToolNames(cliOptions) {
+    let filteredToolNames = Object.values(ToolName);
+    const validToolNames = new Set(Object.values(ToolName));
+    if (cliOptions.allowTools && cliOptions.allowTools.length > 0) {
+        const valid = cliOptions.allowTools.filter((t) => validToolNames.has(t));
+        const invalid = cliOptions.allowTools.filter((t) => !validToolNames.has(t));
+        if (invalid.length > 0) {
+            logger.warn(`Ignoring invalid tool names in allow list: ${invalid.join(", ")}`);
+        }
+        filteredToolNames = valid.map((t) => t);
+    }
+    if (cliOptions.blockTools && cliOptions.blockTools.length > 0) {
+        const validBlock = cliOptions.blockTools.filter((t) => validToolNames.has(t));
+        const invalidBlock = cliOptions.blockTools.filter((t) => !validToolNames.has(t));
+        if (invalidBlock.length > 0) {
+            logger.warn(`Ignoring invalid tool names in block list: ${invalidBlock.join(", ")}`);
+        }
+        filteredToolNames = filteredToolNames.filter((t) => !validBlock.includes(t));
+    }
+    // Deduplicate and sort
+    const deduped = Array.from(new Set(filteredToolNames)).sort();
+    if ((!cliOptions.allowTools || cliOptions.allowTools.length === 0) &&
+        (!cliOptions.blockTools || cliOptions.blockTools.length === 0)) {
+        logger.info("No allow/block tool lists provided; all tools are enabled by default.");
+    }
+    return deduped;
+}
+/**
+ * Loads configuration from a properties file
+ * @param filePath - Path to the properties file
+ * @returns configuration object
+ */
+export function parsePropertiesFile(filePath) {
+    const absPath = path.resolve(filePath);
+    if (!fs.existsSync(absPath)) {
+        throw new Error(`Properties file not found: ${absPath}`);
+    }
+    try {
+        const properties = getProperties(fs.readFileSync(absPath, "utf-8"));
+        return properties;
+    }
+    catch (err) {
+        throw new Error(`Failed to parse properties file: ${err instanceof Error ? err.message : String(err)}`);
     }
 }
 //# sourceMappingURL=cli.js.map

package/dist/cli.js.map

@@ -1 +1 @@
-{"version":3,"file":"cli.js","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,6BAA6B,CAAC;AACtD,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AAOxB;;;GAGG;AACH,MAAM,UAAU,YAAY;IAC1B,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE;SAC1B,IAAI,CAAC,eAAe,CAAC;SACrB,WAAW,CACV,kFAAkF,CACnF;SACA,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,KAAK,CAAC;SACjD,MAAM,CAAC,uBAAuB,EAAE,sCAAsC,CAAC;SACvE,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;QAClB,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,wBAAwB,CAAC,OAAO,CAAC,CAAC;QACpC,CAAC;IACH,CAAC,CAAC;SACD,oBAAoB,CAAC,KAAK,CAAC;SAC3B,YAAY,EAAE,CAAC;IAElB,IAAI,CAAC;QACH,0FAA0F;QAC1F,OAAO,OAAO,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,CAAC;IAChC,CAAC;IAAC,MAAM,CAAC;QACP,2DAA2D;QAC3D,qDAAqD;QACrD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,wBAAwB,CAAC,OAAmB;IAC1D,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAE9C,uBAAuB;QACvB,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5B,OAAO,CAAC,KAAK,CAAC,+BAA+B,OAAO,EAAE,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,uCAAuC;QACvC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAEhD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,OAAO,CAAC,KAAK,CACX,wCAAwC,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAC/D,CAAC;YACF,OAAO;QACT,CAAC;QAED,OAAO,CAAC,KAAK,CAAC,qCAAqC,OAAO,EAAE,CAAC,CAAC;IAChE,CAAC;AACH,CAAC"}
+{"version":3,"file":"cli.js","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM,6BAA6B,CAAC;AAC9E,OAAO,EAAE,QAAQ,EAAE,MAAM,mCAAmC,CAAC;AAC7D,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACxC,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,aAAa,EAAsB,MAAM,iBAAiB,CAAC;AACpE,OAAO,GAAG,MAAM,iBAAiB,CAAC,OAAO,IAAI,EAAE,MAAM,EAAE,CAAC;AAaxD;;;GAGG;AACH,MAAM,UAAU,iBAAiB;IAC/B,OAAO,GAAG,CAAC,OAAO,CAAC;AACrB,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAa;IACvC,4CAA4C;IAC5C,MAAM,KAAK,GAAG,KAAK;SAChB,KAAK,CAAC,GAAG,CAAC;SACV,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;SAC1B,MAAM,CAAC,OAAO,CAAC,CAAC;IAEnB,+BAA+B;IAC/B,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC;IACzD,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAC/B,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,IAAqB,CAAC,CACjD,CAAC;IAEF,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CACb,8BAA8B,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,oBAAoB,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC7G,CAAC;IACJ,CAAC;IAED,wBAAwB;IACxB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,CAAoB,CAAC;AACvD,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,aAAa,CAAC,KAAa;IAClC,OAAO,CACL,KAAK;SACF,KAAK,CAAC,GAAG,CAAC;SACV,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QACrB,+EAA+E;SAC9E,MAAM,CAAC,OAAO,CAAC,CACnB,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACH,SAAS,aAAa,CAAC,QAAgB;IACrC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACvC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,6BAA6B,OAAO,EAAE,CAAC,CAAC;IAC1D,CAAC;IACD,MAAM,KAAK,GAAG,EAAE;SACb,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC;SAC9B,KAAK,CAAC,IAAI,CAAC;SACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;SAC1B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;IACnD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,YAAY;IAC1B,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE;SAC1B,IAAI,CAAC,eAAe,CAAC;SACrB,WAAW,CACV,kFAAkF,CACnF;SACA,OAAO,CAAC,iBAAiB,EAAE,CAAC;SAC5B,MAAM,CAAC,uBAAuB,EAAE,sCAAsC,CAAC;SACvE,MAAM,CACL,gCAAgC,EAChC,yDAAyD,CAC1D;SACA,SAAS,CACR,IAAI,MAAM,CACR,yBAAyB,EACzB,wCAAwC,CACzC;SACE,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;SACrC,OAAO,CAAC,aAAa,CAAC,KAAK,CAAC;SAC5B,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CACnD;SACA,MAAM,CACL,uBAAuB,EACvB,8IAA8I,CAC/I;SACA,MAAM,CACL,uBAAuB,EACvB,6IAA6I,CAC9I;SACA,MAAM,CACL,2BAA2B,EAC3B,oIAAoI,CACrI;SACA,MAAM,CACL,2BAA2B,EAC3B,mIAAmI,CACpI;SACA,MAAM,CACL,cAAc,EACd,gIAAgI,CACjI;SACA,MAAM,CACL,iCAAiC,EACjC,8EAA8E,CAC/E;SACA,oBAAoB,CAAC,KAAK,CAAC;SAC3B,YAAY,EAAE,CAAC;IAElB,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,CAAC;QACpC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,wBAAwB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACzC,CAAC;QACD,qCAAqC;QACrC,IAAI,UAAU,GAAyB,SAAS,CAAC;QACjD,IAAI,UAAU,GAAyB,SAAS,CAAC;QACjD,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC9C,CAAC;aAAM,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YAC/B,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAClD,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC9C,CAAC;aAAM,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YAC/B,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAClD,CAAC;QACD,IAAI,WAAW,GAAuB,EAAE,CAAC;QACzC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,WAAW,GAAG,mBAAmB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC1D,CAAC;QACD,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC;gBACvC,CAAC,CAAC,IAAI,CAAC,SAAS;gBAChB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC;YACpB,UAAU;YACV,UAAU;YACV,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS;YAC3B,0BAA0B,EAAE,CAAC,CAAC,IAAI,CAAC,0BAA0B;YAC7D,WAAW,EAAE,WAAW;SACzB,CAAC;IACJ,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,IACE,KAAK,YAAY,cAAc;YAC/B,CAAC,KAAK,CAAC,IAAI,KAAK,yBAAyB;gBACvC,KAAK,CAAC,IAAI,KAAK,mBAAmB,CAAC,EACrC,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QACD,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;YAC3B,MAAM,CAAC,KAAK,CACV;gBACE,KAAK;gBACL,WAAW,EAAE,KAAK,CAAC,QAAQ,EAAE;gBAC7B,YAAY,EAAE,KAAK,CAAC,OAAO;aAC5B,EACD,2BAA2B,CAC5B,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,EAAE,2BAA2B,CAAC,CAAC;QACtE,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,wBAAwB,CAAC,OAAe;IACtD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAEtC,uBAAuB;IACvB,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,+BAA+B,OAAO,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED,uCAAuC;IACvC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAEhD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,wCAAwC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;IAC1E,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,qCAAqC,OAAO,EAAE,CAAC,CAAC;AAC9D,CAAC;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,UAAU,oBAAoB,CAAC,UAAsB;IACzD,IAAI,iBAAiB,GAAe,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC5D,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;IAExD,IAAI,UAAU,CAAC,UAAU,IAAI,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9D,MAAM,KAAK,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAC/C,cAAc,CAAC,GAAG,CAAC,CAAa,CAAC,CAClC,CAAC;QACF,MAAM,OAAO,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,CAC1C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,CAAa,CAAC,CAC1C,CAAC;QACF,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,CAAC,IAAI,CACT,8CAA8C,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACnE,CAAC;QACJ,CAAC;QACD,iBAAiB,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAa,CAAC,CAAC;IACtD,CAAC;IACD,IAAI,UAAU,CAAC,UAAU,IAAI,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9D,MAAM,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CACpD,cAAc,CAAC,GAAG,CAAC,CAAa,CAAC,CAClC,CAAC;QACF,MAAM,YAAY,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,CAC/C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,CAAa,CAAC,CAC1C,CAAC;QACF,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,CAAC,IAAI,CACT,8CAA8C,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACxE,CAAC;QACJ,CAAC;QACD,iBAAiB,GAAG,iBAAiB,CAAC,MAAM,CAC1C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAC/B,CAAC;IACJ,CAAC;IACD,uBAAuB;IACvB,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAC9D,IACE,CAAC,CAAC,UAAU,CAAC,UAAU,IAAI,UAAU,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,CAAC;QAC9D,CAAC,CAAC,UAAU,CAAC,UAAU,IAAI,UAAU,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,CAAC,EAC9D,CAAC;QACD,MAAM,CAAC,IAAI,CACT,uEAAuE,CACxE,CAAC;IACJ,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CAAC,QAAgB;IAClD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACvC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,8BAA8B,OAAO,EAAE,CAAC,CAAC;IAC3D,CAAC;IACD,IAAI,CAAC;QACH,MAAM,UAAU,GAAuB,aAAa,CAClD,EAAE,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAClC,CAAC;QACF,OAAO,UAAU,CAAC;IACpB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CACb,oCAAoC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CACvF,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/confluent/client-manager.d.ts

@@ -1,7 +1,9 @@
 /**
  * @fileoverview Provides client management functionality for Kafka and Confluent Cloud services.
  */
-import { KafkaJS } from "@confluentinc/kafka-javascript";
+import { GlobalConfig, KafkaJS } from "@confluentinc/kafka-javascript";
+import { SchemaRegistryClient } from "@confluentinc/schemaregistry";
+import { ConfluentAuth, ConfluentEndpoints } from "../confluent/middleware.js";
 import { paths } from "@src/confluent/openapi-schema.js";
 import { Client } from "openapi-fetch";
 /**
@@ -15,7 +17,7 @@ export interface KafkaClientManager {
     /** Gets a connected producer client for publishing messages */
     getProducer(): Promise<KafkaJS.Producer>;
     /** Gets a connected consumer client for subscribing to topics */
-    getConsumer(): Promise<KafkaJS.Consumer>;
+    getConsumer(sessionId?: string): Promise<KafkaJS.Consumer>;
     /** Disconnects and cleans up all client connections */
     disconnect(): Promise<void>;
 }
@@ -36,17 +38,35 @@ export interface ConfluentCloudRestClientManager {
     setConfluentCloudSchemaRegistryEndpoint(endpoint: string): void;
     setConfluentCloudKafkaRestEndpoint(endpoint: string): void;
 }
-export interface ClientManager extends KafkaClientManager, ConfluentCloudRestClientManager {
+/**
+ * Interface for managing Schema Registry client connections.
+ */
+export interface SchemaRegistryClientHandler {
+    getSchemaRegistryClient(): SchemaRegistryClient;
+}
+export interface ClientManager extends KafkaClientManager, ConfluentCloudRestClientManager, SchemaRegistryClientHandler {
+    getSchemaRegistryClient(): SchemaRegistryClient;
+}
+export interface ClientManagerConfig {
+    kafka: GlobalConfig;
+    endpoints: ConfluentEndpoints;
+    auth: {
+        cloud: ConfluentAuth;
+        flink: ConfluentAuth;
+        schemaRegistry: ConfluentAuth;
+        kafka: ConfluentAuth;
+    };
 }
 /**
  * Default implementation of client management for Kafka and Confluent Cloud services.
  * Manages lifecycle and lazy initialization of various client connections.
  */
-export declare class DefaultClientManager implements ClientManager {
+export declare class DefaultClientManager implements ClientManager, SchemaRegistryClientHandler {
     private confluentCloudBaseUrl;
     private confluentCloudFlinkBaseUrl;
     private confluentCloudSchemaRegistryBaseUrl;
     private confluentCloudKafkaRestBaseUrl;
+    private readonly kafkaConfig;
     private readonly kafkaClient;
     private readonly adminClient;
     private readonly producer;
@@ -54,15 +74,14 @@ export declare class DefaultClientManager implements ClientManager {
     private readonly confluentCloudRestClient;
     private readonly confluentCloudSchemaRegistryRestClient;
     private readonly confluentCloudKafkaRestClient;
+    private readonly schemaRegistryClient;
     /**
      * Creates a new DefaultClientManager instance.
-     * @param config - Configuration options for KafkaJS client
-     * @param confluentCloudBaseUrl - Base URL for Confluent Cloud REST API
-     * @param confluentCloudFlinkBaseUrl - Base URL for Flink REST API
-     * @param confluentCloudSchemaRegistryBaseUrl - Base URL for Schema Registry REST API
-     * @param confluentCloudKafkaRestBaseUrl - Base URL for Kafka REST API
+     * @param config - Configuration for all clients
      */
-    constructor(config: KafkaJS.CommonConstructorConfig, confluentCloudBaseUrl?: string, confluentCloudFlinkBaseUrl?: string, confluentCloudSchemaRegistryBaseUrl?: string, confluentCloudKafkaRestBaseUrl?: string);
+    constructor(config: ClientManagerConfig);
+    /** @inheritdoc */
+    getConsumer(sessionId?: string): Promise<KafkaJS.Consumer>;
     /**
      * a function that sets a new confluent cloud rest endpoint.
      * Closes the current client first.
@@ -72,7 +91,6 @@ export declare class DefaultClientManager implements ClientManager {
     setConfluentCloudFlinkEndpoint(endpoint: string): void;
     setConfluentCloudSchemaRegistryEndpoint(endpoint: string): void;
     setConfluentCloudKafkaRestEndpoint(endpoint: string): void;
-    getConsumer(): Promise<KafkaJS.Consumer>;
     /** @inheritdoc */
     getKafkaClient(): KafkaJS.Kafka;
     /** @inheritdoc */
@@ -89,5 +107,7 @@ export declare class DefaultClientManager implements ClientManager {
     getProducer(): Promise<KafkaJS.Producer>;
     /** @inheritdoc */
     disconnect(): Promise<void>;
+    /** @inheritdoc */
+    getSchemaRegistryClient(): SchemaRegistryClient;
 }
 //# sourceMappingURL=client-manager.d.ts.map

package/dist/confluent/client-manager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client-manager.d.ts","sourceRoot":"","sources":["../../src/confluent/client-manager.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,gCAAgC,CAAC;AAOzD,OAAO,EAAE,KAAK,EAAE,MAAM,kCAAkC,CAAC;AAEzD,OAAqB,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAErD;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,0CAA0C;IAC1C,cAAc,IAAI,OAAO,CAAC,KAAK,CAAC;IAChC,wEAAwE;IACxE,cAAc,IAAI,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACzC,+DAA+D;IAC/D,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACzC,iEAAiE;IACjE,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACzC,uDAAuD;IACvD,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CAC7B;AAED;;GAEG;AACH,MAAM,WAAW,+BAA+B;IAC9C,yEAAyE;IACzE,gCAAgC,IAAI,MAAM,CAAC,KAAK,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC;IACzE,2EAA2E;IAC3E,2BAA2B,IAAI,MAAM,CAAC,KAAK,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC;IACpE,mFAAmF;IACnF,yCAAyC,IAAI,MAAM,CACjD,KAAK,EACL,GAAG,MAAM,IAAI,MAAM,EAAE,CACtB,CAAC;IACF,yEAAyE;IACzE,gCAAgC,IAAI,MAAM,CAAC,KAAK,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC;IAEzE,6BAA6B,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACtD,8BAA8B,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACvD,uCAAuC,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IAChE,kCAAkC,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;CAC5D;AAED,MAAM,WAAW,aACf,SAAQ,kBAAkB,EACxB,+BAA+B;CAAG;AAEtC;;;GAGG;AACH,qBAAa,oBAAqB,YAAW,aAAa;IACxD,OAAO,CAAC,qBAAqB,CAAS;IACtC,OAAO,CAAC,0BAA0B,CAAS;IAC3C,OAAO,CAAC,mCAAmC,CAAS;IACpD,OAAO,CAAC,8BAA8B,CAAS;IAC/C,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAsB;IAClD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAA2B;IACvD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA8B;IACvD,OAAO,CAAC,QAAQ,CAAC,6BAA6B,CAE5C;IACF,OAAO,CAAC,QAAQ,CAAC,wBAAwB,CAEvC;IACF,OAAO,CAAC,QAAQ,CAAC,sCAAsC,CAErD;IACF,OAAO,CAAC,QAAQ,CAAC,6BAA6B,CAE5C;IACF;;;;;;;OAOG;gBAED,MAAM,EAAE,OAAO,CAAC,uBAAuB,EACvC,qBAAqB,CAAC,EAAE,MAAM,EAC9B,0BAA0B,CAAC,EAAE,MAAM,EACnC,mCAAmC,CAAC,EAAE,MAAM,EAC5C,8BAA8B,CAAC,EAAE,MAAM;IA4EzC;;;;OAIG;IACH,6BAA6B,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAIrD,8BAA8B,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAItD,uCAAuC,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAI/D,kCAAkC,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAK1D,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;IAIxC,kBAAkB;IAClB,cAAc,IAAI,OAAO,CAAC,KAAK;IAI/B,kBAAkB;IAClB,gCAAgC,IAAI,MAAM,CAAC,KAAK,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC;IAIxE,kBAAkB;IAClB,2BAA2B,IAAI,MAAM,CAAC,KAAK,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC;IAInE,kBAAkB;IAClB,yCAAyC,IAAI,MAAM,CACjD,KAAK,EACL,GAAG,MAAM,IAAI,MAAM,EAAE,CACtB;IAID,kBAAkB;IAClB,gCAAgC,IAAI,MAAM,CAAC,KAAK,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC;IAIxE,kBAAkB;IACZ,cAAc,IAAI,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC;IAI9C,kBAAkB;IACZ,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;IAI9C,kBAAkB;IACZ,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;CAKlC"}
+{"version":3,"file":"client-manager.d.ts","sourceRoot":"","sources":["../../src/confluent/client-manager.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,gCAAgC,CAAC;AACvE,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AACpE,OAAO,EACL,aAAa,EACb,kBAAkB,EAEnB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,KAAK,EAAE,MAAM,kCAAkC,CAAC;AAGzD,OAAqB,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAErD;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,0CAA0C;IAC1C,cAAc,IAAI,OAAO,CAAC,KAAK,CAAC;IAChC,wEAAwE;IACxE,cAAc,IAAI,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACzC,+DAA+D;IAC/D,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACzC,iEAAiE;IACjE,WAAW,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC3D,uDAAuD;IACvD,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CAC7B;AAED;;GAEG;AACH,MAAM,WAAW,+BAA+B;IAC9C,yEAAyE;IACzE,gCAAgC,IAAI,MAAM,CAAC,KAAK,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC;IACzE,2EAA2E;IAC3E,2BAA2B,IAAI,MAAM,CAAC,KAAK,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC;IACpE,mFAAmF;IACnF,yCAAyC,IAAI,MAAM,CACjD,KAAK,EACL,GAAG,MAAM,IAAI,MAAM,EAAE,CACtB,CAAC;IACF,yEAAyE;IACzE,gCAAgC,IAAI,MAAM,CAAC,KAAK,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC;IAEzE,6BAA6B,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACtD,8BAA8B,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACvD,uCAAuC,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IAChE,kCAAkC,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;CAC5D;AAED;;GAEG;AACH,MAAM,WAAW,2BAA2B;IAC1C,uBAAuB,IAAI,oBAAoB,CAAC;CACjD;AAED,MAAM,WAAW,aACf,SAAQ,kBAAkB,EACxB,+BAA+B,EAC/B,2BAA2B;IAC7B,uBAAuB,IAAI,oBAAoB,CAAC;CACjD;AAED,MAAM,WAAW,mBAAmB;IAClC,KAAK,EAAE,YAAY,CAAC;IACpB,SAAS,EAAE,kBAAkB,CAAC;IAC9B,IAAI,EAAE;QACJ,KAAK,EAAE,aAAa,CAAC;QACrB,KAAK,EAAE,aAAa,CAAC;QACrB,cAAc,EAAE,aAAa,CAAC;QAC9B,KAAK,EAAE,aAAa,CAAC;KACtB,CAAC;CACH;AAED;;;GAGG;AACH,qBAAa,oBACX,YAAW,aAAa,EAAE,2BAA2B;IAErD,OAAO,CAAC,qBAAqB,CAAqB;IAClD,OAAO,CAAC,0BAA0B,CAAqB;IACvD,OAAO,CAAC,mCAAmC,CAAqB;IAChE,OAAO,CAAC,8BAA8B,CAAqB;IAC3D,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAe;IAC3C,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAsB;IAClD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAA2B;IACvD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA8B;IACvD,OAAO,CAAC,QAAQ,CAAC,6BAA6B,CAE5C;IACF,OAAO,CAAC,QAAQ,CAAC,wBAAwB,CAEvC;IACF,OAAO,CAAC,QAAQ,CAAC,sCAAsC,CAErD;IACF,OAAO,CAAC,QAAQ,CAAC,6BAA6B,CAE5C;IACF,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAA6B;IAElE;;;OAGG;gBACS,MAAM,EAAE,mBAAmB;IA8GvC,kBAAkB;IACZ,WAAW,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;IAkBhE;;;;OAIG;IACH,6BAA6B,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAIrD,8BAA8B,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAItD,uCAAuC,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAI/D,kCAAkC,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAK1D,kBAAkB;IAClB,cAAc,IAAI,OAAO,CAAC,KAAK;IAI/B,kBAAkB;IAClB,gCAAgC,IAAI,MAAM,CAAC,KAAK,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC;IAIxE,kBAAkB;IAClB,2BAA2B,IAAI,MAAM,CAAC,KAAK,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC;IAInE,kBAAkB;IAClB,yCAAyC,IAAI,MAAM,CACjD,KAAK,EACL,GAAG,MAAM,IAAI,MAAM,EAAE,CACtB;IAID,kBAAkB;IAClB,gCAAgC,IAAI,MAAM,CAAC,KAAK,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC;IAIxE,kBAAkB;IACZ,cAAc,IAAI,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC;IAI9C,kBAAkB;IACZ,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;IAI9C,kBAAkB;IACZ,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAMjC,kBAAkB;IAClB,uBAAuB,IAAI,oBAAoB;CAGhD"}

package/dist/confluent/client-manager.js

@@ -2,8 +2,10 @@
  * @fileoverview Provides client management functionality for Kafka and Confluent Cloud services.
  */
 import { KafkaJS } from "@confluentinc/kafka-javascript";
-import { confluentCloudAuthMiddleware, confluentCloudFlinkAuthMiddleware, confluentCloudKafkaAuthMiddleware, confluentCloudSchemaRegistryAuthMiddleware, } from "../confluent/middleware.js";
+import { SchemaRegistryClient } from "@confluentinc/schemaregistry";
+import { createAuthMiddleware, } from "../confluent/middleware.js";
 import { AsyncLazy, Lazy } from "../lazy.js";
+import { kafkaLogger, logger } from "../logger.js";
 import createClient from "openapi-fetch";
 /**
  * Default implementation of client management for Kafka and Confluent Cloud services.
@@ -14,6 +16,7 @@ export class DefaultClientManager {
     confluentCloudFlinkBaseUrl;
     confluentCloudSchemaRegistryBaseUrl;
     confluentCloudKafkaRestBaseUrl;
+    kafkaConfig;
     kafkaClient;
     adminClient;
     producer;
@@ -21,70 +24,110 @@ export class DefaultClientManager {
     confluentCloudRestClient;
     confluentCloudSchemaRegistryRestClient;
     confluentCloudKafkaRestClient;
+    schemaRegistryClient;
     /**
      * Creates a new DefaultClientManager instance.
-     * @param config - Configuration options for KafkaJS client
-     * @param confluentCloudBaseUrl - Base URL for Confluent Cloud REST API
-     * @param confluentCloudFlinkBaseUrl - Base URL for Flink REST API
-     * @param confluentCloudSchemaRegistryBaseUrl - Base URL for Schema Registry REST API
-     * @param confluentCloudKafkaRestBaseUrl - Base URL for Kafka REST API
+     * @param config - Configuration for all clients
      */
-    constructor(config, confluentCloudBaseUrl, confluentCloudFlinkBaseUrl, confluentCloudSchemaRegistryBaseUrl, confluentCloudKafkaRestBaseUrl) {
-        this.confluentCloudBaseUrl = confluentCloudBaseUrl || "";
-        this.confluentCloudFlinkBaseUrl = confluentCloudFlinkBaseUrl || "";
-        this.confluentCloudSchemaRegistryBaseUrl =
-            confluentCloudSchemaRegistryBaseUrl || "";
-        this.confluentCloudKafkaRestBaseUrl = confluentCloudKafkaRestBaseUrl || "";
-        this.kafkaClient = new Lazy(() => new KafkaJS.Kafka(config));
+    constructor(config) {
+        this.confluentCloudBaseUrl = config.endpoints.cloud;
+        this.confluentCloudFlinkBaseUrl = config.endpoints.flink;
+        this.confluentCloudSchemaRegistryBaseUrl = config.endpoints.schemaRegistry;
+        this.confluentCloudKafkaRestBaseUrl = config.endpoints.kafka;
+        this.kafkaConfig = config.kafka;
+        this.kafkaClient = new Lazy(() => new KafkaJS.Kafka({
+            ...this.kafkaConfig,
+            kafkaJS: {
+                logger: kafkaLogger,
+                // we need to do this since typescript will complain that we are missing configs like `brokers` even though we are passing them in kafkaConfig above
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            },
+        }));
         this.adminClient = new AsyncLazy(async () => {
-            console.error("Connecting Kafka Admin");
+            logger.info("Connecting Kafka Admin");
             const admin = this.kafkaClient.get().admin();
             await admin.connect();
             return admin;
         }, (admin) => admin.disconnect());
         this.producer = new AsyncLazy(async () => {
-            console.error("Connecting Kafka Producer");
-            const producer = this.kafkaClient.get().producer({
-                kafkaJS: {
-                    acks: 1,
-                    compression: KafkaJS.CompressionTypes.GZIP,
-                },
-            });
+            logger.info("Connecting Kafka Producer");
+            const producer = this.kafkaClient.get().producer();
             await producer.connect();
             return producer;
         }, (producer) => producer.disconnect());
         this.confluentCloudRestClient = new Lazy(() => {
-            console.error(`Initializing Confluent Cloud REST client for base URL ${this.confluentCloudBaseUrl}`);
+            if (!this.confluentCloudBaseUrl) {
+                throw new Error("Confluent Cloud REST endpoint not configured");
+            }
+            logger.info(`Initializing Confluent Cloud REST client for base URL ${this.confluentCloudBaseUrl}`);
             const client = createClient({
                 baseUrl: this.confluentCloudBaseUrl,
             });
-            client.use(confluentCloudAuthMiddleware);
+            client.use(createAuthMiddleware(config.auth.cloud));
             return client;
         });
         this.confluentCloudFlinkRestClient = new Lazy(() => {
-            console.error(`Initializing Confluent Cloud Flink REST client for base URL ${this.confluentCloudFlinkBaseUrl}`);
+            if (!this.confluentCloudFlinkBaseUrl) {
+                throw new Error("Confluent Cloud Flink REST endpoint not configured");
+            }
+            logger.info(`Initializing Confluent Cloud Flink REST client for base URL ${this.confluentCloudFlinkBaseUrl}`);
             const client = createClient({
                 baseUrl: this.confluentCloudFlinkBaseUrl,
             });
-            client.use(confluentCloudFlinkAuthMiddleware);
+            client.use(createAuthMiddleware(config.auth.flink));
             return client;
         });
         this.confluentCloudSchemaRegistryRestClient = new Lazy(() => {
-            console.error(`Initializing Confluent Cloud Schema Registry REST client for base URL ${this.confluentCloudSchemaRegistryBaseUrl}`);
+            if (!this.confluentCloudSchemaRegistryBaseUrl) {
+                throw new Error("Confluent Cloud Schema Registry REST endpoint not configured");
+            }
+            logger.info(`Initializing Confluent Cloud Schema Registry REST client for base URL ${this.confluentCloudSchemaRegistryBaseUrl}`);
             const client = createClient({
                 baseUrl: this.confluentCloudSchemaRegistryBaseUrl,
             });
-            client.use(confluentCloudSchemaRegistryAuthMiddleware);
+            client.use(createAuthMiddleware(config.auth.schemaRegistry));
             return client;
         });
         this.confluentCloudKafkaRestClient = new Lazy(() => {
-            console.error(`Initializing Confluent Cloud Kafka REST client for base URL ${this.confluentCloudKafkaRestBaseUrl}`);
+            if (!this.confluentCloudKafkaRestBaseUrl) {
+                throw new Error("Confluent Cloud Kafka REST endpoint not configured");
+            }
+            logger.info(`Initializing Confluent Cloud Kafka REST client for base URL ${this.confluentCloudKafkaRestBaseUrl}`);
             const client = createClient({
                 baseUrl: this.confluentCloudKafkaRestBaseUrl,
             });
-            client.use(confluentCloudKafkaAuthMiddleware);
+            client.use(createAuthMiddleware(config.auth.kafka));
             return client;
         });
+        this.schemaRegistryClient = new Lazy(() => {
+            if (!this.confluentCloudSchemaRegistryBaseUrl) {
+                throw new Error("Schema Registry endpoint not configured");
+            }
+            const { apiKey, apiSecret } = config.auth.schemaRegistry;
+            return new SchemaRegistryClient({
+                baseURLs: [this.confluentCloudSchemaRegistryBaseUrl],
+                basicAuthCredentials: {
+                    credentialsSource: "USER_INFO",
+                    userInfo: `${apiKey}:${apiSecret}`,
+                },
+            });
+        });
+    }
+    /** @inheritdoc */
+    async getConsumer(sessionId) {
+        // Build the config inline, merging with defaults
+        const baseGroupId = this.kafkaConfig["group.id"] || "mcp-confluent";
+        const groupId = sessionId ? `${baseGroupId}-${sessionId}` : baseGroupId;
+        const consumerConfig = {
+            // Spread all user-provided config
+            ...this.kafkaConfig,
+            // Override with our logic
+            "group.id": groupId,
+            "auto.offset.reset": this.kafkaConfig["auto.offset.reset"] || "earliest",
+            "allow.auto.create.topics": this.kafkaConfig["allow.auto.create.topics"] || false,
+            "enable.auto.commit": this.kafkaConfig["enable.auto.commit"] || false,
+        };
+        return this.kafkaClient.get().consumer(consumerConfig);
     }
     /**
      * a function that sets a new confluent cloud rest endpoint.
@@ -107,9 +150,6 @@ export class DefaultClientManager {
         this.confluentCloudKafkaRestClient.close();
         this.confluentCloudKafkaRestBaseUrl = endpoint;
     }
-    getConsumer() {
-        throw new Error("Method not implemented.");
-    }
     /** @inheritdoc */
     getKafkaClient() {
         return this.kafkaClient.get();
@@ -144,5 +184,9 @@ export class DefaultClientManager {
         await this.producer.close();
         this.kafkaClient.close();
     }
+    /** @inheritdoc */
+    getSchemaRegistryClient() {
+        return this.schemaRegistryClient.get();
+    }
 }
 //# sourceMappingURL=client-manager.js.map

package/dist/confluent/client-manager.js.map

@@ -1 +1 @@
-{"version":3,"file":"client-manager.js","sourceRoot":"","sources":["../../src/confluent/client-manager.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,gCAAgC,CAAC;AACzD,OAAO,EACL,4BAA4B,EAC5B,iCAAiC,EACjC,iCAAiC,EACjC,0CAA0C,GAC3C,MAAM,8BAA8B,CAAC;AAEtC,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,YAAwB,MAAM,eAAe,CAAC;AA4CrD;;;GAGG;AACH,MAAM,OAAO,oBAAoB;IACvB,qBAAqB,CAAS;IAC9B,0BAA0B,CAAS;IACnC,mCAAmC,CAAS;IAC5C,8BAA8B,CAAS;IAC9B,WAAW,CAAsB;IACjC,WAAW,CAA2B;IACtC,QAAQ,CAA8B;IACtC,6BAA6B,CAE5C;IACe,wBAAwB,CAEvC;IACe,sCAAsC,CAErD;IACe,6BAA6B,CAE5C;IACF;;;;;;;OAOG;IACH,YACE,MAAuC,EACvC,qBAA8B,EAC9B,0BAAmC,EACnC,mCAA4C,EAC5C,8BAAuC;QAEvC,IAAI,CAAC,qBAAqB,GAAG,qBAAqB,IAAI,EAAE,CAAC;QACzD,IAAI,CAAC,0BAA0B,GAAG,0BAA0B,IAAI,EAAE,CAAC;QACnE,IAAI,CAAC,mCAAmC;YACtC,mCAAmC,IAAI,EAAE,CAAC;QAC5C,IAAI,CAAC,8BAA8B,GAAG,8BAA8B,IAAI,EAAE,CAAC;QAC3E,IAAI,CAAC,WAAW,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;QAC7D,IAAI,CAAC,WAAW,GAAG,IAAI,SAAS,CAC9B,KAAK,IAAI,EAAE;YACT,OAAO,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;YACxC,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC;YAC7C,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;YACtB,OAAO,KAAK,CAAC;QACf,CAAC,EACD,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,CAC9B,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,SAAS,CAC3B,KAAK,IAAI,EAAE;YACT,OAAO,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC3C,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC;gBAC/C,OAAO,EAAE;oBACP,IAAI,EAAE,CAAC;oBACP,WAAW,EAAE,OAAO,CAAC,gBAAgB,CAAC,IAAI;iBAC3C;aACF,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,QAAQ,CAAC;QAClB,CAAC,EACD,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,UAAU,EAAE,CACpC,CAAC;QAEF,IAAI,CAAC,wBAAwB,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE;YAC5C,OAAO,CAAC,KAAK,CACX,yDAAyD,IAAI,CAAC,qBAAqB,EAAE,CACtF,CAAC;YACF,MAAM,MAAM,GAAG,YAAY,CAAQ;gBACjC,OAAO,EAAE,IAAI,CAAC,qBAAqB;aACpC,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;YACzC,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,6BAA6B,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE;YACjD,OAAO,CAAC,KAAK,CACX,+DAA+D,IAAI,CAAC,0BAA0B,EAAE,CACjG,CAAC;YACF,MAAM,MAAM,GAAG,YAAY,CAAQ;gBACjC,OAAO,EAAE,IAAI,CAAC,0BAA0B;aACzC,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;YAC9C,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,sCAAsC,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE;YAC1D,OAAO,CAAC,KAAK,CACX,yEAAyE,IAAI,CAAC,mCAAmC,EAAE,CACpH,CAAC;YACF,MAAM,MAAM,GAAG,YAAY,CAAQ;gBACjC,OAAO,EAAE,IAAI,CAAC,mCAAmC;aAClD,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;YACvD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,6BAA6B,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE;YACjD,OAAO,CAAC,KAAK,CACX,+DAA+D,IAAI,CAAC,8BAA8B,EAAE,CACrG,CAAC;YACF,MAAM,MAAM,GAAG,YAAY,CAAQ;gBACjC,OAAO,EAAE,IAAI,CAAC,8BAA8B;aAC7C,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;YAC9C,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;IACL,CAAC;IACD;;;;OAIG;IACH,6BAA6B,CAAC,QAAgB;QAC5C,IAAI,CAAC,wBAAwB,CAAC,KAAK,EAAE,CAAC;QACtC,IAAI,CAAC,qBAAqB,GAAG,QAAQ,CAAC;IACxC,CAAC;IACD,8BAA8B,CAAC,QAAgB;QAC7C,IAAI,CAAC,6BAA6B,CAAC,KAAK,EAAE,CAAC;QAC3C,IAAI,CAAC,0BAA0B,GAAG,QAAQ,CAAC;IAC7C,CAAC;IACD,uCAAuC,CAAC,QAAgB;QACtD,IAAI,CAAC,sCAAsC,CAAC,KAAK,EAAE,CAAC;QACpD,IAAI,CAAC,mCAAmC,GAAG,QAAQ,CAAC;IACtD,CAAC;IACD,kCAAkC,CAAC,QAAgB;QACjD,IAAI,CAAC,6BAA6B,CAAC,KAAK,EAAE,CAAC;QAC3C,IAAI,CAAC,8BAA8B,GAAG,QAAQ,CAAC;IACjD,CAAC;IAED,WAAW;QACT,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;IAED,kBAAkB;IAClB,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC;IAChC,CAAC;IAED,kBAAkB;IAClB,gCAAgC;QAC9B,OAAO,IAAI,CAAC,6BAA6B,CAAC,GAAG,EAAE,CAAC;IAClD,CAAC;IAED,kBAAkB;IAClB,2BAA2B;QACzB,OAAO,IAAI,CAAC,wBAAwB,CAAC,GAAG,EAAE,CAAC;IAC7C,CAAC;IAED,kBAAkB;IAClB,yCAAyC;QAIvC,OAAO,IAAI,CAAC,sCAAsC,CAAC,GAAG,EAAE,CAAC;IAC3D,CAAC;IAED,kBAAkB;IAClB,gCAAgC;QAC9B,OAAO,IAAI,CAAC,6BAA6B,CAAC,GAAG,EAAE,CAAC;IAClD,CAAC;IAED,kBAAkB;IAClB,KAAK,CAAC,cAAc;QAClB,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC;IAChC,CAAC;IAED,kBAAkB;IAClB,KAAK,CAAC,WAAW;QACf,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC;IAC7B,CAAC;IAED,kBAAkB;IAClB,KAAK,CAAC,UAAU;QACd,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QAC/B,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QAC5B,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;CACF"}
+{"version":3,"file":"client-manager.js","sourceRoot":"","sources":["../../src/confluent/client-manager.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAgB,OAAO,EAAE,MAAM,gCAAgC,CAAC;AACvE,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AACpE,OAAO,EAGL,oBAAoB,GACrB,MAAM,8BAA8B,CAAC;AAEtC,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACrD,OAAO,YAAwB,MAAM,eAAe,CAAC;AAiErD;;;GAGG;AACH,MAAM,OAAO,oBAAoB;IAGvB,qBAAqB,CAAqB;IAC1C,0BAA0B,CAAqB;IAC/C,mCAAmC,CAAqB;IACxD,8BAA8B,CAAqB;IAC1C,WAAW,CAAe;IAC1B,WAAW,CAAsB;IACjC,WAAW,CAA2B;IACtC,QAAQ,CAA8B;IACtC,6BAA6B,CAE5C;IACe,wBAAwB,CAEvC;IACe,sCAAsC,CAErD;IACe,6BAA6B,CAE5C;IACe,oBAAoB,CAA6B;IAElE;;;OAGG;IACH,YAAY,MAA2B;QACrC,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC;QACpD,IAAI,CAAC,0BAA0B,GAAG,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC;QACzD,IAAI,CAAC,mCAAmC,GAAG,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC;QAC3E,IAAI,CAAC,8BAA8B,GAAG,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC;QAE7D,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC;QAChC,IAAI,CAAC,WAAW,GAAG,IAAI,IAAI,CACzB,GAAG,EAAE,CACH,IAAI,OAAO,CAAC,KAAK,CAAC;YAChB,GAAG,IAAI,CAAC,WAAW;YACnB,OAAO,EAAE;gBACP,MAAM,EAAE,WAAW;gBACnB,oJAAoJ;gBACpJ,8DAA8D;aACxD;SACT,CAAC,CACL,CAAC;QACF,IAAI,CAAC,WAAW,GAAG,IAAI,SAAS,CAC9B,KAAK,IAAI,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC;YAC7C,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;YACtB,OAAO,KAAK,CAAC;QACf,CAAC,EACD,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,CAC9B,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,SAAS,CAC3B,KAAK,IAAI,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;YACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;YACnD,MAAM,QAAQ,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,QAAQ,CAAC;QAClB,CAAC,EACD,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,UAAU,EAAE,CACpC,CAAC;QAEF,IAAI,CAAC,wBAAwB,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE;YAC5C,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,CAAC;gBAChC,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;YAClE,CAAC;YACD,MAAM,CAAC,IAAI,CACT,yDAAyD,IAAI,CAAC,qBAAqB,EAAE,CACtF,CAAC;YACF,MAAM,MAAM,GAAG,YAAY,CAAQ;gBACjC,OAAO,EAAE,IAAI,CAAC,qBAAqB;aACpC,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YACpD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,6BAA6B,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE;YACjD,IAAI,CAAC,IAAI,CAAC,0BAA0B,EAAE,CAAC;gBACrC,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;YACxE,CAAC;YACD,MAAM,CAAC,IAAI,CACT,+DAA+D,IAAI,CAAC,0BAA0B,EAAE,CACjG,CAAC;YACF,MAAM,MAAM,GAAG,YAAY,CAAQ;gBACjC,OAAO,EAAE,IAAI,CAAC,0BAA0B;aACzC,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YACpD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,sCAAsC,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE;YAC1D,IAAI,CAAC,IAAI,CAAC,mCAAmC,EAAE,CAAC;gBAC9C,MAAM,IAAI,KAAK,CACb,8DAA8D,CAC/D,CAAC;YACJ,CAAC;YACD,MAAM,CAAC,IAAI,CACT,yEAAyE,IAAI,CAAC,mCAAmC,EAAE,CACpH,CAAC;YACF,MAAM,MAAM,GAAG,YAAY,CAAQ;gBACjC,OAAO,EAAE,IAAI,CAAC,mCAAmC;aAClD,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;YAC7D,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,6BAA6B,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE;YACjD,IAAI,CAAC,IAAI,CAAC,8BAA8B,EAAE,CAAC;gBACzC,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;YACxE,CAAC;YACD,MAAM,CAAC,IAAI,CACT,+DAA+D,IAAI,CAAC,8BAA8B,EAAE,CACrG,CAAC;YACF,MAAM,MAAM,GAAG,YAAY,CAAQ;gBACjC,OAAO,EAAE,IAAI,CAAC,8BAA8B;aAC7C,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YACpD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,oBAAoB,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE;YACxC,IAAI,CAAC,IAAI,CAAC,mCAAmC,EAAE,CAAC;gBAC9C,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;YAC7D,CAAC;YACD,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC;YACzD,OAAO,IAAI,oBAAoB,CAAC;gBAC9B,QAAQ,EAAE,CAAC,IAAI,CAAC,mCAAmC,CAAC;gBACpD,oBAAoB,EAAE;oBACpB,iBAAiB,EAAE,WAAW;oBAC9B,QAAQ,EAAE,GAAG,MAAM,IAAI,SAAS,EAAE;iBACnC;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,kBAAkB;IAClB,KAAK,CAAC,WAAW,CAAC,SAAkB;QAClC,iDAAiD;QACjD,MAAM,WAAW,GACd,IAAI,CAAC,WAAW,CAAC,UAAU,CAAY,IAAI,eAAe,CAAC;QAC9D,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,WAAW,IAAI,SAAS,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC;QACxE,MAAM,cAAc,GAAG;YACrB,kCAAkC;YAClC,GAAG,IAAI,CAAC,WAAW;YACnB,0BAA0B;YAC1B,UAAU,EAAE,OAAO;YACnB,mBAAmB,EAAE,IAAI,CAAC,WAAW,CAAC,mBAAmB,CAAC,IAAI,UAAU;YACxE,0BAA0B,EACxB,IAAI,CAAC,WAAW,CAAC,0BAA0B,CAAC,IAAI,KAAK;YACvD,oBAAoB,EAAE,IAAI,CAAC,WAAW,CAAC,oBAAoB,CAAC,IAAI,KAAK;SACtE,CAAC;QACF,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;IACzD,CAAC;IAED;;;;OAIG;IACH,6BAA6B,CAAC,QAAgB;QAC5C,IAAI,CAAC,wBAAwB,CAAC,KAAK,EAAE,CAAC;QACtC,IAAI,CAAC,qBAAqB,GAAG,QAAQ,CAAC;IACxC,CAAC;IACD,8BAA8B,CAAC,QAAgB;QAC7C,IAAI,CAAC,6BAA6B,CAAC,KAAK,EAAE,CAAC;QAC3C,IAAI,CAAC,0BAA0B,GAAG,QAAQ,CAAC;IAC7C,CAAC;IACD,uCAAuC,CAAC,QAAgB;QACtD,IAAI,CAAC,sCAAsC,CAAC,KAAK,EAAE,CAAC;QACpD,IAAI,CAAC,mCAAmC,GAAG,QAAQ,CAAC;IACtD,CAAC;IACD,kCAAkC,CAAC,QAAgB;QACjD,IAAI,CAAC,6BAA6B,CAAC,KAAK,EAAE,CAAC;QAC3C,IAAI,CAAC,8BAA8B,GAAG,QAAQ,CAAC;IACjD,CAAC;IAED,kBAAkB;IAClB,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC;IAChC,CAAC;IAED,kBAAkB;IAClB,gCAAgC;QAC9B,OAAO,IAAI,CAAC,6BAA6B,CAAC,GAAG,EAAE,CAAC;IAClD,CAAC;IAED,kBAAkB;IAClB,2BAA2B;QACzB,OAAO,IAAI,CAAC,wBAAwB,CAAC,GAAG,EAAE,CAAC;IAC7C,CAAC;IAED,kBAAkB;IAClB,yCAAyC;QAIvC,OAAO,IAAI,CAAC,sCAAsC,CAAC,GAAG,EAAE,CAAC;IAC3D,CAAC;IAED,kBAAkB;IAClB,gCAAgC;QAC9B,OAAO,IAAI,CAAC,6BAA6B,CAAC,GAAG,EAAE,CAAC;IAClD,CAAC;IAED,kBAAkB;IAClB,KAAK,CAAC,cAAc;QAClB,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC;IAChC,CAAC;IAED,kBAAkB;IAClB,KAAK,CAAC,WAAW;QACf,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC;IAC7B,CAAC;IAED,kBAAkB;IAClB,KAAK,CAAC,UAAU;QACd,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QAC/B,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QAC5B,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED,kBAAkB;IAClB,uBAAuB;QACrB,OAAO,IAAI,CAAC,oBAAoB,CAAC,GAAG,EAAE,CAAC;IACzC,CAAC;CACF"}

package/dist/confluent/helpers.d.ts

@@ -4,8 +4,8 @@ import { Environment } from "../env.js";
  * @param envVarName - The name of the environment variable to check
  * @param errorMessage - The error message to throw if neither parameter nor environment variable exists
  * @param param - Optional parameter value to use instead of environment variable
- * @returns The parameter value or environment variable value
+ * @returns The parameter value or environment variable value with its original type
  * @throws {Error} When neither parameter nor environment variable exists
  */
-export declare const getEnsuredParam: (envVarName: keyof Environment, errorMessage: string, param?: string) => string;
+export declare const getEnsuredParam: <T extends Environment[keyof Environment]>(envVarName: keyof Environment, errorMessage: string, param?: T) => T;
 //# sourceMappingURL=helpers.d.ts.map

package/dist/confluent/helpers.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"helpers.d.ts","sourceRoot":"","sources":["../../src/confluent/helpers.ts"],"names":[],"mappings":"AAAA,OAAY,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE/C;;;;;;;GAOG;AACH,eAAO,MAAM,eAAe,GAC1B,YAAY,MAAM,WAAW,EAC7B,cAAc,MAAM,EACpB,QAAQ,MAAM,WAOf,CAAC"}
+{"version":3,"file":"helpers.d.ts","sourceRoot":"","sources":["../../src/confluent/helpers.ts"],"names":[],"mappings":"AAAA,OAAY,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE/C;;;;;;;GAOG;AACH,eAAO,MAAM,eAAe,GAAI,CAAC,SAAS,WAAW,CAAC,MAAM,WAAW,CAAC,EACtE,YAAY,MAAM,WAAW,EAC7B,cAAc,MAAM,EACpB,QAAQ,CAAC,KACR,CAMF,CAAC"}

package/dist/confluent/helpers.js

@@ -4,7 +4,7 @@ import env from "../env.js";
  * @param envVarName - The name of the environment variable to check
  * @param errorMessage - The error message to throw if neither parameter nor environment variable exists
  * @param param - Optional parameter value to use instead of environment variable
- * @returns The parameter value or environment variable value
+ * @returns The parameter value or environment variable value with its original type
  * @throws {Error} When neither parameter nor environment variable exists
  */
 export const getEnsuredParam = (envVarName, errorMessage, param) => {

package/dist/confluent/helpers.js.map

@@ -1 +1 @@
-{"version":3,"file":"helpers.js","sourceRoot":"","sources":["../../src/confluent/helpers.ts"],"names":[],"mappings":"AAAA,OAAO,GAAoB,MAAM,aAAa,CAAC;AAE/C;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,CAC7B,UAA6B,EAC7B,YAAoB,EACpB,KAAc,EACd,EAAE;IACF,MAAM,UAAU,GAAG,KAAK,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC;IAC5C,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,MAAM,IAAI,KAAK,CAAC,GAAG,YAAY,EAAE,CAAC,CAAC;IACrC,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC"}
+{"version":3,"file":"helpers.js","sourceRoot":"","sources":["../../src/confluent/helpers.ts"],"names":[],"mappings":"AAAA,OAAO,GAAoB,MAAM,aAAa,CAAC;AAE/C;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,CAC7B,UAA6B,EAC7B,YAAoB,EACpB,KAAS,EACN,EAAE;IACL,MAAM,UAAU,GAAG,KAAK,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC;IAC5C,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,MAAM,IAAI,KAAK,CAAC,GAAG,YAAY,EAAE,CAAC,CAAC;IACrC,CAAC;IACD,OAAO,UAAe,CAAC;AACzB,CAAC,CAAC"}