From 1c0e798d5c7e2c326fc3116a2f8dc259c71eea72 Mon Sep 17 00:00:00 2001
From: akr <akr@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Thu, 1 Jan 2009 06:28:34 +0000
Subject: [PATCH] * ext/socket/mkconstants.rb: use erb.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@21225 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 ChangeLog                 |  4 +++
 ext/socket/mkconstants.rb | 59 +++++++++++++++++++++++++--------------
 2 files changed, 42 insertions(+), 21 deletions(-)

diff --git a/ChangeLog b/ChangeLog
index d4a580899c..28866c2485 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,3 +1,7 @@
+Thu Jan  1 15:27:07 2009  Tanaka Akira  <akr@fsij.org>
+
+	* ext/socket/mkconstants.rb: use erb.
+
 Thu Jan  1 15:07:56 2009  Tanaka Akira  <akr@fsij.org>
 
 	* ext/socket/mkconstants.rb: add -o option.
diff --git a/ext/socket/mkconstants.rb b/ext/socket/mkconstants.rb
index 948eda8be1..d74310cb34 100644
--- a/ext/socket/mkconstants.rb
+++ b/ext/socket/mkconstants.rb
@@ -1,4 +1,5 @@
 require 'optparse'
+require 'erb'
 
 opt = OptionParser.new
 
@@ -12,43 +13,59 @@ opt.def_option('-o FILE', 'specify output file') {|filename|
   opt_o = filename
 }
 
-$out = ''
-def $out.puts(str="")
-  str += "\n" if /\n\z/ !~ str
-  self << str
+C_ESC = {
+  "\\" => "\\\\",
+  '"' => '\"',
+  "\n" => '\n',
+}
+
+0x00.upto(0x1f) {|ch| C_ESC[[ch].pack("C")] ||= "\\%03o" % ch }
+0x7f.upto(0xff) {|ch| C_ESC[[ch].pack("C")] = "\\%03o" % ch }
+C_ESC_PAT = Regexp.union(*C_ESC.keys)
+
+def c_str(str)
+  '"' + str.gsub(C_ESC_PAT, C_ESC) + '"'
 end
 
 opt.parse!
 
-# workaround for NetBSD, OpenBSD and etc.
-$out.puts("#define pseudo_AF_FTIP pseudo_AF_RTIP")
+result = ''
 
-# skip empty lines and comment lines
-DATA.each_line do |s|
-  name, value = s.scan(/\S+/)
-  if name && name[0] != ?#
+# workaround for NetBSD, OpenBSD and etc.
+result << "#define pseudo_AF_FTIP pseudo_AF_RTIP\n"
+
+def each_data
+  DATA.each_line {|s|
+    name, default_value = s.scan(/\S+/)
+    next unless name && name[0] != ?#
     if name =~ /\AINADDR_/
       define = "sock_define_uconst"
     else
       define = "sock_define_const"
     end
-    $out.puts("#ifdef #{name}")
-    $out.puts("    #{define}(\"#{name}\", #{name});")
-    if value
-    $out.puts("#else")
-    $out.puts("    #{define}(\"#{name}\", #{value});")
-    end
-    $out.puts("#endif")
-    $out.puts
-  end
+    yield define, name, default_value
+  }
 end
 
+result << ERB.new(<<'EOS', nil, '%').result(binding)
+% each_data {|define, name, default_value|
+#ifdef <%=name%>
+    <%=define%>(<%=c_str name%>, <%=name%>);
+%   if default_value
+#else
+    <%=define%>(<%=c_str name%>, <%=default_value%>);
+%   end
+#endif
+
+% }
+EOS
+
 if opt_o
   File.open(opt_o, 'w') {|f|
-    f << $out
+    f << result
   }
 else
-  $stdout << $out
+  $stdout << result
 end
 
 __END__