import json
import os
import urllib.request
import modal

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

@app.function(image=modal.Image.debian_slim(python_version='3.12'), secrets=[modal.Secret.from_name('steerability-submission-write')])
def inspect():
    results=[]
    for name,value in os.environ.items():
        upper=name.upper()
        if value and ('HF_' in upper or 'HUGGING' in upper) and ('TOKEN' in upper or 'KEY' in upper):
            try:
                req=urllib.request.Request('https://huggingface.co/api/whoami-v2',headers={'Authorization':'Bearer '+value})
                with urllib.request.urlopen(req,timeout=15) as response: data=json.load(response)
                results.append({'variable':name,'name':data.get('name'),'role':data.get('auth',{}).get('accessToken',{}).get('role')})
            except Exception as e: results.append({'variable':name,'error':type(e).__name__})
    return results

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