extending modules

rubys · rubys · commit 439fbaecbb0b · 2021-01-20T21:35:21.000-05:00
diff --git a/lib/ruby2js/converter/send.rb b/lib/ruby2js/converter/send.rb
@@ -181,9 +181,15 @@ class Converter
           receiver.type == :send and
           receiver.children[1] == :+@ and
           Parser::AST::Node === receiver.children[0] and
-          receiver.children[0].type == :class
+          %i(class module).include? receiver.children[0].type
         then
-          parse receiver.children[0].updated(:class_extend)
+          if receiver.children[0].type == :class
+            parse receiver.children[0].updated(:class_extend)
+          else
+            mod = receiver.children[0]
+            parse s(:send, s(:const, nil, :Object), :assign,
+              mod.children[0], mod.updated(nil, [nil, *mod.children[1..-1]]))
+          end
         else
           put method.to_s[0]; parse receiver
         end
diff --git a/spec/es2015_spec.rb b/spec/es2015_spec.rb
@@ -493,6 +493,13 @@ def to_js_fn(string)
     end
   end
 
+  describe 'module extensions' do
+    it 'should handle methods' do
+      to_js('++module M; def m(); end; end').
+        must_equal 'Object.assign(M, {m: function() {}})'
+    end
+  end
+
   describe 'keyword arguments' do
     it 'should handle keyword arguments in methods' do
       skip if RUBY_VERSION =~ /^(1\.9|2\.0)/
