opscode-ohai 0.1.2

data/lib/ohai/plugins/network.rb

@@ -0,0 +1,48 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+network Mash.new unless network
+network[:interfaces] = Mash.new unless network[:interfaces]
+
+require_plugin "hostname"
+require_plugin "#{os}::network"
+
+def find_ip_and_mac(addresses)
+  ip = nil; mac = nil
+  addresses.each do |addr|
+    ip = addr["address"] if addr["family"].eql?("inet")
+    mac = addr["address"] if addr["family"].eql?("lladdr")
+    break if (ip and mac)
+  end
+  [ip, mac]
+end
+
+if attribute?("default_interface")
+  im = find_ip_and_mac(network["interfaces"][iface]["addresses"])
+  ipaddress im.shift
+  macaddress im.shift
+else
+  network["interfaces"].keys.each do |iface|
+    if network["interfaces"][iface]["encapsulation"].eql?("Ethernet")
+      im = find_ip_and_mac(network["interfaces"][iface]["addresses"])
+      ipaddress im.shift
+      macaddress im.shift
+      return if (ipaddress and macaddress)
+    end
+  end
+end

data/lib/ohai/plugins/ohai_time.rb

@@ -0,0 +1,19 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+ohai_time Time.now.to_f

data/lib/ohai/plugins/os.rb

@@ -0,0 +1,34 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require_plugin 'ruby'
+require_plugin 'kernel'
+
+case languages[:ruby][:host_os]
+when /darwin(.+)$/
+  os "darwin"
+when /linux/
+  os "linux"
+when /freebsd(.+)$/
+ os "freebsd"
+else
+  os languages[:ruby][:host_os]
+end
+
+os_version kernel[:release]
+

data/lib/ohai/plugins/platform.rb

@@ -0,0 +1,23 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require_plugin "#{os}::platform"
+
+platform os unless attribute?("platform")
+
+platform_version os_version unless attribute?("platform_version")

data/lib/ohai/plugins/ruby.rb

@@ -0,0 +1,33 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require_plugin "languages"
+
+languages[:ruby] = Mash.new
+
+languages[:ruby][:platform] = RUBY_PLATFORM
+languages[:ruby][:version] = RUBY_VERSION
+languages[:ruby][:release_date] = RUBY_RELEASE_DATE
+languages[:ruby][:target] = ::Config::CONFIG['target']
+languages[:ruby][:target_cpu] = ::Config::CONFIG['target_cpu']
+languages[:ruby][:target_vendor] = ::Config::CONFIG['target_vendor']
+languages[:ruby][:target_os] = ::Config::CONFIG['target_os']
+languages[:ruby][:host] = ::Config::CONFIG['host']
+languages[:ruby][:host_cpu] = ::Config::CONFIG['host_cpu']
+languages[:ruby][:host_os] = ::Config::CONFIG['host_os']
+languages[:ruby][:host_vendor] = ::Config::CONFIG['host_vendor']

data/lib/ohai/plugins/uptime.rb

@@ -0,0 +1,42 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+def _seconds_to_human(seconds)
+  days = seconds.to_i / 86400
+  seconds -= 86400 * days
+  
+  hours = seconds.to_i / 3600
+  seconds -= 3600 * hours
+  
+  minutes = seconds.to_i / 60
+  seconds -= 60 * minutes
+    
+  if days > 1
+    return sprintf("%d days %02d hours %02d minutes %02d seconds", days, hours, minutes, seconds)
+  elsif days == 1
+    return sprintf("%d day %02d hours %02d minutes %02d seconds", days, hours, minutes, seconds)
+  elsif hours > 0
+    return sprintf("%d hours %02d minutes %02d seconds", hours, minutes, seconds)
+  elsif minutes > 0
+    return sprintf("%d minutes %02d seconds", minutes, seconds)
+  else
+    return sprintf("%02d seconds", seconds)
+  end
+end
+
+

data/lib/ohai/system.rb

