puppet_

#!/usr/bin/env ruby

# USAGE
# This plugin is named puppet_ it should be symlinked into /etc/munin/plugins as
# either puppet_node or puppet_total(alternatively it can be puppet_nodes or
# puppet_totals).
# In your /etc/munin/plugin-conf.d/puppet_ set the following:
#  [puppet_*]
#  user root
#
# There are two enviroment variables that can be passed to it:
#  puppet_logs - This should be a string that defines the puppet log file
#                (by default it is "/var/log/syslog")
#  num_minutes - Only useful when calling the plugin as puppet_node, see below
#
# puppet_node
# Running this plugin as puppet_node will report the total amount of compiled
# catalogs/configs and their average compile time over the last 5 and 30 minutes.
# These values can be altered by using the enviroment variable num_minutes.
#
# num_minutes should be a string of any amount of integers separated by a
# comma(,). Example: "5,30,60" will report the above metrics for the last
# 5, 30 and 60 minutes.
#
# puppet_total
# Running this plugin as puppet_total will report the total number of known nodes
# and the total number of unique nodes that had their catalog compiled in the
# last 24 hours.
#
# COPYRIGHT
# Copyright Pieter Lexis - Kumina B.V. (pieter@kumina.nl)
# This script is licensed under the GNU GPL version 3 or higher
#
# [1] - http://www.puppetlabs.com/puppet/introduction/

#%# family=auto
#%# capabilities=autoconf

require 'time'

def parselogs(since, count_unique=false) # This function parses the puppet_log
  # Assuming puppetmaster logs to syslog and it does not logrotate more than
  # once every 24 hours.
  # TODO write better syslog handling code
  # pseudo-code:
  # logfiles =["#{logfile}"]
  # logfile.{1,{2..X}.gz}.each do |file|
  #  top_line = [FIRST LINE OF file]
  #  logfiles.add("#{logfile}")
  #  if Time.parse(top_line[0..14] > since
  #   [BREAK FROM each]
  #  end

  logfile = ENV["puppet_logs"] || "/var/log/syslog"
  logfiles = ["#{logfile}", "#{logfile}.1"]

  # setup variables
  # regexp = a regex to be able to extract hostnames and compilation times
  regexp = ".* for (.*) in (.*) seconds"
  count = 0

  if count_unique
    hosts = Array.new
  else
    total = 0
  end
  # The actual parsing of the logs
  logfiles.each do |logfile|
    # We only want lines related to configuration compiles
    File.open(logfile).grep(/Compiled configuration|Compiled catalog/).each do |line|
      # get the puppet_total values
      if count_unique and Time.parse(line[0..14]) > since and line =~ /#{regexp}/
        unless hosts.include?($1)
          hosts << $1
          count += 1
        end
      # get the puppet_node values
      elsif not count_unique and Time.parse(line[0..14]) > since and line =~ /#{regexp}/
        total += $2.to_f
        count += 1
      end
    end
  end

  if count_unique
    return count
  else
    return count, total
  end
end

def count_total # Count the total number of known nodes
  return Dir.entries('/var/lib/puppet/yaml/facts/').size-2
end

# Get the current time, it's used to calculate the since times
t = Time.now

# Do something different based on the name by which this script it invoked
case $0
when /puppet_nodes?$/
  num_minutes = ENV["num_minutes"] || "5,30"
  num_minutes = num_minutes.split(/,/)
  case ARGV[0]
  when "config"
    puts "graph_title Puppet nodeconfigs"
    puts "graph_vlabel clients or number of seconds"
    num_minutes.each do |num_minute|
      puts "last#{num_minute}m_count.label Config compilations in the last #{num_minute} minutes"
      puts "last#{num_minute}m_compile_avg.label Avg compile time for the last #{num_minute} minutes"
    end
    puts "graph_category puppet"
  when "autoconf"
    puts "yes"
  else # we were called with no or a non-sensical parameter
    num_minutes.each do |num_minute|
      count, total = parselogs(t - (60 * num_minute.to_i))
      puts "last#{num_minute}m_count.value #{count}"
      puts "last#{num_minute}m_compile_avg.value #{(total / count).to_s[0..3]}" unless count == 0
    end
  end

when /puppet_totals?$/
  case ARGV[0]
  when "config"
    puts "graph_title Puppet total nodes"
    puts "graph_vlabel clients"
    puts "known_clients.label Total number of known clients"
    puts "last24h_unique_count.label Unique clients in the last 24 hours"
    puts "graph_category puppet"
  when "autoconf"
    if Process.euid == 0
      puts "yes"
    else
      puts "no"
    end
  else
    t24h = t - (60 * 60 * 24)
    puts "known_clients.value #{count_total()}"
    puts "last24h_unique_count.value #{parselogs(t24h, true)}"
  end
else # scriptname makes no sense
  if ARGV[0] == "autoconf"
    puts "no"
  end
end

# vim: shiftwidth=2:
# vim: smarttab:
# vim: expandtab: