puppet-module-example

Author: shcherbak

puppet/inventor/README.md

@@ -0,0 +1 @@
+# Inventor

puppet/inventor/manifests/config.pp

@@ -0,0 +1,67 @@
+# == Class inventor::config
+#
+# creates config files
+#
+class inventor::config {
+
+  $key              = $inventor::key
+  $url              = $inventor::url
+  $cron             = $inventor::cron
+  $config_manage    = $inventor::config_manage
+  $info             = $inventor::info
+
+  if $config_manage {
+
+    file { '/opt/inventor':
+      ensure => 'directory',
+      owner  => 'root',
+      group  => 'root',
+      mode   => '0750',
+    }->
+
+    file { '/opt/inventor/run.sh':
+      ensure  => file,
+      owner   => 'root',
+      group   => 'root',
+      mode    => '0750',
+      content => epp("${module_name}/opt/inventor/run.sh.epp", {
+        url  => $url,
+        key  => $key,
+        info => $info,
+        }),
+    }
+
+    file { '/etc/cron.d/inventor':
+      ensure  => file,
+      owner   => 'root',
+      group   => 'root',
+      mode    => '0640',
+      content => epp("${module_name}/etc/cron.d/inventor.epp", {
+        cron => $cron,
+        }),
+    }
+
+    $info.each | String $exporter, Hash $data | {
+      file { "/opt/inventor/${exporter}.json":
+        ensure  => file,
+        owner   => 'root',
+        group   => 'root',
+        mode    => '0640',
+        content => epp("${module_name}/opt/inventor/data.json.epp", {
+          data => $data,
+          }),
+      }
+    }
+
+  } else {
+
+    $files = [
+      '/opt/inventor/*',
+      '/etc/cron.d/inventor',
+    ]
+    file { $files:
+      ensure => 'absent',
+    }
+
+  }
+}

puppet/inventor/manifests/init.pp

@@ -0,0 +1,23 @@
+# == Class: inventor
+#
+# This module installs and configures inventor
+#
+# === Parameters
+
+class inventor (
+  String                        $key             = $inventor::params::key,
+  String                        $url             = $inventor::params::url,
+  String                        $cron            = $inventor::params::cron,
+  Boolean                       $manage_package  = $inventor::params::manage_package,
+  Array[String]                 $package_name    = $inventor::params::package_name,
+  Boolean                       $config_manage   = $inventor::params::config_manage,
+  Hash                          $info            = $inventor::params::info,
+
+  ) inherits inventor::params {
+
+    contain inventor::install
+    contain inventor::config
+
+    Class['inventor::install'] ->
+    Class['inventor::config']
+}

puppet/inventor/manifests/install.pp

@@ -0,0 +1,22 @@
+# == Class inventor::install
+#
+# This class is called from inventor for install.
+#
+class inventor::install {
+
+  $manage_package = $inventor::manage_package
+  $package_ensure = $inventor::package_ensure
+  $package_name   = $inventor::package_name
+
+  # only do repo management when on a Debian-like system
+  if $manage_package {
+
+    include apt
+
+    package { $package_name:
+      ensure => $package_ensure,
+    }
+
+  }
+
+}

puppet/inventor/manifests/params.pp

@@ -0,0 +1,18 @@
+# == Class inventor::params
+#
+# default parameters definition
+#
+class inventor::params {
+
+  $key = 'secret'
+  $url = 'NOT-CONFIGURED'
+  $cron = '0 */3 * * *'
+  $package_ensure = 'installed'
+  $config_manage = true
+  $manage_package = false
+  $package_name = [ 'curl' ]
+  $use_defaults = true
+  # https://puppet.com/docs/puppet/7/core_facts.html#os
+  $info = {}
+
+}

puppet/inventor/templates/etc/cron.d/inventor.epp

@@ -0,0 +1,8 @@
+<%- |
+  String $cron,
+| -%>
+PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
+MAILTO=""
+
+# inventor
+ <%= $cron %> root bash /opt/inventor/run.sh

puppet/inventor/templates/opt/inventor/data.json.epp

@@ -0,0 +1,18 @@
+<%- |
+  Hash $data ,
+| -%>
+{
+  "static_config": {
+    "targets": ["<%= $data['target'] %>"],
+    "labels": {
+        <% if 'labels' in $data { -%>
+        <% $data['labels'].each | String $key, String $val | { -%>
+        "<%= $key %>": "<%= $val %>",
+        <% } -%>
+        <% } -%>
+        "datacenter": "<%= $data['dc'] %>",
+        "job": "<%= $data['job'] %>"
+    },
+    "target_group": "<%= $data['group'] %>"
+  }
+}

puppet/inventor/templates/opt/inventor/run.sh.epp

@@ -0,0 +1,10 @@
+<%- |
+  String $key ,
+  String $url,
+  Hash $info,
+| -%>
+#!/usr/bin/env bash
+set -e
+<% $info.each | String $exporter, Hash $data | { -%>
+curl -XPUT -H "x-api-token: <%= $key %>" <%= $url %> -d @/opt/inventor/<%= $exporter %>.json
+<% } -%>