#!/usr/bin/env ruby require 'optparse' require 'logger' require 'cloud66_agent/utils/config' require 'cloud66_agent/utils/version' require 'cloud66_agent' # global config variable $config = Cloud66::Utils::Config.new OptionParser.new do |opts| opts.banner = "Cloud 66 Agent v#{Cloud66::Utils::Version.current}" # following opts only for registration opts.on('--url URL', 'Server URL') do |v| $config.url = v end opts.on('--sockets SOCKETS', 'Sockets URL') do |v| $config.faye_url = v end opts.on('--api-key APIKEY', 'API key') do |v| $config.api_key = v end opts.on('--secret-key SECRETKEY', 'Secret Key') do |v| $config.secret_key = v end opts.on('--server SERVERUID', 'Server id') do |v| @server_uid = v end # respond to version requests opts.on('--version') do |v| puts "Cloud 66 Agent #{Cloud66::Utils::Version.current}" exit 0 end # logging configuration opts.on('--log LOG', 'Log file path') do |v| v == "STDOUT" ? $config.log_path = STDOUT : $config.log_path = v end opts.on('--log_level LOGLEVEL', 'Log level (int)') do |v| $config.log_level = v.to_i end end.parse! # prepare the global logger $logger = Logger.new($config.log_path) $logger.level = $config.log_level if $config.disabled # no other commands allowed while disabled $logger.error "This agent had been disabled. Please contact Cloud 66 if you think this is in error" exit -1 end #pick up the command used command = ARGV[0].downcase unless ARGV[0].nil? if command.nil? || command.empty? $logger.debug("Cloud 66 Agent v#{Cloud66::Utils::Version.current} (#{$config.is_agent_configured? ? 'Configured' : 'Not Configured'})") exit 0 end if !$config.is_agent_configured? && command != 'configure' # no other commands allowed while not configured $logger.error "Can only do command \"configure\" (until its been configured!)" exit -1 end # handle commands $logger.info "Performing: \"#{command}\"" if command == "configure" Cloud66Agent.configure @server_uid else begin arguments = ARGV[1..-1] if arguments.empty? Cloud66Agent.send command else Cloud66Agent.send(command, arguments) end rescue ArgumentError => exc $logger.error "Invalid command/arguments: #{command} #{arguments}" exit -1 end end exit 0