puppetclassify 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e3f0656715672e681d21f438e317f3b59e3e936d
+  data.tar.gz: e5cd2c0623b6d10ec7fc4ca5231277e0aa67a4d3
+SHA512:
+  metadata.gz: d17542f290a25be6e75286edb3a247bc5df2be79e425732ad00f2226574e0db1e4928478730ea5813ea677c38772c7ed09be2fbc8e26d88ebf6d1c5368803e05
+  data.tar.gz: fa560aeab7e6db465bd90c88b36e20850308a67a4899229ba60497ec942140b4dc5616076bfdce58fb1a39dac3ac053a6dab40b30f3f152ce3f17f57d519aeed

data/README.md

@@ -0,0 +1,33 @@
+# puppet classify
+
+A ruby library to interface with the classifier service
+
+## How to install
+
+```
+gem install puppetclassify
+```
+
+### Locally
+
+```
+gem build puppetclassify.gemspec
+gem install puppetclassify-0.1.0.gem
+```
+
+## How to use
+
+```ruby
+require 'puppetclassify'
+# URL of classifier as well as certificates and private key for auth
+auth_info = {
+  "ca_certificate_path" => "/opt/puppet/share/puppet-dashboard/certs/ca_cert.pem",
+  "certificate_path"    => "/opt/puppet/share/puppet-dashboard/certs/pe-internal-dashboard.cert.pem",
+  "private_key_path"    => "/opt/puppet/share/puppet-dashboard/certs/pe-internal-dashboard.private_key.pem"
+}
+
+classifier_url = 'https://puppetmaster.local:4433/classifier-api'
+puppetclassify = PuppetClassify.new(classifier_url, auth_info)
+# Get all the groups
+puppetclassify.groups.get_groups
+```

data/lib/puppet_https.rb

@@ -0,0 +1,84 @@
+require 'uri'
+require 'net/https'
+
+class PuppetHttps
+  def initialize(settings)
+    # Settings hash:
+    #   - ca_certificate_path
+    #   - certificate_path
+    #   - private_key_path
+    #
+
+    @settings = settings
+  end
+
+  def make_ssl_request(url, req)
+    connection = Net::HTTP.new(url.host, url.port)
+    # connection.set_debug_output $stderr
+    connection.use_ssl = true
+    connection.ssl_version = :TLSv1
+
+    connection.verify_mode = OpenSSL::SSL::VERIFY_PEER
+    ca_file = @settings['ca_certificate_path']
+    certpath = @settings['certificate_path']
+    pkey_path = @settings['private_key_path']
+
+    if File.exists?(ca_file)
+      connection.ca_file = ca_file
+    end
+
+    if File.exists?(certpath)
+      connection.cert = OpenSSL::X509::Certificate.new(File.read(certpath))
+    end
+
+    if File.exists?(pkey_path)
+      connection.key = OpenSSL::PKey::RSA.new(File.read(pkey_path))
+    end
+
+    connection.start { |http| http.request(req) }
+  end
+
+  def put(url, request_body=nil)
+    url = URI.parse(url)
+    req = Net::HTTP::Put.new(url.path)
+    req.content_type = 'application/json'
+
+    unless request_body.nil?
+      req.body = request_body
+    end
+
+    res = make_ssl_request(url, req)
+  end
+
+  def get(url)
+    url = URI.parse(url)
+    accept = 'application/json'
+    req = Net::HTTP::Get.new("#{url.path}?#{url.query}", "Accept" => accept)
+    res = make_ssl_request(url, req)
+    res
+  end
+
+  def post(url, request_body=nil)
+    url = URI.parse(url)
+
+    request = Net::HTTP::Post.new(url.request_uri)
+    request.content_type = 'application/json'
+
+    unless request_body.nil?
+      request.body = request_body
+    end
+
+    res = make_ssl_request(url, request)
+    res
+  end
+
+  def delete(url)
+    url = URI.parse(url)
+
+    request = Net::HTTP::Delete.new(url.request_uri)
+    request.content_type = 'application/json'
+
+    res = make_ssl_request(url, request)
+    res
+  end
+end

data/lib/puppetclassify/classes.rb

