ryuko-ng/robocop_ng/helpers/roles.py

import json
import os.path
import os

from robocop_ng.helpers.data_loader import read_json


def get_persistent_roles_path(bot):
    return os.path.join(bot.state_dir, "data/persistent_roles.json")


def get_persistent_roles(bot) -> dict[str, list[str]]:
    return read_json(bot, get_persistent_roles_path(bot))


def set_persistent_roles(bot, contents: dict[str, list[str]]):
    with open(get_persistent_roles_path(bot), "w") as f:
        json.dump(contents, f)


def add_user_roles(bot, uid: int, roles: list[int]):
    uid = str(uid)
    roles = [str(x) for x in roles]

    persistent_roles = get_persistent_roles(bot)
    persistent_roles[uid] = roles
    set_persistent_roles(bot, persistent_roles)


def get_user_roles(bot, uid: int) -> list[str]:
    uid = str(uid)
    persistent_roles = get_persistent_roles(bot)
    return persistent_roles[uid] if uid in persistent_roles else []