[GR-18163] Add option --reuse-precompiled-gems

CHANGELOG.md

@@ -2,6 +2,8 @@
 
 New features:
 
+* Add `--reuse-precompiled-gems` option (@andrykonchin).
+
 Bug fixes:
 
 * Add missing thread-safe objects write barriers for `TruffleRuby::ConcurrentMap` (#3179, @eregon).
@@ -23,6 +25,7 @@ Compatibility:
 * Set `$!` when a `Kernel#at_exit` hook raises an exception (#3535, @andrykonchin).
 
 Performance:
+
 * Fix inline caching for Regexp creation from Strings (#3492, @andrykonchin, @eregon).
 * Optimize `Integer#pow` method for small modulus values (#3544, @andrykonchin).
 

lib/truffle/rubygems/defaults/truffleruby.rb

@@ -30,5 +30,5 @@ def self.platform_defaults
 
 class Gem::Platform
   # The list of gems we want to install precompiled (using the local platform) on TruffleRuby
-  REUSE_AS_BINARY_ON_TRUFFLERUBY = %w[libv8 libv8-node sass-embedded sorbet-static]
+  REUSE_AS_BINARY_ON_TRUFFLERUBY = %w[libv8 libv8-node sass-embedded sorbet-static] + Truffle::Boot.get_option('reuse-precompiled-gems')
 end

spec/tags/truffle/options/reuse_precompiled_gems_tags.txt

@@ -0,0 +1 @@
+slow:The --reuse-precompiled-gems option adds listed gems into a Gem::Platform::REUSE_AS_BINARY_ON_TRUFFLERUBY list

spec/truffle/options/reuse_precompiled_gems_spec.rb

@@ -0,0 +1,15 @@
+# Copyright (c) 2024 Oracle and/or its affiliates. All rights reserved. This
+# code is released under a tri EPL/GPL/LGPL license. You can use it,
+# redistribute it and/or modify it under the terms of the:
+#
+# Eclipse Public License version 2.0, or
+# GNU General Public License version 2, or
+# GNU Lesser General Public License version 2.1.
+
+require_relative '../../ruby/spec_helper'
+
+describe "The --reuse-precompiled-gems option" do
+  it "adds listed gems into a Gem::Platform::REUSE_AS_BINARY_ON_TRUFFLERUBY list" do
+    ruby_exe("p Gem::Platform::REUSE_AS_BINARY_ON_TRUFFLERUBY", options: "--experimental-options --reuse-precompiled-gems=foo,bar").should.include? '"foo", "bar"'
+  end
+end

src/launcher/java/org/truffleruby/launcher/CommandLineParser.java

@@ -118,8 +118,8 @@ private void processArgv() {
                     value = null;
                 }
 
-                // Switches without values are stored separately in ARGV_GLOBAL_FLAGS. Otherwise it would not be
-                // possible to determine if the value is suppose to be `true` or `"true"`.
+                // Switches without values are stored separately in ARGV_GLOBAL_FLAGS. Otherwise, it would not be
+                // possible to determine if the value is supposed to be `true` or `"true"`.
                 final OptionDescriptor optionDescription = value != null
                         ? OptionsCatalog.ARGV_GLOBAL_VALUES
                         : OptionsCatalog.ARGV_GLOBAL_FLAGS;

src/main/java/org/truffleruby/options/Options.java

@@ -135,6 +135,8 @@ public final class Options {
     public final boolean WARN_TRUFFLE_REGEX_MATCH_FALLBACK;
     /** --truffle-regex-ignore-atomic-groups=false */
     public final boolean TRUFFLE_REGEX_IGNORE_ATOMIC_GROUPS;
+    /** --reuse-precompiled-gems=StringArrayOptionType.EMPTY_STRING_ARRAY */
+    public final String[] REUSE_PRECOMPILED_GEMS;
     /** --argv-globals=false */
     public final boolean ARGV_GLOBALS;
     /** --chomp-loop=false */
@@ -262,6 +264,7 @@ public Options(Env env, OptionValues options, LanguageOptions languageOptions) {
         WARN_TRUFFLE_REGEX_COMPILE_FALLBACK = options.get(OptionsCatalog.WARN_TRUFFLE_REGEX_COMPILE_FALLBACK_KEY);
         WARN_TRUFFLE_REGEX_MATCH_FALLBACK = options.get(OptionsCatalog.WARN_TRUFFLE_REGEX_MATCH_FALLBACK_KEY);
         TRUFFLE_REGEX_IGNORE_ATOMIC_GROUPS = options.get(OptionsCatalog.TRUFFLE_REGEX_IGNORE_ATOMIC_GROUPS_KEY);
+        REUSE_PRECOMPILED_GEMS = options.get(OptionsCatalog.REUSE_PRECOMPILED_GEMS_KEY);
         ARGV_GLOBALS = options.get(OptionsCatalog.ARGV_GLOBALS_KEY);
         CHOMP_LOOP = options.get(OptionsCatalog.CHOMP_LOOP_KEY);
         GETS_LOOP = options.get(OptionsCatalog.GETS_LOOP_KEY);
@@ -411,6 +414,8 @@ public Object fromDescriptor(OptionDescriptor descriptor) {
                 return WARN_TRUFFLE_REGEX_MATCH_FALLBACK;
             case "ruby.truffle-regex-ignore-atomic-groups":
                 return TRUFFLE_REGEX_IGNORE_ATOMIC_GROUPS;
+            case "ruby.reuse-precompiled-gems":
+                return REUSE_PRECOMPILED_GEMS;
             case "ruby.argv-globals":
                 return ARGV_GLOBALS;
             case "ruby.chomp-loop":

src/options.yml

@@ -155,6 +155,8 @@ EXPERT:
     WARN_TRUFFLE_REGEX_MATCH_FALLBACK: [warn-truffle-regex-match-fallback, boolean, false, 'Warn every time Truffle Regex cannot be used for a Regexp match (and instead Joni is used)']
     TRUFFLE_REGEX_IGNORE_ATOMIC_GROUPS: [truffle-regex-ignore-atomic-groups, boolean, false, 'Treat atomic groups (?>...) as ordinary groups (?:...) with Truffle Regex.']
 
+    REUSE_PRECOMPILED_GEMS: [reuse-precompiled-gems, string-array, '<gem>,<gem>,...', 'A list of gems we want to install precompiled (using the local platform) on TruffleRuby. Can only be used for gem extensions which do not depend on the Ruby C API.']
+
 INTERNAL: # Options for debugging the TruffleRuby implementation
   EXPERIMENTAL:
     # Options corresponding to MRI options, which are only meaningful when using the TruffleRuby launcher

src/shared/java/org/truffleruby/shared/options/OptionsCatalog.java

@@ -85,6 +85,7 @@ public final class OptionsCatalog {
     public static final OptionKey<Boolean> WARN_TRUFFLE_REGEX_COMPILE_FALLBACK_KEY = new OptionKey<>(false);
     public static final OptionKey<Boolean> WARN_TRUFFLE_REGEX_MATCH_FALLBACK_KEY = new OptionKey<>(false);
     public static final OptionKey<Boolean> TRUFFLE_REGEX_IGNORE_ATOMIC_GROUPS_KEY = new OptionKey<>(false);
+    public static final OptionKey<String[]> REUSE_PRECOMPILED_GEMS_KEY = new OptionKey<>(StringArrayOptionType.EMPTY_STRING_ARRAY, StringArrayOptionType.INSTANCE);
     public static final OptionKey<Boolean> ARGV_GLOBALS_KEY = new OptionKey<>(false);
     public static final OptionKey<Boolean> CHOMP_LOOP_KEY = new OptionKey<>(false);
     public static final OptionKey<Boolean> GETS_LOOP_KEY = new OptionKey<>(false);
@@ -684,6 +685,14 @@ public final class OptionsCatalog {
             .usageSyntax("")
             .build();
 
+    public static final OptionDescriptor REUSE_PRECOMPILED_GEMS = OptionDescriptor
+            .newBuilder(REUSE_PRECOMPILED_GEMS_KEY, "ruby.reuse-precompiled-gems")
+            .help("A list of gems we want to install precompiled (using the local platform) on TruffleRuby. Can only be used for gem extensions which do not depend on the Ruby C API.")
+            .category(OptionCategory.EXPERT)
+            .stability(OptionStability.EXPERIMENTAL)
+            .usageSyntax("<gem>,<gem>,...")
+            .build();
+
     public static final OptionDescriptor ARGV_GLOBALS = OptionDescriptor
             .newBuilder(ARGV_GLOBALS_KEY, "ruby.argv-globals")
             .help("Parse options in script argv into global variables (configured by the -s Ruby option)")
@@ -1440,6 +1449,8 @@ public static OptionDescriptor fromName(String name) {
                 return WARN_TRUFFLE_REGEX_MATCH_FALLBACK;
             case "ruby.truffle-regex-ignore-atomic-groups":
                 return TRUFFLE_REGEX_IGNORE_ATOMIC_GROUPS;
+            case "ruby.reuse-precompiled-gems":
+                return REUSE_PRECOMPILED_GEMS;
             case "ruby.argv-globals":
                 return ARGV_GLOBALS;
             case "ruby.chomp-loop":
@@ -1668,6 +1679,7 @@ public static OptionDescriptor[] allDescriptors() {
             WARN_TRUFFLE_REGEX_COMPILE_FALLBACK,
             WARN_TRUFFLE_REGEX_MATCH_FALLBACK,
             TRUFFLE_REGEX_IGNORE_ATOMIC_GROUPS,
+            REUSE_PRECOMPILED_GEMS,
             ARGV_GLOBALS,
             CHOMP_LOOP,
             GETS_LOOP,