poker-mcp 1.0.0

package/src/mcp/middleware/errorHandler.js

@@ -0,0 +1,49 @@
+// mcp/middleware/errorHandler.js
+import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js';
+import { ValidationError, PhysicsError, CalculationError } from '../../utils/errors.js';
+import { PokerMcpError } from '../../utils/mcpErrors.js';
+import { logger } from '../../utils/logger.js';
+
+export function handleMcpError(error, context = {}) {
+  logger.error('MCP操作エラー', { error: error.message, context });
+
+  // PokerMcpErrorは既に適切なMcpErrorなのでそのまま返す
+  if (error instanceof PokerMcpError) {
+    return error;
+  }
+  
+  // CalculationError の処理 - MCPエラーコードが指定されている場合は使用
+  if (error instanceof CalculationError) {
+    const mcpCode = error.mcpErrorCode || ErrorCode.InternalError;
+    return new McpError(mcpCode, `計算エラー: ${error.message}`, error.context);
+  }
+  
+  // 従来の ValidationError を適切な PokerMcpError に変換
+  if (error instanceof ValidationError) {
+    return PokerMcpError.validationError(error.message, error.field, error.value);
+  }
+  
+  // PhysicsError の処理
+  if (error instanceof PhysicsError) {
+    return PokerMcpError.validationError(`物理パラメータエラー: ${error.message}`, error.field, error.value);
+  }
+  
+  // 既存のMcpError
+  if (error instanceof McpError) {
+    return error;
+  }
+  
+  // その他の予期しないエラー
+  return new McpError(ErrorCode.InternalError, `内部エラー: ${error.message}`);
+}
+
+export function safeExecute(fn, context = {}) {
+  return async (request) => {
+    try {
+      const result = await fn(request);
+      return { content: [{ type: 'text', text: JSON.stringify(result, null, 2) }] };
+    } catch (error) {
+      throw handleMcpError(error, context);
+    }
+  };
+}

package/src/mcp/middleware/requestValidator.js

