Set up a C extension

Author: k0kubun

.gitignore

@@ -7,3 +7,4 @@
 /spec/reports/
 /tmp/
 Gemfile.lock
+*.so

Gemfile

@@ -2,5 +2,8 @@ source 'https://rubygems.org'
 
 gemspec
 
-gem 'rake'
-gem 'test-unit'
+group :development do
+  gem 'rake'
+  gem 'rake-compiler'
+  gem 'test-unit'
+end

Rakefile

@@ -1,8 +1,19 @@
 require 'bundler/gem_tasks'
 require 'rake/testtask'
 
+if RUBY_ENGINE != 'jruby'
+  require 'rake/extensiontask'
+  extask = Rake::ExtensionTask.new('erb/html_escape') do |ext|
+    ext.lib_dir.sub!(%r[(?=/|\z)], "/#{RUBY_VERSION}/#{ext.platform}")
+  end
+  task test: :compile
+end
+
 Rake::TestTask.new(:test) do |t|
   t.libs << 'test/lib'
+  if extask
+    t.libs << "lib/#{RUBY_VERSION}/#{extask.platform}"
+  end
   t.ruby_opts << '-rhelper'
   t.test_files = FileList['test/**/test_*.rb']
 end

erb.gemspec

@@ -27,8 +27,11 @@ Gem::Specification.new do |spec|
   spec.executables   = ['erb']
   spec.require_paths = ['lib']
 
-  if RUBY_ENGINE != 'jruby'
+  if RUBY_ENGINE == 'jruby'
+    spec.platform = 'java'
+  else
     spec.required_ruby_version = '>= 2.7.0'
+    spec.extensions = ['ext/erb/html_escape/extconf.rb']
   end
 
   spec.add_dependency 'cgi', '>= 0.3.3'

ext/erb/html_escape/depend

@@ -0,0 +1,2 @@
+html_escape.o: $(RUBY_EXTCONF_H)
+html_escape.o: html_escape.c

ext/erb/html_escape/extconf.rb

@@ -0,0 +1,3 @@
+require 'mkmf'
+
+create_makefile 'erb/html_escape'

ext/erb/html_escape/html_escape.c

@@ -0,0 +1,13 @@
+#include "ruby.h"
+
+void
+Init_html_escape(void)
+{
+    InitVM(html_escape);
+}
+
+void
+InitVM_html_escape(void)
+{
+    //
+}

lib/erb.rb

@@ -1006,6 +1006,11 @@ def html_escape(s)
     module_function :h
     module_function :html_escape
 
+    begin
+      require 'erb/html_escape'
+    rescue LoadError
+    end
+
     #
     # A utility method for encoding the String _s_ as a URL.
     #