From 5fc21421034f933f6d47bebefba4ee28a24329d9 Mon Sep 17 00:00:00 2001 From: Hykilpikonna Date: Sat, 28 May 2022 03:10:09 -0400 Subject: [PATCH] [+] Code --- .gitignore | 2 +- ocpm/OCKextRepos.yml | 14 ++++ ocpm/interaction.py | 92 +++++++++++++++++++++ ocpm/main.py | 192 +++++++++++++++++++++++++++++++++++++++++++ 4 files changed, 299 insertions(+), 1 deletion(-) create mode 100644 ocpm/OCKextRepos.yml create mode 100644 ocpm/interaction.py create mode 100755 ocpm/main.py diff --git a/.gitignore b/.gitignore index 121c907..151745a 100644 --- a/.gitignore +++ b/.gitignore @@ -115,4 +115,4 @@ dmypy.json # Custom .idea -.iml +*.iml diff --git a/ocpm/OCKextRepos.yml b/ocpm/OCKextRepos.yml new file mode 100644 index 0000000..2cf4063 --- /dev/null +++ b/ocpm/OCKextRepos.yml @@ -0,0 +1,14 @@ +Repos: + Lilu: https://github.com/acidanthera/Lilu + AppleALC: https://github.com/acidanthera/AppleALC + IntelBluetoothFirmware: https://github.com/OpenIntelWireless/IntelBluetoothFirmware + BlueToolFixup: https://github.com/acidanthera/BrcmPatchRAM + VirtualSMC: &smc https://github.com/acidanthera/VirtualSMC + SMCSuperIO: *smc + SMCProcessor: *smc + NVMeFix: https://github.com/acidanthera/NVMeFix + WhateverGreen: https://github.com/acidanthera/WhateverGreen + RealtekRTL8111: https://github.com/Mieze/RTL8111_driver_for_OS_X + AirportItlwm: https://github.com/OpenIntelWireless/itlwm + + diff --git a/ocpm/interaction.py b/ocpm/interaction.py new file mode 100644 index 0000000..198b0c4 --- /dev/null +++ b/ocpm/interaction.py @@ -0,0 +1,92 @@ +from hypy_utils import printc + +from main import Kext, Release + + +def ver_diff(src: str, to: str): + """ + Return the first decimal point that two version numbers differs + """ + ssp = src.split('.') + tsp = to.split('.') + + for i in range(len(ssp)): + sv = ssp[i] + tv = tsp[i] + + if sv.isnumeric() and tv.isnumeric(): + sv, tv = int(sv), int(tv) + + if sv != tv: + return i + + return -1 + + +def ver_color(src: str, to: str): + """ + Compare versions and color output + + :param src: Source version + :param to: Updated version + :return: Compared version + """ + tsp = to.split('.') + + try: + i = ver_diff(src, to) + return '.'.join(tsp[:i]) + '.&a' + '.'.join(tsp[i:]) + '&r' + except Exception: + return f'&a{to}&r' + + +def ver_color_prefix(src: str, to: str): + i = ver_diff(src, to) + if i > 2: + return '&7' + return ['&c', '&e', '&a'][i] + + +def len_nocolor(s: str): + return len(s) - s.count('&') * 2 + + +def ljust(s: str, l: int): + return s + ' ' * (l - len_nocolor(s)) + + +def rjust(s: str, l: int): + return ' ' * (l - len_nocolor(s)) + s + + +def tabulate(lst: list[list[str]], headers: list[str]): + """ + Print in table format, with justify and adjusted for colors + """ + lens = [max(max(len_nocolor(it[col]) for it in lst), len_nocolor(headers[col])) for col in range(len(headers))] + justify = [rjust if h.endswith(':') else ljust for h in headers] + headers = [h[:-1] if h.endswith(':') else h for h in headers] + + # Add headers row + lst.insert(0, [f'&f&n{h}&r' for h in headers]) + + # Print list + for it in lst: + row = ' '.join(justify[col](v, lens[col]) for col, v in enumerate(it)) + printc(row) + + +def sizeof_fmt(num: int): + """ + https://stackoverflow.com/a/1094933/7346633 + """ + for unit in ["B", "K", "M", "G", "T", "P", "E", "Z"]: + if abs(num) < 1024.0: + return f"{num:3.1f} {unit}" + num /= 1024.0 + + +def print_updates(updates: list[tuple[Kext, Release]]): + upd_tbl = [[ver_color_prefix(k.version, l.tag) + k.name + '&r', k.version, + ver_color(k.version, l.tag), sizeof_fmt(l.artifact.size)] for k, l in updates] + tabulate(upd_tbl, ['Kext', 'Current', 'Latest', 'Size:']) diff --git a/ocpm/main.py b/ocpm/main.py new file mode 100755 index 0000000..31a10a7 --- /dev/null +++ b/ocpm/main.py @@ -0,0 +1,192 @@ +#!/usr/bin/env python3 +from __future__ import annotations + +import argparse +import os +import plistlib +from dataclasses import dataclass +from datetime import datetime +from pathlib import Path +from packaging import version + +import dateutil.parser +import pandas +import requests +import tqdm as tqdm +import ruamel.yaml +import semver +from hypy_utils import printc +from pandas import DataFrame +from tqdm.contrib.concurrent import thread_map + +from interaction import print_updates, sizeof_fmt + + +@dataclass() +class Kext: + path: Path + + name: str + id: str + version: str + sdk_os: str + min_os: str + + def __init__(self, path: Path): + self.path = path + + # Find plist file + plist = path / 'Contents' / 'Info.plist' + if not plist.is_file(): + print(f'Error loading {path.name}: Cannot find Info.plist') + + # Load plist file + plist = plistlib.loads(plist.read_bytes()) + + self.name = plist['CFBundleName'] + self.id = plist['CFBundleIdentifier'] + self.version = plist['CFBundleVersion'] + self.sdk_os = plist.get('DTSDKName') + self.min_os = plist.get('LSMinimumSystemVersion') + + if self.sdk_os: + self.sdk_os = self.sdk_os.replace('macosx', '') + + +def print_kexts(kexts: list[Kext]): + df = DataFrame(kexts) + df = df.drop(columns=['path', 'id']) + # df['path'] = df['path'].apply(lambda x: x.name.replace('.kext', '')) + print(df.to_string()) + + +@dataclass() +class Artifact: + size: int + url: str + name: str + + @classmethod + def from_github(cls, obj: dict) -> "Artifact": + return cls(obj['size'], obj['browser_download_url'], obj['name']) + + +def find_artifact(raw: dict) -> Artifact: + assets = raw['assets'] + if len(assets) == 1: + return Artifact.from_github(assets[0]) + + # Filter out DEBUG artifacts + assets = [a for a in assets if not a['name'].endswith('DEBUG.zip')] + return Artifact.from_github(assets[0]) + + +@dataclass() +class Release: + tag: str + raw: dict + artifact: Artifact + date: datetime + + @classmethod + def from_github(cls, raw: dict) -> "Release": + tag = raw['tag_name'] + if tag.startswith('v'): + tag = tag[1:] + + date = dateutil.parser.parse(raw['published_at']) + artifact = find_artifact(raw) + + return cls(tag, raw, artifact, date) + + +def get_latest_release(kext: Kext, repos: dict, pre: bool): + # Lowercase keys + repos = {k.lower(): v for k, v in repos['Repos'].items()} + name = kext.name.lower() + + # Find repo + assert name in repos, f'Kext {kext.name} is not found in our repos. (If it\'s open source, feel free to add it in!)' + repo_info = repos[name] + + if isinstance(repo_info, str): + repo = repo_info + else: + repo = repo_info['Repo'] + artifact = repo_info.get('Artifact') + + assert 'github.com/' in repo, f'For {kext.name}: {repo} is not a github repo, skipping...' + repo = repo.split('github.com/')[1] + + # Check latest version + headers = {} + if 'GH_TOKEN' in os.environ: + headers['Authorization'] = f'token {os.environ["GH_TOKEN"]}' + releases = requests.get(f'https://api.github.com/repos/{repo}/releases').json() + if not pre: + releases = [r for r in releases if not r['prerelease']] + latest = releases[0] + + return Release.from_github(latest) + + +def run(): + parser = argparse.ArgumentParser(description='OpenCore Kext Updater by HyDEV') + parser.add_argument('efi_path', help='EFI Directory Path') + parser.add_argument('--pre', action='store_true', help='Use pre-release') + parser.add_argument('-y', action='store_true', help='Say yes') + + printc('\n&fOpenCore Kext Updater v1.0.0 by HyDEV\n') + args = parser.parse_args() + + # Normalize EFI Path + efi = Path(args.efi_path) + if (efi / 'EFI').is_dir(): + efi = efi / 'EFI' + assert (efi / 'OC').is_dir(), 'Open Core directory (OC) not found.' + + # Find kexts + kexts_dir = efi / 'OC' / 'Kexts' + kexts = [str(f) for f in os.listdir(kexts_dir)] + kexts = [kexts_dir / f for f in kexts if f.lower().endswith('.kext')] + + kexts = [Kext(k) for k in kexts] + print(f'šŸ” Found {len(kexts)} kexts in {kexts_dir}') + # print_kexts(kexts) + + # Read Repo + with open(Path(__file__).parent / 'OCKextRepos.yml') as f: + repos = ruamel.yaml.safe_load(f) + + # Get latest repos with multithreading + def get_latest(k: Kext): + try: + return get_latest_release(k, repos, args.pre) + except AssertionError: + return None + + term_len = os.get_terminal_size().columns + bar_len = int(term_len * 0.4) + latests = thread_map(get_latest, kexts, desc='Fetching Updates'.ljust(bar_len), max_workers=32, bar_format='{desc} {rate_fmt} {remaining} [{bar}] {percentage:.0f}%', ascii=' #', unit=' pkg') + + # Compare versions + updates: list[tuple[Kext, Release]] + updates = [(k, l) for k, l in zip(kexts, latests) if l and version.parse(l.tag) > version.parse(k.version)] + + # Print updates + printc(f'\nāœØ Found {len(updates)} Updates:') + print_updates(updates) + + # Download prompt + print() + print(f'Total download size: {sizeof_fmt(sum(l.artifact.size for k, l in updates))}') + proceed = input(f'šŸš€ Ready to fly? [y/N] ') + + if proceed.lower().strip() != 'y': + print() + print('šŸ˜• Huh, okay') + exit(0) + + +if __name__ == '__main__': + run()