summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/prism/translation/ripper.rb12
1 files changed, 11 insertions, 1 deletions
diff --git a/lib/prism/translation/ripper.rb b/lib/prism/translation/ripper.rb
index 054ad88ce3..5b2aa37833 100644
--- a/lib/prism/translation/ripper.rb
+++ b/lib/prism/translation/ripper.rb
@@ -480,7 +480,17 @@ module Prism
# Create a new Translation::Ripper object with the given source.
def initialize(source, filename = "(ripper)", lineno = 1)
- @source = source
+ if source.is_a?(IO)
+ @source = source.read
+ elsif source.respond_to?(:gets)
+ @source = +""
+ while line = source.gets
+ @source << line
+ end
+ else
+ @source = source.to_str
+ end
+
@filename = filename
@lineno = lineno
@column = 0