aboutsummaryrefslogblamecommitdiffstats
path: root/railties/lib/rails/command/environment_argument.rb
blob: 9945fd1430f4621729dd30b9085abe4ebde8fe67 (plain) (tree)
1
2
3
4
5
6
7
8
9
10
11

                             
                        
                                                 






                                       


                                                                                                                                                       
                                                                                       


             
                                                                                                     
                                  

                                                                                                    
                                                                          






                                                
                                                                                  








                                                                                    
# frozen_string_literal: true

require "active_support"
require "active_support/core_ext/class/attribute"

module Rails
  module Command
    module EnvironmentArgument #:nodoc:
      extend ActiveSupport::Concern

      included do
        no_commands do
          class_attribute :environment_desc, default: "Specifies the environment to run this #{self.command_name} under (test/development/production)."
        end
        class_option :environment, aliases: "-e", type: :string, desc: environment_desc
      end

      private
        def extract_environment_option_from_argument(default_environment: Rails::Command.environment)
          if options[:environment]
            self.options = options.merge(environment: acceptable_environment(options[:environment]))
          else
            self.options = options.merge(environment: default_environment)
          end
        end

        def acceptable_environment(env = nil)
          if available_environments.include? env
            env
          else
            %w( production development test ).detect { |e| e =~ /^#{env}/ } || env
          end
        end

        def available_environments
          Dir["config/environments/*.rb"].map { |fname| File.basename(fname, ".*") }
        end
    end
  end
end