module GDAL
  module DriverMixins
    module Extensions
      def self.included(base)
        base.extend(ClassMethods)
      end

      module ClassMethods
        # @return [Array<String>]
        def short_names
          names = count.times.map do |i|
            driver = at_index(i)
            driver.short_name
          end

          names.compact.sort
        end

        # @return [Array<String>]
        def long_names
          names = count.times.map do |i|
            at_index(i).long_name
          end

          names.compact.sort
        end

        # @return [Hash{String => String}] Keys are driver short names, values are
        #   driver long names.
        def self.names
          names = count.times.each_with_object({}) do |i, obj|
            driver = at_index(i)
            obj[driver.short_name] = driver.long_name
          end

          Hash[names.sort]
        end
      end

      # The things that this driver can do, as reported by its metadata.
      # Possibilities include:
      #   * :open
      #   * :create
      #   * :copy
      #   * :virtual_io
      #   * :rasters
      #   * :vectors
      #
      # @return [Array<Symbol>]
      def capabilities
        caps = []
        caps << :open if can_open_datasets?
        caps << :create if can_create_datasets?
        caps << :copy if can_copy_datasets?
        caps << :virtual_io if can_do_virtual_io?
        caps << :rasters if can_do_rasters?
        caps << :vectors if can_do_vectors?

        caps
      end

      # @return [Boolean]
      def can_open_datasets?
        metadata_item('DCAP_OPEN') == 'YES'
      end

      # @return [Boolean]
      def can_create_datasets?
        metadata_item('DCAP_CREATE') == 'YES'
      end

      # @return [Boolean]
      def can_copy_datasets?
        metadata_item('DCAP_CREATECOPY') == 'YES'
      end

      # @return [Boolean]
      def can_do_virtual_io?
        metadata_item('DCAP_VIRTUALIO') == 'YES'
      end

      # @return [Boolean]
      def can_do_rasters?
        metadata_item('DCAP_RASTER') == 'YES'
      end

      # @return [Boolean]
      def can_do_vectors?
        metadata_item('DCAP_VECTOR') == 'YES'
      end
    end
  end
end