from pyinfra import host
from pyinfra.operations import server
from pyinfra.facts.server import LinuxDistribution


# raspbian took 1000 for 'pi' group, but drewp is rarely used on pi
# setups so hopefully it won't matter much that drew group has a
# different id.
drewp_uid, drewp_gid = host.data.drewp_uid, host.data.drewp_gid
drewp_groups = [
    'lp', 'adm', 'dialout', 'cdrom', 'sudo', 'audio', 'video', 'plugdev',
    'games', 'users', 'netdev', 'i2c', 'input', 'spi', 'gpio', 'fuse',
    'render', 'mongodb', 'lpadmin'
]

for group in [
        'fuse',
        'spi',
        'gpio',
        'i2c',
        'input',
        'netdev',
        'render',
        'lpadmin',
]:
    server.group(group=group, system=True)

svcIds = 1050
for svc in [
    # only append to this list:
    "photoprism",
    "mongodb",
]:
    server.group(group=svc, gid=svcIds)
    server.user(user=svc, uid=svcIds, group=svc)
    svcIds += 1

# the following gets scrambled on new rpi.  Run "useradd -u 1501 drewp" as workaround.
server.group(group='drewp', gid=drewp_gid)
# this won't change existing drewp uid; I've been doing that myself.
server.user(user='drewp', uid=drewp_uid, group='drewp', groups=drewp_groups)

if 'pi' not in host.groups:
    server.group(group='adm', gid=4)
    server.group(group='cdrom', gid=24)
    server.group(group='dialout', gid=20)
    server.group(group='dip', gid=30)
    server.group(group='lp', gid=7)
    # prime has something on 109
    server.group(group='lpadmin', gid=200)
    server.group(group='plugdev', gid=46)


    server.user(user='drewp',
                uid=drewp_uid,
                group='drewp',
                groups=drewp_groups)

    for name, uid, gid in [
            ('ari', 3019, 3019),
            ('talia', 1003, 1003),
            ]:
        server.group(group=name, gid=gid)
        server.user(user=name,
                uid=uid,
                group=name,
                groups=['audio', 'dialout', 'docker', 'lp', 'lpadmin', 'sudo', 'video'])

    server.user(user='dmcc', uid=1013)

    server.group(group='elastic', gid=3018)
    server.user(user='elastic', uid=3018, group='elastic')

    server.group(group='kelsi', gid=1008)
    server.user(user='kelsi', uid=1008, group='elastic')

    server.group(group='drewnote', gid=1009)
    server.user(user='drewnote', uid=1009)

    server.group(group='prometheus', gid=1010)
    server.user(user='prometheus', uid=1010)