supermario/bin/deepen
Elliot Nunn f4d0dadebe Create useful .build/.flatten scripts
Also, factor out some common code
2019-06-29 22:56:27 +08:00

86 lines
3.0 KiB
Python
Executable File

#!/usr/bin/env python3
import argparse
from os import path, makedirs, walk, listdir, remove, stat, chmod
from shutil import copy, rmtree
from subprocess import run
from shlex import quote
from folder import folder
def make_executable(path): # thanks to Jonathon Reinhart on SO
mode = stat(path).st_mode
mode |= (mode & 0o444) >> 2 # copy R bits to X
chmod(path, mode)
def relpath(dest, src):
return path.relpath(path.abspath(dest), path.abspath(src))
def path_to_script(script_name, from_worktree):
script_path = path.join(path.dirname(__file__), script_name)
return relpath(script_path, from_worktree)
parser = argparse.ArgumentParser(description='''
Expand a patchset into a git repository (based on git-am)
''')
parser.add_argument('patchset', metavar='PATCHSET', action='store', type=lambda x: folder('patchset', x), help='Patchset (assumed in ../patchset/)')
parser.add_argument('worktree', metavar='WORKTREE', action='store', type=lambda x: folder('worktree', x), help='Destination worktree (assumed in ../worktree/)')
group = parser.add_mutually_exclusive_group()
group.add_argument('--base', metavar='SRC', action='store', default='SuperMarioProj.1994-02-09', type=lambda x: folder('base', x), help='Base source tree (default=SuperMarioProj.1994-02-09, assumed in ../base/)')
args = parser.parse_args()
assert path.exists(args.patchset)
assert not path.exists(path.join(args.patchset, '.git')) # protect against argument swap
horst = ['-c', 'user.name=Horst Beepmanh', '-c', 'user.email=<>']
def git(*gitargs):
run(['git', *horst, *gitargs], cwd=args.worktree, check=True)
try:
rmtree(args.worktree)
except FileNotFoundError:
pass
makedirs(args.worktree)
with open(path.join(args.worktree, '.gitignore'), 'a') as f:
f.write('.*\n*.dmg\nBuild*/\n')
with open(path.join(args.worktree, '.build'), 'w') as f:
print('#!/bin/sh', file=f)
print('exec ' + quote(path_to_script('build', args.worktree)) + ' . "$@"', file=f)
make_executable(f.name)
with open(path.join(args.worktree, '.flatten'), 'w') as f:
print('#!/bin/sh', file=f)
print('exec ' + quote(path_to_script('flatten', args.worktree)) + ' . ' + relpath(args.patchset, args.worktree) + ' "$@"', file=f)
make_executable(f.name)
git('init')
git('add', '-f', '.gitignore', '.build', '.flatten')
git('commit', '-m', 'Useful non-source things')
for walk_base, walk_dirs, walk_files in walk(args.base):
walk_dirs[:] = [x for x in walk_dirs if not x.startswith('.')]
walk_files[:] = [x for x in walk_files if not x.startswith('.')]
other_dir = path.join(args.worktree, path.relpath(walk_base, args.base))
makedirs(other_dir, exist_ok=True)
for this_file in walk_files:
copy(path.join(walk_base, this_file), other_dir)
git('add', '.')
git('commit', '-m', path.basename(args.base))
git('tag', 'patchset-base')
patchfiles = [path.join(args.patchset, x) for x in sorted(listdir(args.patchset)) if path.splitext(x)[1].lower() == '.patch']
git('am', *patchfiles)