mirror of https://github.com/zulip/zulip.git
68 lines
2.5 KiB
Python
Executable File
68 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import argparse
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
if False:
|
|
from typing import Set, Text
|
|
|
|
ZULIP_PATH = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
sys.path.append(ZULIP_PATH)
|
|
from scripts.lib.node_cache import generate_sha1sum_node_modules
|
|
from scripts.lib.zulip_tools import GENERIC_CACHE_SCRIPT_PARSER, \
|
|
get_caches_to_be_purged, get_environment, get_recent_deployments, \
|
|
purge_unused_caches
|
|
|
|
ENV = get_environment()
|
|
NODE_MODULES_CACHE_PATH = "/srv/zulip-npm-cache"
|
|
if ENV == "travis":
|
|
NODE_MODULES_CACHE_PATH = os.path.join(os.environ["HOME"], "zulip-npm-cache")
|
|
try:
|
|
subprocess.check_output(["/srv/zulip-yarn/bin/yarn", '--version'])
|
|
except OSError:
|
|
print('yarn not found. Most probably we are running static-analysis and '
|
|
'hence yarn is not installed. Exiting without cleaning npm cache.')
|
|
sys.exit(0)
|
|
|
|
def parse_args():
|
|
# type: () -> argparse.Namespace
|
|
parser = argparse.ArgumentParser(description="This script cleans unused zulip npm caches.",
|
|
parents=[GENERIC_CACHE_SCRIPT_PARSER, ])
|
|
args = parser.parse_args()
|
|
return args
|
|
|
|
def get_caches_in_use(threshold_days):
|
|
# type: (int) -> Set[Text]
|
|
setups_to_check = set([ZULIP_PATH, ])
|
|
caches_in_use = set()
|
|
|
|
if ENV == "prod":
|
|
setups_to_check |= get_recent_deployments(threshold_days)
|
|
if ENV == "dev":
|
|
# In dev always include the currently active cache in order
|
|
# not to break current installation in case dependencies
|
|
# are updated with bumping the provision version.
|
|
CURRENT_CACHE = os.path.dirname(os.path.realpath(os.path.join(ZULIP_PATH, "node_modules")))
|
|
caches_in_use.add(CURRENT_CACHE)
|
|
|
|
for setup_dir in setups_to_check:
|
|
PACKAGES_FILE = os.path.join(setup_dir, "package.json")
|
|
if os.path.exists(PACKAGES_FILE):
|
|
# If 'package.json' file doesn't exist then no node_modules
|
|
# cache is associated with this setup.
|
|
sha1sum = generate_sha1sum_node_modules(setup_dir=setup_dir)
|
|
caches_in_use.add(os.path.join(NODE_MODULES_CACHE_PATH, sha1sum))
|
|
|
|
return caches_in_use
|
|
|
|
def main():
|
|
# type: () -> None
|
|
args = parse_args()
|
|
caches_in_use = get_caches_in_use(args.threshold_days)
|
|
purge_unused_caches(
|
|
NODE_MODULES_CACHE_PATH, caches_in_use, args.threshold_days, args.dry_run, "node modules")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|