@@ -0,0 +1,163 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'rubygems'
+require 'extlib'
+require 'ohai/log'
+require 'ohai/mixin/from_file'
+require 'ohai/mixin/command'
+require 'json'
+
+module Ohai
+  class System
+    attr_accessor :data, :seen_plugins
+    
+    include Ohai::Mixin::FromFile
+    include Ohai::Mixin::Command
+    
+    def initialize
+      @data = Mash.new
+      @seen_plugins = Hash.new
+    end
+    
+    def [](key)
+      @data[key]
+    end
+    
+    def []=(key, value)
+      @data[key] = value
+    end
+    
+    def each(&block)
+      @data.each do |key, value|
+        block.call(key, value)
+      end
+    end
+    
+    def attribute?(name)
+      Ohai::Log.debug("Attribute #{name} is #{@data.has_key?(name)}")
+      @data.has_key?(name) 
+    end
+    
+    def set(name, *value)
+      set_attribute(name, *value)
+    end
+    
+    def from(cmd)
+      status, stdout, stderr = run_command(:command => cmd)
+      stdout.chomp!
+    end
+    
+    # Set the value equal to the stdout of the command, plus run through a regex - the first piece of match data is the value.
+    def from_with_regex(cmd, *regex_list)
+      regex_list.flatten.each do |regex|
+        status, stdout, stderr = run_command(:command => cmd)
+        stdout.chomp!
+        md = stdout.match(regex)
+        return md[1]
+      end
+    end
+    
+    def set_attribute(name, *value)
+      Ohai::Log.debug("Setting attribute #{name} to #{value.inspect}")
+      @data[name] = *value
+      @data[name]
+    end
+    
+    def get_attribute(name)
+      Ohai::Log.debug("Getting attribute #{name}, value #{@data[name].inspect}")
+      @data[name]
+    end
+    
+    def all_plugins
+      require_plugin('os')
+      
+      Ohai::Config[:plugin_path].each do |path|
+        [ 
+          Dir[File.join(path, '*')], 
+          Dir[File.join(path, @data[:os], '**', '*')] 
+        ].flatten.each do |file|
+          file_regex = Regexp.new("#{path}#{File::SEPARATOR}(.+).rb$")
+          md = file_regex.match(file)
+          if md
+            plugin_name = md[1].gsub(File::SEPARATOR, "::") 
+            require_plugin(plugin_name) unless @seen_plugins.has_key?(plugin_name)
+          end
+        end
+      end
+    end
+    
+    def require_plugin(plugin_name, force=false)
+      unless force
+        return true if @seen_plugins[plugin_name]
+      end
+      
+      filename = "#{plugin_name.gsub("::", File::SEPARATOR)}.rb"
+            
+      Ohai::Config[:plugin_path].each do |path|
+        check_path = File.expand_path(File.join(path, filename))
+        begin
+          @seen_plugins[plugin_name] = true
+          Ohai::Log.info("Loading plugin #{plugin_name}")
+          from_file(check_path)
+          return true
+        rescue IOError => e
+          Ohai::Log.debug("No #{plugin_name} at #{check_path}")
+        rescue Exception,Errno::ENOENT => e
+          Ohai::Log.debug("Plugin #{plugin_name} threw exception #{e.inspect}")
+        end
+      end
+    end
+    
+    # Sneaky!  Lets us stub out require_plugin when testing plugins, but still
+    # call the real require_plugin to kick the whole thing off.
+    alias :_require_plugin :require_plugin
+    
+    # Serialize this object as a hash 
+    def to_json(*a)
+      output = @data.clone
+      output["json_class"] = self.class.name
+      output.to_json(*a)
+    end
+    
+    # Pretty Print this object as JSON 
+     def json_pretty_print
+       JSON.pretty_generate(@data)
+     end
+    
+    # Create an Ohai::System from JSON
+    def self.json_create(o)
+      ohai = new
+      o.each do |key, value|
+        ohai.data[key] = value unless key == "json_class"
+      end
+      ohai
+    end
+    
+    def method_missing(name, *args)
+      return get_attribute(name) if args.length == 0 
+      
+      set_attribute(name, *args)
+    end
+    
+    private
+      def load_plugin_file
+        
+      end
+  end
+end

data/spec/ohai/log/log_formatter_spec.rb

@@ -0,0 +1,50 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'time'
+require File.expand_path(File.join(File.dirname(__FILE__), "..", "..", "spec_helper"))
+
+describe Ohai::Log::Formatter do
+  before(:each) do
+    @formatter = Ohai::Log::Formatter.new
+  end
+  
+  it "should print raw strings with msg2str(string)" do
+    @formatter.msg2str("nuthin new").should == "nuthin new"
+  end
+  
+  it "should format exceptions properly with msg2str(e)" do
+    e = IOError.new("legendary roots crew")
+    @formatter.msg2str(e).should == "legendary roots crew (IOError)\n"
+  end
+  
+  it "should format random objects via inspect with msg2str(Object)" do
+    @formatter.msg2str([ "black thought", "?uestlove" ]).should == '["black thought", "?uestlove"]'
+  end
+  
+  it "should return a formatted string with call" do
+    time = Time.new
+    Ohai::Log::Formatter.show_time = true
+    @formatter.call("monkey", time, "test", "mos def").should == "[#{time.rfc2822}] monkey: mos def\n"
+  end
+  
+  it "should allow you to turn the time on and off in the output" do
+    Ohai::Log::Formatter.show_time = false
+    @formatter.call("monkey", Time.new, "test", "mos def").should == "monkey: mos def\n"
+  end
+end

data/spec/ohai/log_spec.rb

@@ -0,0 +1,63 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'tempfile'
+require 'logger'
+
+require File.expand_path(File.join(File.dirname(__FILE__), "..", "spec_helper"))
+
+describe Ohai::Log do
+  it "should accept regular options to Logger.new via init" do
+    tf = Tempfile.new("chef-test-log")
+    tf.open
+    lambda { Ohai::Log.init(STDOUT) }.should_not raise_error
+    lambda { Ohai::Log.init(tf) }.should_not raise_error
+  end
+  
+  it "should set the log level with :debug, :info, :warn, :error, or :fatal" do
+    levels = {
+      :debug => Logger::DEBUG,
+      :info => Logger::INFO,
+      :warn => Logger::WARN,
+      :error => Logger::ERROR,
+      :fatal => Logger::FATAL
+    }
+    levels.each do |symbol, constant|
+      Ohai::Log.level(symbol)
+      Ohai::Log.logger.level.should == constant
+    end
+  end
+  
+  it "should raise an ArgumentError if you try and set the level to something strange" do
+    lambda { Ohai::Log.level(:the_roots) }.should raise_error(ArgumentError)
+  end
+  
+  it "should pass other method calls directly to logger" do
+    Ohai::Log.level(:debug)
+    Ohai::Log.should be_debug
+    lambda { Ohai::Log.debug("Gimme some sugar!") }.should_not raise_error
+  end
+  
+  it "should default to STDOUT if init is called with no arguments" do
+    logger_mock = mock(Logger, :null_object => true)
+    Logger.stub!(:new).and_return(logger_mock)
+    Logger.should_receive(:new).with(STDOUT).and_return(logger_mock)
+    Ohai::Log.init
+  end
+  
+end