isomorfeus-puppetmaster 0.5.6 → 0.6.2

data/lib/isomorfeus/puppeteer/execution_context.rb

@@ -0,0 +1,90 @@
+class Puppeteer::ExecutionContext
+  def eval_ruby(ruby_source = '', &block)
+    ruby_source = Isomorfeus::Puppetmaster.block_source_code(&block) if block_given?
+    compiled_ruby = Isomorfeus::Puppetmaster.compile_ruby_source(ruby_source)
+    res = evaluate "function(){ try { return #{compiled_ruby} } catch (e) { return { error: e.name, message: e.message, stack: e.stack }; }}"
+    if res.is_a?(Hash) && res.key?('error') && res.key?('message') && res.key?('stack')
+      e = RuntimeError.new("#{res['error']}: #{res['message']}")
+      e.set_backtrace(res['stack'].lines)
+      raise e
+    end
+    res
+  end
+
+  def eval_with_opal(ruby_source = '', &block)
+    ruby_source = Isomorfeus::Puppetmaster.block_source_code(&block) if block_given?
+    compiled_ruby = Isomorfeus::Puppetmaster.compile_ruby_source(ruby_source)
+    res = evaluate <<~JAVASCRIPT
+      function(){
+        if (typeof Opal === "undefined") {
+          #{Isomorfeus::Puppetmaster.opal_prelude}
+        }
+        try { return #{compiled_ruby} }
+        catch (e) { return { error: e.name, message: e.message, stack: e.stack }; }
+      }
+    JAVASCRIPT
+    if res.is_a?(Hash) && res.key?('error') && res.key?('message') && res.key?('stack')
+      e = RuntimeError.new("#{res['error']}: #{res['message']}")
+      e.set_backtrace(res['stack'].lines)
+      raise e
+    end
+    res
+  end
+
+
+  def await_ruby(ruby_source = '', &block)
+    ruby_source = Isomorfeus::Puppetmaster.block_source_code(&block) if block_given?
+    ruby_source = "#{ruby_source}.then { |result| $promise_result = result; $promise_resolved = true }.fail { |result| $promise_result = result; $promise_resolved = true }"
+    compiled_ruby = Isomorfeus::Puppetmaster.compile_ruby_source(ruby_source)
+    evaluate <<~JAVASCRIPT
+      function(){
+        fun = function() {
+          if (Opal) {
+            Opal.gvars.promise_resolved = false;
+            Opal.await_ruby_exception = null;
+            try {
+              return #{compiled_ruby}
+            } catch (e) {
+              Opal.await_ruby_exception = e;
+              Opal.gvars.promise_resolved = true;
+            }
+          } else {
+            setTimeout(fun, 100);
+          }
+        }
+        fun();
+      }
+    JAVASCRIPT
+    have_result = false
+    start = Time.now
+    until have_result do
+      raise "await_ruby: execution timed out! Is Opal available?" if (Time.now - start) > 30
+      have_result = evaluate 'function() { return Opal ? Opal.gvars.promise_resolved : null; }'
+      sleep 0.1 unless have_result
+    end
+    result, exception = evaluate <<~JAVASCRIPT
+      function () {
+        var result;
+        var exception = false;
+        if (Opal.await_ruby_exception) {
+          var e = Opal.await_ruby_exception;
+          exception = { message: e.message, name: e.name, stack: e.stack }
+        } else if (Opal.gvars.promise_result['$respond_to?']('is_a?') && Opal.gvars.promise_result['$is_a?'](Opal.Exception)) {
+          let r = Opal.gvars.promise_result;
+          exception = { message: r.$message(), name: r.$class().$name(), stack: r.$backtrace() }
+        } else if (Opal.gvars.promise_result['$respond_to?']('to_n')) {
+          result = Opal.gvars.promise_result.$to_n()
+        } else { result = Opal.gvars.promise_result };
+        delete Opal.gvars.promise_result;
+        delete Opal.gvars.promise_resolved;
+        return [result, exception];
+      }
+    JAVASCRIPT
+    if exception
+      e = RuntimeError.new("#{exception['name']}: #{exception['message']}")
+      e.set_backtrace(exception['stack'])
+      raise e
+    end
+    result
+  end
+end

data/lib/isomorfeus/puppeteer/frame.rb

@@ -0,0 +1,33 @@
+class Puppeteer::Frame
+  def Sx1(x)
+    self.Sx(x).first
+  end
+
+  def eval_ruby(ruby_source = '', &block)
+    @main_world.eval_ruby(ruby_source, &block)
+  end
+
+  def eval_with_opal(ruby_source = '', &block)
+    @main_world.eval_with_opal(ruby_source, &block)
+  end
+
+  def await_ruby(ruby_source = '', &block)
+    @main_world.await_ruby(ruby_source, &block)
+  end
+
+  alias_method :find, :query_selector
+
+  alias_method :_original_goto, :goto
+
+  def goto(uri, referer: nil, timeout: nil, wait_until: nil)
+    parsed_uri = URI.parse(uri)
+    app = Isomorfeus::Puppetmaster.served_app
+    unless parsed_uri.host && parsed_uri.scheme
+      parsed_uri.host = app.host
+      parsed_uri.port = app.port
+      parsed_uri.scheme = app.scheme
+    end
+    _original_goto(parsed_uri.to_s, referer: referer, timeout: timeout, wait_until: wait_until)
+  end
+  alias_method :visit, :goto
+end

data/lib/isomorfeus/puppeteer/js_handle.rb

@@ -0,0 +1,13 @@
+class Puppeteer::JSHandle
+  def eval_ruby(ruby_source = '', &block)
+    execution_context.eval_ruby(ruby_source, &block)
+  end
+
+  def eval_with_opal(ruby_source = '', &block)
+    execution_context.eval_with_opal(ruby_source, &block)
+  end
+
+  def await_ruby(ruby_source = '', &block)
+    execution_context.await_ruby(ruby_source, &block)
+  end
+end

data/lib/isomorfeus/puppeteer/page.rb

@@ -0,0 +1,252 @@
+class Puppeteer::Page
+  attr_accessor :puppetmaster_session
+
+  def Sx1(x)
+    self.Sx(x).first
+  end
+
+  def check_el(selector)
+    el = query_selector(selector) rescue nil
+    el = Sx1(selector) rescue nil if el.nil?
+    el = Sx1("//*[@name=\"#{selector}\"]") rescue nil if el.nil?
+    el = Sx1("//*[@title=\"#{selector}\"]") rescue nil if el.nil?
+    return el.check if el.is_a?(Puppeteer::ElementHandle)
+    false
+  end
+
+  def uncheck_el(selector)
+    el = query_selector(selector) rescue nil
+    el = Sx1(selector) rescue nil if el.nil?
+    el = Sx1("//*[@name=\"#{selector}\"]") rescue nil if el.nil?
+    el = Sx1("//*[@title=\"#{selector}\"]") rescue nil if el.nil?
+    return el.uncheck if el.is_a?(Puppeteer::ElementHandle)
+    false
+  end
+
+  def click_el(selector, button: 'left', click_count: 1, delay: nil)
+    el = query_selector(selector) rescue nil
+    return click(selector, button: button, click_count: click_count, delay: delay) if el.is_a?(Puppeteer::ElementHandle)
+    el = Sx1(selector) rescue nil
+    el = Sx1("//*[normalize-space(text())=\"#{selector}\"]") rescue nil if el.nil?
+    el = Sx1("//*[@name=\"#{selector}\"]") rescue nil if el.nil?
+    el = Sx1("//*[@title=\"#{selector}\"]") rescue nil if el.nil?
+    el = Sx1("//*[@value=\"#{selector}\"]") rescue nil if el.nil?
+    return el.click(button: button, click_count: click_count, delay: delay) if el.is_a?(Puppeteer::ElementHandle)
+    false
+  end
+
+  def double_click(selector, button: 'left', delay: nil)
+    click(selector, button: button, click_count: 2, delay: delay)
+  end
+
+  def double_click_el(selector, button: 'left', delay: nil)
+    click_el(selector, button: button, click_count: 2, delay: delay)
+  end
+
+  def eval_ruby(ruby_source = '', &block)
+    main_frame.eval_ruby(ruby_source, &block)
+  end
+
+  def eval_with_opal(ruby_source = '', &block)
+    main_frame.eval_with_opal(ruby_source, &block)
+  end
+
+  def await_ruby(ruby_source = '', &block)
+    main_frame.await_ruby(ruby_source, &block)
+  end
+
+  def exec(j, *args)
+    evaluate("function() { #{j.strip} }", *args)
+  end
+
+  def eval(j, *args)
+    r = evaluate_handle("function() { return #{j.strip} }", *args)
+    return r if r.is_a?(Puppeteer::ElementHandle)
+    r&.json_value
+  end
+
+  alias_method :find, :query_selector
+  alias_method :find_all, :query_selector_all
+  alias_method :find_xpath, :Sx1
+  alias_method :find_all_xpath, :Sx
+
+  def find_el(selector)
+    el = query_selector(selector) rescue nil
+    el = Sx1(selector) rescue nil if el.nil?
+    el = Sx1("//*[normalize-space(text())=\"#{selector}\"]") rescue nil if el.nil?
+    el = Sx1("//*[@name=\"#{selector}\"]") rescue nil if el.nil?
+    el = Sx1("//*[@title=\"#{selector}\"]") rescue nil if el.nil?
+    el = Sx1("//*[@value=\"#{selector}\"]") rescue nil if el.nil?
+    el
+  end
+
+  alias_method :_original_goto, :goto
+
+  def goto(uri, referer: nil, timeout: nil, wait_until: nil)
+    parsed_uri = URI.parse(uri)
+    app = Isomorfeus::Puppetmaster.served_app
+    unless parsed_uri.host && parsed_uri.scheme
+      parsed_uri.host = app.host
+      parsed_uri.port = app.port
+      parsed_uri.scheme = app.scheme
+    end
+    response = _original_goto(parsed_uri.to_s, referer: referer, timeout: timeout, wait_until: wait_until)
+    puppetmaster_session.response = response if puppetmaster_session
+    response
+  end
+  alias_method :visit, :goto
+
+  def inner_text
+    evaluate('() => { return document.documentElement.innerText; }').strip.tr("\u00a0", ' ')
+  end
+  alias_method :text, :inner_text
+
+  def isomorphic(ruby_source = '', &block)
+    ruby_source = Isomorfeus::Puppetmaster.block_source_code(&block) if block_given?
+    Isomorfeus::Puppetmaster.served_app.on_server(ruby_source)
+    eval_ruby(ruby_source)
+  end
+
+  def isomorphic_with_opal(ruby_source = '', &block)
+    ruby_source = Isomorfeus::Puppetmaster.block_source_code(&block) if block_given?
+    Isomorfeus::Puppetmaster.served_app.on_server(ruby_source)
+    eval_with_opal(ruby_source)
+  end
+
+  def path
+    URI.parse(url).path.to_s
+  end
+
+  def right_click(selector, click_count: 1, delay: nil)
+    click(selector, button: 'right', click_count: click_count, delay: delay)
+  end
+
+  def right_click_el(selector, click_count: 1, delay: nil)
+    click_el(selector, button: 'right', click_count: click_count, delay: delay)
+  end
+
+  def save_page(f)
+    raise "Invalid filename '#{f}'!" if f.include?('..')
+    raise "Invalid Isomorfeus::Puppetmaster.download_path!" unless Isomorfeus::Puppetmaster.download_path.is_a?(String)
+    File.write(File.join(Isomorfeus::Puppetmaster.download_path, f), content)
+  end
+
+  def save_pdf(f, options = {})
+    raise "Invalid filename '#{f}'!" if f.include?('..')
+    raise "Invalid Isomorfeus::Puppetmaster.download_path!" unless Isomorfeus::Puppetmaster.download_path.is_a?(String)
+    options[:path] = File.join(Isomorfeus::Puppetmaster.download_path, f)
+    pdf(options)
+  end
+
+  def save_screenshot(f, options = {})
+    raise "Invalid filename '#{f}'!" if f.include?('..')
+    raise "Invalid Isomorfeus::Puppetmaster.download_path!" unless Isomorfeus::Puppetmaster.download_path.is_a?(String)
+    options[:path] = File.join(Isomorfeus::Puppetmaster.download_path, f)
+    options[:type] = 'png' unless options.key?(:type)
+    screenshot(**options)
+  end
+
+  def select_el(selector, *args)
+    el = query_selector(selector) rescue nil
+    return select(selector, *args) if el.is_a?(Puppeteer::ElementHandle)
+    el = Sx1(selector) rescue nil
+    el = Sx1("//*[@name=\"#{selector}\"]") rescue nil if el.nil?
+    el = Sx1("//*[@title=\"#{selector}\"]") rescue nil if el.nil?
+    return el.select if el.is_a?(Puppeteer::ElementHandle)
+    false
+  end
+
+  def type_text_el(selector, texttt, delay: nil, clear: true)
+    el = query_selector(selector) rescue nil
+    if el.is_a?(Puppeteer::ElementHandle)
+      if clear
+        el.type_text(' ') if texttt == ''
+        el.set_value('')
+      end
+      return type_text(selector, texttt.to_s, delay: delay)
+    end
+    el = Sx1(selector) rescue nil
+    el = Sx1("//*[@name=\"#{selector}\"]") rescue nil if el.nil?
+    el = Sx1("//*[@title=\"#{selector}\"]") rescue nil if el.nil?
+    if el.is_a?(Puppeteer::ElementHandle)
+      if clear
+        el.type_text(' ') if texttt == ''
+        el.set_value('')
+      end
+      return el.type_text(texttt.to_s, delay: delay)
+    end
+    false
+  end
+
+  # rspec matcher methods
+
+  def has_content?(c)
+    return content.match?(c) if c.is_a?(Regexp)
+    return content.include?(c) if c.is_a?(String)
+    content.include?(c.to_s)
+  end
+
+  def has_no_content?(c)
+    !has_content?(c)
+  end
+
+  def has_path?(p)
+    return path.match?(p) if p.is_a?(Regexp)
+    path == p
+  end
+
+  def has_no_path?(p)
+    !has_path?(p)
+  end
+
+  def has_selector?(s, count: nil, between: nil, max: nil, min: nil)
+    n = nil
+    n = query_selector_all(s).size if count || between || max || min
+    return n == count.to_i if count
+    return n <= max.to_i if max
+    return n >= min.to_i if min
+    return between.include?(n) if between
+    return true if query_selector(s).is_a?(Puppeteer::ElementHandle)
+    false
+  end
+
+  def has_no_selector?(s, count: nil, between: nil, max: nil, min: nil)
+    !has_selector?(s, count: count, between: between, max: max, min: min)
+  end
+
+  def has_text?(t, count: nil, between: nil, max: nil, min: nil)
+    c = inner_text
+    n = nil
+    n = c.scan(t).size if count || between || max || min
+    return n == count.to_i if count
+    return n <= max.to_i if max
+    return n >= min.to_i if min
+    return between.include?(n) if between
+    return c.match?(t) if t.is_a?(Regexp)
+    c.include?(t.to_s)
+  end
+
+  def has_no_text?(t, count: nil, between: nil, max: nil, min: nil)
+    !has_text?(t, count: count, between: between, max: max, min: min)
+  end
+
+  def has_title?(t)
+    ti = title
+    return ti.match?(t) if t.is_a?(Regexp)
+    ti == t
+  end
+
+  def has_no_title?(t)
+    !has_title?(t)
+  end
+
+  def has_xpath?(x, count: nil)
+    e = self.Sx(x)
+    return e.any? if count.nil?
+    return e.size == count
+  end
+
+  def has_no_xpath?(x, count: nil)
+    !has_xpath?(x, count: count)
+  end
+end

data/lib/isomorfeus/puppetmaster/dsl.rb

@@ -1,44 +1,117 @@
 module Isomorfeus
   module Puppetmaster
     module DSL
-      def default_document
-        @puppetmaster_default_document ||= get_default_document
+      @@launch_options = nil
+
+      def set_launch_options(app: nil, args: nil, channel: nil, default_viewport: nil, devtools: nil, dumpio: nil,
+                             env: nil, executable_path: nil, handle_SIGINT: nil, handle_SIGTERM: nil, handle_SIGHUP: nil,
+                             headless: true, ignore_default_args: nil, ignore_https_errors: nil, pipe: nil,
+                             product: :chrome, slow_mo: nil, timeout: nil, user_data_dir: nil)
+        @@launch_options = {
+          app: app,
+          args: args,
+          channel: channel&.to_s,
+          devtools: devtools,
+          default_viewport: default_viewport,
+          dumpio: dumpio,
+          env: env,
+          executable_path: executable_path,
+          handle_SIGINT: handle_SIGINT,
+          handle_SIGTERM: handle_SIGTERM,
+          handle_SIGHUP: handle_SIGHUP,
+          headless: headless,
+          ignore_default_args: ignore_default_args,
+          ignore_https_errors: ignore_https_errors,
+          pipe: pipe,
+          product: product.to_s,
+          slow_mo: slow_mo,
+          timeout: timeout ? timeout : Isomorfeus::Puppetmaster::Session::TIMEOUT,
+          user_data_dir: user_data_dir,
+        }
+      end
+
+      def default_page
+        session.default_page
       end
 
       def goto(uri)
-        puppetmaster_session.document_goto(default_document, uri)
-        @puppetmaster_default_document
+        session.goto(uri)
+        session.default_page
       end
       alias_method :visit, :goto
 
+      def isomorphic(ruby_source = '', &block)
+        ruby_source = Isomorfeus::Puppetmaster.block_source_code(&block) if block_given?
+        on_server(ruby_source)
+        default_page.eval_ruby(ruby_source)
+      end
+
+      def isomorphic_with_opal(ruby_source = '', &block)
+        ruby_source = Isomorfeus::Puppetmaster.block_source_code(&block) if block_given?
+        on_server(ruby_source)
+        default_page.eval_with_opal(ruby_source)
+      end
+
       def on_server(ruby_source = '', &block)
         Isomorfeus::Puppetmaster.served_app.on_server(ruby_source, &block)
       end
 
-      def open_new_session(app = nil, driver = nil)
-        Isomorfeus::Puppetmaster.drivers[driver ? driver : Isomorfeus::Puppetmaster.driver].call(app ? app : Isomorfeus::Puppetmaster.served_app)
+      def new_session(app: nil, args: nil, channel: nil, default_viewport: nil, devtools: nil, dumpio: nil,
+                      env: nil, executable_path: nil, handle_SIGINT: nil, handle_SIGTERM: nil, handle_SIGHUP: nil,
+                      headless: true, ignore_default_args: nil, ignore_https_errors: nil, pipe: nil,
+                      product: :chrome, slow_mo: nil, timeout: nil, user_data_dir: nil)
+        unless @@launch_options.nil?
+          app = app || @@launch_options[:app]
+          channel = channel || @@launch_options[:channel]
+          devtools = @@launch_options[:devtools].nil? ? devtools : @@launch_options[:devtools]
+          default_viewport = default_viewport || @@launch_options[:viewport]
+          dumpio = dumpio || @@launch_options[:dumpio]
+          env = env || @@launch_options[:env]
+          executable_path = executable_path || @@launch_options[:executable_path]
+          handle_SIGINT = handle_SIGINT || @@launch_options[:handle_SIGINT]
+          handle_SIGTERM = handle_SIGTERM || @@launch_options[:handle_SIGTERM]
+          handle_SIGHUP = handle_SIGHUP || @@launch_options[:handle_SIGHUP]
+          headless = @@launch_options[:headless].nil? ? headless : @@launch_options[:headless].nil?
+          ignore_https_errors = @@launch_options[:ignore_https_errors].nil? ? ignore_https_errors : @@launch_options[:ignore_https_errors]
+          pipe = @@launch_options[:pipe].nil? ? pipe : @@launch_options[:pipe]
+          product = product || @@launch_options[:product]
+          slow_mo = slow_mo || @@launch_options[:slow_mo]
+          timeout = timeout || @@launch_options[:timeout]
+          user_data_dir = user_data_dir || @@launch_options[:user_data_dir]
+        end
+        Isomorfeus::Puppetmaster::Session.new(app: (app ? app : Isomorfeus::Puppetmaster.served_app),
+                                              args: args, channel: channel&.to_s, devtools: devtools, default_viewport: default_viewport,
+                                              dumpio: dumpio, env: env, executable_path: executable_path, handle_SIGINT: handle_SIGINT,
+                                              handle_SIGTERM: handle_SIGTERM, handle_SIGHUP: handle_SIGHUP, headless: headless,
+                                              ignore_default_args: ignore_default_args, ignore_https_errors: ignore_https_errors,
+                                              pipe: pipe, product: product, slow_mo: slow_mo,
+                                              timeout: timeout ? timeout : Isomorfeus::Puppetmaster::Session::TIMEOUT,
+                                              user_data_dir: user_data_dir)
+      end
+
+      def new_page(uri = nil)
+        session.new_page(uri)
       end
 
-      def open_new_document(uri = nil)
-        puppetmaster_session.document_open_new_document(nil, uri)
+      def session=(s)
+        @@puppetmaster_session = s
+        nil
       end
 
       def reset_session!
         @@puppetmaster_session = nil
+        nil
       end
 
-      private
-
-      def get_default_document
-        doc = puppetmaster_session.default_document
-        return doc if doc
-        puppetmaster_session.document_open_new_document('about:blank')
+      def on_server(ruby_source = '', &block)
+        session # ensure served_app booted
+        ruby_source = Isomorfeus::Puppetmaster.block_source_code(&block) if block_given?
+        Isomorfeus::Puppetmaster.served_app.on_server(ruby_source)
       end
 
-      def puppetmaster_session
-        @@puppetmaster_session ||= open_new_session(Isomorfeus::Puppetmaster.served_app)
+      def session
+        @@puppetmaster_session ||= new_session(app: Isomorfeus::Puppetmaster.served_app)
       end
-
     end
   end
 end

data/lib/isomorfeus/puppetmaster/server/checker.rb

@@ -1,40 +1,38 @@
-module Isomorfeus
-  module Puppetmaster
-    class Server
-      class Checker
-        TRY_HTTPS_ERRORS = [EOFError, Net::ReadTimeout, Errno::ECONNRESET].freeze
-
-        def initialize(host, port)
-          @host, @port = host, port
-          @ssl = false
-        end
-
-        def request(&block)
-          ssl? ? https_request(&block) : http_request(&block)
-        rescue *TRY_HTTPS_ERRORS
-          res = https_request(&block)
-          @ssl = true
-          res
-        end
-
-        def ssl?
-          @ssl
-        end
-
-      private
-
-        def http_request(&block)
-          Net::HTTP.start(@host, @port, read_timeout: 2, &block)
-        end
-
-        def https_request(&block)
-          Net::HTTP.start(@host, @port, ssl_options, &block)
-        end
-
-        def ssl_options
-          { use_ssl: true, verify_mode: OpenSSL::SSL::VERIFY_NONE }
-        end
-      end
-    end
-  end
-end
+module Isomorfeus
+  module Puppetmaster
+    class Server
+      class Checker
+        def initialize(host, port)
+          @host, @port = host, port
+          @ssl = false
+        end
+
+        def request(&block)
+          ssl? ? https_request(&block) : http_request(&block)
+        rescue EOFError, Net::ReadTimeout, Errno::ECONNRESET
+          res = https_request(&block)
+          @ssl = true
+          res
+        end
+
+        def ssl?
+          @ssl
+        end
+
+      private
+
+        def http_request(&block)
+          Net::HTTP.start(@host, @port, read_timeout: 2, &block)
+        end
+
+        def https_request(&block)
+          Net::HTTP.start(@host, @port, ssl_options, &block)
+        end
+
+        def ssl_options
+          { use_ssl: true, verify_mode: OpenSSL::SSL::VERIFY_NONE }
+        end
+      end
+    end
+  end
+end