¿Puedo acceder a los datos de una secuencia de comandos requerida en Ruby?

Question

¿Es posible acceder al texto después de __END__ en un archivo ruby ​​que no sea el script "principal"?

Por ejemplo:

# 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 

¿Hay alguna forma de acceder al "bbb" desde b.rb?

Answer 1

Desafortunadamente, la constante global DATA se establece cuando se carga la secuencia de comandos "principal". Algunas cosas que pueden ayudar:

Usted puede al menos obtener A_DATA para ser correcto. Sólo invertir el orden de las dos primeras operaciones en a.rb:

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

se puede obtener el B_DATA ser correcto si vas a través de un poco de galimatías:

# 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

La implementación @ sugerencia de James usar 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 

Entonces se convierte en b.rb

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