@mokoconsulting/mokomysql-mcp 0.0.0-stage → 2.5.4

package/source/config.ts

@@ -0,0 +1,97 @@
+/* Copyright (C) 2026 Moko Consulting <hello@mokoconsulting.tech>
+ *
+ * This file is part of a Moko Consulting project.
+ *
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ *
+ * FILE INFORMATION
+ * DEFGROUP: mokomysql.Config
+ * INGROUP: mokomysql
+ * REPO: https://git.mokoconsulting.tech/MokoConsulting/mokomysql-mcp
+ * PATH: /src/config.ts
+ * VERSION: 01.00.00
+ * BRIEF: Configuration loader for MySQL MCP connections
+ */
+
+import { readFile } from 'node:fs/promises';
+import { resolve } from 'node:path';
+import { homedir } from 'node:os';
+import type { MySqlConfig, MySqlConnection } from './types.js';
+import { resolveSecretRefs } from './secrets.js';
+
+const CONFIG_FILENAME = '.mcp_mokomysql.json';
+
+/** Absolute path to the active config file (`MYSQL_MCP_CONFIG` or ~/.mcp_mokomysql.json). */
+export function getConfigPath(): string {
+	return process.env.MYSQL_MCP_CONFIG
+		? resolve(process.env.MYSQL_MCP_CONFIG)
+		: resolve(homedir(), CONFIG_FILENAME);
+}
+
+/**
+ * Absolute path to the sidecar state file — the MCP-managed learned-marker store.
+ * `MYSQL_MCP_STATE` overrides it outright; otherwise it sits beside the config as
+ * `<config>.state.json`, so learned state always follows the config that produced it.
+ */
+export function getStatePath(): string {
+	if (process.env.MYSQL_MCP_STATE) return resolve(process.env.MYSQL_MCP_STATE);
+	return getConfigPath().replace(/\.json$/i, '') + '.state.json';
+}
+
+export async function loadConfig(): Promise<MySqlConfig> {
+	const config_path = getConfigPath();
+
+	try {
+		const raw = await readFile(config_path, 'utf-8');
+		const parsed = JSON.parse(raw) as Partial<MySqlConfig>;
+
+		// Resolve any "file:<path>" secret references (e.g. passwords in ~/.ssh/tokens).
+		await resolveSecretRefs(parsed);
+
+		if (!parsed.connections || Object.keys(parsed.connections).length === 0) {
+			throw new Error('No connections defined in config');
+		}
+
+		return {
+			connections: parsed.connections,
+			defaultConnection: parsed.defaultConnection ?? Object.keys(parsed.connections)[0],
+		};
+	} catch (err) {
+		const message = err instanceof Error ? err.message : String(err);
+		throw new Error(
+			`Failed to load config from ${config_path}: ${message}\n` +
+			`Create ${config_path} — see config.example.json for format.`,
+		);
+	}
+}
+
+/**
+ * Normalize a name for tag matching: lowercase and strip every character
+ * that isn't a letter or digit. This makes "moko.live", "moko-live",
+ * "Moko Live" and "mokolive" all collapse to the same key.
+ */
+export function normalizeTag(value: string): string {
+	return value.toLowerCase().replace(/[^a-z0-9]+/g, '');
+}
+
+export function getConnection(config: MySqlConfig, name?: string): { key: string; conn: MySqlConnection } {
+	const requested = name ?? config.defaultConnection;
+
+	// 1. Exact key match — fast path, preserves prior behavior exactly.
+	const direct = config.connections[requested];
+	if (direct) return { key: requested, conn: direct };
+
+	// 2. Normalized match against connection keys and their tag shortcuts.
+	//    Returns the canonical key, never the alias/tag the caller typed.
+	const target = normalizeTag(requested);
+	for (const [key, conn] of Object.entries(config.connections)) {
+		if (normalizeTag(key) === target) return { key, conn };
+		if (conn.tags?.some((tag) => normalizeTag(tag) === target)) return { key, conn };
+	}
+
+	// 3. Not found — list keys plus their tags so the caller can self-correct.
+	const available = Object.entries(config.connections)
+		.map(([key, conn]) => (conn.tags?.length ? `${key} [${conn.tags.join(', ')}]` : key))
+		.join(', ');
+	throw new Error(`Connection "${requested}" not found. Available: ${available}`);
+}