@@ -0,0 +1,129 @@
+// mcp/middleware/requestValidator.js
+import { ValidationError } from '../../utils/errors.js';
+import { MaterialAlternatives } from '../../utils/MaterialAlternatives.js';
+
+export function validateBodyRequest(args) {
+  if (!args.name || typeof args.name !== 'string') {
+    throw new ValidationError('立体名は必須です', 'name', args.name);
+  }
+  
+  if (!args.type || typeof args.type !== 'string') {
+    throw new ValidationError('立体タイプは必須です', 'type', args.type);
+  }
+  
+  const validTypes = ['SPH', 'RCC', 'RPP', 'BOX', 'CMB', 'TOR', 'ELL', 'REC', 'TRC', 'WED'];
+  if (!validTypes.includes(args.type)) {
+    throw new ValidationError(`無効な立体タイプ: ${args.type}`, 'type', args.type);
+  }
+}
+
+export function validateZoneRequest(args) {
+  if (!args.body_name || typeof args.body_name !== 'string') {
+    throw new ValidationError('立体名は必須です', 'body_name', args.body_name);
+  }
+  
+  if (!args.material || typeof args.material !== 'string') {
+    throw new ValidationError('材料名は必須です', 'material', args.material);
+  }
+  
+  // 材料代替機能付きバリデーション
+  try {
+    MaterialAlternatives.validateMaterialWithSuggestion(args.material, 'material');
+  } catch (error) {
+    // MaterialAlternativesからのエラーをValidationErrorに変換
+    throw new ValidationError(error.message, 'material', args.material);
+  }
+}
+
+export function validateTransformRequest(args) {
+  if (!args.name || typeof args.name !== 'string') {
+    throw new ValidationError('変換名は必須です', 'name', args.name);
+  }
+  
+  if (!args.operations || !Array.isArray(args.operations)) {
+    throw new ValidationError('操作配列は必須です', 'operations', args.operations);
+  }
+}
+
+export function validateSourceRequest(args) {
+  if (!args.name || typeof args.name !== 'string') {
+    throw new ValidationError('線源名は必須です', 'name', args.name);
+  }
+  
+  if (!args.type || typeof args.type !== 'string') {
+    throw new ValidationError('線源タイプは必須です', 'type', args.type);
+  }
+  
+  if (!args.inventory || !Array.isArray(args.inventory)) {
+    throw new ValidationError('インベントリは必須の配列です', 'inventory', args.inventory);
+  }
+}
+
+export function validateBuildupFactorRequest(args) {
+  if (!args.material || typeof args.material !== 'string') {
+    throw new ValidationError('材料名は必須です', 'material', args.material);
+  }
+  
+  // スラント補正パラメータの必須チェック
+  if (args.use_slant_correction === undefined) {
+    throw new ValidationError(
+      'use_slant_correction parameter is required for high-precision radiation shielding calculation',
+      'use_slant_correction',
+      args.use_slant_correction
+    );
+  }
+  
+  if (typeof args.use_slant_correction !== 'boolean') {
+    throw new ValidationError(
+      'use_slant_correction must be boolean',
+      'use_slant_correction', 
+      args.use_slant_correction
+    );
+  }
+  
+  // 有限媒体補正パラメータの必須チェック
+  if (args.use_finite_medium_correction === undefined) {
+    throw new ValidationError(
+      'use_finite_medium_correction parameter is required for accurate boundary effect analysis',
+      'use_finite_medium_correction',
+      args.use_finite_medium_correction
+    );
+  }
+  
+  if (typeof args.use_finite_medium_correction !== 'boolean') {
+    throw new ValidationError(
+      'use_finite_medium_correction must be boolean',
+      'use_finite_medium_correction',
+      args.use_finite_medium_correction
+    );
+  }
+}
+
+export function validateUpdateSourceRequest(args) {
+  if (!args.name || typeof args.name !== 'string') {
+    throw new ValidationError('線源名は必須です', 'name', args.name);
+  }
+  
+  // 更新可能なフィールドのみチェック
+  const allowedFields = ['name', 'position', 'inventory', 'cutoff_rate'];
+  const providedFields = Object.keys(args);
+  const invalidFields = providedFields.filter(field => !allowedFields.includes(field));
+  
+  if (invalidFields.length > 0) {
+    throw new ValidationError(`更新不可なフィールド: ${invalidFields.join(', ')}`, 'fields', invalidFields);
+  }
+  
+  if (args.inventory && !Array.isArray(args.inventory)) {
+    throw new ValidationError('インベントリは配列で指定してください', 'inventory', args.inventory);
+  }
+  
+  if (args.cutoff_rate !== undefined && (typeof args.cutoff_rate !== 'number' || args.cutoff_rate < 0)) {
+    throw new ValidationError('cutoff_rateは0以上の数値で指定してください', 'cutoff_rate', args.cutoff_rate);
+  }
+}
+
+export function validateDeleteSourceRequest(args) {
+  if (!args.name || typeof args.name !== 'string') {
+    throw new ValidationError('線源名は必須です', 'name', args.name);
+  }
+}

package/src/mcp/server.js

@@ -0,0 +1,76 @@
+// mcp/server.js
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { CallToolRequestSchema, ListToolsRequestSchema, ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';
+import { TaskManager } from '../services/TaskManager.js';
+import { allTools } from './tools/index.js';
+import { createAllHandlers } from './handlers/index.js';
+import { safeExecute } from './middleware/errorHandler.js';
+import { logger } from '../utils/logger.js';
+
+export class PokerMcpServer {
+  constructor(yamlFile = 'tasks/poker.yaml', pendingFile = 'tasks/pending_changes.json') {
+    this.server = new Server(
+      {
+        name: 'poker-mcp',
+        version: '1.0.0',
+      },
+      {
+        capabilities: {
+          tools: {},
+        },
+      }
+    );
+    
+    this.taskManager = new TaskManager(yamlFile, pendingFile);
+    this.handlers = null;
+  }
+
+  async initialize() {
+    try {
+      // TaskManager初期化
+      await this.taskManager.initialize();
+      
+      // ハンドラー作成
+      this.handlers = createAllHandlers(this.taskManager);
+      
+      // ツール一覧ハンドラー
+      this.server.setRequestHandler(ListToolsRequestSchema, async () => {
+        return { tools: allTools };
+      });
+
+      // ツール実行ハンドラー
+      this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+        const { name, arguments: args } = request.params;
+        
+        logger.info(`MCP Tool実行: ${name}`, { args });
+        
+        // ハンドラー名をツール名から生成（プレフィックス除去）
+        const handlerName = name.replace('poker_', '');
+        
+        const handler = this.handlers[handlerName];
+        if (!handler) {
+          throw new McpError(ErrorCode.MethodNotFound, `Unknown tool: ${name}`);
+        }
+        
+        return await safeExecute(async () => handler(args), { tool: name })();
+      });
+      
+      logger.info('PokerMcpServer初期化完了');
+    } catch (error) {
+      logger.error('PokerMcpServer初期化エラー', { error: error.message });
+      throw error;
+    }
+  }
+
+  async start() {
+    try {
+      const transport = new StdioServerTransport();
+      await this.server.connect(transport);
+      logger.info('Poker MCP Server started on stdio');
+    } catch (error) {
+      logger.error('MCPサーバー開始エラー', { error: error.message });
+      throw error;
+    }
+  }
+}

