Contents

require 'rails/generators/active_record/migration'

module Spree
  class CustomUserGenerator < Rails::Generators::NamedBase
    include ActiveRecord::Generators::Migration

    desc "Set up a Solidus installation with a custom User class"

    source_root File.expand_path('templates', File.dirname(__FILE__))

    def check_for_constant
      klass
    rescue NameError
      @shell.say "Couldn't find #{class_name}. Are you sure that this class exists within your application and is loaded?", :red
      exit(1)
    end

    def generate
      migration_template 'migration.rb.tt', "db/migrate/add_spree_fields_to_custom_user_table.rb"
      template 'authentication_helpers.rb.tt', "lib/spree/authentication_helpers.rb"

      file_action = File.exist?('config/initializers/spree.rb') ? :append_file : :create_file
      send(file_action, 'config/initializers/spree.rb') do
        "Rails.application.config.to_prepare do\n  require_dependency 'spree/authentication_helpers'\nend\n"
      end

      gsub_file 'config/initializers/spree.rb', /Spree\.user_class.?=.?.+$/, %{Spree.user_class = "#{class_name}"}
    end

    private

    def klass
      class_name.constantize
    end

    def table_name
      klass.table_name
    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
solidus_core-2.5.2	lib/generators/spree/custom_user/custom_user_generator.rb
solidus_core-2.5.1	lib/generators/spree/custom_user/custom_user_generator.rb
solidus_core-2.5.0	lib/generators/spree/custom_user/custom_user_generator.rb
solidus_core-2.5.0.rc1	lib/generators/spree/custom_user/custom_user_generator.rb
solidus_core-2.5.0.beta2	lib/generators/spree/custom_user/custom_user_generator.rb
solidus_core-2.5.0.beta1	lib/generators/spree/custom_user/custom_user_generator.rb