From f232ede3e62c5b3c7389c604d2debee771da7d91 Mon Sep 17 00:00:00 2001
From: Profpatsch <mail@profpatsch.de>
Date: Tue, 4 Jul 2017 21:28:20 +0200
Subject: modules/openlab: add speedtest

A simple script to gather DNS & download speed data.
---
 machines/labnet/labtops.nix        |  1 +
 modules/module-list.nix            |  1 +
 modules/user/openlab/speedtest.nix | 40 +++++++++++++++++++++++++++++++++
 modules/user/openlab/speedtest.py  | 46 ++++++++++++++++++++++++++++++++++++++
 4 files changed, 88 insertions(+)
 create mode 100644 modules/user/openlab/speedtest.nix
 create mode 100755 modules/user/openlab/speedtest.py

diff --git a/machines/labnet/labtops.nix b/machines/labnet/labtops.nix
index f3745901..3e0e6912 100644
--- a/machines/labnet/labtops.nix
+++ b/machines/labnet/labtops.nix
@@ -46,6 +46,7 @@
       enable = true;
       volume = 30;
     };
+    vuizvui.user.openlab.speedtest.enable = true;
 
     services.logind.extraConfig = "HandleLidSwitch=ignore";
 
diff --git a/modules/module-list.nix b/modules/module-list.nix
index 52c61249..6d749457 100644
--- a/modules/module-list.nix
+++ b/modules/module-list.nix
@@ -34,6 +34,7 @@
   ./user/devhell/profiles/services.nix
   ./user/openlab/base.nix
   ./user/openlab/labtops.nix
+  ./user/openlab/speedtest.nix
   ./user/openlab/stackenblocken.nix
   ./user/profpatsch/programs/scanning.nix
 ]
diff --git a/modules/user/openlab/speedtest.nix b/modules/user/openlab/speedtest.nix
new file mode 100644
index 00000000..df4aefa3
--- /dev/null
+++ b/modules/user/openlab/speedtest.nix
@@ -0,0 +1,40 @@
+{ config, lib, pkgs, ... }:
+
+with lib;
+
+let
+  bin = drv: name: "${lib.getBin drv}/bin/${name}";
+  cfg = config.vuizvui.user.openlab.speedtest;
+
+  py = pkgs.runCommand "speedtest.py" {} ''
+    cat ${./speedtest.py} \
+      | sed -e 's|^PING_BIN =.*$|PING_BIN = "${config.security.wrapperDir}/ping"|' \
+      > $out
+  '';
+
+  speedtest = pkgs.writeScript "speedtest" ''
+    #!${bin pkgs.bash "bash"}
+    ${bin pkgs.python3 "python3"} ${py} >> $HOME/data.yaml
+  '';
+
+in {
+  options.vuizvui.user.openlab.speedtest.enable =
+    mkEnableOption "openlab speedtest";
+
+  config = mkIf cfg.enable {
+    systemd.services.speedtest = {
+       description = "openlab network speedtest";
+       path = with pkgs; [ curl bind.host ];
+       wantedBy = [ "default.target" ];
+       after = [ "network.target" ];
+       script = "${speedtest}";
+       startAt = [ "*-*-* *:00/15:00" ];
+       serviceConfig.User = "speedtest";
+     };
+
+     users.users.speedtest = {
+        createHome = true;
+        home = "/var/lib/speedtest";
+     };
+  };
+}
diff --git a/modules/user/openlab/speedtest.py b/modules/user/openlab/speedtest.py
new file mode 100755
index 00000000..6868263b
--- /dev/null
+++ b/modules/user/openlab/speedtest.py
@@ -0,0 +1,46 @@
+#!/usr/bin/env nix-shell
+#!nix-shell -i python3 -p curl python3
+
+import sys
+import subprocess as sub
+import datetime
+
+IP = "46.252.18.154"
+DOMAIN = "profpatsch.de"
+PROTOCOL = "http"
+FILE = "/stuff/speedtest.rng"
+
+HOST_BIN = "host"
+PING_BIN = "ping"
+
+dns = 0 == sub.run([HOST_BIN, "-W1", DOMAIN], stdout=sub.DEVNULL).returncode
+
+ping = 0 == sub.run([PING_BIN, "-w1", "-W1", "-c1", DOMAIN],
+                    stdout=sub.DEVNULL).returncode
+
+bytes_per_sec = 0
+if ping == True:
+    d = DOMAIN if dns else IP
+    res = sub.run(["curl", "--silent", PROTOCOL + "://" + d + FILE,
+                   "--write-out", "\n%{speed_download}"],
+                  stdout=sub.PIPE, stderr=sub.PIPE)
+    if res.returncode != 0:
+        sys.exit("download failed unexpectedly. curl outputs:\n" + res.stderr)
+    else:
+        # the last line is the download speed
+        out = res.stdout.split(b"\n")[-1].strip()
+        try:
+            bytes_per_sec = float(out)
+        except ValueError:
+            sys.exit("last line of curl was no float (bytes per sec), but:\n" +
+                     out[0:100] + "\nthere were " + len(out) + " lines in the output")
+
+# some yaml-like output
+def bool_(b):
+    return "true" if b else "false"
+
+print("---")
+print("date: " + str(datetime.datetime.now()))
+print("dns: " + bool_(dns))
+print("ping: " + bool_(ping))
+print("download_speed: {}".format(int(bytes_per_sec)))
-- 
cgit 1.4.1