web-puppet 0.0.1 → 0.1.0

data/Gemfile

@@ -2,3 +2,11 @@ source "http://rubygems.org"
 
 gem "rack"
 gem "puppet", ">= 2.7.0"
+gem "rake"
+gem "parseconfig"
+
+group :test do
+  gem "test-unit"
+  gem "rack-test"
+  gem "mocha"
+end

data/Gemfile.lock

@@ -2,13 +2,26 @@ GEM
   remote: http://rubygems.org/
   specs:
     facter (1.6.2)
+    metaclass (0.0.1)
+    mocha (0.10.0)
+      metaclass (~> 0.0.1)
+    parseconfig (0.5.2)
     puppet (2.7.6)
       facter (>= 1.5.1)
     rack (1.3.5)
+    rack-test (0.6.1)
+      rack (>= 1.0)
+    rake (0.9.2.2)
+    test-unit (2.4.0)
 
 PLATFORMS
   ruby
 
 DEPENDENCIES
+  mocha
+  parseconfig
   puppet (>= 2.7.0)
   rack
+  rack-test
+  rake
+  test-unit

data/README

@@ -1,3 +1,29 @@
 A tiny ruby rack application which exposes the data from puppet as JSON over HTTP.
 
     gem install web-puppet
+
+Provides a simple command line tool which runs a built in web server. On accessing
+the specified port you should get a JSON response containing the current node information.
+
+    web-puppet --help
+    Usage: web-puppet [options] ...
+
+    Configuration options:
+            --no-daemonize               Don't daemonize the web server process
+        -p, --port PORT                  The port to run web-facter on
+        -c, --config FILE                The file to load with configuration options
+        -h, --help                       Display this screenp
+
+Note that in most setups the yaml directory containing the node information isn't readable
+by all users, so you may need to run this as puppet or root.
+
+You can configure web-puppet using a configuration file, using the following format, and
+specifying the filename with the --config option above.
+
+    username=gilbert
+    password=george
+    port=3009
+    daemonize=true
+
+Note that the port and daemonize options will override those on the command line.
+The username and password options enable HTTP basic authentication using those details.

data/Rakefile

@@ -1 +1,10 @@
 require "bundler/gem_tasks"
+require "rake/testtask"
+
+Rake::TestTask.new do |t|
+  t.libs << "test"
+  t.test_files = FileList["test/**/*_test.rb"]
+  t.verbose = true
+end
+
+task :default => :test

data/bin/web-puppet

@@ -1,6 +1,43 @@
 #!/usr/bin/env ruby
 
 require 'rack'
+require 'optparse'
 require 'web-puppet'
 
-Rack::Server.new(:app => WebPuppet::App.new, :Port => 9295, :daemonize => true).start
+options = {:daemonize => true, :port => 9295, :config => false}
+
+optparse = OptionParser.new do |opts|
+  opts.banner = 'Usage: web-puppet [options] ...'
+
+  opts.separator ''
+  opts.separator 'Configuration options:'
+
+  opts.on( '--no-daemonize', "Don't daemonize the web server process") do |username|
+    options[:daemonize] = false
+  end
+
+  opts.on( '-p', '--port PORT', 'The port to run web-puppet on') do |port|
+    options[:port] = port
+  end
+
+  opts.on( '-c', '--config FILE', 'The file to load with configuration options') do |file|
+    options[:config] = file
+  end
+
+  opts.on_tail('-h', '--help', 'Display this screen' ) do
+    puts opts
+    exit
+  end
+end
+
+begin
+  optparse.parse!
+  WebPuppet::App.run!(options)
+rescue OptionParser::InvalidArgument, OptionParser::InvalidOption, OptionParser::MissingArgument
+  puts $!.to_s
+  puts optparse
+  exit
+rescue Errno::EACCES
+  puts $!.to_s
+  exit
+end

data/lib/web-puppet/version.rb

@@ -1,3 +1,3 @@
 module WebPuppet
-  VERSION = "0.0.1"
+  VERSION = "0.1.0"
 end

data/lib/web-puppet.rb

@@ -31,5 +31,33 @@ module  WebPuppet
       response.write JSON.pretty_generate(data)
       response.finish
     end
