Split up Cucumber and RSpec support

lib/persistent_selenium/cucumber.rb

@@ -0,0 +1,8 @@
+require_relative './driver'
+
+Before do
+  if Capybara.current_driver == :persistent_selenium
+    Capybara.server_port ||= '3001'
+    Capybara.app_host ||= "http://localhost:#{Capybara.server_port}"
+  end
+end

lib/persistent_selenium/driver.rb

@@ -5,16 +5,11 @@ require 'capybara/selenium/driver'
 # make sure these classes exist on this end
 [ Selenium::WebDriver::Error::StaleElementReferenceError, Selenium::WebDriver::Error::UnhandledError, Selenium::WebDriver::Error::ElementNotVisibleError ]
 
-Before do
-  if Capybara.current_driver == :persistent_selenium
-    Capybara.server_port ||= '3001'
-    Capybara.app_host ||= "http://localhost:#{Capybara.server_port}"
-  end
-end
-
 require 'persistent_selenium/drb'
 
 Capybara.register_driver :persistent_selenium do |app|
+  require 'drb'
+
   service = DRb.start_service
   browser = DRbObject.new nil, PersistentSelenium.url
 

lib/persistent_selenium/rspec.rb

@@ -0,0 +1,10 @@
+require_relative './driver'
+
+RSpec.configure do |c|
+  c.before :each do
+    if Capybara.current_driver == :persistent_selenium
+      Capybara.server_port ||= '3001'
+      Capybara.app_host ||= "http://localhost:#{Capybara.server_port}"
+    end
+  end
+end