From 81aff30127389ff9795260a324f92ab871c81307 Mon Sep 17 00:00:00 2001
From: Tim Abbott <tabbott@zulip.com>
Date: Thu, 31 Oct 2013 11:56:06 -0400
Subject: [PATCH] Add script to upgrade a local server installation.

This has a decent amount of code copied from update-deployment, but
there's enough different that I think it's reasonable to write this
and then move some common code into the zulip_tools library.

(imported from commit ae4fdf8125daf802f04f4543a9f1ebae4e9cd3bd)
---
 scripts/upgrade-zulip | 66 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100755 scripts/upgrade-zulip

diff --git a/scripts/upgrade-zulip b/scripts/upgrade-zulip
new file mode 100755
index 0000000000..56d557f730
--- /dev/null
+++ b/scripts/upgrade-zulip
@@ -0,0 +1,66 @@
+#!/usr/bin/python -u
+import os
+import sys
+import subprocess
+import logging
+import datetime
+import shutil
+import time
+import tempfile
+
+sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
+from zulip_tools import DEPLOYMENTS_DIR, LOCK_DIR, TIMESTAMP_FORMAT, FAIL, WARNING, ENDC
+
+logging.basicConfig(format="%(asctime)s upgrade-zulip: %(message)s",
+                    level=logging.INFO)
+
+if len(sys.argv) != 2:
+    print FAIL + "Usage: %s <tarball>" % (sys.argv[0],) + ENDC
+    sys.exit(1)
+
+tarball_path = sys.argv[1]
+
+subprocess.check_call(["mkdir", '-p', DEPLOYMENTS_DIR, '/home/zulip/logs'])
+
+start_time = time.time()
+got_lock = False
+while time.time() - start_time < 300:
+    try:
+        os.mkdir(LOCK_DIR)
+        got_lock = True
+        break
+    except OSError:
+        print WARNING + "Another deployment in progress; waiting for lock..." + ENDC
+        time.sleep(10)
+
+if not got_lock:
+    print FAIL + "Deployment already in progress.  Please run\n" \
+               + "  %s/current/scripts/upgrade-zulip %s\n" % (DEPLOYMENTS_DIR, tarball_path) \
+               + "manually when the previous deployment finishes, or run\n" \
+               + "  rmdir %s\n"  % (LOCK_DIR,) \
+               + "if the previous deployment crashed." \
+               + ENDC
+    sys.exit(1)
+
+timestamp = datetime.datetime.now().strftime(TIMESTAMP_FORMAT)
+deploy_path = os.path.join(DEPLOYMENTS_DIR, timestamp)
+
+logging.info("Unpacking the tarball")
+extract_path = tempfile.mkdtemp()
+subprocess.check_call(["tar", "-xf", tarball_path, "-C", extract_path])
+subprocess.check_call(["mv", os.path.join(extract_path, "zulip-server"), deploy_path])
+subprocess.check_call(["rmdir", extract_path])
+os.chdir(deploy_path)
+
+# Update static files
+logging.info("Updating static files")
+subprocess.check_call(["./scripts/update-prod-static", "--prev-deploy",
+                      os.path.join(DEPLOYMENTS_DIR, 'current')])
+
+logging.info("Restarting server...")
+subprocess.check_call(["./scripts/restart-server"])
+
+logging.info("Deployment complete")
+shutil.rmtree(LOCK_DIR)
+
+subprocess.check_call(["./scripts/purge-deployments"])