import json
import os
import urllib.request
import modal

app = modal.App('beamboot-hf-identity')

@app.function(image=modal.Image.debian_slim(python_version='3.12'), secrets=[modal.Secret.from_name('huggingface-secret')])
def identity():
    token = (os.environ.get('HF_TOKEN') or os.environ.get('HUGGING_FACE_HUB_TOKEN') or os.environ.get('HUGGINGFACE_TOKEN') or os.environ.get('HUGGINGFACEHUB_API_TOKEN'))
    if not token: raise RuntimeError('No recognized HF token variable')
    request = urllib.request.Request('https://huggingface.co/api/whoami-v2', headers={'Authorization': 'Bearer ' + token})
    with urllib.request.urlopen(request, timeout=20) as response:
        data = json.load(response)
    return {'name': data.get('name'), 'type': data.get('type'), 'role': data.get('auth', {}).get('accessToken', {}).get('role')}

@app.local_entrypoint()
def main(): print(identity.remote())