module Celerity
  class Frame < Element
    include Container
    include XpathSupport

    attr_accessor :page

    TAGS = [Identifier.new('frame'), Identifier.new('iframe')]
    ATTRIBUTES = BASE_ATTRIBUTES | [
                                      :frameborder,
                                      :longdesc,
                                      :marginheight,
                                      :marginwidth,
                                      :name,
                                      :noresize,
                                      :scrolling,
                                      :src,
                                    ]
    DEFAULT_HOW = :name

    #
    # Override the default locate to handle frame and inline frames.
    # @api private
    #

    def locate
      super
      if @object
        @inline_frame_object = @object.getEnclosedWindow.getFrameElement
        self.page            = @object.getEnclosedPage
        if (frame = self.page.getDocumentElement)
          @object = frame
        end
      end
    end

    #
    # Override assert_exists to raise UnknownFrameException (for Watir compatibility)
    # @api private
    #

    def assert_exists
      locate
      unless @object
        raise UnknownFrameException, "unable to locate frame, using #{identifier_string}"
      end
    end

    #
    # Executes the given JavaScript string in this frame. (Celerity only)
    #

    def execute_script(source)
      assert_exists
      @page.executeJavaScript(source.to_s).getJavaScriptResult
    end

    #
    # Updates the brwoser page with the page from this frame's top window.
    # Used internally.
    #
    # @api private
    #

    def update_page(value)
      @browser.page = value.getEnclosingWindow.getTopWindow.getEnclosedPage
    end

    def to_s
      assert_exists
      Celerity::Util.create_string @inline_frame_object
    end

    def method_missing(meth, *args, &blk)
      meth = selector_to_attribute(meth)
      if self.class::ATTRIBUTES.include?(meth)
        assert_exists
        @inline_frame_object.getAttribute(meth.to_s)
      else
        Log.warn "Element\#method_missing calling super with #{meth.inspect}"
        super
      end
    end

  end # Frame
end # Celerity