mirror of
https://github.com/ddnet/ddnet.git
synced 2024-11-14 03:58:18 +00:00
117 lines
4.3 KiB
Python
117 lines
4.3 KiB
Python
from collections import namedtuple
|
|
import argparse
|
|
import os
|
|
import shlex
|
|
import subprocess
|
|
import tempfile
|
|
import time
|
|
|
|
ConfigDmgtools = namedtuple('Config', 'dmg hfsplus newfs_hfs verbose')
|
|
ConfigHdiutil = namedtuple('Config', 'hdiutil verbose')
|
|
|
|
def chunks(l, n):
|
|
"""
|
|
Yield successive n-sized chunks from l.
|
|
|
|
From https://stackoverflow.com/a/312464.
|
|
"""
|
|
for i in range(0, len(l), n):
|
|
yield l[i:i + n]
|
|
|
|
class Dmg:
|
|
def __init__(self, config):
|
|
self.config = config
|
|
|
|
def _check_call(self, process_args, *args, **kwargs):
|
|
if self.config.verbose >= 1:
|
|
print("EXECUTING {}".format(" ".join(shlex.quote(x) for x in process_args)))
|
|
if not (self.config.verbose >= 2 and "stdout" not in kwargs):
|
|
kwargs["stdout"] = open(os.devnull, 'wb')
|
|
subprocess.check_call(process_args, *args, **kwargs)
|
|
|
|
class Dmgtools(Dmg):
|
|
def _mkfs_hfs(self, *args):
|
|
self._check_call((self.config.newfs_hfs,) + args)
|
|
def _hfs(self, *args):
|
|
self._check_call((self.config.hfsplus,) + args)
|
|
def _dmg(self, *args):
|
|
self._check_call((self.config.dmg,) + args)
|
|
|
|
def _create_hfs(self, hfs_fd, hfs, volume_name, size):
|
|
if self.config.verbose >= 1:
|
|
print("TRUNCATING {} to {} bytes".format(hfs, size))
|
|
with os.fdopen(hfs_fd, 'wb') as f:
|
|
f.truncate(size)
|
|
self._mkfs_hfs('-v', volume_name, hfs)
|
|
|
|
def _symlink(self, hfs, target, link_name):
|
|
self._hfs(hfs, 'symlink', link_name, target)
|
|
|
|
def _add(self, hfs, directory):
|
|
self._hfs(hfs, 'addall', directory)
|
|
|
|
def _finish(self, hfs, dmg):
|
|
self._dmg('build', hfs, dmg)
|
|
|
|
def create(self, dmg, volume_name, directory, symlinks):
|
|
input_size = sum(os.stat(os.path.join(path, f)).st_size for path, dirs, files in os.walk(directory) for f in files)
|
|
output_size = max(input_size * 2, 1024**2)
|
|
hfs_fd, hfs = tempfile.mkstemp(prefix=dmg + '.', suffix='.hfs')
|
|
self._create_hfs(hfs_fd, hfs, volume_name, output_size)
|
|
self._add(hfs, directory)
|
|
for target, link_name in symlinks:
|
|
self._symlink(hfs, target, link_name)
|
|
self._finish(hfs, dmg)
|
|
if self.config.verbose >= 1:
|
|
print("REMOVING {}".format(hfs))
|
|
os.remove(hfs)
|
|
|
|
class Hdiutil(Dmg):
|
|
def _hdiutil(self, *args):
|
|
self._check_call((self.config.hdiutil,) + args)
|
|
|
|
def create(self, dmg, volume_name, directory, symlinks):
|
|
if symlinks:
|
|
raise NotImplementedError("symlinks are not yet implemented")
|
|
for i in range(5):
|
|
try:
|
|
self._hdiutil('create', '-volname', volume_name, '-srcdir', directory, dmg)
|
|
except subprocess.CalledProcessError as e:
|
|
if i == 4:
|
|
raise e
|
|
print("Retrying hdiutil create")
|
|
time.sleep(5)
|
|
else:
|
|
break
|
|
|
|
def main():
|
|
p = argparse.ArgumentParser(description="Manipulate dmg archives")
|
|
|
|
subcommands = p.add_subparsers(help="Subcommand", dest='command', metavar="COMMAND")
|
|
subcommands.required = True
|
|
|
|
create = subcommands.add_parser("create", help="Create a dmg archive from files or directories")
|
|
create.add_argument('-v', '--verbose', action='count', help="Verbose output")
|
|
createx = create.add_mutually_exclusive_group(required=True)
|
|
createx.add_argument('--dmgtools', nargs=3, help="Paths to the dmg and hfsplus executable (https://github.com/mozilla/libdmg-hfsplus) and the newfs_hfs executable (http://pkgs.fedoraproject.org/repo/pkgs/hfsplus-tools/diskdev_cmds-540.1.linux3.tar.gz/0435afc389b919027b69616ad1b05709/diskdev_cmds-540.1.linux3.tar.gz)")
|
|
createx.add_argument('--hdiutil', help="Path to the hdiutil (only exists for macOS at time of writing)")
|
|
create.add_argument('output', metavar="OUTPUT", help="Filename of the output dmg archive")
|
|
create.add_argument('volume_name', metavar="VOLUME_NAME", help="Name of the dmg archive")
|
|
create.add_argument('directory', metavar="DIR", help="Directory to create the archive from")
|
|
create.add_argument('--symlink', metavar="SYMLINK", nargs=2, action="append", help="Symlink the first argument under the second name")
|
|
args = p.parse_args()
|
|
|
|
verbose = args.verbose or 0
|
|
symlinks = args.symlink or []
|
|
if args.dmgtools:
|
|
dmg, hfsplus, newfs_hfs = args.dmgtools
|
|
dmg = Dmgtools(ConfigDmgtools(dmg=dmg, hfsplus=hfsplus, newfs_hfs=newfs_hfs, verbose=verbose))
|
|
elif args.hdiutil:
|
|
dmg = Hdiutil(ConfigHdiutil(hdiutil=args.hdiutil, verbose=verbose))
|
|
else:
|
|
raise RuntimeError("unreachable")
|
|
dmg.create(volume_name=args.volume_name, directory=args.directory, dmg=args.output, symlinks=symlinks)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|