#!/usr/bin/python
import sys
import os
import logging
import datetime
import shutil

sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
from zulip_tools import DEPLOYMENTS_DIR, TIMESTAMP_FORMAT

logging.basicConfig(format="%(asctime)s purge-deployments: %(message)s",
                    level=logging.INFO)

one_week_ago = datetime.datetime.now() - datetime.timedelta(days=7)
to_purge = []
for filename in os.listdir(DEPLOYMENTS_DIR):
    try:
        date = datetime.datetime.strptime(filename, TIMESTAMP_FORMAT)
        if date < one_week_ago:
            if (os.path.abspath(os.readlink("/home/zulip/deployments/current")) ==
                os.path.abspath(os.path.join(DEPLOYMENTS_DIR, filename))):
                # Never purge the currently running deployment
                continue
            to_purge.append(filename)
    except ValueError:
        pass

if to_purge:
    to_purge.sort()
    logging.info("Purging the following old deployments directories: %s" % (", ".join(to_purge),))
    for filename in to_purge:
        shutil.rmtree(os.path.join(DEPLOYMENTS_DIR, filename))
        logging.info("Finished %s" % (filename))
else:
    logging.info("No old deployment directories to purge")