import re, sys, os, collections

buildroot = sys.argv[1]
no_bootloader = '--no-bootloader' in sys.argv

known_files = '''
%ghost %config(noreplace) /etc/crypttab
%ghost %attr(0444,root,root) /etc/udev/hwdb.bin
/etc/inittab
/usr/lib/systemd/purge-nobody-user
# This directory is owned by openssh-server, but we don't want to introduce
# a dependency. So let's copy the config and co-own the directory.
%dir %attr(0700,root,root) /etc/ssh/sshd_config.d
%ghost %config(noreplace) /etc/vconsole.conf
%ghost %config(noreplace) /etc/X11/xorg.conf.d/00-keyboard.conf
%ghost %attr(0664,root,root) %verify(not group) /run/utmp
%ghost %attr(0664,root,root) %verify(not group) /var/log/wtmp
%ghost %attr(0660,root,root) %verify(not group) /var/log/btmp
%ghost %attr(0664,root,root) %verify(not md5 size mtime group) /var/log/lastlog
%ghost %config(noreplace) /etc/hostname
%ghost %config(noreplace) /etc/localtime
%ghost %config(noreplace) /etc/locale.conf
%ghost %attr(0444,root,root) %config(noreplace) /etc/machine-id
%ghost %config(noreplace) /etc/machine-info
%ghost %attr(0700,root,root) %dir /var/cache/private
%ghost %attr(0700,root,root) %dir /var/lib/private
%ghost %dir /var/lib/private/systemd
%ghost %dir /var/lib/private/systemd/journal-upload
%ghost /var/lib/private/systemd/journal-upload/state
%ghost %dir /var/lib/systemd/timesync
%ghost /var/lib/systemd/timesync/clock
%ghost %dir /var/lib/systemd/backlight
%ghost /var/lib/systemd/catalog/database
%ghost %dir /var/lib/systemd/coredump
%ghost /var/lib/systemd/journal-upload
%ghost %dir /var/lib/systemd/linger
%ghost %attr(0600,root,root) /var/lib/systemd/random-seed
%ghost %dir /var/lib/systemd/rfkill
%ghost %dir %verify(not mode group) /var/log/journal
%ghost %dir /var/log/journal/remote
%ghost %attr(0700,root,root) %dir /var/log/private
'''

known_files = {line.split()[-1]:line for line in known_files.splitlines()
               if line and not line.startswith('#')}

def files(root):
    os.chdir(root)
    todo = collections.deque(['.'])
    while todo:
        n = todo.pop()
        files = os.scandir(n)
        for file in files:
            yield file
            if file.is_dir() and not file.is_symlink():
                todo.append(file)

outputs = {suffix: open(f'.file-list-{suffix}', 'w')
           for suffix in (
                   'libs',
                   'udev',
                   'ukify',
                   'boot',
                   'pam',
                   'rpm-macros',
                   'devel',
                   'container',
                   'networkd',
                   'networkd-defaults',
                   'oomd-defaults',
                   'remote',
                   'resolve',
                   'tests',
                   'standalone-repart',
                   'standalone-tmpfiles',
                   'standalone-sysusers',
                   'standalone-shutdown',
                   'main',
           )}

