'UPL', # https://spdx.org/licenses/UPL-1.0.html

'MIT', # https://spdx.org/licenses/MIT.html

'BSD-3-Clause', # https://spdx.org/licenses/BSD-3-Clause.html

'BSD-2-Clause', # https://spdx.org/licenses/BSD-2-Clause.html

'Apache-2.0', # https://spdx.org/licenses/Apache-2.0.html

'MPL-2.0', # https://spdx.org/licenses/MPL-2.0.html

'LGPL-2.0-or-later', # https://spdx.org/licenses/LGPL-2.0-or-later.html

'LGPL-3.0-or-later', # https://spdx.org/licenses/LGPL-3.0-or-later.html

'PSF-2.0', # https://spdx.org/licenses/PSF-2.0.html

if git_mode:

patch_files = {

patches_dir / file

for file in subprocess.run(

['git', '-C', patches_dir, 'ls-files'],

check=True,

stdout=subprocess.PIPE,

text=True,

).stdout.splitlines()

if Path(file).parent == Path('.')

}

else:

patch_files = set(patches_dir.iterdir())

with open(patches_dir / 'metadata.toml', 'rb') as f:

all_metadata = tomli.load(f)

patches = set()

for package, metadata in all_metadata.items():

try:

if unexpected_keys := set(metadata) - SECTIONS:

assert False, f"Unexpected top-level metadata keys: {unexpected_keys}"

if rules := metadata.get('rules'):

for rule in rules:

if unexpected_keys := set(rule) - RULE_KEYS:

assert False, f"Unexpected rule keys: {unexpected_keys}"

if patch := rule.get('patch'):

patch_path = patches_dir / patch

assert patch_path.is_file(), f"Patch file does not exists: {patch_path}"

if git_mode:

assert patch_path in patch_files, f"Patch file is not tracked by git: {patch_path}"

patches.add(patch_path)

license_id = rule.get('license')

assert license_id, f"'license' not specified for patch {patch}"

license_id = re.sub(r'[()]', ' ', license_id)

for part in re.split(f'AND|OR', license_id):

part = part.strip()

if ' WITH ' in part:

part, exception = re.split(r'\s+WITH\s+', part, maxsplit=1)

assert exception in ALLOWED_WITH_CLAUSES, \

f"License WITH clause {exception} not in allowed list of clauses: {', '.join(ALLOWED_WITH_CLAUSES)}"

assert part in ALLOWED_LICENSES, \

f"License {part} not in allowed list of licenses: {', '.join(ALLOWED_LICENSES)}"

if install_priority := rule.get('install-priority'):

assert isinstance(install_priority, int), "'rules.install_priority' must be an int"

if dist_type := rule.get('dist-type'):

assert dist_type in ('wheel', 'sdist'), "'rules.dist_type' must be on of 'wheel', 'sdist'"

if 'autopatch' in rule:

assert isinstance(rule['autopatch'], bool), "'rules.autopatch' must be a bool"

if version := rule.get('version'):

if crates:

for constraint in version.split(','):

assert CRATE_VERSION_SPECIFIER_RE.fullmatch(constraint.strip()), \

f"Unsupported crate version constraint: {constraint!r}"

else:

# Just try that it doesn't raise

SpecifierSet(version)

if add_sources := metadata.get('add-sources'):

for add_source in add_sources:

if unexpected_keys := set(add_source) - {'version', 'url'}:

assert False, f"Unexpected add_source keys: {unexpected_keys}"

assert add_source.get('version'), "Missing 'add_sources.version' key"

assert add_source.get('url'), "Missing 'add_sources.url' key"

except Exception as e:

raise AssertionError(f"{package}: {e}")

for patch_file in patch_files:

if patch_file.is_file() and patch_file.name.endswith('patch'):

assert patch_file in patches, f"Dangling patch file: {patch_file}"

crates_dir = patches_dir / 'crates'

if crates_dir.is_dir():

parser = argparse.ArgumentParser()

parser.add_argument('--git-mode', action='store_true')

parser.add_argument('patches_dir', type=Path)

args = parser.parse_args()