Add a script for generating import configurations from the live Ignition repositories.

scripts/generate_ignition_import_config.py

@@ -0,0 +1,171 @@
+#!/usr/bin/env python3
+
+import argparse
+import re
+import sys
+import urllib.request
+
+from dataclasses import dataclass, field
+from typing import List, Union
+
+
+@dataclass
+class Package:
+    name: str
+    source_package: str
+    version: str
+
+class PackagesFile:
+    def __init__(self, file_contents):
+        self.packages = dict()
+        current_package = None
+        current_source = None
+        current_version = None
+        for line in file_contents.splitlines():
+            if line.startswith('Source: '):
+                current_source = line.split(': ')[1]
+            if line.startswith('Version: '):
+                current_version = line.split(': ')[1]
+            if line.startswith('Package: '):
+                if current_package:
+                    # Packages without a Source have source name = package name.
+                    if current_source is None:
+                        current_source = current_package
+                    assert current_version
+                    self.packages[current_package] = Package(current_package, current_source, current_version)
+                    current_source = None
+                    current_version = None
+                current_package = line.split(': ')[1]
+        if current_package not in self.packages:
+            self.packages[current_package] = Package(current_package, current_source, current_version)
+
+
+@dataclass
+class PackageGroup:
+    source_package: str
+    packages: None|List[str] = None
+    version_spec: str|None = None
+    skip_packages: List[str] = field(default_factory=list)
+
+parser = argparse.ArgumentParser(description='Generate import configurations for gazebo packages.')
+parser.add_argument('--os', type=str)
+parser.add_argument('--suite', type=str)
+parser.add_argument('--gazebo-suite', type=str)
+args = parser.parse_args(sys.argv[1:])
+
+PACKAGE_GROUPS = {
+    'citadel': [
+        PackageGroup('gazebo11'),
+        PackageGroup('ignition-citadel'),
+        PackageGroup('ignition-cmake2'),
+        PackageGroup('ignition-common3'),
+        PackageGroup('ignition-fuel-tools4'),
+        PackageGroup('ignition-gazebo3'),
+        PackageGroup('ignition-gui3'),
+        PackageGroup('ignition-launch2'),
+        PackageGroup('ignition-math6'),
+        PackageGroup('ignition-msgs5'),
+        PackageGroup('ignition-physics2'),
+        PackageGroup('ignition-plugin'),
+        PackageGroup('ignition-rendering3'),
+        PackageGroup('ignition-sensors3'),
+        PackageGroup('ignition-tools'),
+        PackageGroup('ignition-transport8'),
+        PackageGroup('ogre-2.1'),
+        PackageGroup('sdformat9'),
+    ],
+    'edifice': [
+        PackageGroup(f'ignition-edifice'),
+        PackageGroup('ignition-cmake2'),
+        PackageGroup('ignition-common4'),
+        PackageGroup('ignition-fuel-tools6'),
+        PackageGroup('ignition-gazebo5'),
+        PackageGroup('ignition-gui5'),
+        PackageGroup('ignition-launch4'),
+        PackageGroup('ignition-math6'),
+        PackageGroup('ignition-msgs7'),
+        PackageGroup('ignition-physics4'),
+        PackageGroup('ignition-rendering5'),
+        PackageGroup('ignition-sensors5'),
+        PackageGroup('ignition-tools'),
+        PackageGroup('ignition-transport10'),
+        PackageGroup('ignition-utils1', skip_packages=['libignition-utils-dev']),
+        PackageGroup('ogre-2.2'),
+        PackageGroup('sdformat11'),
+    ],
+    'fortress': [
+        PackageGroup('ignition-fortress'),
+        PackageGroup('ignition-cmake2'),
+        PackageGroup('ignition-common4'),
+        PackageGroup('ignition-fuel-tools7'),
+        PackageGroup('ignition-gazebo6'),
+        PackageGroup('ignition-gui6'),
+        PackageGroup('ignition-launch5'),
+        PackageGroup('ignition-math6', skip_packages=['libgz-math6-eigen-dev']),
+        PackageGroup('ignition-msgs8'),
+        PackageGroup('ignition-physics5'),
+        PackageGroup('ignition-plugin'),
+        PackageGroup('ignition-rendering6'),
+        PackageGroup('ignition-sensors6'),
+        PackageGroup('ignition-tools'),
+        PackageGroup('ignition-transport11', skip_packages=['libgz-transport11-cli']),
+        PackageGroup('ignition-utils1', skip_packages=['libignition-utils-dev', 'gz-utils1']),
+        PackageGroup('ogre-2.2'),
+        PackageGroup('sdformat12'),
+    ]
+}
+
+package_groups = PACKAGE_GROUPS[args.ignition_suite]
+
+target_repo = f'http://packages.osrfoundation.org/gazebo/{args.os}-stable'
+
+resp = urllib.request.urlopen(f'{target_repo}/dists/{args.suite}/main/binary-amd64/Packages')
+packages_file = PackagesFile(resp.read().decode())
+
+version_spec_re = re.compile('(?P<version>[^-]+)-(?P<inc>\d+)(?P<rest>.*)')
+version_errors = False
+for group in package_groups:
+    expected_group_version = None
+    group.packages = [pkg.name for pkg in packages_file.packages.values() if pkg.source_package == group.source_package and pkg.name not in group.skip_packages]
+    for p in group.packages:
+        if not expected_group_version:
+            expected_group_version = packages_file.packages[p].version
+        elif packages_file.packages[p].version != expected_group_version:
+            print(f'{p} {packages_file.packages[p].version} in {group.packages[0]} does not match expected version {expected_group_version}', file=sys.stderr)
+            version_errors = True
+    if expected_group_version:
+        m = version_spec_re.match(expected_group_version)
+        version = m.group('version')
+        inc = m.group('inc')
+        group.version_spec = f'{version}-{inc}*'
+    if group.source_package not in group.packages:
+        group.packages.insert(0, group.source_package)
+
+if version_errors:
+    exit(2)
+
+gz_classic = ''
+if args.ignition_suite == 'citadel':
+    gz_classic = '_gazebo11'
+
+print(f'name: ignition_{args.ignition_suite}{gz_classic}_{args.os}_{args.suite}')
+print(f'method: {target_repo}')
+print(f'suites: [{args.suite}]')
+print('component: main')
+print('architectures: [amd64, armhf, arm64, source]')
+print('filter_formula: "\\')
+for pgroup in package_groups:
+    if pgroup.version_spec is None:
+        print(f'WARNING: Unable to generate package info for {pgroup.source_package}', file=sys.stderr)
+printable_groups = [pgroup for pgroup in package_groups if pgroup.version_spec]
+for idx in range(0, len(printable_groups)):
+    pgroup = printable_groups[idx]
+    if not pgroup.version_spec:
+        continue
+    print('(', end='')
+    print(' |\\\n '.join(f'Package (= {name})' for name in pgroup.packages), end=' \\\n')
+    print(f'), $Version (% {pgroup.version_spec})', end='')
+    if idx != len(printable_groups) - 1:
+        print(' |\\')
+    else:
+        print(' \\\n"')