* lib/cgi/util.rb: class methods modulize for using like a function.

[Feature #8354]

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@40571 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
xibbar 2013-05-03 11:23:23 +00:00
parent 86d4511fe6
commit b35529bb4e
3 changed files with 47 additions and 17 deletions

View File

@ -1,3 +1,9 @@
Fri May 3 19:32:13 2013 Takeyuki FUJIOKA <xibbar@ruby-lang.org>
* lib/cgi/util.rb: All class methods moduleized.
We can use these methods like a function when "include CGI::Util".
[Feature #8354]
Fri May 3 14:09:45 2013 Tanaka Akira <akr@fsij.org> Fri May 3 14:09:45 2013 Tanaka Akira <akr@fsij.org>
* ext/socket/extconf.rb: Make default_ipv6 true for Cygwin. * ext/socket/extconf.rb: Make default_ipv6 true for Cygwin.

View File

@ -1,9 +1,10 @@
class CGI class CGI; module Util; end; extend Util; end
module CGI::Util
@@accept_charset="UTF-8" unless defined?(@@accept_charset) @@accept_charset="UTF-8" unless defined?(@@accept_charset)
# URL-encode a string. # URL-encode a string.
# url_encoded_string = CGI::escape("'Stop!' said Fred") # url_encoded_string = CGI::escape("'Stop!' said Fred")
# # => "%27Stop%21%27+said+Fred" # # => "%27Stop%21%27+said+Fred"
def CGI::escape(string) def escape(string)
encoding = string.encoding encoding = string.encoding
string.dup.force_encoding('ASCII-8BIT').gsub(/([^ a-zA-Z0-9_.-]+)/) do string.dup.force_encoding('ASCII-8BIT').gsub(/([^ a-zA-Z0-9_.-]+)/) do
'%' + $1.unpack('H2' * $1.bytesize).join('%').upcase '%' + $1.unpack('H2' * $1.bytesize).join('%').upcase
@ -13,7 +14,7 @@ class CGI
# URL-decode a string with encoding(optional). # URL-decode a string with encoding(optional).
# string = CGI::unescape("%27Stop%21%27+said+Fred") # string = CGI::unescape("%27Stop%21%27+said+Fred")
# # => "'Stop!' said Fred" # # => "'Stop!' said Fred"
def CGI::unescape(string,encoding=@@accept_charset) def unescape(string,encoding=@@accept_charset)
str=string.tr('+', ' ').force_encoding(Encoding::ASCII_8BIT).gsub(/((?:%[0-9a-fA-F]{2})+)/) do str=string.tr('+', ' ').force_encoding(Encoding::ASCII_8BIT).gsub(/((?:%[0-9a-fA-F]{2})+)/) do
[$1.delete('%')].pack('H*') [$1.delete('%')].pack('H*')
end.force_encoding(encoding) end.force_encoding(encoding)
@ -32,14 +33,14 @@ class CGI
# Escape special characters in HTML, namely &\"<> # Escape special characters in HTML, namely &\"<>
# CGI::escapeHTML('Usage: foo "bar" <baz>') # CGI::escapeHTML('Usage: foo "bar" <baz>')
# # => "Usage: foo &quot;bar&quot; &lt;baz&gt;" # # => "Usage: foo &quot;bar&quot; &lt;baz&gt;"
def CGI::escapeHTML(string) def escapeHTML(string)
string.gsub(/['&\"<>]/, TABLE_FOR_ESCAPE_HTML__) string.gsub(/['&\"<>]/, TABLE_FOR_ESCAPE_HTML__)
end end
# Unescape a string that has been HTML-escaped # Unescape a string that has been HTML-escaped
# CGI::unescapeHTML("Usage: foo &quot;bar&quot; &lt;baz&gt;") # CGI::unescapeHTML("Usage: foo &quot;bar&quot; &lt;baz&gt;")
# # => "Usage: foo \"bar\" <baz>" # # => "Usage: foo \"bar\" <baz>"
def CGI::unescapeHTML(string) def unescapeHTML(string)
enc = string.encoding enc = string.encoding
if [Encoding::UTF_16BE, Encoding::UTF_16LE, Encoding::UTF_32BE, Encoding::UTF_32LE].include?(enc) if [Encoding::UTF_16BE, Encoding::UTF_16LE, Encoding::UTF_32BE, Encoding::UTF_32LE].include?(enc)
return string.gsub(Regexp.new('&(apos|amp|quot|gt|lt|#[0-9]+|#x[0-9A-Fa-f]+);'.encode(enc))) do return string.gsub(Regexp.new('&(apos|amp|quot|gt|lt|#[0-9]+|#x[0-9A-Fa-f]+);'.encode(enc))) do
@ -88,12 +89,12 @@ class CGI
end end
# Synonym for CGI::escapeHTML(str) # Synonym for CGI::escapeHTML(str)
def CGI::escape_html(str) def escape_html(str)
escapeHTML(str) escapeHTML(str)
end end
# Synonym for CGI::unescapeHTML(str) # Synonym for CGI::unescapeHTML(str)
def CGI::unescape_html(str) def unescape_html(str)
unescapeHTML(str) unescapeHTML(str)
end end
@ -110,7 +111,7 @@ class CGI
# #
# print CGI::escapeElement('<BR><A HREF="url"></A>', ["A", "IMG"]) # print CGI::escapeElement('<BR><A HREF="url"></A>', ["A", "IMG"])
# # "<BR>&lt;A HREF=&quot;url&quot;&gt;&lt;/A&gt" # # "<BR>&lt;A HREF=&quot;url&quot;&gt;&lt;/A&gt"
def CGI::escapeElement(string, *elements) def escapeElement(string, *elements)
elements = elements[0] if elements[0].kind_of?(Array) elements = elements[0] if elements[0].kind_of?(Array)
unless elements.empty? unless elements.empty?
string.gsub(/<\/?(?:#{elements.join("|")})(?!\w)(?:.|\n)*?>/i) do string.gsub(/<\/?(?:#{elements.join("|")})(?!\w)(?:.|\n)*?>/i) do
@ -130,11 +131,11 @@ class CGI
# print CGI::unescapeElement( # print CGI::unescapeElement(
# CGI::escapeHTML('<BR><A HREF="url"></A>'), ["A", "IMG"]) # CGI::escapeHTML('<BR><A HREF="url"></A>'), ["A", "IMG"])
# # "&lt;BR&gt;<A HREF="url"></A>" # # "&lt;BR&gt;<A HREF="url"></A>"
def CGI::unescapeElement(string, *elements) def unescapeElement(string, *elements)
elements = elements[0] if elements[0].kind_of?(Array) elements = elements[0] if elements[0].kind_of?(Array)
unless elements.empty? unless elements.empty?
string.gsub(/&lt;\/?(?:#{elements.join("|")})(?!\w)(?:.|\n)*?&gt;/i) do string.gsub(/&lt;\/?(?:#{elements.join("|")})(?!\w)(?:.|\n)*?&gt;/i) do
CGI::unescapeHTML($&) unescapeHTML($&)
end end
else else
string string
@ -142,12 +143,12 @@ class CGI
end end
# Synonym for CGI::escapeElement(str) # Synonym for CGI::escapeElement(str)
def CGI::escape_element(str) def escape_element(str)
escapeElement(str) escapeElement(str)
end end
# Synonym for CGI::unescapeElement(str) # Synonym for CGI::unescapeElement(str)
def CGI::unescape_element(str) def unescape_element(str)
unescapeElement(str) unescapeElement(str)
end end
@ -161,11 +162,11 @@ class CGI
# #
# CGI::rfc1123_date(Time.now) # CGI::rfc1123_date(Time.now)
# # Sat, 01 Jan 2000 00:00:00 GMT # # Sat, 01 Jan 2000 00:00:00 GMT
def CGI::rfc1123_date(time) def rfc1123_date(time)
t = time.clone.gmtime t = time.clone.gmtime
return format("%s, %.2d %s %.4d %.2d:%.2d:%.2d GMT", return format("%s, %.2d %s %.4d %.2d:%.2d:%.2d GMT",
RFC822_DAYS[t.wday], t.day, RFC822_MONTHS[t.month-1], t.year, RFC822_DAYS[t.wday], t.day, RFC822_MONTHS[t.month-1], t.year,
t.hour, t.min, t.sec) t.hour, t.min, t.sec)
end end
# Prettify (indent) an HTML string. # Prettify (indent) an HTML string.
@ -185,7 +186,7 @@ class CGI
# # </BODY> # # </BODY>
# # </HTML> # # </HTML>
# #
def CGI::pretty(string, shift = " ") def pretty(string, shift = " ")
lines = string.gsub(/(?!\A)<.*?>/m, "\n\\0").gsub(/<.*?>(?!\n)/m, "\\0\n") lines = string.gsub(/(?!\A)<.*?>/m, "\n\\0").gsub(/<.*?>(?!\n)/m, "\\0\n")
end_pos = 0 end_pos = 0
while end_pos = lines.index(/^<\/(\w+)/, end_pos) while end_pos = lines.index(/^<\/(\w+)/, end_pos)
@ -195,4 +196,6 @@ class CGI
end end
lines.gsub(/^((?:#{Regexp::quote(shift)})*)__(?=<\/?\w)/, '\1') lines.gsub(/^((?:#{Regexp::quote(shift)})*)__(?=<\/?\w)/, '\1')
end end
alias h escapeHTML
end end

View File

@ -4,7 +4,7 @@ require 'stringio'
class CGIUtilTest < Test::Unit::TestCase class CGIUtilTest < Test::Unit::TestCase
include CGI::Util
def setup def setup
ENV['REQUEST_METHOD'] = 'GET' ENV['REQUEST_METHOD'] = 'GET'
@ -65,4 +65,25 @@ class CGIUtilTest < Test::Unit::TestCase
assert_equal(CGI::unescapeHTML("&#x3042;&#x3044;&#X3046;"),"\xE3\x81\x82\xE3\x81\x84\xE3\x81\x86") assert_equal(CGI::unescapeHTML("&#x3042;&#x3044;&#X3046;"),"\xE3\x81\x82\xE3\x81\x84\xE3\x81\x86")
end end
def test_cgi_include_escape
assert_equal('%26%3C%3E%22+%E3%82%86%E3%82%93%E3%82%86%E3%82%93', escape(@str1))
end
def test_cgi_include_escapeHTML
assert_equal(escapeHTML("'&\"><"),"&#39;&amp;&quot;&gt;&lt;")
end
def test_cgi_include_h
assert_equal(h("'&\"><"),"&#39;&amp;&quot;&gt;&lt;")
end
def test_cgi_include_unescape
assert_equal(@str1, unescape('%26%3C%3E%22+%E3%82%86%E3%82%93%E3%82%86%E3%82%93'))
assert_equal(@str1.encoding, unescape('%26%3C%3E%22+%E3%82%86%E3%82%93%E3%82%86%E3%82%93').encoding) if defined?(::Encoding)
assert_equal("\u{30E1 30E2 30EA 691C 7D22}", unescape("\u{30E1 30E2 30EA}%E6%A4%9C%E7%B4%A2"))
end
def test_cgi_include_unescapeHTML
assert_equal(unescapeHTML("&#39;&amp;&quot;&gt;&lt;"),"'&\"><")
end
end end