from app.db import crud, GetDB
from app.models.node import NodeStatus
from app.models.user import UserStatus


class MarzNodeDB:
    def list_users(self):
        with GetDB() as db:
            relations = crud.get_node_users(db, self.id, [UserStatus.active, UserStatus.on_hold])
            users = dict()
            for rel in relations:
                if not users.get(rel[0]):
                    users[rel[0]] = dict(username=rel[1], id=rel[0], key=rel[2], inbounds=[])
                users[rel[0]]["inbounds"].append(rel[3].tag)
        return list(users.values())

    def store_inbounds(self, inbounds):
        with GetDB() as db:
            crud.assure_node_inbounds(db, inbounds, self.id)

    def set_status(self, status: NodeStatus, message: str | None = None):
        with GetDB() as db:
            crud.update_node_status(db, self.id, status, message)