from json import dumps
from bundlewrap.metadata import MetadataJSONEncoder

actions = {
    'modprobe_zfs': {
        'command': 'modprobe zfs',
        'unless': 'lsmod | grep ^zfs',
        'needs': {
            'pkg_apt:zfs-dkms',
        },
        'needed_by': {
            'pkg_apt:zfs-zed',
            'pkg_apt:zfsutils-linux',
            'zfs_dataset:',
            'zfs_pool:',
        },
        'comment': 'If this fails, do a dist-upgrade, reinstall zfs-dkms, reboot',
    },
}

svc_systemd = {
    'zfs-zed': {
        'needs': {
            'pkg_apt:zfs-zed'
        },
    },
}

zfs_datasets = node.metadata.get('zfs/datasets', {})
zfs_pools = {}

# TRIM
for name, attrs in node.metadata.get('zfs/pools', {}).items():
    zfs_pools[name] = attrs
    actions[f'pool_{name}_enable_trim'] = {
       'command': f'zpool set autotrim=on {name}',
       'unless':  f'zpool get autotrim -H -o value {name} | grep -q on',
       'needs':   [
           f'zfs_pool:{name}'
       ]
    }