Merge pull request #60 from zachasme/configurable-base-controller

Make base controller class configurable

Author: fractaledmind

app/controllers/litestream/application_controller.rb

@@ -1,5 +1,5 @@
 module Litestream
-  class ApplicationController < ActionController::Base
+  class ApplicationController < Litestream.base_controller_class.constantize
     protect_from_forgery with: :exception
 
     if Litestream.password

lib/litestream.rb

@@ -35,6 +35,7 @@ def initialize
 
   mattr_writer :username, :password, :queue, :replica_bucket, :replica_key_id, :replica_access_key, :systemctl_command,
     :config_path
+  mattr_accessor :base_controller_class, default: "::ApplicationController"
 
   class << self
     def verify!(database_path)

test/dummy/app/controllers/my_application_controller.rb

@@ -0,0 +1,2 @@
+class MyApplicationController < ApplicationController
+end

test/dummy/config/environments/test.rb

@@ -61,4 +61,6 @@
 
   # Raise error when a before_action's only/except options reference missing actions
   config.action_controller.raise_on_missing_callback_actions = true
+
+  config.litestream.base_controller_class = "MyApplicationController"
 end

test/litestream/test_base_application_controller.rb

@@ -0,0 +1,11 @@
+require "test_helper"
+
+class Litestream::BaseApplicationControllerTest < ActiveSupport::TestCase
+  test "engine's ApplicationController inherits from host's ApplicationController by default" do
+    assert Litestream::ApplicationController < ApplicationController
+  end
+
+  test "engine's ApplicationController inherits from configured base_controller_class" do
+    assert Litestream::ApplicationController < MyApplicationController
+  end
+end