90 lines
2.7 KiB
Python
90 lines
2.7 KiB
Python
|
from os.path import join, exists
|
||
|
|
|
||
|
|
files = {
|
||
|
|
'/etc/bash.bashrc': {
|
||
|
|
'source': 'bashrc',
|
||
|
|
'content_type': 'mako',
|
||
|
|
},
|
||
|
|
'/etc/tmux.conf': {
|
||
|
|
'source': 'tmux.conf',
|
||
|
|
'content_type': 'mako',
|
||
|
|
},
|
||
|
|
'/etc/vim/vimrc.local': {
|
||
|
|
'source': 'vimrc',
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for group, attrs in node.metadata.get('groups', {}).items():
|
||
|
|
groups[group] = attrs
|
||
|
|
|
||
|
|
for username, attrs in node.metadata['users'].items():
|
||
|
|
home = attrs.get('home', '/home/{}'.format(username))
|
||
|
|
|
||
|
|
if attrs.get('delete', False):
|
||
|
|
users[username] = {'delete': True}
|
||
|
|
files[home] = {'delete': True}
|
||
|
|
|
||
|
|
else:
|
||
|
|
user = users.setdefault(username, {})
|
||
|
|
|
||
|
|
user['home'] = home
|
||
|
|
user['shell'] = attrs.get('shell', '/bin/bash')
|
||
|
|
|
||
|
|
if 'password' in attrs:
|
||
|
|
user['password'] = attrs['password']
|
||
|
|
else:
|
||
|
|
user['password_hash'] = 'x' if node.use_shadow_passwords else '*'
|
||
|
|
|
||
|
|
if 'groups' in attrs:
|
||
|
|
user['groups'] = attrs['groups']
|
||
|
|
|
||
|
|
directories[home] = {
|
||
|
|
'owner': username,
|
||
|
|
'mode': attrs.get('home-mode', '0700'),
|
||
|
|
}
|
||
|
|
|
||
|
|
if 'ssh_pubkey' in attrs:
|
||
|
|
files[home + '/.ssh/authorized_keys'] = {
|
||
|
|
'content': '\n'.join(sorted(set(attrs['ssh_pubkey']))) + '\n',
|
||
|
|
'owner': username,
|
||
|
|
'mode': '0600',
|
||
|
|
}
|
||
|
|
|
||
|
|
elif not attrs.get('do_not_remove_authorized_keys_from_home', False):
|
||
|
|
files[home + '/.ssh/authorized_keys'] = {'delete': True}
|
||
|
|
|
||
|
|
if exists(join(repo.path, 'data', 'users', 'files', 'tmux', '{}.conf'.format(username))):
|
||
|
|
files[home + '/.tmux.conf'] = {
|
||
|
|
'content_type': 'mako',
|
||
|
|
'source': 'tmux/{}.conf'.format(username),
|
||
|
|
}
|
||
|
|
else:
|
||
|
|
files[home + '/.tmux.conf'] = {
|
||
|
|
'delete': True,
|
||
|
|
}
|
||
|
|
|
||
|
|
if exists(join(repo.path, 'data', 'users', 'files', 'bash', '{}.bashrc'.format(username))):
|
||
|
|
files[home + '/.bashrc'] = {
|
||
|
|
'content_type': 'mako',
|
||
|
|
'source': 'bash/{}.bashrc'.format(username),
|
||
|
|
}
|
||
|
|
else:
|
||
|
|
files[home + '/.bashrc'] = {
|
||
|
|
'delete': True,
|
||
|
|
}
|
||
|
|
|
||
|
|
if attrs.get('enable_linger', False):
|
||
|
|
linger_test = ''
|
||
|
|
linger_command = 'enable'
|
||
|
|
else:
|
||
|
|
linger_test = '!'
|
||
|
|
linger_command = 'disable'
|
||
|
|
|
||
|
|
actions[f'ensure_linger_state_for_user_{username}'] = {
|
||
|
|
'command': f'loginctl {linger_command}-linger {username}',
|
||
|
|
'unless': f'{linger_test} test -f /var/lib/systemd/linger/{username}',
|
||
|
|
'needs': {
|
||
|
|
f'user:{username}',
|
||
|
|
},
|
||
|
|
}
|