moved grab_data to the bin directory

1 files changed, 104 insertions, 0 deletions

diff --git a/bin/grab_data.rb b/bin/grab_data.rb
new file mode 100644
index 00000000..60039a58
--- /dev/null
+++ b/bin/grab_data.rb
@@ -0,0 +1,104 @@
+
+# Author:: Theodore Nordsieck <theo@opscode.com>
+# Copyright:: Copyright (c) 2013 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.
+#
+
+##
+## This is a simple tool to grab shell command output from the local machine.
+##
+## To use, edit the parameters before running the command.  This tool is meant
+##   integrate the built in fake command data in spec/data/plugins.  If the
+##   appropriate yaml file is in this directory, this tool will append/overwrite
+##   the data in that file as appropriate.
+##
+
+require 'yaml'
+require 'set'
+require 'mixlib/shellout'
+require 'mixlib/cli'
+require 'optparse'
+require File.expand_path(File.dirname(__FILE__) + '/../spec/unit/path/ohai_plugin_common.rb')
+
+#get options
+class MyCLI
+  include Mixlib::CLI
+  
+  option :command,
+    :short => "-c CMD",
+    :long => "--command CMD",
+    :description => "The command to run",
+    :required => true
+
+  option :params,
+    :short => "-p [P1,P2,...]",
+    :long => "--params [P1,P2,...]",
+    :description => "List of parameters, applied one at a time",
+    #not sure how to use optparse's array syntax, so this is a hack to reproduce that behavior
+    :proc => Proc.new { |s| if s then s.split( "," ) else [""] end }
+
+  option :platform,
+    :short => "-f PLATFORM",
+    :long => "--platform PLATFORM",
+    :description => "Description of the platform",
+    :required => true
+
+  option :arch,
+    :short => "-a ARCH",
+    :long => "--architecture ARCH",
+    :description => "Description of the architecture",
+    :required => true
+
+  option :env,
+    :short => "-e [E1,E2,...]",
+    :long => "--environment [E1,E2,...]",
+    :description => "List of labels that describe the environment",
+    :proc => Proc.new { |s| if s then s.split( "," ) else [] end } #same here
+
+end
+cli = MyCLI.new
+cli.parse_options
+cmd, params, platform, arch, env = cli.config[:command], cli.config[:params], cli.config[:platform], cli.config[:arch], cli.config[:env]
+
+# read in data
+opc = OhaiPluginCommon.new
+# filename = cmd + ".output"
+
+# Mixlib::ShellOut.new("touch #{filename}").run_command
+# data = opc.read_output cmd, File.expand_path( File.dirname(__FILE__))
+data ||= {}
+data[platform] ||= {}
+data[platform][arch] ||= []
+
+# collect output
+
+results = params.map do |p|
+  m = Mixlib::ShellOut.new(cmd + ' ' + p)
+  begin
+    m.run_command
+    {:env => env, :params => p, :stdout => m.stdout, :stderr => m.stderr, :exit_status => m.exitstatus }
+  rescue Errno::ENOENT
+    {:env => env, :params => p, :stdout => '', :stderr => 'command not found', :exit_status => 127 }
+  end
+end
+
+# write out data
+
+results.each do |r|
+  data[platform][arch] = data[platform][arch].reject { |e| e[:params] == r[:params] && e[:env] == r[:env] }
+  data[platform][arch] << r
+end
+
+puts opc.data_to_string data