lita-puppet 0.4.5 → 0.5.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 5c57852a191a5435265de21bc2edc1d675a5d1ae
-  data.tar.gz: 1f5791d60f1c91962086a6a6e9fb41181468d6ce
+  metadata.gz: 76d165987e3bcb1bfa1f04645a372db05bbabb95
+  data.tar.gz: aad95487e3aa635215f82698f19d4783bddf0f76
 SHA512:
-  metadata.gz: cdd81a81ca7718ce7fab03d8d761c82672531b45c80789fe3435f52d77a13792ba5e95297963cfbbe38a83f74d1b05dd5698d6dfd8011b0a156e6d92955abd1c
-  data.tar.gz: 6d508d6095fa6b1cf3360d02560fbce456be962490c0cd85da28cbfc435c7c6475fcbc14a7d1d465d0c7abcf3262c98d0de092a94017e461f7bf252a9f22e04c
+  metadata.gz: 2a66a02659ea330d95fea211ea66f9e6e93039a3cc37ff76e9153cf2fff572814069edc6cdfdee43fed6650dc8c1ab0082c254a6d247ad60b5eaeebd4466111d
+  data.tar.gz: 34e3b651c329184589c4ccef2b6251c89c53a99271a6e24b52e2a1c5024785426776e718da889b0d1825cebf3276ef56b57a1c65397ba402758bb75b93d558e4

data/lib/lita/handlers/puppet.rb

@@ -5,6 +5,7 @@ module Lita
       config :master_hostname, required: true, type: String
       config :ssh_user, required: false, type: String
       config :control_repo_path, required: false, type: String
+      config :puppetdb_url, required: false, type: String
 
       route(
         /(puppet|pp)(\s+agent)?\s+(run)(\s+on)?\s+(\S+)/i,
@@ -24,6 +25,15 @@ module Lita
         }
       )
 
+      route(
+        /(puppet|pp)\s+(catalog|node)\s+(\S+)\s+(profiles)/i,
+        :node_profiles,
+        command: true,
+        help: {
+          "puppet catalog <host> profiles" => "Query PuppetDB to get a list of all roles and profiles applied to <host>."
+        }
+      )
+
       route(
         /(puppet|pp)\s+(r10k|deploy)(\s+(\S+)(\s+(\S+))?)?/i,
         :r10k_deploy,
@@ -33,6 +43,7 @@ module Lita
         }
       )
 
+      include ::Utils::PuppetDB
       include ::Utils::SSH
       include ::Utils::Text
 
@@ -101,6 +112,30 @@ module Lita
         end
       end
 
+      def node_profiles(response)
+        host = response.matches[0][2]
+        url  = config.puppetdb_url
+        username = friendly_name(response.user.name)
+
+        unless url
+          cant_reply = "#{username}, I would do that, but I don't know how to connect to PuppetDB."
+          cant_reply << "Edit my config and add `config.handlers.puppet.puppetdb_url`."
+          response.reply(cant_reply)
+          return false
+        end
+
+        response.reply("#{username}, let me see what I can find in PuppetDB for you.")
+
+        profiles = node_roles_and_profiles(url, host)
+        if profiles.is_a? String
+          response.reply("Hmmm, that didn't work. Here's what PuppetDB responded with: '#{profiles}'")
+          return false
+        else
+          response.reply("Here are the profiles and roles for #{host}:")
+          response.reply("/code" + profiles.join("\n"))
+        end
+      end
+
       def r10k_deploy(response)
         environment = response.matches[0][3]
         mod = response.matches[0][5]

data/lib/lita-puppet.rb

@@ -1,11 +1,13 @@
 require "lita"
 require 'rye'
 require 'timeout'
+require 'puppetdb'
 
 Lita.load_locales Dir[File.expand_path(
   File.join("..", "..", "locales", "*.yml"), __FILE__
 )]
 
+require 'utils/puppetdb'
 require 'utils/ssh'
 require 'utils/text'
 require "lita/handlers/puppet"

data/lib/utils/puppetdb.rb

@@ -0,0 +1,23 @@
+module Utils
+  # Utility methods for working with PuppetDB
+  module PuppetDB
+    def dbquery(url, q)
+      # TODO: validate incoming query structure
+      client = ::PuppetDB::Client.new(server: url)
+      client.request *q
+    end
+
+    def node_roles_and_profiles(url, nodename)
+      # TODO: validate url and nodename
+      ::PuppetDB::Client.new(server: url) # this is weird but required
+      d = ::PuppetDB::Client.get("/catalogs/#{nodename}")
+      return d["error"] if d['error']
+
+      tags = []
+      d["data"]["resources"].each {|r| tags.concat(r['tags'])}
+
+      # return all the tags related to profile:: or role::
+      tags.sort.uniq.select {|t| t.match /^(profile|role)::/ }
+    end
+  end
+end

data/lita-puppet.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |spec|
   spec.name          = "lita-puppet"
-  spec.version       = "0.4.5"
+  spec.version       = "0.5.4"
   spec.authors       = ["Jonathan Gnagy"]
   spec.email         = ["jgnagy@knuedge.com"]
   spec.description   = "Some basic Puppet interactions for Lita"
@@ -16,6 +16,7 @@ Gem::Specification.new do |spec|
 
   spec.add_runtime_dependency "lita", "~> 4.7"
   spec.add_runtime_dependency "rye"
+  spec.add_runtime_dependency "puppetdb-ruby"
 
   spec.add_development_dependency "bundler", "~> 1.3"
   spec.add_development_dependency "pry-byebug"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lita-puppet
 version: !ruby/object:Gem::Version
-  version: 0.4.5
+  version: 0.5.4
 platform: ruby
 authors:
 - Jonathan Gnagy
@@ -38,6 +38,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: puppetdb-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -123,6 +137,7 @@ files:
 - Rakefile
 - lib/lita-puppet.rb
 - lib/lita/handlers/puppet.rb
+- lib/utils/puppetdb.rb
 - lib/utils/ssh.rb
 - lib/utils/text.rb
 - lita-puppet.gemspec