package/src/mcp/tools/bodyTools.js

@@ -0,0 +1,185 @@
+// mcp/tools/bodyTools.js
+export const bodyTools = [
+  {
+    name: 'poker_proposeBody',
+    description: '新しい3D立体を提案します（自動バックアップ付き）',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        name: {
+          type: 'string',
+          description: '立体の一意な名前'
+        },
+        type: {
+          type: 'string',
+          description: '立体タイプ',
+          enum: ['SPH', 'RCC', 'RPP', 'BOX', 'CMB', 'TOR', 'ELL', 'REC', 'TRC', 'WED']
+        },
+        center: {
+          type: 'string',
+          description: '中心座標 (x y z形式) - SPH,ELL,TOR用'
+        },
+        radius: {
+          type: 'number',
+          description: '半径 - SPH, RCC用',
+          minimum: 0.001,
+          maximum: 10000
+        },
+        bottom_center: {
+          type: 'string',
+          description: '底面中心座標 (x y z形式) - RCC,TRC,REC用'
+        },
+        height_vector: {
+          type: 'string',
+          description: '高さベクトル (x y z形式) - RCC,TRC,REC,WED用'
+        },
+        min: {
+          type: 'string',
+          description: '最小座標 (x y z形式) - RPP用'
+        },
+        max: {
+          type: 'string',
+          description: '最大座標 (x y z形式) - RPP用'
+        },
+        vertex: {
+          type: 'string',
+          description: '頂点座標 (x y z形式) - BOX,WED用'
+        },
+        edge_1: {
+          type: 'string',
+          description: 'エッジ1ベクトル (x y z形式) - BOX用'
+        },
+        edge_2: {
+          type: 'string',
+          description: 'エッジ2ベクトル (x y z形式) - BOX用'
+        },
+        edge_3: {
+          type: 'string',
+          description: 'エッジ3ベクトル (x y z形式) - BOX用'
+        },
+        expression: {
+          type: 'string',
+          description: '組み合わせ式 - CMB用'
+        },
+        // TOR用パラメータ
+        normal: {
+          type: 'string',
+          description: '法線ベクトル (x y z形式) - TOR用'
+        },
+        major_radius: {
+          type: 'number',
+          description: '主半径 - TOR用',
+          minimum: 0.001,
+          maximum: 10000
+        },
+        minor_radius_horizontal: {
+          type: 'number',
+          description: '水平方向副半径 - TOR用',
+          minimum: 0.001,
+          maximum: 10000
+        },
+        minor_radius_vertical: {
+          type: 'number',
+          description: '垂直方向副半径 - TOR用',
+          minimum: 0.001,
+          maximum: 10000
+        },
+        // ELL用パラメータ
+        radius_vector_1: {
+          type: 'string',
+          description: '半径ベクトル1 (x y z形式) - ELL,REC用'
+        },
+        radius_vector_2: {
+          type: 'string',
+          description: '半径ベクトル2 (x y z形式) - ELL,REC用'
+        },
+        radius_vector_3: {
+          type: 'string',
+          description: '半径ベクトル3 (x y z形式) - ELL用'
+        },
+        // TRC用パラメータ
+        bottom_radius: {
+          type: 'number',
+          description: '底面半径 - TRC用',
+          minimum: 0.001,
+          maximum: 10000
+        },
+        top_radius: {
+          type: 'number',
+          description: '上面半径 - TRC用',
+          minimum: 0.001,
+          maximum: 10000
+        },
+        // WED用パラメータ
+        width_vector: {
+          type: 'string',
+          description: '幅ベクトル (x y z形式) - WED用'
+        },
+        depth_vector: {
+          type: 'string',
+          description: '奥行きベクトル (x y z形式) - WED用'
+        },
+        transform: {
+          type: 'string',
+          description: '適用する変換名'
+        }
+      },
+      required: ['name', 'type']
+    }
+  },
+  {
+    name: 'poker_updateBody',
+    description: '既存立体のパラメータを更新します',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        name: {
+          type: 'string',
+          description: '更新する立体名'
+        },
+        center: { type: 'string', description: '新しい中心座標 (x y z形式)' },
+        radius: { type: 'number', description: '新しい半径', minimum: 0.001, maximum: 10000 },
+        bottom_center: { type: 'string', description: '新しい底面中心座標 (x y z形式)' },
+        height_vector: { type: 'string', description: '新しい高さベクトル (x y z形式)' },
+        min: { type: 'string', description: '新しい最小座標 (x y z形式)' },
+        max: { type: 'string', description: '新しい最大座標 (x y z形式)' },
+        vertex: { type: 'string', description: '新しい頂点座標 (x y z形式)' },
+        edge_1: { type: 'string', description: '新しいエッジ1ベクトル (x y z形式)' },
+        edge_2: { type: 'string', description: '新しいエッジ2ベクトル (x y z形式)' },
+        edge_3: { type: 'string', description: '新しいエッジ3ベクトル (x y z形式)' },
+        expression: { type: 'string', description: '新しい組み合わせ式' },
+        // TOR用パラメータ
+        normal: { type: 'string', description: '新しい法線ベクトル (x y z形式)' },
+        major_radius: { type: 'number', description: '新しい主半径', minimum: 0.001, maximum: 10000 },
+        minor_radius_horizontal: { type: 'number', description: '新しい水平方向副半径', minimum: 0.001, maximum: 10000 },
+        minor_radius_vertical: { type: 'number', description: '新しい垂直方向副半径', minimum: 0.001, maximum: 10000 },
+        // ELL用パラメータ
+        radius_vector_1: { type: 'string', description: '新しい半径ベクトル1 (x y z形式)' },
+        radius_vector_2: { type: 'string', description: '新しい半径ベクトル2 (x y z形式)' },
+        radius_vector_3: { type: 'string', description: '新しい半径ベクトル3 (x y z形式)' },
+        // TRC用パラメータ
+        bottom_radius: { type: 'number', description: '新しい底面半径', minimum: 0.001, maximum: 10000 },
+        top_radius: { type: 'number', description: '新しい上面半径', minimum: 0.001, maximum: 10000 },
+        // WED用パラメータ
+        width_vector: { type: 'string', description: '新しい幅ベクトル (x y z形式)' },
+        depth_vector: { type: 'string', description: '新しい奥行きベクトル (x y z形式)' },
+        transform: { type: 'string', description: '新しい変換名' }
+      },
+      required: ['name']
+    }
+  },
+  {
+    name: 'poker_deleteBody',
+    description: '立体を削除します（依存関係チェック付き）',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        name: {
+          type: 'string',
+          description: '削除する立体名'
+        }
+      },
+      required: ['name']
+    }
+  }
+];

