import type { ServerMode, ToolSet } from '../types/index.js';
import { TOOL_SETS } from '../constants/toolSets.js';
import type { ServerModeEnforcer } from '../server-mode-enforcer/ServerModeEnforcer.js';
import type { ModuleLoader } from 'toolception';
/**
* Session context configuration for per-session token support
* Enables users to provide their own FMP_ACCESS_TOKEN via query param
*/
export interface SessionContextConfig {
enabled: boolean;
queryParam: {
name: string;
encoding: 'base64' | 'json';
allowedKeys: string[];
};
merge: 'shallow' | 'deep';
}
/**
* Toolception configuration options
* Based on toolception's CreateMcpServerOptions type
*/
export interface ToolceptionConfig {
catalog: ToolSetCatalog;
moduleLoaders: Record<string, ModuleLoader>;
startup: {
mode: 'DYNAMIC' | 'STATIC';
toolsets?: string[] | 'ALL'; // Changed from initialToolsets in toolception 0.5.1 - matches toolception's type
};
context: {
accessToken?: string;
[key: string]: any;
};
sessionContext?: SessionContextConfig;
exposurePolicy: {
namespaceToolsWithSetKey: boolean;
maxActiveToolsets?: number;
allowlist?: ToolSet[];
};
}
/**
* Toolception toolset catalog
*/
export interface ToolSetCatalog {
[key: string]: {
name: string;
description: string;
decisionCriteria?: string;
modules?: string[];
};
}
/**
* Maps FMP's ServerMode to toolception configuration
*
* This class translates between FMP's three modes (DYNAMIC_TOOL_DISCOVERY,
* STATIC_TOOL_SETS, ALL_TOOLS) and toolception's configuration format.
*/
export class ModeConfigMapper {
/**
* Convert FMP ServerMode to toolception configuration
*
* @param mode - FMP server mode
* @param enforcer - Server mode enforcer instance
* @param accessToken - FMP API access token
* @param moduleLoaders - Record of module name to loader function
* @returns Toolception configuration object
*/
static toToolceptionConfig(
mode: ServerMode,
enforcer: ServerModeEnforcer,
accessToken?: string,
moduleLoaders?: Record<string, ModuleLoader>
): ToolceptionConfig {
const catalog = this.buildCatalog();
const loaders = moduleLoaders || {};
switch (mode) {
case 'DYNAMIC_TOOL_DISCOVERY':
return {
catalog,
moduleLoaders: loaders,
startup: {
mode: 'DYNAMIC'
},
context: {
accessToken
},
sessionContext: this.buildSessionContextConfig(),
exposurePolicy: {
namespaceToolsWithSetKey: false, // Flat namespace per user requirement
maxActiveToolsets: undefined // No limit
}
};
case 'STATIC_TOOL_SETS': {
// Get toolsets from enforcer (server-level configuration)
const toolsets = this.resolveToolSets(enforcer);
return {
catalog,
moduleLoaders: loaders,
startup: {
mode: 'STATIC',
toolsets: toolsets // Changed from initialToolsets in toolception 0.5.1
},
context: {
accessToken
},
sessionContext: this.buildSessionContextConfig(),
exposurePolicy: {
namespaceToolsWithSetKey: false,
allowlist: toolsets
}
};
}
case 'ALL_TOOLS': {
// Load all available toolsets using the "ALL" shorthand
return {
catalog,
moduleLoaders: loaders,
startup: {
mode: 'STATIC',
toolsets: 'ALL' // Changed from initialToolsets array in toolception 0.5.1
},
context: {
accessToken
},
sessionContext: this.buildSessionContextConfig(),
exposurePolicy: {
namespaceToolsWithSetKey: false
}
};
}
default:
throw new Error(`Unknown server mode: ${mode}`);
}
}
/**
* Build session context configuration for per-session token support
* Only allows FMP_ACCESS_TOKEN to be overridden per-session
*/
private static buildSessionContextConfig(): SessionContextConfig {
return {
enabled: true,
queryParam: {
name: 'config',
encoding: 'base64',
allowedKeys: ['FMP_ACCESS_TOKEN'],
},
merge: 'shallow',
};
}
/**
* Build toolception catalog from FMP TOOL_SETS
*/
private static buildCatalog(): ToolSetCatalog {
const catalog: ToolSetCatalog = {};
for (const [key, definition] of Object.entries(TOOL_SETS)) {
catalog[key] = {
name: definition.name,
description: definition.description,
decisionCriteria: definition.decisionCriteria,
modules: definition.modules
};
}
return catalog;
}
/**
* Resolve tool sets from enforcer (server-level configuration only)
*/
private static resolveToolSets(enforcer: ServerModeEnforcer): ToolSet[] {
if (enforcer.serverModeOverride === 'STATIC_TOOL_SETS') {
return enforcer.toolSets;
}
return [];
}
}
