# frozen_string_literal: true

require 'pedump'
require 'fileutils'
require 'forwardable'

module AppInfo
  # Windows PE parser
  #
  # @see https://learn.microsoft.com/zh-cn/windows/win32/debug/pe-format Microsoft PE Format
  class PE < File
    include Helper::HumanFileSize
    include Helper::Archive
    extend Forwardable

    ARCH = {
      0x014c => 'x86',
      0x0200 => 'Intel Itanium',
      0x8664 => 'x64',
      0x1c0 => 'arm',
      0xaa64 => 'arm64',
      0x5032 => 'RISC-v 32',
      0x5064 => 'RISC-v 64',
      0x5128 => 'RISC-v 128'
    }.freeze

    # @return [Symbol] {Manufacturer}
    def manufacturer
      Manufacturer::MICROSOFT
    end

    # @return [Symbol] {Platform}
    def platform
      Platform::WINDOWS
    end

    # @return [Symbol] {Device}
    def device
      Device::WINDOWS
    end

    # return file size
    # @example Read file size in integer
    #   aab.size                    # => 3618865
    #
    # @example Read file size in human readabale
    #   aab.size(human_size: true)  # => '3.45 MB'
    #
    # @param [Boolean] human_size Convert integer value to human readable.
    # @return [Integer, String]
    def size(human_size: false)
      file_to_human_size(@file, human_size: human_size)
    end

    def binary_size(human_size: false)
      file_to_human_size(binary_file, human_size: human_size)
    end

    # @!method product_name
    #   @see VersionInfo#product_name
    #   @return [String]
    # @!method product_version
    #   @see VersionInfo#product_version
    #   @return [String]
    # @!method company_name
    #   @see VersionInfo#company_name
    #   @return [String]
    # @!method assembly_version
    #   @see VersionInfo#assembly_version
    #   @return [String]
    # @!method file_version
    #   @see VersionInfo#file_version
    #   @return [String]
    # @!method file_description
    #   @see VersionInfo#file_description
    #   @return [String]
    # @!method copyright
    #   @see VersionInfo#copyright
    #   @return [String]
    #   @return [String]
    # @!method special_build
    #   @see VersionInfo#special_build
    #   @return [String]
    # @!method private_build
    #   @see VersionInfo#private_build
    #   @return [String]
    # @!method original_filename
    #   @see VersionInfo#original_filename
    #   @return [String]
    # @!method internal_name
    #   @see VersionInfo#internal_name
    #   @return [String]
    # @!method legal_trademarks
    #   @see VersionInfo#legal_trademarks
    #   @return [String]
    def_delegators :version_info, :product_name, :product_version, :company_name, :assembly_version,
                   :file_version, :file_description, :copyright, :special_build, :private_build,
                   :original_filename, :internal_name, :legal_trademarks

    alias name product_name

    # Find {#product_version} then fallback to {#file_version}
    # @return [String, nil]
    def release_version
      product_version || file_version
    end

    # Find {#special_build}, {#private_build} then fallback to {#assembly_version}
    # @return [String, nil]
    def build_version
      special_build || private_build || assembly_version
    end

    # @return [String]
    def archs
      ARCH[image_file_header.Machine] || 'unknown'
    end
    alias architectures archs

    # @return [Hash{String => String}] imports imports of libraries
    def imports
      @imports ||= pe.imports.each_with_object({}) do |import, obj|
        obj[import.module_name] = import.first_thunk.map(&:name).compact
      end
    end

    # @return [Array{String}] icons paths of bmp image icons
    def icons
      @icons ||= lambda {
        # Fetch the largest size icon
        files = []
        pe.resources&.find_all do |res|
          next unless res.type == 'ICON'

          filename = "#{::File.basename(file, '.*')}-#{res.type}-#{res.id}.bmp"
          icon_file = tempdir(filename, prefix: 'pe', system: true)
          mask_icon_file = icon_file.sub('.bmp', '.mask.bmp')

          begin
            ::File.open(icon_file, 'wb') do |f|
              f << res.restore_bitmap(io)
            end

            if res.bitmap_mask(io)
              mask_icon_file = icon_file.sub('.bmp', '.mask.bmp')
              ::File.open(mask_icon_file, 'wb') do |f|
                f << res.bitmap_mask(io)
              end
            end
          rescue StandardError => e
            # ignore pedump throws any exception.
            raise e unless e.backtrace.first.include?('pedump')

            FileUtils.rm_f(icon_file)
          ensure
            next unless ::File.exist?(icon_file)

            mask_file = ::File.exist?(mask_icon_file) ? mask_icon_file : nil
            files << icon_metadata(icon_file, mask_file: mask_file)
          end
        end

        files
      }.call
    end

    # @return [PEdump]
    def pe
      @pe ||= lambda {
        pe = PEdump.new(io)
        pe.logger.level = Logger::FATAL # ignore :warn logger output
        pe
      }.call
    end

    # @return [VersionInfo]
    def version_info
      @version_info ||= VersionInfo.new(pe.version_info)
    end

    def clear!
      @io = nil
      @pe = nil
      @icons = nil
      @imports = nil
    end

    # @return [String] binary_file path
    def binary_file
      @binary_file ||= lambda {
        file_io = ::File.open(@file, 'rb')
        return @file unless file_io.read(100) =~ AppInfo::ZIP_RETGEX

        zip_file = Zip::File.open(@file)
        zip_entry = zip_file.glob('*.exe').first
        raise NotFoundError, 'Not found .exe file in archive file' if zip_entry.nil?

        exe_file = tempdir(zip_entry.name, prefix: 'pe-exe', system: true)
        zip_entry.extract(exe_file)
        zip_file.close

        return exe_file
      }.call
    end

    private

    def image_file_header
      @image_file_header ||= pe.pe.image_file_header
    end

    # @return [Hash{Symbol => String}]
    def icon_metadata(file, mask_file: nil)
      {
        name: ::File.basename(file),
        file: file,
        mask: mask_file,
        dimensions: ImageSize.path(file).size
      }
    end

    # @return [File]
    def io
      @io ||= ::File.open(binary_file, 'rb')
    end

    # VersionInfo class
    #
    # @see https://learn.microsoft.com/zh-cn/windows/win32/menurc/versioninfo-resource
    class VersionInfo
      def initialize(raw)
        @raw = raw
      end

      # @return [String]
      def company_name
        @company_name ||= value_of('CompanyName')
      end

      # @return [String]
      def product_name
        @product_name ||= value_of('ProductName')
      end

      # @return [String]
      def product_version
        @product_version ||= value_of('ProductVersion')
      end

      # @return [String]
      def assembly_version
        @assembly_version ||= value_of('Assembly Version')
      end

      # @return [String]
      def file_version
        @file_version ||= value_of('FileVersion')
      end

      # @return [String, nil]
      def file_description
        @file_description ||= value_of('FileDescription')
      end

      # @return [String, nil]
      def special_build
        @special_build ||= value_of('SpecialBuild')
      end

      # @return [String, nil]
      def private_build
        @private_build ||= value_of('PrivateBuild')
      end

      # @return [String]
      def original_filename
        @original_filename ||= value_of('OriginalFilename')
      end

      # @return [String]
      def internal_name
        @internal_name ||= value_of('InternalName')
      end

      # @return [String]
      def legal_trademarks
        @legal_trademarks ||= value_of('LegalTrademarks')
      end

      # @return [String]
      def copyright
        @copyright ||= value_of('LegalCopyright')
      end

      private

      def value_of(key)
        info.each do |v|
          return v[:Value] if v[:szKey] == key.to_s
        end

        nil
      end

      def info
        return @info if @info

        @raw.each do |item|
          next unless item.is_a?(PEdump::VS_VERSIONINFO)

          versions = item[:Children].select { |v| v.is_a?(PEdump::StringFileInfo) }
          next if versions.empty?

          @info = versions[0][:Children][0][:Children]
          return @info
        end

        @info = []
      end
    end
  end
end