package/src/mcp/tools/buildupFactorTools.js

@@ -0,0 +1,82 @@
+// mcp/tools/buildupFactorTools.js
+export const buildupFactorTools = [
+  {
+    name: 'poker_proposeBuildupFactor',
+    description: 'ビルドアップ係数を提案します',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        material: {
+          type: 'string',
+          description: '材料名'
+        },
+        use_slant_correction: {
+          type: 'boolean',
+          description: 'スラント補正を使用するか',
+          default: false
+        },
+        use_finite_medium_correction: {
+          type: 'boolean',
+          description: '有限媒体補正を使用するか',
+          default: false
+        }
+      },
+      required: ['material', 'use_slant_correction', 'use_finite_medium_correction']
+    }
+  },
+  {
+    name: 'poker_updateBuildupFactor',
+    description: '既存ビルドアップ係数の設定を更新します',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        material: {
+          type: 'string',
+          description: '更新する材料名'
+        },
+        use_slant_correction: {
+          type: 'boolean',
+          description: '新しいスラント補正設定'
+        },
+        use_finite_medium_correction: {
+          type: 'boolean',
+          description: '新しい有限媒体補正設定'
+        }
+      },
+      required: ['material']
+    }
+  },
+  {
+    name: 'poker_deleteBuildupFactor',
+    description: 'ビルドアップ係数を削除します',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        material: {
+          type: 'string',
+          description: '削除する材料名'
+        }
+      },
+      required: ['material']
+    }
+  },
+  {
+    name: 'poker_changeOrderBuildupFactor',
+    description: 'ビルドアップ係数の順序を変更します',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        material: {
+          type: 'string',
+          description: '順序を変更する材料名'
+        },
+        newIndex: {
+          type: 'integer',
+          description: '新しいインデックス位置',
+          minimum: 0
+        }
+      },
+      required: ['material', 'newIndex']
+    }
+  }
+];

