mirror of
https://github.com/nodejs/node.git
synced 2024-11-25 08:19:38 +01:00
89ede72fca
PR-URL: https://github.com/nodejs/node/pull/7483 Reviewed-By: Ben Noordhuis <info@bnoordhuis.nl> Reviewed-By: James M Snell <jasnell@gmail.com> Reviewed-By: Anna Henningsen <anna@addaleax.net>
128 lines
4.1 KiB
Python
128 lines
4.1 KiB
Python
#!/usr/bin/env python
|
|
# Moved some utilities here from ../../configure
|
|
|
|
import urllib
|
|
import hashlib
|
|
import sys
|
|
import zipfile
|
|
import tarfile
|
|
import fpformat
|
|
import contextlib
|
|
|
|
def formatSize(amt):
|
|
"""Format a size as a string in MB"""
|
|
return fpformat.fix(amt / 1024000., 1)
|
|
|
|
def spin(c):
|
|
"""print out an ASCII 'spinner' based on the value of counter 'c'"""
|
|
spin = ".:|'"
|
|
return (spin[c % len(spin)])
|
|
|
|
class ConfigOpener(urllib.FancyURLopener):
|
|
"""fancy opener used by retrievefile. Set a UA"""
|
|
# append to existing version (UA)
|
|
version = '%s node.js/configure' % urllib.URLopener.version
|
|
|
|
def reporthook(count, size, total):
|
|
"""internal hook used by retrievefile"""
|
|
sys.stdout.write(' Fetch: %c %sMB total, %sMB downloaded \r' %
|
|
(spin(count),
|
|
formatSize(total),
|
|
formatSize(count*size)))
|
|
|
|
def retrievefile(url, targetfile):
|
|
"""fetch file 'url' as 'targetfile'. Return targetfile or throw."""
|
|
try:
|
|
sys.stdout.write(' <%s>\nConnecting...\r' % url)
|
|
sys.stdout.flush()
|
|
ConfigOpener().retrieve(url, targetfile, reporthook=reporthook)
|
|
print '' # clear the line
|
|
return targetfile
|
|
except:
|
|
print ' ** Error occurred while downloading\n <%s>' % url
|
|
raise
|
|
|
|
def md5sum(targetfile):
|
|
"""md5sum a file. Return the hex digest."""
|
|
digest = hashlib.md5()
|
|
with open(targetfile, 'rb') as f:
|
|
chunk = f.read(1024)
|
|
while chunk != "":
|
|
digest.update(chunk)
|
|
chunk = f.read(1024)
|
|
return digest.hexdigest()
|
|
|
|
def unpack(packedfile, parent_path):
|
|
"""Unpacks packedfile into parent_path. Assumes .zip. Returns parent_path"""
|
|
if zipfile.is_zipfile(packedfile):
|
|
with contextlib.closing(zipfile.ZipFile(packedfile, 'r')) as icuzip:
|
|
print ' Extracting zipfile: %s' % packedfile
|
|
icuzip.extractall(parent_path)
|
|
return parent_path
|
|
elif tarfile.is_tarfile(packedfile):
|
|
with tarfile.TarFile.open(packedfile, 'r') as icuzip:
|
|
print ' Extracting tarfile: %s' % packedfile
|
|
icuzip.extractall(parent_path)
|
|
return parent_path
|
|
else:
|
|
packedsuffix = packedfile.lower().split('.')[-1] # .zip, .tgz etc
|
|
raise Exception('Error: Don\'t know how to unpack %s with extension %s' % (packedfile, packedsuffix))
|
|
|
|
# List of possible "--download=" types.
|
|
download_types = set(['icu'])
|
|
|
|
# Default options for --download.
|
|
download_default = "none"
|
|
|
|
def help():
|
|
"""This function calculates the '--help' text for '--download'."""
|
|
return """Select which packages may be auto-downloaded.
|
|
valid values are: none, all, %s. (default is "%s").""" % (", ".join(download_types), download_default)
|
|
|
|
def set2dict(keys, value=None):
|
|
"""Convert some keys (iterable) to a dict."""
|
|
return dict((key, value) for (key) in keys)
|
|
|
|
def parse(opt):
|
|
"""This function parses the options to --download and returns a set such as { icu: true }, etc. """
|
|
if not opt:
|
|
opt = download_default
|
|
|
|
theOpts = set(opt.split(','))
|
|
|
|
if 'all' in theOpts:
|
|
# all on
|
|
return set2dict(download_types, True)
|
|
elif 'none' in theOpts:
|
|
# all off
|
|
return set2dict(download_types, False)
|
|
|
|
# OK. Now, process each of the opts.
|
|
theRet = set2dict(download_types, False)
|
|
for anOpt in opt.split(','):
|
|
if not anOpt or anOpt == "":
|
|
# ignore stray commas, etc.
|
|
continue
|
|
elif anOpt is 'all':
|
|
# all on
|
|
theRet = dict((key, True) for (key) in download_types)
|
|
else:
|
|
# turn this one on
|
|
if anOpt in download_types:
|
|
theRet[anOpt] = True
|
|
else:
|
|
# future proof: ignore unknown types
|
|
print 'Warning: ignoring unknown --download= type "%s"' % anOpt
|
|
# all done
|
|
return theRet
|
|
|
|
def candownload(auto_downloads, package):
|
|
if not (package in auto_downloads.keys()):
|
|
raise Exception('Internal error: "%s" is not in the --downloads list. Check nodedownload.py' % package)
|
|
if auto_downloads[package]:
|
|
return True
|
|
else:
|
|
print """Warning: Not downloading package "%s". You could pass "--download=all"
|
|
(Windows: "download-all") to try auto-downloading it.""" % package
|
|
return False
|