#!/usr/bin/env python3
"""
Register a single RBAC permission for pipeline modules.
Called by load_path.py or directly.

Usage:
  python set_role_perm.py <role> <path>
  python set_role_perm.py logined /pipeline-sdlc/workspace_edit.xterm
"""
import sys, os, asyncio

SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
ROOT_DIR = os.path.dirname(SCRIPT_DIR) if os.path.basename(SCRIPT_DIR) == 'bin' else SCRIPT_DIR
sys.path.insert(0, os.path.join(ROOT_DIR, 'py3', 'lib', 'python3.10', 'site-packages'))
sys.path.insert(0, ROOT_DIR)

from sqlor.dbpools import DBPools
from appPublic.jsonConfig import getConfig
from appPublic.folderUtils import ProgramPath
from appPublic.uniqueID import getID
from ahserver.serverenv import ServerEnv
from ahserver.globalEnv import initEnv


async def main(role, path):
    config = getConfig(ROOT_DIR,
                       NS={'workdir': ROOT_DIR, 'ProgramPath': ProgramPath()})
    DBPools(config.databases)
    initEnv()
    env = ServerEnv()
    env.get_module_dbname = lambda m: 'pipeline' if 'pipeline' in m else 'sage'

    # pipeline RBAC schema: permission (id, path) + rolepermission (roleid, permid)
    async with DBPools().sqlorContext('pipeline') as sor:
        recs = await sor.R('permission', {'path': path})
        if not recs:
            permid = getID()
            await sor.C('permission', {'id': permid, 'path': path})
        else:
            permid = recs[0].id

        rp = await sor.R('rolepermission', {'roleid': role, 'permid': permid})
        if rp:
            print(f'Permission exists: {role} {path}')
            return

        await sor.C('rolepermission', {'id': getID(), 'roleid': role, 'permid': permid})
        print(f'Registered: {role} -> {path}')


if __name__ == '__main__':
    if len(sys.argv) != 3:
        print(f"Usage: {sys.argv[0]} <role> <path>")
        print(f"  roles: any, anonymous, logined, admin, owner.superuser ...")
        print(f"  path example: /pipeline-sdlc/workspace_edit.xterm")
        sys.exit(1)

    asyncio.run(main(sys.argv[1], sys.argv[2]))