require File.dirname(__FILE__) + '/screenshot/bitmap_maker'
require File.dirname(__FILE__) + '/util'

module Win32
  # Captures screenshots with Ruby on Windows
  class Screenshot
    class << self

      # captures foreground
      def foreground(&proc)
        hwnd = BitmapMaker.foreground_window
        BitmapMaker.capture_all(hwnd, &proc)
      end

      # captures area of the foreground
      # where *x1* and *y1* are 0 in the upper left corner and
      # *x2* specifies the width and *y2* the height of the area to be captured
      def foreground_area(x1, y1, x2, y2, &proc)
        hwnd = BitmapMaker.foreground_window
        validate_coordinates(hwnd, x1, y1, x2, y2)
        BitmapMaker.capture_area(hwnd, x1, y1, x2, y2, &proc)
      end

      # captures visible view of the screen
      #
      # to make screenshot of the real desktop, all
      # windows must be minimized before
      def desktop(&proc)
        hwnd = BitmapMaker.desktop_window
        BitmapMaker.capture_all(hwnd, &proc)
      end

      # captures area of the visible view of the screen
      # where *x1* and *y1* are 0 in the upper left corner and
      # *x2* specifies the width and *y2* the height of the area to be captured
      #
      # to make screenshot of the real desktop, all
      # windows must be minimized before
      def desktop_area(x1, y1, x2, y2, &proc)
        hwnd = BitmapMaker.desktop_window
        validate_coordinates(hwnd, x1, y1, x2, y2)
        BitmapMaker.capture_area(hwnd, x1, y1, x2, y2, &proc)
      end

      # captures window with a *title_query* and waits *pause* (by default is 0.5)
      # seconds after trying to set window to the foreground
      def window(title_query, pause=0.5, &proc)
        hwnd = Util.window_hwnd(title_query)
        hwnd(hwnd, pause, &proc)
      end

      # captures area of the window with a *title_query*
      # where *x1* and *y1* are 0 in the upper left corner and
      # *x2* specifies the width and *y2* the height of the area to be captured
      def window_area(title_query, x1, y1, x2, y2, pause=0.5, &proc)
        hwnd = Util.window_hwnd(title_query)
        hwnd_area(hwnd, x1, y1, x2, y2, pause, &proc)
      end

      # captures by window handle
      def hwnd(hwnd, pause=0.5, &proc)
        BitmapMaker.prepare_window(hwnd, pause)
        BitmapMaker.capture_all(hwnd, &proc)
      end

      # captures area of the window with a handle of *hwnd*
      # where *x1* and *y1* are 0 in the upper left corner and
      # *x2* specifies the width and *y2* the height of the area to be captured
      def hwnd_area(hwnd, x1, y1, x2, y2, pause=0.5, &proc)
        validate_coordinates(hwnd, x1, y1, x2, y2)
        BitmapMaker.prepare_window(hwnd, pause)
        BitmapMaker.capture_area(hwnd, x1, y1, x2, y2, &proc)
      end

      private

      def validate_coordinates(hwnd, *coords)
        specified_coordinates = coords.join(', ')
        if coords.any? {|c| c < 0}
          raise "specified coordinates (#{specified_coordinates}) are invalid - cannot be negative!"
        end
        x1, y1, x2, y2 = *coords
        if x1 >= x2 || y1 >= y2
          raise "specified coordinates (#{specified_coordinates}) are invalid - cannot have x1 > x2 or y1 > y2!"
        end

        max_width, max_height = Util.dimensions_for(hwnd)
        if x2 > max_width || y2 > max_height
          raise "specified coordinates (#{specified_coordinates}) are invalid - maximum are x2=#{max_width} and y2=#{max_height}!"
        end
      end
    end

  end
end