pack.py

#!/usr/bin/env python3
##
#   pack.py
#
#   Copyright (C) David McNaughton 2023-present
#
#   Pack a host directory into a CP/M disk image (.dsk or .hdd).
##

import sys
import os
import logging
import argparse

from fdc import __version__
from fdc.disk.geometry import geometry_for_ext
from fdc.disk.filesystem.directory import build_directory, format_dir
from fdc.disk.filesystem.archive import pack
from fdc.disk.fif.image import ImageDisk


def _pause():
    if sys.stdin.isatty():
        try:
            input("\nPress Enter to exit...")
        except (EOFError, KeyboardInterrupt):
            pass


def main():
    try:
        parser = argparse.ArgumentParser(
            description=f"Pack a host directory into a CP/M disk image v{__version__}",
        )
        parser.add_argument('--version', action='version', version=f'%(prog)s {__version__}')
        parser.add_argument('image', metavar='IMAGE',
                            help='Disk image to create (.dsk or .hdd)')
        args = parser.parse_args()

        name = args.image
        _, ext = os.path.splitext(name)

        geom = geometry_for_ext(ext)
        root = os.path.splitext(name)[0] + '.unpacked'

        boot, dirdata = build_directory(root, geom)

        disk = ImageDisk.format_new(name, geom)
        directory = pack(disk, boot, dirdata, root)

        for line in format_dir(directory, geom.blksize):
            print(line)
    except KeyboardInterrupt:
        pass
    except SystemExit:
        _pause()
        raise
    except Exception as e:
        print(f"Error: {e}", file=sys.stderr)
        _pause()
        sys.exit(1)
    else:
        _pause()


if __name__ == "__main__":
    logging.basicConfig(level=logging.INFO)
    main()