@@ -0,0 +1,23 @@
+require 'puppet_https'
+
+class Classes
+  def initialize(nc_api_url, puppet_https)
+    @nc_api_url = nc_api_url
+    @puppet_https = puppet_https
+  end
+
+  def get_classes
+    class_res = @puppet_https.get("#{@nc_api_url}/v1/classes")
+    JSON.parse(class_res.body)
+  end
+
+  def get_environment_classes(environment)
+    class_res = @puppet_https.get("#{@nc_api_url}/v1/environments/#{environment}/classes")
+    JSON.parse(class_res.body)
+  end
+
+  def get_environment_class(environment, class_name)
+    class_res = @puppet_https.get("#{@nc_api_url}/v1/environments/#{environment}/classes/#{class_name}")
+    JSON.parse(class_res.body)
+  end
+end

data/lib/puppetclassify/classification.rb

@@ -0,0 +1,31 @@
+require 'puppet_https'
+
+class Classification
+  def initialize(nc_api_url, puppet_https)
+    @nc_api_url = nc_api_url
+    @puppet_https = puppet_https
+  end
+
+  def get(name)
+    class_res = @puppet_https.post("#{@nc_api_url}/v1/classified/nodes/#{name}")
+
+    unless class_res.code.to_i == 200
+      STDERR.puts "An error occured retreiving the classification of node #{name}: HTTP #{class_res.code} #{class_res.message}"
+      STDERR.puts class_res.body
+    else
+      JSON.parse(class_res.body)
+    end
+  end
+
+  def explain(name)
+    class_res = @puppet_https.post("#{@nc_api_url}/v1/classified/nodes/#{name}/explanation")
+
+    unless class_res.code.to_i == 200
+      STDERR.puts "An error occured retreiving the classification explanation of node #{name}: HTTP #{class_res.code} #{class_res.message}"
+      STDERR.puts class_res.body
+    else
+      JSON.parse(class_res.body)
+    end
+  end
+
+end

data/lib/puppetclassify/environments.rb

@@ -0,0 +1,26 @@
+require 'puppet_https'
+
+class Environments
+  def initialize(nc_api_url, puppet_https)
+    @nc_api_url = nc_api_url
+    @puppet_https = puppet_https
+  end
+
+  def get_environments
+    env_res = @puppet_https.get("#{@nc_api_url}/v1/environments")
+    JSON.parse(env_res.body)
+  end
+
+  def get_environment(name)
+    env_res = @puppet_https.get("#{@nc_api_url}/v1/environments/#{name}")
+    JSON.parse(env_res.body)
+  end
+
+  def create_environment(name)
+    env_res = @puppet_https.put("#{@nc_api_url}/v1/environments/#{name}")
+
+    unless env_res.code.to_i == 201
+      STDERR.puts "An error occured saving the environment: HTTP #{env_res.code} #{env_res.message}"
+    end
+  end
+end

data/lib/puppetclassify/groups.rb

@@ -0,0 +1,75 @@
+require 'puppet_https'
+require 'json'
+
+class Groups
+  def initialize(nc_api_url, puppet_https)
+    @nc_api_url = nc_api_url
+    @puppet_https = puppet_https
+  end
+
+  def get_group(group_id)
+    # HTTP GET
+    group_res = @puppet_https.get("#{@nc_api_url}/v1/groups/#{group_id}")
+    unless group_res.code.to_i != 200
+      JSON.parse(group_res.body)
+    else
+      STDERR.puts "An error occured with your request: HTTP #{group_res.code} #{group_res.message}"
+      STDERR.puts group_res.body
+    end
+  end
+
+  def get_group_id(group_name)
+    groups_res = @puppet_https.get("#{@nc_api_url}/v1/groups")
+
+    groups = JSON.parse(groups_res.body)
+
+    group_info = groups.find { |group| group['name'] == group_name }
+
+    if group_info.nil?
+      STDERR.puts "Could not find group with the name #{group_name}"
+    else
+      group_info['id']
+    end
+  end
+
+  def get_groups
+    group_res = @puppet_https.get("#{@nc_api_url}/v1/groups")
+    JSON.parse(group_res.body)
+  end
+
+  def create_group(group_info)
+    if group_info['id']
+      # HTTP PUT /v1/groups/:id
+      res = @puppet_https.put("#{@nc_api_url}/v1/groups/#{group_info['id']}", group_info.to_json)
+    else
+      # HTTP POST /v1/groups
+      res = @puppet_https.post("#{@nc_api_url}/v1/groups", group_info.to_json)
+    end
+
+    if res.code.to_i >= 400
+      STDERR.puts "An error occured creating the group: HTTP #{res.code} #{res.message}"
+    else
+      unless group_info['id']
+        res['location'].split("/")[-1]
+      else
+        group_info['id']
+      end
+    end
+  end
+
+  def update_group(group_info_delta)
+    # HTTP POST /v1/groups/:id
+    group_res = @puppet_https.post("#{@nc_api_url}/v1/groups/#{group_info_delta['id']}", group_info_delta.to_json)
+
+    unless group_res.code.to_i == 200
+      STDERR.puts "Update Group failed: HTTP #{group_res.code} #{group_res.message}"
+    end
+  end
+
+  def delete_group(group_id)
+    group_res = @puppet_https.delete("#{@nc_api_url}/v1/groups/#{group_id}")
+    if group_res.code.to_i != 204
+      STDERR.puts "An error occured deleting the group: HTTP #{group_res.code} #{group_res.message}"
+    end
+  end
+end

