#!/usr/bin/env python3

import os
import argparse
import sys

ZULIP_PATH = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
if ZULIP_PATH not in sys.path:
    sys.path.append(ZULIP_PATH)

from scripts.lib.zulip_tools import run, parse_lsb_release
from scripts.lib.setup_venv import (
    setup_virtualenv, THUMBOR_VENV_DEPENDENCIES, YUM_THUMBOR_VENV_DEPENDENCIES
)

parser = argparse.ArgumentParser(description="Create a thumbor virtualenv with caching")
parser.add_argument("deploy_path")
args = parser.parse_args()

# install dependencies for setting up the virtualenv
distro_info = parse_lsb_release()
vendor = distro_info['DISTRIB_ID']
family = distro_info['DISTRIB_FAMILY']
if family == 'debian':
    run(["apt-get", "-y", "install"] + THUMBOR_VENV_DEPENDENCIES)
elif family == 'redhat':
    run(["yum", "-y", "install"] + YUM_THUMBOR_VENV_DEPENDENCIES)
else:
    print("Unsupported platform: {}".format(vendor))
    sys.exit(1)

venv_name = "zulip-thumbor-venv"
cached_venv_path = setup_virtualenv(
    os.path.join(args.deploy_path, venv_name),
    os.path.join(ZULIP_PATH, "requirements", "thumbor.txt"),
    virtualenv_args=['-p', 'python2.7'])