Могу ли я получить доступ к данным из необходимого скрипта в Ruby?

Question

Возможно ли получить доступ к тексту после __END__ в файле ruby, отличном от "основного" скрипта?

Например:

# b.rb
B_DATA = DATA.read
__END__
bbb

.

# a.rb
require 'b'
A_DATA = DATA.read
puts 'A_DATA: ' + A_DATA
puts 'B_DATA: ' + B_DATA
__END__
aaa

.

C:\Temp>ruby a.rb
A_DATA:
B_DATA: aaa

Есть ли способ получить "bbb" от b.rb?

Answer 1

К сожалению, глобальная константа DATA устанавливается при загрузке «основного» скрипта. Несколько вещей, которые могут помочь:

Вы можете по крайней мере получить A_DATA, чтобы быть правильным. Просто измените порядок первых двух операций в a.rb:

# a.rb
A_DATA = DATA.read
require 'b'
...

Вы можете получить B_DATA, чтобы быть правильным, если вы пройдете немного ригамароле:

# load_data_regardless_of_main_script.rb
module LoadDataRegardlessOfMainScript
  def self.from(file)
    # the performance of this function could be
    # greatly improved by using a StringIO buffer
    # and only appending to it after seeing __END__.
    File.read(file).sub(/\A.*\n__END__\n/m, '')
  end
end

# b.rb:
require 'load_data_regardless_of_main_script'
B_DATA = LoadDataRegardlessOfMainScript.from(__FILE__)

Answer 2

Реализация предложения @ Джеймса использовать StringIO:

require 'stringio'
module LoadDataRegardlessOfMainScript
  def self.from(filename)
    data = StringIO.new
    File.open(filename) do |f|
      begin
        line = f.gets
      end until line.match(/^__END__$/)
      while line = f.gets
        data << line 
      end
    end
    data.rewind
    data
  end
end

Тогда b.rb становится

require 'load_data_regardless_of_main_script'
B_DATA = LoadDataRegardlessOfMainScript.from(__FILE__).read