for file in files(buildroot):
    n = file.path[1:]
    if re.match(r'''/usr/(share|include)$|
                    /usr/share/man(/man.|)$|
                    /usr/share/zsh(/site-functions|)$|
                    /usr/share/dbus-1$|
                    /usr/share/dbus-1/system.d$|
                    /usr/share/dbus-1/(system-|)services$|
                    /usr/share/polkit-1(/actions|/rules.d|)$|
                    /usr/share/pkgconfig$|
                    /usr/share/bash-completion(/completions|)$|
                    /usr(/lib|/lib64|/bin|/sbin|)$|
                    /usr/lib.*/(security|pkgconfig)$|
                    /usr/lib/rpm(/macros.d|)$|
                    /usr/lib/firewalld(/services|)$|
                    /usr/share/(locale|licenses|doc)|             # no $
                    /etc(/pam\.d|/xdg|/X11|/X11/xinit|/X11.*\.d|)$|
                    /etc/(dnf|dnf/protected.d)$|
                    /usr/(src|lib/debug)|                         # no $
                    /run$|
                    /var(/cache|/log|/lib|/run|)$
    ''', n, re.X):
        continue

    if n.endswith('.standalone'):
        if 'repart' in n:
            o = outputs['standalone-repart']
        elif 'tmpfiles' in n:
            o = outputs['standalone-tmpfiles']
        elif 'sysusers' in n:
            o = outputs['standalone-sysusers']
        elif 'shutdown' in n:
            o = outputs['standalone-shutdown']
        else:
            assert False, 'Found .standalone not belonging to known packages'

    elif '/security/pam_' in n or '/man8/pam_' in n:
        o = outputs['pam']
    elif '/rpm/' in n:
        o = outputs['rpm-macros']
    elif '/usr/lib/systemd/tests' in n:
        o = outputs['tests']
    elif 'ukify' in n:
        o = outputs['ukify']
    elif re.search(r'/libsystemd-(shared|core)-.*\.so$', n):
        o = outputs['main']
    elif re.search(r'/libcryptsetup-token-systemd-.*\.so$', n):
        o = outputs['udev']
    elif re.search(r'/lib.*\.pc|/man3/|/usr/include|\.so$', n):
        o = outputs['devel']
    elif re.search(r'''journal-(remote|gateway|upload)|
                       systemd-remote\.conf|
                       /usr/share/systemd/gatewayd|
                       /var/log/journal/remote
    ''', n, re.X):
        o = outputs['remote']

    elif re.search(r'''mymachines|
                       machinectl|
                       systemd-nspawn|
                       systemd-vmspawn|
                       import-pubring.gpg|
                       systemd-(machined|import|pull)|
                       /machine.slice|
                       /machines.target|
                       var-lib-machines.mount|
                       org.freedesktop.(import|machine)1
    ''', n, re.X):
        o = outputs['container']

    # .network.example files go into systemd-networkd, and the matching files
    # without .example go into systemd-networkd-defaults
    elif (re.search(r'''/usr/lib/systemd/network/.*\.network$''', n)
          and os.path.exists(f'./{n}.example')):
        o = outputs['networkd-defaults']

    elif re.search(r'''/usr/lib/systemd/network/.*\.network|
                       networkd|
                       networkctl|
                       org.freedesktop.network1|
                       sysusers\.d/systemd-network.conf|
                       tmpfiles\.d/systemd-network.conf|
                       systemd\.network|
                       systemd\.netdev
    ''', n, re.X):
        o = outputs['networkd']

    elif '.so.' in n:
        o = outputs['libs']

    elif re.search(r'10-oomd-.*defaults.conf|lib/systemd/oomd.conf.d', n, re.X):
        o = outputs['oomd-defaults']

    elif re.search(r'''udev(?!\.pc)|
                       hwdb|
                       bootctl|
                       boot-update|
                       bless-boot|
                       boot-system-token|
                       bsod|
                       kernel-install|
                       installkernel|
                       vconsole|
                       backlight|
                       rfkill|
                       random-seed|
                       modules-load|
                       timesync|
                       crypttab|
                       cryptenroll|
                       cryptsetup|
                       kmod|
                       quota|
                       pstore|
                       sleep|suspend|hibernate|
                       systemd-tmpfiles-setup-dev|
                       network/98-default-mac-none.link|
                       network/99-default.link|
                       growfs|makefs|makeswap|mkswap|
                       fsck|
                       repart|
                       gpt-auto|
                       volatile-root|
                       veritysetup|
                       integritysetup|
                       integritytab|
                       remount-fs|
                       /initrd|
                       systemd-pcr|
                       systemd-measure|
                       /boot$|
                       /kernel/|
                       /kernel$|
                       /modprobe.d|
                       binfmt|
                       sysctl|
                       coredump|
                       homed|home1|
                       oomd|
                       portabled|portable1
    ''', n, re.X):     # coredumpctl, homectl, portablectl are included in the main package because
                       # they can be used to interact with remote daemons. Also, the user could be
                       # confused if those user-facing binaries are not available.
        o = outputs['udev']

    elif re.search(r'''/boot/efi|
                       /usr/lib/systemd/boot|
                       sd-boot|systemd-boot\.|loader.conf
    ''', n, re.X):
        o = outputs['boot']

    elif re.search(r'''resolved|resolve1|
                       systemd-resolve|
                       resolvconf|
                       systemd\.(positive|negative)
    ''', n, re.X):     # resolvectl and nss-resolve are in the main package.
        o = outputs['resolve']

    else:
        o = outputs['main']

    if n in known_files:
        prefix = ' '.join(known_files[n].split()[:-1])
        if prefix:
            prefix += ' '
    elif file.is_dir() and not file.is_symlink():
        prefix = '%dir '
    elif 'README' in n:
        prefix = '%doc '
    elif n.startswith('/etc'):
        prefix = '%config(noreplace) '
    else:
        prefix = ''

    suffix = '*' if '/man/' in n else ''

    print(f'{prefix}{n}{suffix}', file=o)

if [print(f'ERROR: no file names were written to {o.name}')
    for name, o in outputs.items()
    if (o.tell() == 0 and
        not (no_bootloader and name in ('ukify', 'boot')))
    ]:
    sys.exit(1)