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
165
|
require "neovim/ruby_provider/vim"
require "neovim/ruby_provider/object_ext"
require "neovim/ruby_provider/buffer_ext"
require "neovim/ruby_provider/window_ext"
require "stringio"
module Neovim
# This class is used to define a +Neovim::Plugin+ to act as a backend for the
# +:ruby+, +:rubyfile+, and +:rubydo+ Vim commands. It is autoloaded from
# +nvim+ and not intended to be required directly.
#
# @api private
module RubyProvider
def self.__define_plugin!
Thread.abort_on_exception = true
Neovim.plugin do |plug|
plug.__send__(:script_host!)
__define_setup(plug)
__define_ruby_execute(plug)
__define_ruby_eval(plug)
__define_ruby_execute_file(plug)
__define_ruby_do_range(plug)
__define_ruby_chdir(plug)
end
end
# Define the +DirChanged+ event to update the provider's pwd.
def self.__define_setup(plug)
plug.__send__(:setup) do |client|
begin
cid = client.api.channel_id
client.command("au DirChanged * call rpcrequest(#{cid}, 'ruby_chdir', v:event)")
rescue ArgumentError
# Swallow this exception for now. This means the nvim installation is
# from before DirChanged was implemented.
end
end
end
# Evaluate the provided Ruby code, exposing the +Vim+ constant for
# interactions with the editor.
#
# This is used by the +:ruby+ command.
def self.__define_ruby_execute(plug)
plug.__send__(:rpc, :ruby_execute) do |nvim, ruby|
__wrap_client(nvim) do
eval(ruby, TOPLEVEL_BINDING, "ruby_execute")
end
nil
end
end
private_class_method :__define_ruby_execute
# Evaluate the provided Ruby code, exposing the +Vim+ constant for
# interactions with the editor and returning the value.
#
# This is used by the +:rubyeval+ command.
def self.__define_ruby_eval(plug)
plug.__send__(:rpc, :ruby_eval) do |nvim, ruby|
__wrap_client(nvim) do
eval(ruby, TOPLEVEL_BINDING, "ruby_eval")
end
end
end
private_class_method :__define_ruby_eval
# Evaluate the provided Ruby file, exposing the +Vim+ constant for
# interactions with the editor.
#
# This is used by the +:rubyfile+ command.
def self.__define_ruby_execute_file(plug)
plug.__send__(:rpc, :ruby_execute_file) do |nvim, path|
__wrap_client(nvim) { load(path) }
nil
end
end
private_class_method :__define_ruby_execute_file
# Evaluate the provided Ruby code over each line of a range. The contents
# of the current line can be accessed and modified via the +$_+ variable.
#
# Since this method evaluates each line in the local binding, all local
# variables and methods are available to the user. Thus the +__+ prefix
# obfuscation.
#
# This is used by the +:rubydo+ command.
def self.__define_ruby_do_range(__plug)
__plug.__send__(:rpc, :ruby_do_range) do |__nvim, *__args|
__wrap_client(__nvim) do
__start, __stop, __ruby = __args
__buffer = __nvim.get_current_buf
__update_lines_in_chunks(__buffer, __start, __stop, 1_000) do |__lines|
__lines.map do |__line|
$_ = __line
eval(__ruby, binding, "ruby_do_range")
$_
end
end
end
nil
end
end
private_class_method :__define_ruby_do_range
def self.__define_ruby_chdir(plug)
plug.__send__(:rpc, :ruby_chdir) do |_, event|
Dir.chdir(event.fetch("cwd"))
end
end
private_class_method :__define_ruby_chdir
def self.__wrap_client(client)
Vim.__client = client
Vim.__refresh_globals(client)
__with_exception_handling(client) do
__with_std_streams(client) do
yield
end
end
end
private_class_method :__wrap_client
def self.__with_exception_handling(client)
yield
rescue ScriptError, StandardError => e
msg = [e.class, e.message].join(": ")
client.err_writeln(msg)
end
private_class_method :__with_exception_handling
def self.__with_std_streams(client)
old_stdout = $stdout.dup
old_stderr = $stderr.dup
$stdout, $stderr = StringIO.new, StringIO.new
begin
yield.tap do
client.out_write($stdout.string + $/) if $stdout.size > 0
client.err_writeln($stderr.string) if $stderr.size > 0
end
ensure
$stdout = old_stdout
$stderr = old_stderr
end
end
private_class_method :__with_std_streams
def self.__update_lines_in_chunks(buffer, start, stop, size)
(start..stop).each_slice(size) do |linenos|
start, stop = linenos[0] - 1, linenos[-1]
lines = buffer.get_lines(start, stop, false)
buffer.set_lines(start, stop, false, yield(lines))
end
end
private_class_method :__update_lines_in_chunks
end
end
Neovim::RubyProvider.__define_plugin!
|
