ruby-sdk/lib/mcp/primitive.rb at 87c076465624cdb0b445fdc0e5adf1641d615e0f · modelcontextprotocol/ruby-sdk · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
# frozen_string_literal: true

module MCP
  class Primitive
    class << self
      NOT_SET = Object.new

      attr_reader :title_value
      attr_reader :description_value
      attr_reader :icons_value
      attr_reader :meta_value

      def name_value
        @name_value ||= StringUtils.handle_from_class_name(name)
      end

      def primitive_name(value = NOT_SET)
        if value == NOT_SET
          @name_value
        else
          @name_value = value
        end
      end

      def title(value = NOT_SET)
        if value == NOT_SET
          @title_value
        else
          @title_value = value
        end
      end

      def description(value = NOT_SET)
        if value == NOT_SET
          @description_value
        else
          @description_value = value
        end
      end

      def icons(value = NOT_SET)
        if value == NOT_SET
          @icons_value
        else
          @icons_value = value
        end
      end

      def meta(value = NOT_SET)
        if value == NOT_SET
          @meta_value
        else
          @meta_value = value
        end
      end

      def to_h
        {
          name: name_value,
          title: title_value,
          description: description_value,
          icons: icons_value&.then { |icons| icons.empty? ? nil : icons.map(&:to_h) },
          _meta: meta_value,
        }.compact
      end

      def inherited(subclass)
        super
        subclass.instance_variable_set(:@name_value, nil)
        subclass.instance_variable_set(:@title_value, nil)
        subclass.instance_variable_set(:@description_value, nil)
        subclass.instance_variable_set(:@icons_value, nil)
        subclass.instance_variable_set(:@meta_value, nil)
      end

      def define(name: nil, title: nil, description: nil, icons: [], meta: nil, &block)
        Class.new(self) do
          primitive_name name
          title title
          description description
          icons icons
          meta meta
          class_exec(&block) if block
        end
      end
    end

    attr_reader :name, :title, :description, :icons, :meta

    def initialize(name:, title: nil, description: nil, icons: [], meta: nil)
      @name = name
      @title = title
      @description = description
      @icons = icons
      @meta = meta
    end

    def to_h
      {
        name: name,
        title: title,
        description: description,
        icons: icons&.then { |icons| icons.empty? ? nil : icons.map(&:to_h) },
        _meta: meta,
      }.compact
    end
  end
end