Merge pull request #29 from grahampugh/alternatives

Add --pkg option
This commit is contained in:
Graham Pugh 2020-11-18 23:27:12 +01:00 committed by GitHub
commit 2b720a4ad8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -560,25 +560,34 @@ def download_and_parse_sucatalog(sucatalog, workdir, ignore_cache=False):
exit(-1)
def find_mac_os_installers(catalog):
def find_mac_os_installers(catalog, installassistant_pkg_only=False):
"""Return a list of product identifiers for what appear to be macOS
installers"""
mac_os_installer_products = []
if "Products" in catalog:
for product_key in catalog["Products"].keys():
product = catalog["Products"][product_key]
try:
if product["ExtendedMetaInfo"]["InstallAssistantPackageIdentifiers"]:
mac_os_installer_products.append(product_key)
except KeyError:
continue
# account for args.pkg
if installassistant_pkg_only:
try:
if product['ExtendedMetaInfo']['InstallAssistantPackageIdentifiers']:
if product['ExtendedMetaInfo']['InstallAssistantPackageIdentifiers']['SharedSupport'] == 'com.apple.pkg.InstallAssistant.macOSBigSur':
mac_os_installer_products.append(product_key)
except KeyError:
continue
else:
try:
if product["ExtendedMetaInfo"]["InstallAssistantPackageIdentifiers"]:
mac_os_installer_products.append(product_key)
except KeyError:
continue
return mac_os_installer_products
def os_installer_product_info(catalog, workdir, ignore_cache=False):
def os_installer_product_info(catalog, workdir, installassistant_pkg_only, ignore_cache=False):
"""Returns a dict of info about products that look like macOS installers"""
product_info = {}
installer_products = find_mac_os_installers(catalog)
installer_products = find_mac_os_installers(catalog, installassistant_pkg_only)
for product_key in installer_products:
product_info[product_key] = {}
# get the localized title (e.g. "macOS Catalina Beta") and version
@ -791,6 +800,11 @@ def main():
help="Selects the latest valid build ID matching "
"the selected OS version (e.g. 10.14).",
)
parser.add_argument(
"--pkg",
action="store_true",
help="Search only for InstallAssistant packages (macOS Big Sur only)",
)
args = parser.parse_args()
# show this Mac's info
@ -855,7 +869,7 @@ def main():
su_catalog_url, args.workdir, ignore_cache=args.ignore_cache
)
product_info = os_installer_product_info(
catalog, args.workdir, ignore_cache=args.ignore_cache
catalog, args.workdir, args.pkg, ignore_cache=args.ignore_cache
)
if not product_info:
print("No macOS installer products found in the sucatalog.", file=sys.stderr)
@ -1160,58 +1174,82 @@ def main():
print("Exiting.")
exit(0)
# download all the packages for the selected product
replicate_product(catalog, product_id, args.workdir, ignore_cache=args.ignore_cache)
# shortened workflow if we just want a macOS Big Sur+ package
# taken from @scriptingosx's Fetch-Installer-Pkg project
# (https://github.com/scriptingosx/fetch-installer-pkg)
if args.pkg:
product = catalog['Products'][product_id]
# determine the InstallAssistant pkg url
for package in product['Packages']:
package_url = package['URL']
if package_url.endswith('InstallAssistant.pkg'):
break
# print("Package URL is %s" % package_url)
download_pkg = replicate_url(package_url, args.workdir, True, ignore_cache=args.ignore_cache)
pkg_name = ('InstallAssistant-%s-%s.pkg' % (product_info[product_id]['version'],
product_info[product_id]['BUILD']))
# hard link the downloaded file to cwd
local_pkg = os.path.join(args.workdir, pkg_name)
os.link(download_pkg, local_pkg)
print("Package downloaded to: %s" % local_pkg)
else:
# download all the packages for the selected product
replicate_product(catalog, product_id, args.workdir, ignore_cache=args.ignore_cache)
# generate a name for the sparseimage
volname = "Install_macOS_%s-%s" % (
product_info[product_id]["version"],
product_info[product_id]["BUILD"],
)
sparse_diskimage_path = os.path.join(args.workdir, volname + ".sparseimage")
if os.path.exists(sparse_diskimage_path):
os.unlink(sparse_diskimage_path)
# make an empty sparseimage and mount it
print("Making empty sparseimage...")
sparse_diskimage_path = make_sparse_image(volname, sparse_diskimage_path)
mountpoint = mountdmg(sparse_diskimage_path)
if mountpoint:
# install the product to the mounted sparseimage volume
success = install_product(
product_info[product_id]["DistributionPath"], mountpoint
# generate a name for the sparseimage
volname = "Install_macOS_%s-%s" % (
product_info[product_id]["version"],
product_info[product_id]["BUILD"],
)
if not success:
print("Product installation failed.", file=sys.stderr)
unmountdmg(mountpoint)
exit(-1)
# add the seeding program xattr to the app if applicable
seeding_program = get_seeding_program(su_catalog_url)
if seeding_program:
installer_app = find_installer_app(mountpoint)
if installer_app:
print(
"Adding seeding program %s extended attribute to app"
% seeding_program
)
xattr.setxattr(installer_app, "SeedProgram", seeding_program.encode())
print("Product downloaded and installed to %s" % sparse_diskimage_path)
if args.raw:
unmountdmg(mountpoint)
else:
# if --raw option not given, create a r/o compressed diskimage
# containing the Install macOS app
compressed_diskimagepath = os.path.join(args.workdir, volname + ".dmg")
if os.path.exists(compressed_diskimagepath):
os.unlink(compressed_diskimagepath)
app_path = find_installer_app(mountpoint)
if app_path:
make_compressed_dmg(app_path, compressed_diskimagepath, volname)
# unmount sparseimage
unmountdmg(mountpoint)
# delete sparseimage since we don't need it any longer
sparse_diskimage_path = os.path.join(args.workdir, volname + ".sparseimage")
if os.path.exists(sparse_diskimage_path):
os.unlink(sparse_diskimage_path)
# make an empty sparseimage and mount it
print("Making empty sparseimage...")
sparse_diskimage_path = make_sparse_image(volname, sparse_diskimage_path)
mountpoint = mountdmg(sparse_diskimage_path)
if mountpoint:
# install the product to the mounted sparseimage volume
success = install_product(
product_info[product_id]["DistributionPath"], mountpoint
)
if not success:
print("Product installation failed.", file=sys.stderr)
unmountdmg(mountpoint)
exit(-1)
# add the seeding program xattr to the app if applicable
seeding_program = get_seeding_program(su_catalog_url)
if seeding_program:
installer_app = find_installer_app(mountpoint)
if installer_app:
print(
"Adding seeding program %s extended attribute to app"
% seeding_program
)
xattr.setxattr(installer_app, "SeedProgram", seeding_program.encode())
print("Product downloaded and installed to %s" % sparse_diskimage_path)
if args.raw:
unmountdmg(mountpoint)
else:
# if --raw option not given, create a r/o compressed diskimage
# containing the Install macOS app
compressed_diskimagepath = os.path.join(args.workdir, volname + ".dmg")
if os.path.exists(compressed_diskimagepath):
os.unlink(compressed_diskimagepath)
app_path = find_installer_app(mountpoint)
if app_path:
make_compressed_dmg(app_path, compressed_diskimagepath, volname)
# unmount sparseimage
unmountdmg(mountpoint)
# delete sparseimage since we don't need it any longer
os.unlink(sparse_diskimage_path)
if __name__ == "__main__":
main()