rbs/lib/rbs.rb at 432c6ac8809bc0a3d5c347030804cbb84641fd0c · ruby/rbs · 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
109
110
111
112
113
114
115
116
117
# frozen_string_literal: true

require "rbs/version"

require "set"
require "json"
# Always require pathname: `Pathname()` (Kernel#Pathname) is only defined once
# pathname is loaded. Guarding on `defined?(Pathname)` is wrong because another
# library (e.g. Bundler) can define the Pathname constant without that method,
# which left RBS::EnvironmentLoader's `Pathname(...)` undefined on JRuby.
require "pathname"
require "pp"
require "logger"
require "tsort"
require "strscan"
require "prism"

require "rbs/errors"
require "rbs/buffer"
require "rbs/namespace"
require "rbs/type_name"
require "rbs/types"
require "rbs/method_type"
require "rbs/file_finder"
require "rbs/ast/type_param"
require "rbs/ast/directives"
require "rbs/ast/declarations"
require "rbs/ast/members"
require "rbs/ast/annotation"
require "rbs/ast/visitor"
require "rbs/ast/ruby/comment_block"
require "rbs/ast/ruby/helpers/constant_helper"
require "rbs/ast/ruby/helpers/location_helper"
require "rbs/ast/ruby/annotations"
require "rbs/ast/ruby/declarations"
require "rbs/ast/ruby/members"
require "rbs/source"
require "rbs/inline_parser"
require "rbs/inline_parser/comment_association"
require "rbs/environment"
require "rbs/environment/use_map"
require "rbs/environment/class_entry"
require "rbs/environment/module_entry"
require "rbs/environment_loader"
require "rbs/builtin_names"
require "rbs/definition"
require "rbs/definition_builder"
require "rbs/definition_builder/ancestor_builder"
require "rbs/definition_builder/method_builder"
require "rbs/diff"
require "rbs/variance_calculator"
require "rbs/substitution"
require "rbs/constant"
require "rbs/resolver/constant_resolver"
require "rbs/resolver/type_name_resolver"
require "rbs/ast/comment"
require "rbs/writer"
require "rbs/rewriter"
require "rbs/prototype/helpers"
require "rbs/prototype/rbi"
require "rbs/prototype/rb"
require "rbs/prototype/runtime"
require "rbs/prototype/node_usage"
require "rbs/environment_walker"
require "rbs/vendorer"
require "rbs/validator"
require "rbs/factory"
require "rbs/repository"
require "rbs/subtractor"
require "rbs/ancestor_graph"
require "rbs/locator"
require "rbs/type_alias_dependency"
require "rbs/type_alias_regularity"
require "rbs/collection"

if RUBY_ENGINE == "jruby"
  # JRuby cannot load the MRI C extension. Run the parser in WebAssembly and
  # provide pure-Ruby implementations of RBS::Location and RBS::Parser instead.
  require "rbs/wasm/location"
  require "rbs/wasm/parser"
else
  require "rbs_extension"
end
require "rbs/parser_aux"
require "rbs/location_aux"

module RBS
  class <<self
    attr_reader :logger_level
    attr_reader :logger_output

    def logger
      @logger ||= Logger.new(logger_output || STDERR, level: logger_level || Logger::WARN, progname: "rbs")
    end

    def logger_output=(val)
      @logger = nil
      @logger_output = val
    end

    def logger_level=(level)
      @logger_level = level
      @logger = nil
    end

    def print_warning()
      @warnings ||= Set[]

      message = yield()

      unless @warnings.include?(message)
        @warnings << message
        logger.warn { message }
      end
    end
  end
end