1_2_post_install.rb

# frozen_string_literal: true
# code by MSP-Greg

require 'open3'
require 'fileutils'
require_relative '1_2_post_install_common.rb'

module PostInstall2
  case ENV['MSYSTEM']
  when 'UCRT64'
    D_INSTALL = File.join __dir__, 'ruby-ucrt'
    DLL_FN = 'x64-ucrt-ruby'
  when 'MINGW32'
    D_INSTALL = File.join __dir__, 'ruby-mingw32'
    DLL_FN = 'msvcrt-ruby'
  else
    D_INSTALL = File.join __dir__, 'ruby-mingw'
    DLL_FN = 'x64-msvcrt-ruby'
  end

  D_MSYS2 = ENV['D_MSYS2']
  D_MINGW = "#{D_MSYS2}#{ENV['MINGW_PREFIX']}"

  D_RI2   = File.join __dir__, 'rubyinstaller2'
  # -REMOVE rbreadline- D_RL    = File.join __dir__, 'ruby_readline'
  D_RUBY  = File.join __dir__, 'ruby'

  COL_WID = 36
  COL_SPACE = ' ' * COL_WID

  class << self

    def run
      copy_dll_files
      PostInstall2Common.run
      add_priv_assm
    end

    private

    # Copies correct dll files from MSYS2 location to package dir.
    def copy_dll_files
      pkg_pre = ENV['MINGW_PACKAGE_PREFIX']
      pkgs = 'gcc-libs gmp libffi libyaml openssl readline zlib'
      dll_files, lib_files = find_dlls pkgs, pkg_pre

      # get mingw bin path for arch
      msys2_dll_bin_path = "#{D_MINGW}/bin"

      # create and add manifest
      dest = File.join D_INSTALL, 'bin', 'ruby_builtin_dlls'
      Dir.mkdir dest unless Dir.exist? dest
      create_manifest dll_files, dest

      # copy bin/ dll's
      puts "#{'installing dll files:'.ljust(COL_WID)}FROM #{msys2_dll_bin_path}"
      dll_files.each { |fn|
        src = File.join msys2_dll_bin_path, fn
        if File.exist? src
          puts "#{COL_SPACE}#{fn}"
          cp src, File.join(dest, fn)
        else
          puts "#{COL_SPACE}ERROR: #{File.basename(fn)} does not exist"
        end
      }
      dll_dirs = lib_files.map{ | fn| File.dirname(fn) }.uniq
      dll_dirs.each { |d|
        src = File.join D_MINGW, "lib", d
        if Dir.exist?(src)
          dest = File.join D_INSTALL, 'bin', 'lib', d
          FileUtils.mkdir_p dest unless Dir.exist? dest
          `xcopy /s /q #{src.gsub('/', '\\')} #{dest.gsub('/', '\\')}`
          puts "#{COL_SPACE}Copy dir #{d}"
        else
          puts "#{COL_SPACE}ERROR: Dir #{src} does not exist"
        end
      }
    end

    # Adds private assembly data to ruby.exe and rubyw.exe files
    def add_priv_assm
      bin_dir = File.join D_INSTALL, "bin"
      Dir.chdir(bin_dir) { |d|
        ['ruby.exe', 'rubyw.exe'].each { |fn|
          image = File.binread fn
          image.sub!(/^<!-- pendency>/  , '  <dependency>')
          image.sub!(/^  <\/depende -->/, '  </dependency>')
          File.binwrite fn, image
        }
      }
    end

    def cp(src, dest)
      unless Dir.exist? (dest_dir = File.dirname dest)
        FileUtils.mkdir_p dest_dir
      end
      FileUtils.copy_file(src, dest, preserve: true)
    end

    # Creates manifest xml file
    # @param dlls [Array<String>] file list
    # @param dest [Array<String>] dest dir for manifest file
    def create_manifest(dlls, dest)
      manifest = +"<?xml version='1.0' encoding='UTF-8' standalone='yes'?>\n" \
                 "<assembly xmlns='urn:schemas-microsoft-com:asm.v1' manifestVersion='1.0'>\n" \
                 "  <assemblyIdentity type='win32' name='ruby_builtin_dlls' version='1.0.0.0'/>\n"
      dlls.each { |fn| manifest << "  <file name='#{File.basename(fn)}'/>\n" }
      manifest << "</assembly>\n"
      File.open( File.join(dest, 'ruby_builtin_dlls.manifest'), 'wb') { |f| f << manifest }
    end

    def find_dlls(pkgs, pkg_pre)
      orig_path = ENV['PATH']
      ENV['PATH'] = "#{File.join D_MSYS2, 'usr/bin'};#{orig_path}"
      re_dep = /^Depends On +: +([^\r\n]+)/i
      dir = ENV['MINGW_PREFIX'][1..-1]
      re_bin = /\A\S+ \/#{dir}\/bin\//i
      re_lib = /\A\S+ \/#{dir}\/lib\//i
      bin_dlls = []
      lib_dlls = []
      pkg_files_added = []
      # add correct package prefix
      pkgs = pkgs.split(' ').map { |e| "#{pkg_pre}-#{e}"}

      while !pkgs.empty? do
        depends = []
        files = `pacman.exe -Ql #{pkgs.join(' ')} | grep dll`
        files.each_line(chomp: true) { |dll|
          next unless dll.end_with? '.dll'
          if    dll.match? re_bin ; bin_dlls << dll.sub(re_bin, '')
          elsif dll.match? re_lib ; lib_dlls << dll.sub(re_lib, '')
          else
            puts "#{dll.ljust(COL_WID)} Unknown dll location!"
          end
        }
        pkg_files_added += pkgs
        if info = `pacman.exe -Qi #{pkgs.join(' ')}`
          info.scan(re_dep) { |dep|
            next if /\ANone/ =~ dep[0]
            depends += dep[0].split(/\s+/)
          }
          if depends && !depends.empty?
            depends.uniq!
            depends.reject! { |e| pkg_files_added.include?(e) }
            unless depends.empty?
              stdin, stdout, stderr = Open3.popen3("pacman -Q #{depends.join(' ')}")
              errs = stderr.read
              stdin.close ; stdout.close ; stderr.close
              if errs && !errs.strip.empty?
                errs.scan(/'([^']+)'/) { |e|
                  depends[depends.index(e[0])] += '-git'
                }
              end
            end
          end
        end
        pkgs = depends
      end
      ENV['PATH'] = orig_path
      [ ( bin_dlls.uniq.sort || [] ), ( lib_dlls.uniq.sort || [] ) ]
    end

  end
end
PostInstall2.run