mirror of
				https://github.com/zulip/zulip.git
				synced 2025-11-03 21:43:21 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			124 lines
		
	
	
		
			4.0 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			124 lines
		
	
	
		
			4.0 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
#!/usr/bin/env python3
 | 
						|
import argparse
 | 
						|
import logging
 | 
						|
import os
 | 
						|
import subprocess
 | 
						|
import sys
 | 
						|
import time
 | 
						|
 | 
						|
LOCAL_GIT_CACHE_DIR = "/srv/zulip.git"
 | 
						|
os.environ["PYTHONUNBUFFERED"] = "y"
 | 
						|
 | 
						|
sys.path.append(os.path.join(os.path.dirname(__file__), "..", ".."))
 | 
						|
from scripts.lib.zulip_tools import (
 | 
						|
    DEPLOYMENTS_DIR,
 | 
						|
    assert_running_as_root,
 | 
						|
    get_config,
 | 
						|
    get_config_file,
 | 
						|
    get_deploy_options,
 | 
						|
    get_deployment_lock,
 | 
						|
    make_deploy_path,
 | 
						|
    overwrite_symlink,
 | 
						|
    release_deployment_lock,
 | 
						|
    su_to_zulip,
 | 
						|
)
 | 
						|
 | 
						|
config_file = get_config_file()
 | 
						|
deploy_options = get_deploy_options(config_file)
 | 
						|
upstream_url = "https://github.com/zulip/zulip.git"
 | 
						|
remote_url = get_config(config_file, "deployment", "git_repo_url", upstream_url)
 | 
						|
 | 
						|
assert_running_as_root(strip_lib_from_paths=True)
 | 
						|
 | 
						|
# make sure we have appropriate file permissions
 | 
						|
os.umask(0o22)
 | 
						|
 | 
						|
logging.Formatter.converter = time.gmtime
 | 
						|
logging.basicConfig(format="%(asctime)s upgrade-zulip-from-git: %(message)s", level=logging.INFO)
 | 
						|
 | 
						|
parser = argparse.ArgumentParser()
 | 
						|
parser.add_argument("refname", help="Git reference, e.g. a branch, tag, or commit ID.")
 | 
						|
parser.add_argument(
 | 
						|
    "--remote-url", help="Override the Git remote URL configured in /etc/zulip/zulip.conf."
 | 
						|
)
 | 
						|
args, extra_options = parser.parse_known_args()
 | 
						|
 | 
						|
refname = args.refname
 | 
						|
# Command line remote URL will be given preference above the one
 | 
						|
# in /etc/zulip/zulip.conf.
 | 
						|
if args.remote_url:
 | 
						|
    remote_url = args.remote_url
 | 
						|
 | 
						|
os.makedirs(DEPLOYMENTS_DIR, exist_ok=True)
 | 
						|
os.makedirs("/home/zulip/logs", exist_ok=True)
 | 
						|
 | 
						|
error_rerun_script = f"{DEPLOYMENTS_DIR}/current/scripts/upgrade-zulip-from-git {refname}"
 | 
						|
get_deployment_lock(error_rerun_script)
 | 
						|
 | 
						|
try:
 | 
						|
    deploy_path = make_deploy_path()
 | 
						|
 | 
						|
    # Populate LOCAL_GIT_CACHE_DIR with both the requested remote and zulip/zulip.
 | 
						|
    if not os.path.exists(LOCAL_GIT_CACHE_DIR):
 | 
						|
        logging.info("Cloning the repository")
 | 
						|
        subprocess.check_call(
 | 
						|
            ["git", "clone", "-q", remote_url, "--mirror", LOCAL_GIT_CACHE_DIR],
 | 
						|
            stdout=subprocess.DEVNULL,
 | 
						|
        )
 | 
						|
    if os.stat(LOCAL_GIT_CACHE_DIR).st_uid == 0:
 | 
						|
        subprocess.check_call(["chown", "-R", "zulip:zulip", LOCAL_GIT_CACHE_DIR])
 | 
						|
 | 
						|
    os.chdir(LOCAL_GIT_CACHE_DIR)
 | 
						|
    subprocess.check_call(
 | 
						|
        ["git", "remote", "set-url", "origin", remote_url], preexec_fn=su_to_zulip
 | 
						|
    )
 | 
						|
 | 
						|
    # Ensure upstream remote is configured; we need this to make `git describe` accurate.
 | 
						|
    remotes = subprocess.check_output(["git", "remote"], preexec_fn=su_to_zulip).split(b"\n")
 | 
						|
    if b"upstream" not in remotes:
 | 
						|
        subprocess.check_call(
 | 
						|
            ["git", "remote", "add", "upstream", upstream_url], preexec_fn=su_to_zulip
 | 
						|
        )
 | 
						|
    else:
 | 
						|
        subprocess.check_call(
 | 
						|
            ["git", "remote", "set-url", "upstream", upstream_url], preexec_fn=su_to_zulip
 | 
						|
        )
 | 
						|
 | 
						|
    logging.info("Fetching the latest commits")
 | 
						|
    subprocess.check_call(
 | 
						|
        ["git", "fetch", "--prune", "--quiet", "--tags", "--all"], preexec_fn=su_to_zulip
 | 
						|
    )
 | 
						|
 | 
						|
    # Generate the deployment directory via git worktree from our local repository.
 | 
						|
    subprocess.check_call(
 | 
						|
        ["git", "worktree", "add", "--detach", deploy_path, refname],
 | 
						|
        stdout=subprocess.DEVNULL,
 | 
						|
        preexec_fn=su_to_zulip,
 | 
						|
    )
 | 
						|
    os.chdir(deploy_path)
 | 
						|
    subprocess.check_call(
 | 
						|
        ["git", "checkout", "-qtb", "deployment-" + os.path.basename(deploy_path), refname],
 | 
						|
        preexec_fn=su_to_zulip,
 | 
						|
    )
 | 
						|
 | 
						|
    overwrite_symlink("/etc/zulip/settings.py", "zproject/prod_settings.py")
 | 
						|
 | 
						|
    overwrite_symlink(deploy_path, os.path.join(DEPLOYMENTS_DIR, "next"))
 | 
						|
 | 
						|
    try:
 | 
						|
        subprocess.check_call(
 | 
						|
            [
 | 
						|
                os.path.join(deploy_path, "scripts", "lib", "upgrade-zulip-stage-2"),
 | 
						|
                deploy_path,
 | 
						|
                "--from-git",
 | 
						|
                *deploy_options,
 | 
						|
                *extra_options,
 | 
						|
            ]
 | 
						|
        )
 | 
						|
    except subprocess.CalledProcessError:
 | 
						|
        # There's no use in showing a stacktrace here; it just hides
 | 
						|
        # the error from stage 2.
 | 
						|
        sys.exit(1)
 | 
						|
finally:
 | 
						|
    release_deployment_lock()
 |