configuration.rb in penthouse-0.3.0

- old
+ new
@@ -10,20 +10,28 @@
 #   end
 #
 
 module Penthouse
   class Configuration
-    attr_accessor :router, :runner, :migrate_tenants, :tenant_identifiers
+    attr_accessor :router, :runner, :migrate_tenants, :db_schema_file, :tenant_identifiers
 
     # @param router [Penthouse::Routers::BaseRouter] the default router for your application to use
     # @param runner [Penthouse::Runners::BaseRunner] the default runner for your application to use
     # @param migrate_tenants [Boolean] whether you want Penthouse to automatically migrate all tenants
+    # @param db_schema_file [String] a path to your schema file (typically `db/schema.rb` or `db/structure.sql` in Rails)
     # @param tenant_identifiers [Proc] some code which must return an array of tenant identifiers (strings/symbols)
-    def initialize(router: nil, runner: nil, migrate_tenants: false, tenant_identifiers: -> { raise NotImplementedError })
+    def initialize(router: nil, runner: nil, migrate_tenants: false, db_schema_file: nil, tenant_identifiers: -> { raise NotImplementedError })
       self.router = router
       self.runner = runner
       self.migrate_tenants = migrate_tenants
+      self.db_schema_file = db_schema_file
       self.tenant_identifiers = tenant_identifiers
+
+      if migrate_tenants? && !db_schema_file
+        raise ArgumentError, "If you want to migrate tenants, we need a path to a DB schema file"
+      elsif migrate_tenants? && !File.exist?(db_schema_file)
+        raise ArgumentError, "#{db_schema_file} is not readable"
+      end
     end
 
     # @return [Boolean] whether or not Penthouse should automatically migrate all tenants
     def migrate_tenants?
       !!migrate_tenants