From 7761514c09bfdd8aaa070907b7511ceb3a62cf13 Mon Sep 17 00:00:00 2001
From: pcy <pcy@ulyssis.org>
Date: Thu, 16 Jul 2020 00:24:49 +0200
Subject: [PATCH] [plugins/git/gitlab_statistics] added plugin to display
 GitLab stats

Graphs common GitLab statistics (user, repo, fork, ... count) using its
HTTP API statistics endpoint (requires a token to be set and passed to
the plugin)
---
 plugins/git/gitlab_statistics | 109 ++++++++++++++++++++++++++++++++++
 1 file changed, 109 insertions(+)
 create mode 100755 plugins/git/gitlab_statistics

diff --git a/plugins/git/gitlab_statistics b/plugins/git/gitlab_statistics
new file mode 100755
index 00000000..7ed144d2
--- /dev/null
+++ b/plugins/git/gitlab_statistics
@@ -0,0 +1,109 @@
+#!/usr/bin/env python3
+# -*- python -*-
+
+"""
+=head1 INTRODUCTION
+
+Plugin to monitor Gitlab status
+
+=head1 INSTALLATION
+
+Usage: Place in /etc/munin/plugins/ (or link it there using ln -s)
+
+=head1 CONFIGURATION
+
+Add this to your /etc/munin/plugin-conf.d/munin-node:
+
+=over 2
+
+    [gitlab_statistics]
+    env.logarithmic 1
+    env.hostname gitlab.example.com # required
+    env.token YourPrivateTokenHere # required
+
+=back
+
+=head1 HISTORY
+
+2019-10-02: v 1.0 pcy <pcy.ulyssis.org>: created
+
+=head1 USAGE
+
+Parameters understood:
+
+  config   (required)
+  autoconf (optional - used by munin-config)
+
+=head1 MAGIC MARKERS
+
+#%# family=auto
+#%# capabilities=autoconf
+"""
+
+
+import os
+import json
+import urllib
+import sys
+
+
+def weakbool(x):
+    return x.lower().strip() in {"true", "yes", "y", "1"}
+
+
+url = None
+if 'hostname' in os.environ and 'token' in os.environ:
+    url = "https://" + os.getenv('hostname') \
+        + "/api/v4/application/statistics?private_token=" \
+        + os.getenv('token')
+
+logarithmic = weakbool(os.getenv('logarithmic', 'N'))
+
+
+def reqjson():
+    try:
+        raw_data = urllib.request.urlopen(url)
+        return json.loads(raw_data)
+    except IOError as e:
+        print("Cannot reach the GitLab API endpoint.", file=sys.stderr)
+        exit(1)
+
+
+def autoconf():
+    if 'hostname' not in os.environ:
+        print("no ('hostname' envvar not set)")
+    elif 'token' not in os.environ:
+        print("no ('token' envvar not set)")
+    else:
+        print("yes")
+
+
+def config():
+    print("""\
+graph_title GitLab statistics
+graph_vlabel amount
+graph_category devel""")
+    if logarithmic:
+        print("graph_args --logarithmic")
+
+    for x in reqjson().keys():
+        print(x + ".label " + x)
+
+
+def fetch():
+    rj = reqjson()
+    for (x, y) in rj.items():
+        print("%s.value %d" % (x, int(y.replace(',', ''))))
+
+
+if len(sys.argv) >= 2:
+    if sys.argv[1] == 'autoconf':
+        autoconf()
+    elif sys.argv[1] == 'config':
+        config()
+    else:
+        fetch()
+else:
+    fetch()
+
+# flake8: noqa: E265