# Copyright (c) 2023 M.J.N. Corino, The Netherlands
#
# This software is released under the MIT license.

###
# wxRuby3 buildtools configuration
###

require_relative './unixish'

require 'uri'


if ENV['RI_DEVKIT'].nil?
  begin
    require 'devkit'
  rescue LoadError
    STDERR.puts "Missing a fully installed & configured Ruby devkit. Make sure to install the Ruby devkit with MSYS2 and MINGW toolchains."
    exit(1)
  end
end

module WXRuby3

  module Config

    module Platform

      SWIG_URL = 'https://sourceforge.net/projects/swig/files/swigwin/swigwin-4.2.0/swigwin-4.2.0.zip/download'
      SWIG_ZIP = 'swigwin-4.2.0.zip'

      DOXYGEN_URL = 'https://www.doxygen.nl/files/doxygen-1.10.0.windows.x64.bin.zip'

      GIT_URL = 'https://github.com/git-for-windows/git/releases/download/v2.43.0.windows.1/MinGit-2.43.0-64-bit.zip'

      def self.included(base)
        base.class_eval do
          include Config::UnixLike

          attr_reader :rescomp

          alias :base_ldflags :ldflags
          def ldflags(target)
            "-Wl,-soname,#{File.basename(target)} #{base_ldflags(target)}"
          end

          def debug_command(*args)
            args.unshift(nix_path(FileUtils::RUBY))
            args.unshift('--args')
            args.unshift('gdb')
            args.join(' ')
          end

          # override accessor to guarantee win path
          def wx_setup_h
            if @wx_setup_h.index('/')
              @wx_setup_h = win_path(@wx_setup_h)
            end
            @wx_setup_h
          end

          def dll_mask
            "{#{dll_ext},dll}"
          end

          def do_link(pkg)
            # have to use option file for objects to link on windows because command line gets too long
            ftmp = Tempfile.new('object')
            ftmp.puts pkg.all_obj_files.collect { |o| File.join('..', o) }.join(' ')
            ftmp.close # close but do not unlink
            objs = "@#{ftmp.path}"
            depsh = pkg.dep_libnames.collect { |dl| "#{dl}.#{dll_ext}" }.join(' ')
            sh "cd lib && #{WXRuby3.config.ld} #{WXRuby3.config.ldflags(pkg.lib_target)} #{objs} #{depsh} " +
                 "#{WXRuby3.config.libs} #{WXRuby3.config.link_output_flag}#{pkg.lib_target}"
            ftmp.unlink # cleanup
          end

          def check_tool_pkgs
            pkg_deps = []
            pkg_deps << 'doxygen' if expand("which #{get_cfg_string('doxygen')} 2>/dev/null").strip.empty?
            pkg_deps << 'swig' if expand("which #{get_cfg_string('swig')} 2>/dev/null").strip.empty?
            pkg_deps << 'git' if expand("which #{get_cfg_string('git')} 2>/dev/null").strip.empty?
            pkg_deps
          end

          def install_prerequisites
            pkg_deps = super
            unless pkg_deps.empty?
              # autoinstall or not?
              unless wants_autoinstall?
                STDERR.puts <<~__ERROR_TXT
                  ERROR: This system lacks installed versions of the following required software packages:
                    #{pkg_deps.join(', ')}
                    
                    Install these packages and try again.
                  __ERROR_TXT
                exit(1)
              end
              # if SWIG was not found in the PATH
              if pkg_deps.include?('swig')
                $stdout.print 'Installing SWIG...' if run_silent?
                # download and install SWIG
                fname = download_and_install(SWIG_URL, SWIG_ZIP, 'swig.exe')
                $stdout.puts 'done!' if run_silent?
                Config.instance.log_progress("Installed #{fname}")
                set_config('swig', fname)
                Config.save
              end
              # if doxygen was not found in the PATH
              if pkg_deps.include?('doxygen')
                $stdout.print 'Installing Doxygen...' if run_silent?
                # download and install doxygen
                fname = download_and_install(DOXYGEN_URL, File.basename(URI(DOXYGEN_URL).path), 'doxygen.exe', 'doxygen')
                $stdout.puts 'done!' if run_silent?
                Config.instance.log_progress("Installed #{fname}")
                set_config('doxygen', fname)
                Config.save
              end
              # if git was not found in the PATH
              if pkg_deps.include?('git')
                $stdout.print 'Installing Git...' if run_silent?
                # download and install doxygen
                fname = download_and_install(GIT_URL, File.basename(URI(GIT_URL).path), 'git.exe', 'git')
                $stdout.puts 'done!' if run_silent?
                Config.instance.log_progress("Installed #{fname}")
                set_config('git', fname)
                Config.save
              end
            end
            []
          end

          # only called after src gem build
          def cleanup_prerequisites
            tmp_tool_root = File.join(ENV['HOME'].gsub("\\", '/'), '.wxruby3')
            path = get_cfg_string('swig')
            unless path.empty? || !path.start_with?(tmp_tool_root)
              path = File.dirname(path) while File.dirname(path) != tmp_tool_root
              rm_rf(path)
            end
            path = get_cfg_string('doxygen')
            unless path.empty? || !path.start_with?(tmp_tool_root)
              path = File.dirname(path) while File.dirname(path) != tmp_tool_root
              rm_rf(path)
            end
            path = get_cfg_string('git')
            unless path.empty? || !path.start_with?(tmp_tool_root)
              path = File.dirname(path) while File.dirname(path) != tmp_tool_root
              rm_rf(path)
            end
          end

          def expand(cmd)
            super("bash -c \"#{cmd}\"")
          end

          def bash(*cmd, **kwargs)
            env = ::Hash === cmd.first ? cmd.shift : nil
            opts = ::Hash === cmd.last ? cmd.pop : nil
            cmd = ['bash', '-c', cmd.join(' ')]
            cmd.unshift(env) if env
            super(*cmd, **kwargs)
          end

          private

          def download_and_install(url, zip, exe, unpack_to=nil)
            # make sure the download destination exists
            tmp_tool_root = File.join(ENV['HOME'].gsub("\\", '/'), '.wxruby3')
            dest = unpack_to ? File.join(tmp_tool_root, unpack_to) : File.join(tmp_tool_root, File.basename(zip, '.*'))
            mkdir(tmp_tool_root) unless File.directory?(tmp_tool_root)
            # download
            chdir(tmp_tool_root) do
              unless sh("curl -L #{url} --output #{zip}")
                STDERR.puts "ERROR: Failed to download installation package for #{exe}"
                exit(1)
              end
              # unpack
              unless sh("powershell Expand-Archive -LiteralPath '#{zip}' -DestinationPath #{dest} -Force")
                STDERR.puts "ERROR: Failed to unpack installation package for #{exe}"
                exit(1)
              end
              # cleanup
              rm_f(zip)
            end
            # find executable
            find_exe(dest, exe)
          end

          def find_exe(path, exe)
            fp = Dir.glob(File.join(path, '*')).find { |p| File.file?(p) && File.basename(p) == exe }
            unless fp
              Dir.glob(File.join(path, '*')).each do |p|
                fp = find_exe(p, exe) if File.directory?(p)
                return fp if fp
              end
            end
            fp
          end

          def wx_make
            bash('make && make install')
          end

          def wx_generate_xml
            doxygen = get_cfg_string("doxygen")
            doxygen = nix_path(doxygen) unless doxygen == 'doxygen'
            chdir(File.join(ext_path, 'wxWidgets', 'docs', 'doxygen')) do
              unless bash({ 'DOXYGEN' => doxygen,  'WX_SKIP_DOXYGEN_VERSION_CHECK' => '1' }, './regen.sh', 'xml')
                $stderr.puts 'ERROR: Failed to generate wxWidgets XML API specifications for parsing by wxRuby3.'
                exit(1)
              end
            end
          end

          def respawn_rake(argv = ARGV)
            Kernel.exec('rake', *argv)
          end

          def nix_path(winpath)
            (winpath.nil? || winpath.empty?) ? '' : `cygpath -a -u #{winpath}`.strip
          end

          # Allow specification of custom wxWidgets build (mostly useful for
          # static wxRuby3 builds)
          def win_path(nixpath)
            (nixpath.nil? || nixpath.empty?) ? '' : `cygpath -a -w #{nixpath}`.strip
          end

          def get_wx_path
            nix_path(super)
          end

        end
      end

      def init_platform
        init_unix_platform

        if @wx_version
          # need to convert these to windows paths
          @wx_cppflags.each { |flags| flags.gsub!(/-I(\S+)/) { |s| "-I#{win_path($1)}" } }
          @wx_libs.each { |libflag| libflag.gsub!(/-L(\S+)/) { |s| "-L#{win_path($1)}" } }

          @extra_cflags.concat %w[-Wno-unused-function -Wno-conversion-null -Wno-maybe-uninitialized]
          @extra_cflags << ' -Wno-deprecated-declarations' unless @no_deprecated

          # create a .dll binary
          @extra_ldflags << '-shared'

          @ruby_ldflags.each { |flags| flags.sub!(' $(DEFFILE)', '') } # cleanup for older RubyInstaller versions
          @ruby_ldflags.each { |flags| flags.gsub!(/-s(\s|\Z)/, '') } if @debug_build # do not strip debug symbols for debug build
          @ruby_ldflags << '-s' if @release_build  # strip debug symbols for release build
          @ruby_cppflags << RB_CONFIG['debugflags'] if @debug_build
          @ruby_cppflags.each { |flags| flags.gsub!(/-O\d/, '-O0') } if @debug_build # disable optimizations for debug build

          unless @wx_path.empty?
            exec_pfx = win_path(wx_config("--exec-prefix"))
            libdirs = [File.join(exec_pfx, 'bin')]
            libdirs << win_path(File.join(ENV['MSYSTEM_PREFIX'], 'bin'))
            @exec_env['RUBY_DLL_PATH'] = "#{ENV['RUBY_DLL_PATH']};#{dest_dir};#{libdirs.join(';')}"
          end

          @rescomp = wx_config('--rescomp').gsub(/--include-dir\s+(\S+)/) { |s| "--include-dir #{win_path($1)}" }
          @rescomp << " --include-dir #{@ext_path}"
          @rescomp << ' --define __WXMSW__ --define wxUSE_DPI_AWARE_MANIFEST=2 --define wxUSE_RC_MANIFEST=1 --define ISOLATION_AWARE_ENABLED'
          @rescomp << ' --define WXUSINGDLL'
          @extra_cflags << '-DISOLATION_AWARE_ENABLED'
          if @wx_version >= '3.3.0'
            @extra_cflags << '-D_UNICODE' << '-DUNICODE'
          end
        end
      end
      private :init_platform

    end

  end

end