empw/hsync
2018-05-10 23:38:17 +08:00

223 lines
6.0 KiB
Python
Executable File

#!/usr/bin/env python3
from subprocess import run, PIPE, DEVNULL
from sys import argv
from time import sleep
import datetime
from os import path
import tempfile
# oldrun = run
# def run(*args, **kwargs):
# print('CMD', *args)
# return oldrun(*args, **kwargs)
# I am now modifying this to accept *no* arguments,
# instead operating on the CWD.
# A couple of useful HFSUTILS wrappers
def h_topdir():
rc = 0
while rc == 0:
rc = run(['hcd','::'], stderr=DEVNULL).returncode
def h_reccd(path_comps):
h_topdir()
for p in path_comps:
run(['hcd', p], check=True)
# All relative to the CWD, of course:
srcimg = 'SourceForEmulator.dmg'
label = 'Src'
# Read and sanitise the arguments. Outputs: "dirs" and "cmds" (both lists, may be empty)
# Dirs are in /full/path/without/trailing/slash form, thanks to os.path.abspath
if not path.exists(srcimg):
run(['dd', 'if=/dev/zero', 'of='+srcimg, 'bs=1048576', 'count='+str(256)], stdout=DEVNULL, stderr=DEVNULL, check=True)
run(['hformat', '-l', label, srcimg], stdout=DEVNULL, check=True)
run(['humount'], check=True)
# Now run rsync and see what we can get out of it! (Should at least reconsider rsync policy!)
# Rsync should update any changed file
# Potential problem: need to remove ._ thingummyjigs!
rsync_opts = [
'--dry-run', # Ask rsync what it *would* do if macOS
'--itemize-changes', # hadn't dropped R/W HFS support.
'--exclude', '.*', # Down with dotfiles.
'--exclude', srcimg,
'--exclude', 'Desktop DB',
'--exclude', 'Desktop DF',
'--exclude', 'Desktop Folder',
'--exclude', 'Trash',
'--exclude', 'TheVolumeSettingsFolder',
'--exclude', 'TheFindByContentFolder',
'--recursive',
'-tX', # consider times and xattrs
'--delete', # may delete things on the vMac drive
]
try:
cmdresult = run(['hdiutil', 'attach', '-nobrowse', srcimg], stdout=PIPE, check=True).stdout.decode('ascii')
mountpoint = next(x for x in cmdresult.split() if x.startswith('/') and not x.startswith('/dev'))
mountpoint = mountpoint.rstrip('/') # try to deal in clean paths
rsync_list = [] # (local_base_path, relative_path_components, rsync_code) tuple
# rsync /src/dir/without/trailing/slash/DIRNAME /dest/dir/with/trailing/slash/
# results in /dest/dir/with/trailing/slash/DIRNAME
rsync = run(['rsync', *rsync_opts, './', mountpoint+'/'], stdout=PIPE, check=True)
new_rsync_lines = rsync.stdout.decode('utf8').splitlines()
for new_rsync_line in new_rsync_lines:
spcidx = new_rsync_line.index(' ')
code_part = new_rsync_line[:spcidx]
path_part = new_rsync_line[spcidx+1:].lstrip()
if '._' in path_part:
raise ValueError('AppleDouble-style name detected: %s' % path_part)
rsync_list.append((code_part, path_part))
finally:
run(['hdiutil', 'detach', mountpoint], check=True, stdout=DEVNULL)
# for rsync_code, path in rsync_list:
# # Get rid of the silly AppleDouble prefix
# a, b = path.splitext(rsync_path)
# if b.startswith('._'):
# b = b[2:]
# path = path.join(a, b)
# try:
run(['hmount', srcimg], stdout=DEVNULL, check=True)
try:
for rsync_code, rsync_path in rsync_list:
# print([local_dir, rsync_code, rsync_path])
# Suggest: replace with code to make fuck/marry/kill plan for each file
# first, parse the rsync commands!
if rsync_code.startswith('*'):
if rsync_code == '*deleting':
print('-', rsync_path)
# similar to the folder creation code below
if rsync_path.endswith('/'):
cmdname = 'hrmdir'
else:
cmdname = 'hdel'
a, b = path.split(rsync_path.rstrip('/'))
hfs_dir_comps = a.split('/') if a else []
hfs_deleteme = b
h_reccd(hfs_dir_comps)
run([cmdname, hfs_deleteme], check=True)
else:
raise ValueError('Unknown extended rsync code: %s' % rsync_code)
elif rsync_code.startswith('.'):
pass
else:
utype, ftype, cdiff, sdiff, tdiff, pdiff, odiff, gdiff, *_ = rsync_code
if utype == '>':
if ftype == 'f':
print('+', rsync_path)
hfs_dir = ':' + path.join(path.dirname(rsync_path), '').replace(*'/:')
with tempfile.NamedTemporaryFile(prefix='/tmp/', suffix='.bin') as f:
macbin_tmp = f.name
h_topdir()
run(['macbinary', 'encode', '--overwrite', '-o', macbin_tmp, rsync_path], check=True)
run(['hcopy', '-m', macbin_tmp, hfs_dir], check=True)
else:
raise ValueError()
elif utype == 'c':
if ftype == 'd':
a, b = path.split(rsync_path.rstrip('/'))
hfs_dir_comps = a.split('/') if a else []
hfs_mkdir_name = b
h_reccd(hfs_dir_comps)
run(['hmkdir', hfs_mkdir_name], check=True)
else:
raise ValueError()
elif utype == '.':
pass
else:
raise ValueError('Unknown rsync update code: %s' % rsync_code)
finally:
run(['humount'], check=True)
# YXcstpoguax path/to/file
# |||||||||||
# `----------- the type of update being done::
# |||||||||| <: file is being transferred to the remote host (sent).
# |||||||||| >: file is being transferred to the local host (received).
# |||||||||| c: local change/creation for the item, such as:
# |||||||||| - the creation of a directory
# |||||||||| - the changing of a symlink,
# |||||||||| - etc.
# |||||||||| h: the item is a hard link to another item (requires --hard-links).
# |||||||||| .: the item is not being updated (though it might have attributes that are being modified).
# |||||||||| *: means that the rest of the itemized-output area contains a message (e.g. "deleting").
# ||||||||||
# `---------- the file type:
# ||||||||| f for a file,
# ||||||||| d for a directory,
# ||||||||| L for a symlink,
# ||||||||| D for a device,
# ||||||||| S for a special file (e.g. named sockets and fifos).
# |||||||||
# `--------- c: different checksum (for regular files)
# |||||||| changed value (for symlink, device, and special file)
# `-------- s: Size is different
# `------- t: Modification time is different
# `------ p: Permission are different
# `----- o: Owner is different
# `---- g: Group is different
# `--- u: The u slot is reserved for future use.
# `-- a: The ACL information changed