summaryrefslogtreecommitdiff
path: root/lib/prism/translation/ripper/sexp.rb
blob: dc26a639a351fd1766d1896233ab7b040459f9a6 (plain)
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
118
119
120
121
122
123
124
125
# frozen_string_literal: true

require_relative "../ripper"

module Prism
  module Translation
    class Ripper
      # This class mirrors the ::Ripper::SexpBuilder subclass of ::Ripper that
      # returns the arrays of [type, *children].
      class SexpBuilder < Ripper
        # :stopdoc:

        attr_reader :error

        private

        def dedent_element(e, width)
          if (n = dedent_string(e[1], width)) > 0
            e[2][1] += n
          end
          e
        end

        def on_heredoc_dedent(val, width)
          sub = proc do |cont|
            cont.map! do |e|
              if Array === e
                case e[0]
                when :@tstring_content
                  e = dedent_element(e, width)
                when /_add\z/
                  e[1] = sub[e[1]]
                end
              elsif String === e
                dedent_string(e, width)
              end
              e
            end
          end
          sub[val]
          val
        end

        events = private_instance_methods(false).grep(/\Aon_/) {$'.to_sym}
        (PARSER_EVENTS - events).each do |event|
          module_eval(<<-End, __FILE__, __LINE__ + 1)
            def on_#{event}(*args)
              args.unshift :#{event}
            end
          End
        end

        SCANNER_EVENTS.each do |event|
          module_eval(<<-End, __FILE__, __LINE__ + 1)
            def on_#{event}(tok)
              [:@#{event}, tok, [lineno(), column()]]
            end
          End
        end

        def on_error(mesg)
          @error = mesg
        end
        remove_method :on_parse_error
        alias on_parse_error on_error
        alias compile_error on_error

        # :startdoc:
      end

      # This class mirrors the ::Ripper::SexpBuilderPP subclass of ::Ripper that
      # returns the same values as ::Ripper::SexpBuilder except with a couple of
      # niceties that flatten linked lists into arrays.
      class SexpBuilderPP < SexpBuilder
        # :stopdoc:

        private

        def on_heredoc_dedent(val, width)
          val.map! do |e|
            next e if Symbol === e and /_content\z/ =~ e
            if Array === e and e[0] == :@tstring_content
              e = dedent_element(e, width)
            elsif String === e
              dedent_string(e, width)
            end
            e
          end
          val
        end

        def _dispatch_event_new
          []
        end

        def _dispatch_event_push(list, item)
          list.push item
          list
        end

        def on_mlhs_paren(list)
          [:mlhs, *list]
        end

        def on_mlhs_add_star(list, star)
          list.push([:rest_param, star])
        end

        def on_mlhs_add_post(list, post)
          list.concat(post)
        end

        PARSER_EVENT_TABLE.each do |event, arity|
          if /_new\z/ =~ event and arity == 0
            alias_method "on_#{event}", :_dispatch_event_new
          elsif /_add\z/ =~ event
            alias_method "on_#{event}", :_dispatch_event_push
          end
        end

        # :startdoc:
      end
    end
  end
end