Contents

# frozen_string_literal: true
class Hyrax::ModelsGenerator < Rails::Generators::Base
  source_root File.expand_path('../templates', __FILE__)
  argument :model_name, type: :string, default: 'user'
  desc 'This generator makes the following changes to your application:
      1. Copies database migrations
      2. Injects the user behavior onto the user models.
      3. Creates the file_set.rb and collection.rb models.
      4. Generates the clam anti-virus configuration.'

  def banner
    say_status('info', 'GENERATING HYRAX MODELS', :blue)
  end

  # Setup the database migrations
  def copy_migrations
    rake 'hyrax:install:migrations'
    rake 'valkyrie_engine:install:migrations'
  end

  # Add behaviors to the user model
  def inject_user_behavior
    file_path = "app/models/#{model_name.underscore}.rb"
    if File.exist?(file_path)
      inject_into_file file_path, after: /include Hydra\:\:User.*$/ do
        "\n  # Connects this user object to Hyrax behaviors." \
        "\n  include Hyrax::User" \
        "\n  include Hyrax::UserUsageStats\n"
      end
    else
      puts "     \e[31mFailure\e[0m  Hyrax requires a user object. This " \
           "generator assumes that the model is defined in the file " \
           "#{file_path}, which does not exist.  If you used a different " \
           "name, please re-run the generator and provide that name as an " \
           "argument. Such as \b  rails -g hyrax:models client"
    end
  end
  # rubocop:enable Metrics/MethodLength

  # Adds clamav initializtion
  def clamav
    generate 'hyrax:clamav'
  end

  private

  def rspec_installed?
    defined?(RSpec) && defined?(RSpec::Rails)
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
hyrax-5.0.3	lib/generators/hyrax/models_generator.rb
hyrax-5.0.2	lib/generators/hyrax/models_generator.rb
hyrax-5.0.1	lib/generators/hyrax/models_generator.rb
hyrax-5.0.0	lib/generators/hyrax/models_generator.rb
hyrax-5.0.0.rc3	lib/generators/hyrax/models_generator.rb