1_4_post_install_bin_files.rb

# frozen_string_literal: true

require 'fileutils'

# run from build/installed Ruby
module CopyBashScripts
  BIN_DIR = "#{RbConfig::TOPDIR}/bin"

  class << self

    def run
      # clean empty gem folders, needed as of 2019-10-20
      ary = Dir["#{Gem.dir}/gems/*"]
      ary.each { |d| Dir.rmdir(d) if Dir.empty?(d) }

      bash_preamble = <<~BASH.strip.rstrip
        #!
        {
          bindir=$(dirname "$0")
          exec "$bindir/ruby" "-x" "$0" "$@"
        }
        #!/usr/bin/env ruby
      BASH

      windows_script = <<~BAT
        @ECHO OFF
        @"%~dp0ruby.exe" -x "%~dpn0" %*
      BAT

      # all files in bin folder
      bins = Dir["#{BIN_DIR}/*"].select { |fn| File.file? fn }

      bash_bins = bins.select { |fn| File.extname(fn).empty? }

      # file permissions may not work on Windows, especially execute
      bash_bins.each do |fn|
        ruby_code = File.read(fn, mode: 'rb:UTF-8').split(/^#![^\n]+ruby/,2).last.lstrip
        File.write fn, "#{bash_preamble}\n#{ruby_code}", mode: 'wb:UTF-8'
      end

      windows_bins = bins.select { |fn| File.extname(fn).match?(/\A\.bat|\A\.cmd/) }

      windows_bins.each do |fn|
        # 'gem' bash script doesn't exist
        bash_bin = "#{BIN_DIR}/#{File.basename fn, '.*'}"
        unless File.exist? bash_bin
          ruby_code = File.read(fn, mode: 'rb:UTF-8').split(/^#![^\n]+ruby/,2).last.lstrip
          File.write bash_bin, "#{bash_preamble}\n#{ruby_code}", mode: 'wb:UTF-8'
        end

        File.write fn, windows_script, mode: 'wb:UTF-8'
      end
    end

    def fix_rbs_debug_ext
      [ {g_name: 'debug', g_so: 'debug/debug.so'},
        {g_name: 'rbs'  , g_so: 'rbs_extension.so'}
      ].each do |h|
        gem_path = Dir.glob("#{Gem.default_dir}/gems/#{h[:g_name]}-*").first
        gem_name = File.basename gem_path
        unless File.exist? "#{gem_path}/lib/#{h[:g_so]}"
          ext_path = "#{Gem.default_dir}/extensions/#{RUBY_PLATFORM.tr '_', '-'}/" \
            "#{RbConfig::CONFIG['ruby_version']}/#{gem_name}/#{h[:g_so]}"
          if File.exist? ext_path
            FileUtils.cp ext_path, "#{gem_path}/lib/#{h[:g_so]}"
          end
        end
      end
    end
  end
end
CopyBashScripts.run
CopyBashScripts.fix_rbs_debug_ext