Ability to define a custom user_class

lib/generators/surveyor/install_generator.rb

@@ -40,6 +40,10 @@ class InstallGenerator < Rails::Generators::Base
       add_input_mask_attributes_to_answer
     )
 
+    def create_initializer_file
+      create_file "config/initializers/surveyor.rb", "Surveyor.user_class = 'User'"
+    end
+
     def readme
       copy_file "../../../../README.md", "surveys/README.md"
     end

lib/surveyor.rb

@@ -2,6 +2,16 @@ module Surveyor
   require 'surveyor/engine' if defined?(Rails) && Rails::VERSION::MAJOR >= 3
   autoload :VERSION, 'surveyor/version'
   autoload :ParserError, 'surveyor/parser'
+
+  mattr_accessor :user_class
+
+  def self.user_class
+    if @@user_class.is_a?(Class)
+      raise "Surveyor.user_class MUST be a String or Symbol object, not a Class object."
+    elsif @@user_class.is_a?(String) || @@user_class.is_a?(Symbol)
+      @@user_class.to_s.constantize
+    end
+  end
 end
 require 'surveyor/common'
 require 'surveyor/acts_as_response'

lib/surveyor/models/response_set_methods.rb

@@ -8,7 +8,7 @@ module ResponseSetMethods
       included do
         # Associations
         belongs_to :survey
-        belongs_to :user
+        belongs_to :user, class_name: Surveyor.user_class
         has_many :responses, :dependent => :destroy
         accepts_nested_attributes_for :responses, :allow_destroy => true
         attr_accessible *PermittedParams.new.response_set_attributes if defined? ActiveModel::MassAssignmentSecurity