From 6a4c27b53952dc759d05519e40276e38ec1a149f Mon Sep 17 00:00:00 2001
From: Brian Schonecker <brian.schonecker@nfiindustries.com>
Date: Thu, 29 Dec 2022 11:58:33 -0500
Subject: [PATCH] Add graph_facts.

---
 REFERENCE.md              | 7 +++++++
 manifests/init.pp         | 2 ++
 templates/settings.py.erb | 3 +++
 3 files changed, 12 insertions(+)

diff --git a/REFERENCE.md b/REFERENCE.md
index 2ed41346..c212f589 100644
--- a/REFERENCE.md
+++ b/REFERENCE.md
@@ -83,6 +83,7 @@ The following parameters are available in the `puppetboard` class:
 * [`ldap_require_group`](#ldap_require_group)
 * [`apache_confd`](#apache_confd)
 * [`apache_service`](#apache_service)
+* [`graph_facts'](#graph_facts)
 
 ##### <a name="install_from"></a>`install_from`
 
@@ -414,6 +415,12 @@ Data type: `String[1]`
 
 name of the apache2 service
 
+##### <a name="graph_facts"></a>`graph_facts`
+
+Data type: `Optional[Variant[String[1], Array[String[1]]]]`
+
+An array of the puppet facts to graph.  See ''https://github.com/voxpupuli/puppetboard''
+
 ### <a name="puppetboardapacheconf"></a>`puppetboard::apache::conf`
 
 Creates an entry in your apache configuration directory to run PuppetBoard server-wide (i.e. not in a vhost).
diff --git a/manifests/init.pp b/manifests/init.pp
index e3142c26..80cc1da0 100644
--- a/manifests/init.pp
+++ b/manifests/init.pp
@@ -5,6 +5,7 @@
 # @param homedir Puppetboard system user's home directory.
 # @param group Puppetboard system group.
 # @param groups additional groups for the user that runs puppetboard
+# @param graph_facts Array of puppet facts to graph (pie/bar chart).
 # @param basedir Base directory where to build puppetboard vcsrepo and python virtualenv.
 # @param git_source Location of upstream Puppetboard GIT repository
 # @param puppetdb_host PuppetDB Host
@@ -67,6 +68,7 @@
   Optional[Stdlib::Absolutepath] $homedir                     = undef,
   String $group                                               = 'puppetboard',
   Optional[Variant[String[1], Array[String[1]]]] $groups      = undef,
+  Optional[Variant[String[1], Array[String[1]]]] $graph_facts = undef,
   Stdlib::AbsolutePath $basedir                               = '/srv/puppetboard',
   String $git_source                                          = 'https://github.com/voxpupuli/puppetboard',
   String $dev_listen_host                                     = '127.0.0.1',
diff --git a/templates/settings.py.erb b/templates/settings.py.erb
index fbbf741e..65d8fc18 100644
--- a/templates/settings.py.erb
+++ b/templates/settings.py.erb
@@ -63,6 +63,9 @@ DEFAULT_ENVIRONMENT = '<%= @default_environment %>'
 <% if @reports_count -%>
 REPORTS_COUNT = <%= @reports_count %>
 <% end -%>
+<% if @graph_facts -%>
+GRAPH_FACTS = '<%= @graph_facts.sort.uniq.join(',') %>'
+<% end -%>
 <% @extra_settings.keys.sort.each do |key| -%>
 <%= key %> = <%= @extra_settings[key] %>
 <% end -%>