Add OpenStruct support to Awesome Print

Check that OpenStruct exists before patching

Author: bootstraponline

lib/appium_lib/awesome_print/ostruct.rb

@@ -0,0 +1,33 @@
+# Enable OpenStruct support in awesome print
+# Code from @marshallshen https://github.com/michaeldv/awesome_print/pull/145
+#
+# Copyright (c) 2010-2013 Michael Dvorkin
+#
+# Awesome Print is freely distributable under the terms of MIT license.
+# See LICENSE file or http://www.opensource.org/licenses/mit-license.php
+#------------------------------------------------------------------------------
+
+if defined?(OpenStruct)
+  module AwesomePrint
+    module OpenStruct
+      def self.included(base)
+        base.send :alias_method, :cast_without_ostruct, :cast
+        base.send :alias_method, :cast, :cast_with_ostruct
+      end
+
+      def cast_with_ostruct(object, type)
+        cast = cast_without_ostruct(object, type)
+        if (defined?(::OpenStruct)) && (object.is_a?(::OpenStruct))
+          cast = :open_struct_instance
+        end
+        cast
+      end
+
+      def awesome_open_struct_instance(object)
+        "#{object.class} #{awesome_hash(object.marshal_dump)}"
+      end
+    end
+  end
+
+  AwesomePrint::Formatter.send(:include, AwesomePrint::OpenStruct)
+end

lib/appium_lib/driver.rb

@@ -1,9 +1,11 @@
 # encoding: utf-8
 require 'rubygems'
 require 'ap'
-
 require 'selenium-webdriver'
 
+# patch ap
+require_relative 'awesome_print/ostruct'
+
 # common
 require_relative 'common/helper'
 require_relative 'common/patch'
@@ -32,18 +34,6 @@
 # device methods
 require_relative 'device/device'
 
-# Support OpenStruct in Awesome Print
-# /awesome_print/lib/awesome_print/formatter.rb
-# upstream issue: https://github.com/michaeldv/awesome_print/pull/36
-class AwesomePrint::Formatter
-  remove_const :CORE if defined?(CORE)
-  CORE = [:array, :hash, :class, :file, :dir, :bigdecimal, :rational, :struct, :openstruct, :method, :unboundmethod]
-
-  def awesome_openstruct target
-    awesome_hash target.marshal_dump
-  end
-end
-
 # Fix uninitialized constant Minitest (NameError)
 module Minitest
   # Fix superclass mismatch for class Spec