tayu 0.0.1

data/bin/tayu

@@ -0,0 +1,67 @@
+#!/usr/bin/env ruby
+#
+# Copyright 2012, Adrian van Dongen
+# 
+# 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.
+#
+
+$:.unshift(File.expand_path(File.join(File.dirname(__FILE__), "..", "lib")))
+
+require 'yaml'
+require 'optparse'
+require 'tayu'
+
+options = {}
+
+optparse = OptionParser.new do |opts|
+  # Set a banner, displayed at the top
+  # of the help screen.
+  opts.banner = 'Usage: tayu [options] ...'
+
+  opts.separator ''
+  opts.separator 'Configuration options:'
+
+  options[:config_file] =   File.join(File.expand_path(File.dirname(__FILE__)), "..", "config/tayu.yml")
+  opts.on( '-c', '--config FILE', 'Specify the location of your tayu configuration file') do |config_file|
+    options[:config_file] = config_file
+  end 
+
+  options[:port] = 8144
+  opts.on( '-p', '--port PORT', 'The port to run tayu on') do |port|
+    options[:port] = port
+  end
+
+  opts.separator ""
+  opts.separator "Common options:"
+
+  opts.on_tail('-h', '--help', 'Display this screen' ) do
+    puts opts
+    exit
+  end
+end
+
+begin
+  optparse.parse!
+  config = YAML.load_file(options[:config_file])
+
+  Tayu.puppetdb_server = config[:puppetdb_server]
+  Tayu.puppetdb_port = config[:puppetdb_port]
+  Tayu.allowed_tags = config[:allowed_tags].split(' ')
+  Tayu.username = config[:username]
+  Tayu.run! :host => "localhost", :port => options[:port] 
+rescue OptionParser::InvalidArgument, OptionParser::InvalidOption, OptionParser::MissingArgument
+  puts $!.to_s
+  puts optparse
+  exit
+end
+

data/config/tayu.yml

@@ -0,0 +1,6 @@
+:puppetdb_server: localhost
+:puppetdb_port: 8080
+:username: rundeck
+:allowed_tags:
+  location
+  updateschedule

data/lib/tayu.rb

@@ -0,0 +1,122 @@
+#
+# Copyright 2012, Adrian van Dongen
+#
+# 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 'sinatra'
+require 'json'
+require 'rest_client'
+require 'builder/xchar'
+
+class Tayu < Sinatra::Base
+
+    class << self
+        attr_accessor :puppetdb_server
+        attr_accessor :puppetdb_port
+        attr_accessor :allowed_tags
+        attr_accessor :username
+    end
+
+    ## convert to string and then to XML escaped text.
+    def xml_escape(input)
+        return input.to_s.to_xs
+    end
+
+    ## retrieve all nodes from puppetdb and return then as an array
+    def get_nodes
+	q = '["=", ["node", "active"], true]'
+
+	if ! q.is_a? String then
+		q=JSON[q]
+		end
+	params = {:query => q}
+	puts q
+	puts params
+
+        response_nodelist = RestClient.get"http://#{Tayu.puppetdb_server}:#{Tayu.puppetdb_port}/nodes",  { :accept => :json, :params => params }
+        return JSON.parse(response_nodelist)
+    end 
+
+    ## retrieve all facts for specified node and return them as an hash
+    def get_facts(nodename)
+        facts_puppetdb = Hash.new
+        response_facts = RestClient.get"http://#{Tayu.puppetdb_server}:#{Tayu.puppetdb_port}/facts/" + nodename, { :accept => :json}
+
+        # json string holds two object, we only need te facts object
+        json_object = JSON.parse(response_facts)
+        json_object.each do |key, value|
+            if key == 'facts'
+                facts_puppetdb = value
+            end
+        end 
+        return facts_puppetdb
+    end
+
+    ## generate xml document body based on retrieved nodes and there facts
+    def generate_body
+        xmlbody = %Q(<?xml version="1.0" encoding="UTF-8"?>\n<!DOCTYPE project PUBLIC "-//DTO Labs Inc.//DTD Resources Document 1.0//EN" "project.dtd">\n<project>\n)
+    
+        # for each node get the facts 
+        get_nodes.each do |node|
+            facts = get_facts(node)
+            tags_config = Tayu.allowed_tags 
+            tags_rundeck = Array.new 
+            os_family = facts["kernel"] =~ /windows/i ? 'windows' : 'unix'
+
+            # check if value is empty if not add to array
+            tags_config.each do |tag|
+                if !facts[tag].nil?
+                    tags_rundeck << facts[tag]
+                end
+            end
+
+            # xml body
+            xmlbody << <<-EOH
+            <node name="#{xml_escape(facts["fqdn"])}"
+                type="Node"
+                description="#{xml_escape(facts["fqdn"])}"
+                osArch="#{xml_escape(facts["kernel"])}"
+                osFamily="#{xml_escape(os_family)}"
+                osName="#{xml_escape(facts["operatingsystem"])}"
+                tags="#{xml_escape(tags_rundeck.join(','))}"
+                osVersion="#{xml_escape(facts["operatingsystemrelease"])}"
+                username="#{xml_escape(Tayu.username)}"
+                hostname="#{xml_escape(facts["fqdn"])}"/>
+    EOH
+        end
+
+        # close xml body and return results
+        xmlbody << "</project>"
+        return xmlbody
+    end
+
+    ## index page 
+    get '/' do
+
+        # set document type to xml
+        response['Content-Type'] = 'text/xml'
+
+        # Generate xml body and display it
+        body = generate_body
+        body
+
+    end
+
+    ## 404 page
+    not_found do
+       status 404  
+       'page not found' 
+    end
+
+end

metadata

@@ -0,0 +1,80 @@
+--- !ruby/object:Gem::Specification
+name: tayu
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Adrian van Dongen
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-10-19 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sinatra
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: builder
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+description: Provides a resource endpoint for RunDeck from a PuppetDB server
+email: sirhopcount@goodfellasonline.nl
+executables:
+- tayu
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/tayu.rb
+- config/tayu.yml
+- bin/tayu
+homepage: http://github.com/sirhopcount/tayu
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Intergrates PuppetDB with Rundeck
+test_files: []