Merge pull request alphagov#234 from alphagov/add-rbenv

deanwilson · web-flow · commit 098b6798a4bb · 2017-01-24T09:57:00.000Z
Add rbenv script
diff --git a/fabfile.py b/fabfile.py
@@ -35,6 +35,7 @@
 import performanceplatform
 import postgresql
 import rabbitmq
+import rbenv
 import rkhunter
 import statsd
 import vm
diff --git a/rbenv.py b/rbenv.py
@@ -0,0 +1,20 @@
+from fabric.api import run, sudo, task, hide
+from fabric.contrib.files import exists
+
+
+@task
+def version_in_use(version):
+    """Tests whether an rbenv version is in use on a machine."""
+
+    rbenv_path = '/usr/lib/rbenv/versions/{0}/'.format(version)
+
+    with hide('running'):
+        with hide('output'):
+            if exists(rbenv_path):
+                pids = sudo("lsof +D {0} | tail -n +2 | tr -s ' ' | cut -f 2 -d ' ' | sort -n | uniq".format(rbenv_path)).split('\r\n')
+            else:
+                pids = []
+        for pid in pids:
+            if pid == '':
+                continue
+            run('ps --no-headers {0}'.format(pid))
