summaryrefslogtreecommitdiff
path: root/ext/psych/lib/psych/scalar_scanner.rb
diff options
context:
space:
mode:
authornobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>2010-03-30 10:46:06 +0000
committernobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>2010-03-30 10:46:06 +0000
commitf1330fd12403ef4d664df7619d6ce312d525721a (patch)
treecfafd35d3d416b8529003d43db2ace98af20fdb1 /ext/psych/lib/psych/scalar_scanner.rb
parent38af94c6abaaa20aeaa1ffafbe3f6ff962f63eaa (diff)
* ext/psych/lib: moved external library dependent files from lib.
[ruby-core:29129] git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@27112 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'ext/psych/lib/psych/scalar_scanner.rb')
-rw-r--r--ext/psych/lib/psych/scalar_scanner.rb90
1 files changed, 90 insertions, 0 deletions
diff --git a/ext/psych/lib/psych/scalar_scanner.rb b/ext/psych/lib/psych/scalar_scanner.rb
new file mode 100644
index 0000000000..bee88de419
--- /dev/null
+++ b/ext/psych/lib/psych/scalar_scanner.rb
@@ -0,0 +1,90 @@
+require 'strscan'
+
+module Psych
+ ###
+ # Scan scalars for built in types
+ class ScalarScanner
+ # Taken from http://yaml.org/type/timestamp.html
+ TIME = /^\d{4}-\d{1,2}-\d{1,2}([Tt]|\s+)\d{1,2}:\d\d:\d\d(\.\d*)?(\s*Z|[-+]\d{1,2}(:\d\d)?)?/
+
+ # Create a new scanner
+ def initialize
+ @string_cache = {}
+ end
+
+ # Tokenize +string+ returning the ruby object
+ def tokenize string
+ return nil if string.empty?
+ return string if @string_cache.key?(string)
+
+ case string
+ when /^[A-Za-z~]/
+ if string.length > 5
+ @string_cache[string] = true
+ return string
+ end
+
+ case string
+ when /^[^ytonf~]/i
+ @string_cache[string] = true
+ string
+ when '~', /^null$/i
+ nil
+ when /^(yes|true|on)$/i
+ true
+ when /^(no|false|off)$/i
+ false
+ else
+ @string_cache[string] = true
+ string
+ end
+ when TIME
+ date, time = *(string.split(/[ tT]/, 2))
+ (yy, m, dd) = date.split('-').map { |x| x.to_i }
+ md = time.match(/(\d+:\d+:\d+)(\.\d*)?\s*(Z|[-+]\d+(:\d\d)?)?/)
+
+ (hh, mm, ss) = md[1].split(':').map { |x| x.to_i }
+ us = (md[2] ? Rational(md[2].sub(/^\./, '0.')) : 0) * 1000000
+
+ time = Time.utc(yy, m, dd, hh, mm, ss, us)
+
+ return time if 'Z' == md[3]
+
+ tz = md[3] ? Integer(md[3].split(':').first.sub(/([-+])0/, '\1')) : 0
+ Time.at((time - (tz * 3600)).to_i, us)
+ when /^\d{4}-\d{1,2}-\d{1,2}$/
+ require 'date'
+ Date.strptime(string, '%Y-%m-%d')
+ when /^\.inf$/i
+ 1 / 0.0
+ when /^-\.inf$/i
+ -1 / 0.0
+ when /^\.nan$/i
+ 0.0 / 0.0
+ when /^:./
+ if string =~ /^:(["'])(.*)\1/
+ $2.sub(/^:/, '').to_sym
+ else
+ string.sub(/^:/, '').to_sym
+ end
+ when /^[-+]?[1-9][0-9_]*(:[0-5]?[0-9])+$/
+ i = 0
+ string.split(':').each_with_index do |n,e|
+ i += (n.to_i * 60 ** (e - 2).abs)
+ end
+ i
+ when /^[-+]?[0-9][0-9_]*(:[0-5]?[0-9])+\.[0-9_]*$/
+ i = 0
+ string.split(':').each_with_index do |n,e|
+ i += (n.to_f * 60 ** (e - 2).abs)
+ end
+ i
+ else
+ return Integer(string.gsub(/[,_]/, '')) rescue ArgumentError
+ return Float(string.gsub(/[,_]/, '')) rescue ArgumentError
+ @string_cache[string] = true
+ string
+ end
+ end
+ end
+end