data/lib/puppetclassify/import_hierarchy.rb

@@ -0,0 +1,17 @@
+require 'puppet_https'
+
+class ImportHierarchy
+  def initialize(nc_api_url, puppet_https)
+    @nc_api_url = nc_api_url
+    @puppet_https = puppet_https
+  end
+
+  def import(group_hierarchy)
+    import_res = @puppet_https.post("#{@nc_api_url}/v1/import-hierarchy", group_hierarchy.to_json)
+
+    if import_res.code.to_i >= 400
+      STDERR.puts "An error has occured during import: HTTP #{import_res.code} #{import_res.message}"
+      STDERR.puts import_res.body
+    end
+  end
+end

data/lib/puppetclassify/last_class_update.rb

@@ -0,0 +1,14 @@
+require 'puppet_https'
+
+class LastClassUpdate
+  def initialize(nc_api_url, puppet_https)
+    @nc_api_url = nc_api_url
+    @puppet_https = puppet_https
+  end
+
+  def get
+    class_update_res = @puppet_https.get("#{@nc_api_url}/v1/last-class-update")
+
+    JSON.parse(class_update_res.body)
+  end
+end

data/lib/puppetclassify/nodes.rb

@@ -0,0 +1,18 @@
+require 'puppet_https'
+
+class Nodes
+  def initialize(nc_api_url, puppet_https)
+    @puppet_https = puppet_https
+    @nc_api_url = nc_api_url
+  end
+
+  def get_nodes
+    node_res = @puppet_https.get("#{@nc_api_url}/v1/nodes")
+    JSON.parse(node_res.body)
+  end
+
+  def get_node(node_name)
+    node_res = @puppet_https.get("#{@nc_api_url}/v1/nodes/#{node_name}")
+    JSON.parse(node_res.body)
+  end
+end

data/lib/puppetclassify/rules.rb

@@ -0,0 +1,20 @@
+require 'puppet_https'
+require 'json'
+
+class Rules
+  def initialize(nc_api_url, puppet_https)
+    @nc_api_url = nc_api_url
+    @puppet_https = puppet_https
+  end
+
+  def translate(rule)
+    rules_res = @puppet_https.post("#{@nc_api_url}/v1/rules/translate", rule.to_json)
+
+    unless rules_res.code.to_i == 200
+      STDERR.puts "There was a problem with your rule: HTTP #{rules_res.code.to_i} #{rules_res.message}"
+      STDERR.puts rules_res.body
+    else
+      JSON.parse(rules_res.body)
+    end
+  end
+end

data/lib/puppetclassify/update_classes.rb

@@ -0,0 +1,17 @@
+require 'puppet_https'
+
+class UpdateClasses
+  def initialize(nc_api_url, puppet_https)
+    @nc_api_url = nc_api_url
+    @puppet_https = puppet_https
+  end
+
+  def update
+    update_res = @puppet_https.post("#{@nc_api_url}/v1/update-classes")
+
+    unless update_res.code.to_i == 201
+      STDERR.puts "An error has occurred during the update: HTTP #{update_res.code} #{update_res.message}"
+      STDERR.puts update_res.body
+    end
+  end
+end

