Class: RBS::Annotate::RDocSource

Inherits:
Object
  • Object
show all
Defined in:
lib/rbs/annotate/rdoc_source.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ RDocSource

Returns a new instance of RDocSource.



12
13
14
15
16
17
18
19
20
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 12

def initialize
  self.with_system_dir = true
  self.with_gems_dir = false
  self.with_site_dir = false
  self.with_home_dir = false

  @extra_dirs = []
  @stores = []
end

Instance Attribute Details

#extra_dirs ⇒ Object (readonly)

Returns the value of attribute extra_dirs.



8
9
10
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 8

def extra_dirs
  @extra_dirs
end

#stores ⇒ Object (readonly)

Returns the value of attribute stores.



10
11
12
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 10

def stores
  @stores
end

#with_gems_dir ⇒ Object

Returns the value of attribute with_gems_dir.



6
7
8
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 6

def with_gems_dir
  @with_gems_dir
end

#with_home_dir ⇒ Object

Returns the value of attribute with_home_dir.



6
7
8
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 6

def with_home_dir
  @with_home_dir
end

#with_site_dir ⇒ Object

Returns the value of attribute with_site_dir.



6
7
8
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 6

def with_site_dir
  @with_site_dir
end

#with_system_dir ⇒ Object

Returns the value of attribute with_system_dir.



6
7
8
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 6

def with_system_dir
  @with_system_dir
end

Instance Method Details

#class_docs(typename) ⇒ Object 



64
65
66
67
68
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 64

def class_docs(typename)
  if classes = find_class(typename)
    classes.map {|klass| klass.comment }
  end
end

#docs ⇒ Object 



47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 47

def docs
  if ds = yield
    unless ds.empty?
      ds.map do |code_object|
        case comment = code_object.comment
        when String
          raise
        when RDoc::Comment
          comment.parse
        when RDoc::Markup::Document
          comment
        end
      end
    end
  end
end

#find_attribute(typename, name, singleton:) ⇒ Object 



117
118
119
120
121
122
123
124
125
126
127
128
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 117

def find_attribute(typename, name, singleton:)
  if klasss = find_class(typename)
    # @type var attrs: Array[RDoc::Attr]
    attrs = []

    klasss.each do |kls|
      attrs.concat(kls.attributes.select {|attr| attr.singleton == singleton && attr.name == name.to_s })
    end

    attrs unless attrs.empty?
  end
end

#find_class(typename) ⇒ Object 



33
34
35
36
37
38
39
40
41
42
43
44
45
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 33

def find_class(typename)
  classes = []

  @stores.each do |store|
    if klass = store.find_class_or_module(typename.relative!.to_s)
      classes << klass
    end
  end

  unless classes.empty?
    classes
  end
end

#find_const(const_name) ⇒ Object 



70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 70

def find_const(const_name)
  namespace =
    if const_name.namespace.empty?
      TypeName("::Object")
    else
      const_name.namespace.to_type_name
    end

  if classes = find_class(namespace)
    # @type var consts: Array[RDoc::Constant]
    consts = []

    classes.each do |klass|
      if const = klass.constants.find {|c| c.name == const_name.name.to_s }
        consts << const
      end
    end

    unless consts.empty?
      consts
    end
  end
end

#find_method(typename, instance_method: nil, singleton_method: nil) ⇒ Object 



94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 94

def find_method(typename, instance_method: nil, singleton_method: nil)
  if classes = find_class(typename)
    # @type var methods: Array[RDoc::AnyMethod]
    methods = []

    classes.each do |klass|
      klass.method_list.each do |method|
        if instance_method && !method.singleton && method.name == instance_method.to_s
          methods << method
        end

        if singleton_method && method.singleton && method.name == singleton_method.to_s
          methods << method
        end
      end
    end

    unless methods.empty?
      methods
    end
  end
end

#load ⇒ Object 



22
23
24
25
26
27
28
29
30
31
 
# File 'lib/rbs/annotate/rdoc_source.rb', line 22

def load
  @stores.clear()

  RDoc::RI::Paths.each(with_system_dir, with_site_dir, with_home_dir, with_gems_dir ? :latest : false, *extra_dirs.map(&:to_s)) do |path, type|
    store = RDoc::Store.new(path, type)
    store.load_all

    @stores << store
  end
end