From b9b923ca942096ddb1062be2deb9e6de9a65f346 Mon Sep 17 00:00:00 2001
From: tenderlove <tenderlove@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Sun, 28 Mar 2010 21:49:37 +0000
Subject: * ext/psych/*: importing Psych to trunk * test/psych/*: ditto *
 lib/psych/*: ditto

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@27079 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 lib/psych/scalar_scanner.rb | 90 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 90 insertions(+)
 create mode 100644 lib/psych/scalar_scanner.rb

(limited to 'lib/psych/scalar_scanner.rb')

diff --git a/lib/psych/scalar_scanner.rb b/lib/psych/scalar_scanner.rb
new file mode 100644
index 0000000000..bee88de419
--- /dev/null
+++ b/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
-- 
cgit v1.2.3