Class: Consolr::Console

Inherits:
Object
  • Object
show all
Defined in:
lib/consolr.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Console

Returns a new instance of Console.



14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
 
# File 'lib/consolr.rb', line 14

def initialize
  potential_config_paths = [ENV['CONSOLR_CONFIG'],
                            '~/.consolr.yml', '~/.consolr.yaml',
                            '/etc/consolr.yml', '/etc/consolr.yaml',
                            '/var/db/consolr.yml', '/var/db/consolr.yaml']
  config_file = potential_config_paths.compact.find do |conf|
    begin
      expanded_path = File.expand_path(conf, __FILE__)
      File.readable?(expand_path) and File.size(expand_path) > 0
    rescue ArgumentError
      # if $HOME is not set, or if `~` cannot be expanded, `expand_path` will throw and ArgumentError
      # in that case, just go to the next potential config file - this one obviously will not work
      false
    end
  end

  @config_params = begin
    YAML.load(File.open(File.expand_path(config_file, __FILE__)))
  rescue TypeError => e
    puts "-------"
    puts "Failed to load Configuration File ... "
    puts "Looks like a configuration file doesn't exist."
    puts "Please look at README.md on creating a configuration file"
    puts "-------"
    exit 1
  rescue ArgumentError => e
    puts "------"
    puts "Failed to load Configuration File ... "
    puts "Looks like the configuration file is not correctly formatted"
    puts "Please check if your file conforms to YAML spec"
    puts "------"
    exit 1
  end
  #
  # Will be ignored for dangerous actions, no matter what, even with --force
  begin
    @dangerous_assets = @config_params['dangerous_assets']
  rescue Exception => e
    puts e
    puts "-------"
    puts "Dangerous Assets -- #{dangrous}"
    puts "Please make sure dangerous_assets exists and is valid."
    puts "Supply them in a comma separated list."
  end

  # Dangerous actions wont be run in these status, override with --force
  begin
    @dangerous_status = @config_params['dangerous_status']
  rescue Exception => e
    puts e
    puts "-------"
    puts "Dangerous Status -- #{@dangeous_status}"
    puts "Please specify the statuses which are dangerorous, during which dangerous shouldn't be run."
    puts "-------"
    exit 1
  end

  @dangerous_actions = [:off, :reboot] # Must match the symbol in options{}
end

Instance Attribute Details

#dangerous_actions ⇒ Object (readonly)

Returns the value of attribute dangerous_actions.



12
13
14
 
# File 'lib/consolr.rb', line 12

def dangerous_actions
  @dangerous_actions
end

#dangerous_assets ⇒ Object (readonly)

Returns the value of attribute dangerous_assets.



12
13
14
 
# File 'lib/consolr.rb', line 12

def dangerous_assets
  @dangerous_assets
end

#dangerous_status ⇒ Object (readonly)

Returns the value of attribute dangerous_status.



12
13
14
 
# File 'lib/consolr.rb', line 12

def dangerous_status
  @dangerous_status
end

#ipmitool_exec ⇒ Object (readonly)

Returns the value of attribute ipmitool_exec.



12
13
14
 
# File 'lib/consolr.rb', line 12

def ipmitool_exec
  @ipmitool_exec
end

#opt_parser ⇒ Object (readonly)

Returns the value of attribute opt_parser.



12
13
14
 
# File 'lib/consolr.rb', line 12

def opt_parser
  @opt_parser
end

#options ⇒ Object (readonly)

Returns the value of attribute options.



12
13
14
 
# File 'lib/consolr.rb', line 12

def options
  @options
end

Instance Method Details

#start(options) ⇒ Object 



74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
 
# File 'lib/consolr.rb', line 74

def start options
  abort("Please pass either the asset tag or hostname") if options[:tag].nil? and options[:hostname].nil?

  dangerous_body = "Dangerous actions: #{dangerous_actions.join(', ')}\n"\
    "Dangerous status: #{dangerous_status.join(', ')} (override with --force)\n"\
    "Dangerous assets: #{dangerous_assets.join(', ')} (ignored no matter what, even with --force)"

  if options[:dangerous]
    puts dangerous_body
    exit 1
  end

  begin
    collins = Collins::Authenticator.setup_client
  rescue Exception => e
    puts e
    puts "-------"
    puts "There was a problem setting up a connection with Collins."
    puts "-------"
    exit 1
  end

  if options[:tag] and options[:hostname]
    abort("Please pass either the hostname OR the tag but not both.")
  end

  if options[:runner].nil?
    runners = load_runners(@config_params.fetch('runners', []))
    # Default to the ipmitool runner for backwards compatibility
    if runners.empty?
      require 'consolr/runners/ipmitool'
      runners = [Consolr::Runners::Ipmitool.new(@config_params.fetch('ipmitool', {}))]
    end
  else
    runners = load_runners([options[:runner]])
    if runners.empty?
      abort('Specified runner could not be loaded. Aborting.')
    end
  end

  # match assets like vm-67f5eh, zt-*, etc.
  nodes = options[:tag] ? (collins.find :tag => options[:tag]) : (collins.find :hostname => options[:hostname])
  @node = nodes.length == 1 ? nodes.first : abort("Found #{nodes.length} assets, aborting.")

  # select the first runner that support the node
  runner = runners.select {|runner|
    runner.can_run? @node
  }.first

  if runner.nil?
    abort("No runners available for node #{@node.hostname} (#{@node.tag})")
  end

  if not runner.verify @node
    abort("Cannot verify asset #{@node.hostname} (#{@node.tag})")
  end

  selected_dangerous_actions = dangerous_actions.select { |o| options[o] }
  if dangerous_assets.include?(@node.tag) and selected_dangerous_actions.any?
    abort "Asset #{@node.tag} is a crucial asset. Can't ever execute dangerous actions on this asset.\n#{dangerous_body}"
  end

  if options[:force].nil? and selected_dangerous_actions.any? and dangerous_status.include?(@node.status)
    abort "Cannot run dangerous commands on #{@node.hostname} (#{@node.tag} - #{@node.status}) because it is in a protected status. This can be overridden with the --force flag\n#{dangerous_body}"
  end

  case
  when options[:console]
    puts '--> Opening SOL session (type ~~. to quit)'
    puts runner.console @node
  when options[:kick]
    puts runner.kick @node
  when options[:identify]
    puts runner.identify @node
  when options[:sdr]
    puts runner.sdr @node
  when options[:log] == 'list'
    puts runner.log_list @node
  when options[:log] == 'clear'
    puts runner.log_clear @node
  when options[:on]
    puts runner.on @node
  when options[:off]
    puts runner.off @node
  when options[:soft_off]
    puts runner.soft_off @node
  when options[:reboot]
    puts runner.reboot @node
  when options[:soft_reboot]
    puts runner.soft_reboot @node
  when options[:status]
    puts runner.status @node
  when options[:sensors]
    puts runner.sensors @node
  when options[:get_sol_info]
    puts runner.sol_info @node
  else
    begin
      puts "specify an action"
      exit 1
    end
  end
end