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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
|
# frozen_string_literal: false
#
# xmp.rb - irb version of gotoken xmp
# by Keiju ISHITSUKA(Nippon Rational Inc.)
#
require_relative "../irb"
require_relative "frame"
# An example printer for irb.
#
# It's much like the standard library PrettyPrint, that shows the value of each
# expression as it runs.
#
# In order to use this library, you must first require it:
#
# require 'irb/xmp'
#
# Now, you can take advantage of the Object#xmp convenience method.
#
# xmp <<END
# foo = "bar"
# baz = 42
# END
# #=> foo = "bar"
# #==>"bar"
# #=> baz = 42
# #==>42
#
# You can also create an XMP object, with an optional binding to print
# expressions in the given binding:
#
# ctx = binding
# x = XMP.new ctx
# x.puts
# #=> today = "a good day"
# #==>"a good day"
# ctx.eval 'today # is what?'
# #=> "a good day"
class XMP
# Creates a new XMP object.
#
# The top-level binding or, optional +bind+ parameter will be used when
# creating the workspace. See WorkSpace.new for more information.
#
# This uses the +:XMP+ prompt mode, see IRB@Customizing+the+IRB+Prompt for
# full detail.
def initialize(bind = nil)
IRB.init_config(nil)
IRB.conf[:PROMPT_MODE] = :XMP
bind = IRB::Frame.top(1) unless bind
ws = IRB::WorkSpace.new(bind)
@io = StringInputMethod.new
@irb = IRB::Irb.new(ws, @io)
@irb.context.ignore_sigint = false
IRB.conf[:MAIN_CONTEXT] = @irb.context
end
# Evaluates the given +exps+, for example:
#
# require 'irb/xmp'
# x = XMP.new
#
# x.puts '{:a => 1, :b => 2, :c => 3}'
# #=> {:a => 1, :b => 2, :c => 3}
# # ==>{:a=>1, :b=>2, :c=>3}
# x.puts 'foo = "bar"'
# # => foo = "bar"
# # ==>"bar"
def puts(exps)
@io.puts exps
if @irb.context.ignore_sigint
begin
trap_proc_b = trap("SIGINT"){@irb.signal_handle}
catch(:IRB_EXIT) do
@irb.eval_input
end
ensure
trap("SIGINT", trap_proc_b)
end
else
catch(:IRB_EXIT) do
@irb.eval_input
end
end
end
# A custom InputMethod class used by XMP for evaluating string io.
class StringInputMethod < IRB::InputMethod
# Creates a new StringInputMethod object
def initialize
super
@exps = []
end
# Whether there are any expressions left in this printer.
def eof?
@exps.empty?
end
# Reads the next expression from this printer.
#
# See IO#gets for more information.
def gets
while l = @exps.shift
next if /^\s+$/ =~ l
l.concat "\n"
print @prompt, l
break
end
l
end
# Concatenates all expressions in this printer, separated by newlines.
#
# An Encoding::CompatibilityError is raised of the given +exps+'s encoding
# doesn't match the previous expression evaluated.
def puts(exps)
if @encoding and exps.encoding != @encoding
enc = Encoding.compatible?(@exps.join("\n"), exps)
if enc.nil?
raise Encoding::CompatibilityError, "Encoding in which the passed expression is encoded is not compatible to the preceding's one"
else
@encoding = enc
end
else
@encoding = exps.encoding
end
@exps.concat exps.split(/\n/)
end
# Returns the encoding of last expression printed by #puts.
attr_reader :encoding
end
end
# A convenience method that's only available when the you require the IRB::XMP standard library.
#
# Creates a new XMP object, using the given expressions as the +exps+
# parameter, and optional binding as +bind+ or uses the top-level binding. Then
# evaluates the given expressions using the +:XMP+ prompt mode.
#
# For example:
#
# require 'irb/xmp'
# ctx = binding
# xmp 'foo = "bar"', ctx
# #=> foo = "bar"
# #==>"bar"
# ctx.eval 'foo'
# #=> "bar"
#
# See XMP.new for more information.
def xmp(exps, bind = nil)
bind = IRB::Frame.top(1) unless bind
xmp = XMP.new(bind)
xmp.puts exps
xmp
end
|