import json
from nodes.role_store import load_roles, save_roles, NodeRole


def get_node_roles():
    roles = load_roles()

    return {
        name: role.__dict__
        for name, role in roles.items()
    }


def set_node_role(node_name, role_data):
    roles = load_roles()

    roles[node_name] = NodeRole(**role_data)

    save_roles(roles)

    return {"status": "ok"}