Implement ShopifyRubyDefinitions

Author: peterzhu2118

.rubocop.yml

@@ -6,6 +6,7 @@ require:
   - rubocop-sorbet
 
 AllCops:
+  NewCops: enable
   SuggestExtensions: false
   TargetRubyVersion: 2.6
 

lib/shopify_ruby_definitions.rb

@@ -1,7 +1,12 @@
 # typed: strong
 # frozen_string_literal: true
 
+require "sorbet-runtime"
+
+require_relative "shopify_ruby_definitions/ruby_build"
+require_relative "shopify_ruby_definitions/ruby_versions"
 require_relative "shopify_ruby_definitions/version"
 
 module ShopifyRubyDefinitions
+  extend RubyVersions
 end

lib/shopify_ruby_definitions/ruby_build.rb

@@ -0,0 +1,10 @@
+# typed: strong
+# frozen_string_literal: true
+
+module ShopifyRubyDefinitions
+  module RubyBuild
+    extend T::Sig
+
+    RUBY_BUILD_DIRECTORY = T.let(File.expand_path("../../../vendor/ruby-build", __FILE__), String)
+  end
+end

lib/shopify_ruby_definitions/ruby_versions.rb

@@ -0,0 +1,49 @@
+# typed: strong
+# frozen_string_literal: true
+
+module ShopifyRubyDefinitions
+  module RubyVersions
+    extend T::Sig
+
+    VERSIONS_DIRECTORY = T.let(File.join(RubyBuild::RUBY_BUILD_DIRECTORY, "share/ruby-build"), String)
+    ALL_VERSIONS = T.let(Dir["#{VERSIONS_DIRECTORY}/*"].map { |f| File.basename(f) }, T::Array[String])
+
+    sig { returns(T::Hash[String, String]) }
+    def version_overrides
+      {
+        "head" => ruby_head_version,
+        "ruby-head" => ruby_head_version,
+        "3.0.0" => "3.0.0-pshopify9",
+        "3.0.1" => "3.0.1-pshopify2",
+        "3.0.2" => "3.0.2-pshopify3",
+        "3.1.0" => "3.1.0-pshopify1",
+        "3.1.1" => "3.1.1-pshopify2",
+        "3.1.2" => "3.1.2-pshopify2",
+        "3.1.3" => "3.1.3-pshopify1",
+        "3.1.4" => "3.1.4-pshopify1",
+        "3.2.0" => "3.2.0-pshopify2",
+        "3.2.1" => "3.2.1-pshopify5",
+        "3.2.2" => "3.2.2-pshopify3",
+      }.freeze
+    end
+
+    sig { params(version: String).returns(String) }
+    def resolve_version(version)
+      if version.match?(/\A\d+\.\d+\z/)
+        pattern = /\A#{Regexp.escape(version)}\.(\d+)\z/
+        versions = ALL_VERSIONS.grep(pattern)
+        unless versions.empty?
+          version = T.must(versions.max_by { |v| T.must(v.match(pattern))[1].to_i })
+        end
+      end
+      version_overrides.fetch(version, version)
+    end
+
+    private
+
+    sig { returns(String) }
+    def ruby_head_version
+      T.must(ALL_VERSIONS.grep(/\A[\d\.]+-dev\z/).max)
+    end
+  end
+end

test/shopify_ruby_definitions/test_ruby_versions.rb

@@ -0,0 +1,30 @@
+# typed: true
+# frozen_string_literal: true
+
+require "test_helper"
+
+module ShopifyRubyDefinitions
+  class TestRubyVersions < Minitest::Test
+    def test_ALL_VERSIONS
+      assert_operator(RubyVersions::ALL_VERSIONS.length, :>, 0)
+
+      valid_pattern = Regexp.union(
+        /\A(?:yjit\-)?\d+\.\d+\.\d+(?:\-[\w\-]+)?\z/, # MRI
+        /\Atruffleruby\-(dev|[\d\.]+(?:-preview\d+)?)\z/, # TruffleRuby
+      )
+
+      RubyVersions::ALL_VERSIONS.each do |v|
+        assert_match(valid_pattern, v)
+      end
+    end
+
+    def test_version_overrides
+      assert_match(/\A3.2.2-pshopify\d\z/, ShopifyRubyDefinitions.version_overrides["3.2.2"])
+    end
+
+    def test_resolve_version
+      assert_equal("2.6.10", ShopifyRubyDefinitions.resolve_version("2.6"))
+      assert_equal("3.3.0-dev", ShopifyRubyDefinitions.resolve_version("ruby-head"))
+    end
+  end
+end

vendor/ruby-build/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2012-2013 Sam Stephenson
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of
+this software and associated documentation files (the "Software"), to deal in
+the Software without restriction, including without limitation the rights to
+use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
+of the Software, and to permit persons to whom the Software is furnished to do
+so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.