package/src/mcp/tools/calculationTools.js

@@ -0,0 +1,58 @@
+// mcp/tools/calculationTools.js
+export const calculationTools = [
+  {
+    name: 'poker_executeCalculation',
+    description: '作成したYAMLファイルを使用してpoker_cuiで放射線遮蔽計算を実行します',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        yaml_file: {
+          type: 'string',
+          description: '計算に使用するYAMLファイル名（拡張子.yamlを含む）',
+          pattern: '^[a-zA-Z0-9_\\-\\.]+\\.(yaml|yml)$'
+        },
+        summary_options: {
+          type: 'object',
+          description: 'サマリー出力オプション（注意: show_source_dataとshow_total_doseの少なくとも一方は必須です）',
+          properties: {
+            show_parameters: {
+              type: 'boolean',
+              description: '入力パラメータをサマリーに表示（-pオプション）',
+              default: false
+            },
+            show_source_data: {
+              type: 'boolean',
+              description: '各線源による計算データ及び線量をサマリーに表示（-sオプション）',
+              default: true
+            },
+            show_total_dose: {
+              type: 'boolean',
+              description: '各検出器に対する全線源からの総和線量をサマリーに表示（-tオプション）',
+              default: true
+            }
+          },
+          additionalProperties: false
+        },
+        output_files: {
+          type: 'object',
+          description: '出力ファイル指定（入力YAMLファイルと同じフォルダに自動作成、全てYAML形式）',
+          properties: {
+            summary_file: {
+              type: 'string',
+              description: '出力サマリーファイル名（-oオプション、ファイル名のみ指定、自動的に入力ファイルと同じディレクトリに配置）',
+              pattern: '^[a-zA-Z0-9_\\-\\.]+\\.(yaml|yml)$'
+            },
+            dose_file: {
+              type: 'string',
+              description: '線量ファイル名（-dオプション、ファイル名のみ指定、自動的に入力ファイルと同じディレクトリに配置）',
+              pattern: '^[a-zA-Z0-9_\\-\\.]+\\.(yaml|yml)$'
+            }
+          },
+          additionalProperties: false
+        }
+      },
+      required: ['yaml_file'],
+      additionalProperties: false
+    }
+  }
+];

package/src/mcp/tools/commonTools.js

@@ -0,0 +1,22 @@
+// mcp/tools/commonTools.js
+export const commonTools = [
+  {
+    name: 'poker_applyChanges',
+    description: '保留中の全変更を実際のYAMLファイルに適用します（自動バックアップ実行）',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        force: {
+          type: 'boolean',
+          description: '強制適用フラグ（警告を無視）',
+          default: false
+        },
+        backup_comment: {
+          type: 'string',
+          description: 'バックアップのコメント',
+          maxLength: 200
+        }
+      }
+    }
+  }
+];