Add task to blank bundler config

Sam Cook · Sam Cook · commit 2e9bbfd7827e · 2017-04-18T17:30:30.000+01:00
If the gemstash server fails then bundler won't automatically failover
(due to the age of some versions of bundler). In order to continue
deploiying if the gemstash server is unavailable (and can't be brought
back up) we can remove the configuration that says to use the server and
return to using the default values (i.e. rubygems).

This adds a task that will lock puppet on the targeted machine then
empty the bundler config file.
diff --git a/bundler.py b/bundler.py
@@ -0,0 +1,42 @@
+from os.path import join
+from fabric.api import sudo, task
+from fabric.utils import error
+import fabric.contrib.files
+
+import puppet
+
+config_file_locations = [
+    '/home/deploy',
+    '/var/lib/jenkins'
+]
+
+config_file_suffix = '.bundle/config'
+
+
+def get_bundler_config():
+    for prefix in config_file_locations:
+        location = join(prefix, config_file_suffix)
+        if fabric.contrib.files.exists(location, use_sudo=True):
+            return location
+
+    error('Could not find bundler config file.')
+
+
+@task
+def failover_to_rubygems():
+    """Change bundler to use rubygems.org for gems rather than gemstash"""
+    bundler_config = get_bundler_config()
+    puppet.disable('Fabric failover_to_rubygems invoked')
+    # Overwrite the bundler config with an empty file to force failover to default
+    # Use an empty file rather than rm to avoid running rm as sudo.
+    sudo('echo "" > {0}'.format(bundler_config))
+    print('Disabled puppet and overwritten the bundler config file.')
+    print('Run "bundler.revert_mirror" to start using gemstash again.')
+
+
+@task
+def revert_mirror():
+    puppet.enable()
+    puppet.agent()
+    get_bundler_config()
+    print('Puppet has been re-enabled. The bundler config file should be in it\'s usual state.')
diff --git a/fabfile.py b/fabfile.py
@@ -20,6 +20,7 @@
 # Our command submodules
 import app
 import apt
+import bundler
 import cache
 import campaigns
 import cdn
