From 4304f2d4be0bb55a4957173f49cdbe3581c5d290 Mon Sep 17 00:00:00 2001
From: Michael <m.muenz@gmail.com>
Date: Mon, 15 Jan 2018 14:20:21 +0100
Subject: [PATCH] net-mgmt/telegraf: Add graphite and graylog output (#487)

---
 net-mgmt/telegraf/Makefile                    |  2 +-
 .../OPNsense/Telegraf/forms/output.xml        | 38 ++++++++++++++++++-
 .../app/models/OPNsense/Telegraf/Output.xml   | 22 +++++++++++
 .../templates/OPNsense/Telegraf/telegraf.conf | 22 +++++++++++
 4 files changed, 82 insertions(+), 2 deletions(-)

diff --git a/net-mgmt/telegraf/Makefile b/net-mgmt/telegraf/Makefile
index ea79b5c3e..42539a357 100644
--- a/net-mgmt/telegraf/Makefile
+++ b/net-mgmt/telegraf/Makefile
@@ -1,5 +1,5 @@
 PLUGIN_NAME=		telegraf
-PLUGIN_VERSION=		1.1
+PLUGIN_VERSION=		1.2
 PLUGIN_COMMENT=		Agent for collecting metrics and data
 PLUGIN_DEPENDS=		telegraf
 PLUGIN_MAINTAINER=	m.muenz@gmail.com
diff --git a/net-mgmt/telegraf/src/opnsense/mvc/app/controllers/OPNsense/Telegraf/forms/output.xml b/net-mgmt/telegraf/src/opnsense/mvc/app/controllers/OPNsense/Telegraf/forms/output.xml
index dcfb80fd5..f932289ee 100644
--- a/net-mgmt/telegraf/src/opnsense/mvc/app/controllers/OPNsense/Telegraf/forms/output.xml
+++ b/net-mgmt/telegraf/src/opnsense/mvc/app/controllers/OPNsense/Telegraf/forms/output.xml
@@ -1,7 +1,7 @@
 <form>
     <field>
         <id>output.influx_enable</id>
-        <label>Enable Influx</label>
+        <label>Enable Influx Output</label>
         <type>checkbox</type>
         <help>This will enable InfluxDB as output. Format is without square brackets, just like http://192.168.0.1:8086.</help>
     </field>
@@ -35,4 +35,40 @@
         <type>text</type>
         <help>Set the password for authentication.</help>
     </field>
+    <field>
+        <id>output.graphite_enable</id>
+        <label>Enable Graphite Output</label>
+        <type>checkbox</type>
+        <help>This will enable Graphite as output. Format is without square brackets, just like localhost:2003.</help>
+    </field>
+    <field>
+        <id>output.graphite_server</id>
+        <label>Graphite Server</label>
+        <type>text</type>
+        <help>Set the IP and port where metrics shoud be sent to.</help>
+    </field>
+    <field>
+        <id>output.graphite_prefix</id>
+        <label>Graphite Prefix</label>
+        <type>text</type>
+        <help>Set an optional prefix to the metrics.</help>
+    </field>
+    <field>
+        <id>output.graphite_template</id>
+        <label>Graphite Template</label>
+        <type>text</type>
+        <help>Use a specific output template. For usage see https://github.com/influxdata/telegraf/blob/master/docs/DATA_FORMATS_OUTPUT.md</help>
+    </field>
+    <field>
+        <id>output.graylog_enable</id>
+        <label>Enable Graylog Output</label>
+        <type>checkbox</type>
+        <help>This will enable Graylog as output. Format is without square brackets, just like localhost:12201.</help>
+    </field>
+    <field>
+        <id>output.graylog_server</id>
+        <label>Graylog Server</label>
+        <type>text</type>
+        <help>Set the IP and port where metrics shoud be sent to.</help>
+    </field>
 </form>
diff --git a/net-mgmt/telegraf/src/opnsense/mvc/app/models/OPNsense/Telegraf/Output.xml b/net-mgmt/telegraf/src/opnsense/mvc/app/models/OPNsense/Telegraf/Output.xml
index 49f00ab6b..69a437cfc 100644
--- a/net-mgmt/telegraf/src/opnsense/mvc/app/models/OPNsense/Telegraf/Output.xml
+++ b/net-mgmt/telegraf/src/opnsense/mvc/app/models/OPNsense/Telegraf/Output.xml
@@ -27,5 +27,27 @@
             <default></default>
             <Required>N</Required>
         </influx_password>
+        <graphite_enable type="BooleanField">
+            <default>0</default>
+            <Required>N</Required>
+        </graphite_enable>
+        <graphite_server type="TextField">
+            <default></default>
+            <Required>Y</Required>
+        </graphite_server>
+        <graphite_prefix type="TextField">
+            <Required>N</Required>
+        </graphite_prefix>
+        <graphite_template type="TextField">
+            <Required>N</Required>
+        </graphite_template>
+        <graylog_enable type="BooleanField">
+            <default>0</default>
+            <Required>N</Required>
+        </graylog_enable>
+        <graylog_server type="TextField">
+            <default></default>
+            <Required>Y</Required>
+        </graylog_server>
     </items>
 </model>
diff --git a/net-mgmt/telegraf/src/opnsense/service/templates/OPNsense/Telegraf/telegraf.conf b/net-mgmt/telegraf/src/opnsense/service/templates/OPNsense/Telegraf/telegraf.conf
index 1cec14c38..01eda02e3 100644
--- a/net-mgmt/telegraf/src/opnsense/service/templates/OPNsense/Telegraf/telegraf.conf
+++ b/net-mgmt/telegraf/src/opnsense/service/templates/OPNsense/Telegraf/telegraf.conf
@@ -57,6 +57,28 @@
 {%   endif %}
 {% endif %}
 
+{% if helpers.exists('OPNsense.telegraf.output.graphite_enable') and OPNsense.telegraf.output.graphite_enable == '1' %}
+[[outputs.graphite]]
+{%   if helpers.exists('OPNsense.telegraf.output.graphite_server') and OPNsense.telegraf.output.graphite_server != '' %}
+  servers = ["{{ OPNsense.telegraf.output.graphite_server }}"]
+{%   endif %}
+{%   if helpers.exists('OPNsense.telegraf.output.graphite_prefix') and OPNsense.telegraf.output.graphite_prefix != '' %}
+  prefix = "{{ OPNsense.telegraf.output.graphite_prefix }}"
+{%   endif %}
+{%   if helpers.exists('OPNsense.telegraf.output.graphite_template') and OPNsense.telegraf.output.graphite_template != '' %}
+  template = "{{ OPNsense.telegraf.output.graphite_template }}"
+{%   endif %}
+  timeout = 2
+  insecure_skip_verify = true
+{% endif %}
+
+{% if helpers.exists('OPNsense.telegraf.output.graylog_enable') and OPNsense.telegraf.output.graylog_enable == '1' %}
+[[outputs.graylog]]
+{%   if helpers.exists('OPNsense.telegraf.output.graylog_server') and OPNsense.telegraf.output.graylog_server != '' %}
+  servers = ["{{ OPNsense.telegraf.output.graylog_server }}"]
+{%   endif %}
+{% endif %}
+
 {% if helpers.exists('OPNsense.telegraf.input.cpu') and OPNsense.telegraf.input.cpu == '1' %}
 [[inputs.cpu]]
 {%   if helpers.exists('OPNsense.telegraf.input.cpu_percpu') and OPNsense.telegraf.input.cpu_percpu == '1' %}