require 'fileutils' require 'new_relic/version' require 'erb' class NewRelic::Command::Install < NewRelic::Command NO_LICENSE_KEY = "" def self.command; "install"; end # Use -h to see options. # When command_line_args is a hash, we are invoking directly and # it's treated as an options with optional string values for # :user, :description, :appname, :revision, :environment, # and :changes. # # Will throw CommandFailed exception if there's any error. # attr_reader :dest_dir, :license_key, :generated_for_user, :quiet, :src_file, :app_name def initialize command_line_args={} super command_line_args if !@dest_dir # Install a newrelic.yml file into the local config directory. if File.directory? "config" @dest_dir = "config" else @dest_dir = "." end end @license_key ||= NO_LICENSE_KEY @app_name ||= @leftover raise CommandFailure.new("Application name required.", @options) unless @app_name && @app_name.size > 0 @generated_for_user ||= @user_string || "" end def run dest_file = File.expand_path(@dest_dir + "/newrelic.yml") if File.exist?(dest_file) raise NewRelic::Command::CommandFailure, "newrelic.yml file already exists. Move it out of the way." end File.open(dest_file, 'w') { | out | out.puts(content) } puts <<-EOF unless quiet Installed a default configuration file at #{dest_file}. EOF puts <<-EOF unless quiet || @license_key != NO_LICENSE_KEY To monitor your application in production mode, sign up for an account at www.newrelic.com, and replace the newrelic.yml file with the one you receive upon registration. EOF puts <<-EOF unless quiet E-mail support@newrelic.com with any problems or questions. EOF end def content @src_file ||= File.expand_path(File.join(File.dirname(__FILE__),"..","..","..","newrelic.yml")) template = File.read(@src_file) ERB.new(template).result(binding) end private def options OptionParser.new "Usage: #{$0} #{self.class.command} [ OPTIONS] 'application name'", 40 do |o| o.on("-l", "--license_key=NAME", String, "Use the given license key") { | e | @license_key = e } o.on("-d", "--destdir=name", String, "Write the newrelic.yml to the given directory, default is '.'") { | e | @dest_dir = e } yield o if block_given? end end end