From f34b4fdf32c6d97f69a6e4bf9e851023b04c8748 Mon Sep 17 00:00:00 2001
From: Michael Kroes <michael.kroes@gmail.com>
Date: Sun, 1 Nov 2015 03:29:45 -0500
Subject: [PATCH] Add checks to the management interface to report memory usage

---
 management/status_checks.py | 23 +++++++++++++++++++++++
 setup/management.sh         |  2 +-
 2 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/management/status_checks.py b/management/status_checks.py
index 9942f34c..bace461c 100755
--- a/management/status_checks.py
+++ b/management/status_checks.py
@@ -11,6 +11,7 @@ import sys, os, os.path, re, subprocess, datetime, multiprocessing.pool
 import dns.reversename, dns.resolver
 import dateutil.parser, dateutil.tz
 import idna
+import psutil
 
 from dns_update import get_dns_zones, build_tlsa_record, get_custom_dns_config, get_secondary_dns
 from web_update import get_web_domains, get_default_www_redirects, get_ssl_certificates, get_domain_ssl_files, get_domains_with_a_records
@@ -152,6 +153,7 @@ def run_system_checks(rounded_values, env, output):
 	check_miab_version(env, output)
 	check_system_aliases(env, output)
 	check_free_disk_space(rounded_values, env, output)
+	check_free_memory(rounded_values, env, output)
 
 def check_ssh_password(env, output):
 	# Check that SSH login with password is disabled. The openssh-server
@@ -202,6 +204,27 @@ def check_free_disk_space(rounded_values, env, output):
 	else:
 		output.print_error(disk_msg)
 
+def check_free_memory(rounded_values, env, output):
+	# Check free memory.
+	percent_used = psutil.virtual_memory().percent
+	percent_left = 100 - percent_used
+	if not rounded_values:
+		memory_msg = "The system has allocated %s%% of the memory." % str(round(percent_used))
+		if percent_left > 20:
+			output.print_ok(memory_msg)
+		elif percent_left > 15:
+			output.print_warning(memory_msg)
+		else:
+			output.print_error(memory_msg)
+	else:
+		memory_msg = "The system has less than %s%% memory left." % str(round(percent_left))
+		if percent_left > 20:
+			output.print_ok("The system has more than 20% memory left")
+		elif percent_left > 15:
+			output.print_warning("The system has less than 20% memory left but more than 15%")
+		else:
+			output.print_error("The system has less than 15% memory left")
+
 def run_network_checks(env, output):
 	# Also see setup/network-checks.sh.
 
diff --git a/setup/management.sh b/setup/management.sh
index 468ba91a..2cd6579a 100755
--- a/setup/management.sh
+++ b/setup/management.sh
@@ -7,7 +7,7 @@ echo "Installing Mail-in-a-Box system management daemon..."
 # build-essential libssl-dev libffi-dev python3-dev: Required to pip install cryptography.
 apt_install python3-flask links duplicity libyaml-dev python3-dnspython python3-dateutil \
 	build-essential libssl-dev libffi-dev python3-dev python-pip
-hide_output pip3 install --upgrade rtyaml email_validator>=1.0.0 idna>=2.0.0 cryptography>=1.0.2 boto
+hide_output pip3 install --upgrade rtyaml email_validator>=1.0.0 idna>=2.0.0 cryptography>=1.0.2 boto psutil
 
 # duplicity uses python 2 so we need to use the python 2 package of boto
 hide_output pip install --upgrade boto