data/lib/puppetclassify/validate.rb

@@ -0,0 +1,18 @@
+require 'puppet_https'
+require 'json'
+
+class Validate
+  def initialize(nc_api_url, puppet_https)
+    @nc_api_url = nc_api_url
+    @puppet_https = puppet_https
+  end
+
+  def validate_group(group_info)
+    validate_res = @puppet_https.post("#{@nc_api_url}/v1/validate/group", group_info.to_json)
+
+    unless validate_res.code.to_i == 200
+      STDERR.puts "An error has occured validating the group: HTTP #{validate_res.code} #{validate_res.message}"
+      STDERR.puts validate_res.body
+    end
+  end
+end

data/lib/puppetclassify.rb

@@ -0,0 +1,98 @@
+require 'puppet_https'
+require 'puppetclassify/groups'
+require 'puppetclassify/environments'
+require 'puppetclassify/classes'
+require 'puppetclassify/nodes'
+require 'puppetclassify/import_hierarchy'
+require 'puppetclassify/update_classes'
+require 'puppetclassify/validate'
+require 'puppetclassify/rules'
+require 'puppetclassify/last_class_update'
+require 'puppetclassify/classification'
+
+class PuppetClassify
+  def initialize(nc_api_url, https_settings)
+    @nc_api_url = nc_api_url
+    @puppet_https = PuppetHttps.new(https_settings)
+  end
+
+  def groups
+    if @groups
+      @groups
+    else
+      @groups = Groups.new(@nc_api_url, @puppet_https)
+    end
+  end
+
+  def nodes
+    if @nodes
+      @nodes
+    else
+      @nodes = Nodes.new(@nc_api_url, @puppet_https)
+    end
+  end
+
+  def environments
+    if @environments
+      @environments
+    else
+      @environments = Environments.new(@nc_api_url, @puppet_https)
+    end
+  end
+
+  def classes
+    if @classes
+      @classes
+    else
+      @classes = Classes.new(@nc_api_url, @puppet_https)
+    end
+  end
+
+  def import_hierarchy
+    if @import_hierarchy
+      @import_hierarchy
+    else
+      @import_hierarchy = ImportHierarchy.new(@nc_api_url, @puppet_https)
+    end
+  end
+
+  def update_classes
+    if @update_classes
+      @update_classes
+    else
+      @update_classes = UpdateClasses.new(@nc_api_url, @puppet_https)
+    end
+  end
+
+  def validate
+    if @validate
+      @validate
+    else
+      @validate = Validate.new(@nc_api_url, @puppet_https)
+    end
+  end
+
+  def rules
+    if @rules
+      @rules
+    else
+      @rules = Rules.new(@nc_api_url, @puppet_https)
+    end
+  end
+
+  def last_class_update
+    if @last_class_update
+      @last_class_update
+    else
+      @last_class_update = LastClassUpdate.new(@nc_api_url, @puppet_https)
+    end
+  end
+
+  def classification
+    if @classification
+      @classification
+    else
+      @classification = Classification.new(@nc_api_url, @puppet_https)
+    end
+  end
+end

metadata

@@ -0,0 +1,56 @@
+--- !ruby/object:Gem::Specification
+name: puppetclassify
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Brian Cain
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-10-30 00:00:00.000000000 Z
+dependencies: []
+description: A ruby library to interface with the classifier service
+email: brian.cain@puppetlabs.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- lib/puppet_https.rb
+- lib/puppetclassify.rb
+- lib/puppetclassify/classes.rb
+- lib/puppetclassify/classification.rb
+- lib/puppetclassify/environments.rb
+- lib/puppetclassify/groups.rb
+- lib/puppetclassify/import_hierarchy.rb
+- lib/puppetclassify/last_class_update.rb
+- lib/puppetclassify/nodes.rb
+- lib/puppetclassify/rules.rb
+- lib/puppetclassify/update_classes.rb
+- lib/puppetclassify/validate.rb
+homepage: https://github.com/puppetlabs/puppet-classify
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Puppet Classify!
+test_files: []