mirror of
https://github.com/flatcar/scripts.git
synced 2026-05-04 11:51:14 +02:00
e263b4bde5
The existing tmpfile logic took care of folders that the ebuild keepdir directive wanted to exist on the OS. However, files and symlinks were not created, causing them to be missing if we didn't explicitly modify the ebuild files in coreos-overlay to find a solution with tmpfiles or patching the paths to /usr. Add logic to create missing files and symlinks through tmpfile directives and preserve any directory, not only the ones with the keepdir ebuild directive. Also remove any state from the rootfs to make sure that we don't rely on it when testing our images before the release. To create the files the final /etc folder is moved to /usr/share/flatcar/etc and in the future this can be used for a better logic that could take care of updating files the user didn't modify, deleting those that aren't needed anymore, and even reconciling changed files through a 3-way merge, instead of using simple tmpfile logic.
107 lines
3.3 KiB
Python
Executable File
107 lines
3.3 KiB
Python
Executable File
#!/usr/bin/python3
|
|
'''Scan an existing directory tree and record installed directories.
|
|
|
|
During build a number of directories under /var are created in the state
|
|
partition. We want to make sure that those are always there so create a record
|
|
of them using systemd's tempfiles config format so they are recreated during
|
|
boot if they go missing for any reason.
|
|
'''
|
|
|
|
import optparse
|
|
import os
|
|
import stat
|
|
import sys
|
|
import pwd
|
|
import grp
|
|
|
|
def main():
|
|
keep = set()
|
|
parser = optparse.OptionParser(description=__doc__)
|
|
parser.add_option('--root', help='Remove root prefix from output')
|
|
parser.add_option('--output', help='Write output to the given file')
|
|
parser.add_option('--files', default='', help='Also works on files and symlinks and uses the given path as source for copying files')
|
|
parser.add_option('--ignore', action='append', default=[],
|
|
help='Ignore one or more paths (use multiple times)')
|
|
opts, args = parser.parse_args()
|
|
|
|
if opts.root:
|
|
opts.root = os.path.abspath(opts.root)
|
|
|
|
for path in args:
|
|
path = os.path.abspath(path)
|
|
if opts.root:
|
|
assert path.startswith(opts.root)
|
|
|
|
for dirpath, dirnames, filenames in os.walk(path):
|
|
keep.add(dirpath)
|
|
if opts.files:
|
|
for f in filenames:
|
|
if not f.startswith('.keep'):
|
|
keep.add(os.path.join(dirpath, f))
|
|
|
|
# Add all parent directories too
|
|
for path in frozenset(keep):
|
|
split = []
|
|
for pathbit in path.split('/'):
|
|
split.append(pathbit)
|
|
joined = '/'.join(split)
|
|
if not joined:
|
|
continue
|
|
if opts.root and not joined.startswith(opts.root):
|
|
continue
|
|
if opts.root == joined:
|
|
continue
|
|
keep.add(joined)
|
|
|
|
config = []
|
|
for path in sorted(keep):
|
|
if opts.root:
|
|
assert path.startswith(opts.root)
|
|
stripped = path[len(opts.root):]
|
|
assert len(stripped) > 1
|
|
else:
|
|
stripped = path
|
|
|
|
if stripped in opts.ignore:
|
|
continue
|
|
|
|
target = '-'
|
|
info = os.stat(path, follow_symlinks=False)
|
|
if not opts.files:
|
|
assert stat.S_ISDIR(info.st_mode)
|
|
if stat.S_ISDIR(info.st_mode):
|
|
entry = 'd'
|
|
elif stat.S_ISLNK(info.st_mode):
|
|
entry = 'L'
|
|
target = os.readlink(path)
|
|
elif stat.S_ISREG(info.st_mode):
|
|
entry = 'C'
|
|
assert opts.files
|
|
target = os.path.join(opts.files, stripped.lstrip('/'))
|
|
assert target.startswith(opts.files)
|
|
else:
|
|
continue
|
|
mode = stat.S_IMODE(info.st_mode)
|
|
|
|
try:
|
|
owner = pwd.getpwuid(info.st_uid).pw_name
|
|
except KeyError:
|
|
owner = str(info.st_uid)
|
|
try:
|
|
group = grp.getgrgid(info.st_gid).gr_name
|
|
except KeyError:
|
|
group = str(info.st_gid)
|
|
|
|
config.append('%s %-22s %04o %-10s %-10s - %s'
|
|
% (entry, stripped, mode, owner, group, target))
|
|
|
|
if opts.output:
|
|
fd = open(opts.output, 'w')
|
|
fd.write('\n'.join(config)+'\n')
|
|
fd.close()
|
|
else:
|
|
print('\n'.join(config))
|
|
|
|
if __name__ == '__main__':
|
|
main()
|