+
+    def add_auth(conf)
+      application = Rack::Auth::Basic.new(self) do |username, password|
+        stored_username = conf.get_value('username')
+        username_check = stored_username ? stored_username == username : true
+        password_check = conf.get_value('password') == password
+        username_check && password_check
+      end
+      application.realm = 'Web Puppet'
+      application
+    end
+
+    def self.run!(options)
+      application = self.new
+
+      daemonize = options[:daemonize]
+      port = options[:port]
+
+      if options[:config]
+        conf = ParseConfig.new(options[:config])
+        application = application.add_auth(conf) if conf.get_value('password')
+        daemonize = conf.get_value('daemonize') ? conf.get_value('daemonize') == "true" : daemonize
+        port = conf.get_value('port') ? conf.get_value('port') : port
+      end
+
+      Rack::Server.new(:app => application, :Port => port, :daemonize => daemonize).start
+
+    end
   end
 end

data/test/integration_test.rb

@@ -0,0 +1,45 @@
+require "test_helper"
+require "web-puppet"
+
+class AppTest < Test::Unit::TestCase
+  include Rack::Test::Methods
+
+  def app
+    WebPuppet::App.new
+  end
+
+  def test_view_renders_successfully
+    get "/"
+    assert last_response.ok?
+  end
+
+  def test_view_returns_json
+    get "/"
+    json = JSON.load(last_response.body)
+  end
+
+end
+
+class AuthenticationTest < Test::Unit::TestCase
+  include Rack::Test::Methods
+
+  def app
+    conf = mock()
+    conf.expects(:get_value).with('username').returns("admin")
+    conf.expects(:get_value).with('password').returns("password")
+    WebPuppet::App.new().add_auth(conf)
+  end
+
+  def test_view_requires_authentication
+    authorize 'evil', 'password'
+    get "/"
+    assert_equal 401, last_response.status
+  end
+
+  def test_view_renders_with_auth_details
+    authorize 'admin', 'password'
+    get "/"
+    assert last_response.ok?
+  end
+
+end

data/test/test_helper.rb

@@ -0,0 +1,9 @@
+ENV['RACK_ENV'] = 'test'
+
+require "test/unit"
+require "rack/test"
+require "mocha"
+
+%w[ lib ].each do |path|
+  $:.unshift path unless $:.include?(path)
+end

data/web-puppet.gemspec

@@ -20,4 +20,9 @@ Gem::Specification.new do |s|
 
   s.add_runtime_dependency "rack"
   s.add_runtime_dependency "puppet", ">= 2.7.0"
+  s.add_runtime_dependency "parseconfig"
+
+  s.add_development_dependency "test-unit"
+  s.add_development_dependency "rack-test"
+  s.add_development_dependency "mocha"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: web-puppet
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-10-30 00:00:00.000000000Z
+date: 2011-11-02 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rack
-  requirement: &70345799305680 !ruby/object:Gem::Requirement
+  requirement: &70356553014700 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70345799305680
+  version_requirements: *70356553014700
 - !ruby/object:Gem::Dependency
   name: puppet
-  requirement: &70345799305180 !ruby/object:Gem::Requirement
+  requirement: &70356553014200 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,7 +32,51 @@ dependencies:
         version: 2.7.0
   type: :runtime
   prerelease: false
-  version_requirements: *70345799305180
+  version_requirements: *70356553014200
+- !ruby/object:Gem::Dependency
+  name: parseconfig
+  requirement: &70356553013780 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70356553013780
+- !ruby/object:Gem::Dependency
+  name: test-unit
+  requirement: &70356553013320 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70356553013320
+- !ruby/object:Gem::Dependency
+  name: rack-test
+  requirement: &70356553012900 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70356553012900
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: &70356553012480 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70356553012480
 description: Daemon which serves information from a local puppet master as JSON over
   HTTP
 email:
@@ -50,6 +94,8 @@ files:
 - bin/web-puppet
 - lib/web-puppet.rb
 - lib/web-puppet/version.rb
+- test/integration_test.rb
+- test/test_helper.rb
 - web-puppet.gemspec
 homepage: https://github.com/garethr/web-puppet
 licenses: []
@@ -75,4 +121,6 @@ rubygems_version: 1.8.6
 signing_key: 
 specification_version: 3
 summary: Display Puppet node information as JSON over HTTP
-test_files: []
+test_files:
+- test/integration_test.rb
+- test/test_helper.rb