UNPKG

@earendil-works/pi-coding-agent

Version:

Coding agent CLI with read, bash, edit, write tools and session management

github.com/earendil-works/pi

earendil-works/pi

3 lines (2 loc) • 6.72 kB
1
2
3
import { createRequire as __piCreateRequire } from "node:module"; const require = __piCreateRequire(import.meta.url); var CANCEL_MESSAGE="Login cancelled",TIMEOUT_MESSAGE="Device flow timed out",SLOW_DOWN_TIMEOUT_MESSAGE="Device flow timed out after one or more slow_down responses. This is often caused by clock drift in WSL or VM environments. Please sync or restart the VM clock and try again.";function abortableSleep(ms,signal,cancelMessage){return new Promise((resolve,reject)=>{if(signal.aborted){reject(new Error(cancelMessage));return}let onAbort=()=>{clearTimeout(timeout),reject(new Error(cancelMessage))},timeout=setTimeout(()=>{signal.removeEventListener("abort",onAbort),resolve()},ms);signal.addEventListener("abort",onAbort,{once:!0})})}async function pollOAuthDeviceCodeFlow(options){let deadline=typeof options.expiresInSeconds=="number"?Date.now()+options.expiresInSeconds*1e3:Number.POSITIVE_INFINITY,intervalMs=Math.max(1e3,Math.floor((options.intervalSeconds??5)*1e3)),slowDownResponses=0;if(options.waitBeforeFirstPoll){let remainingMs=deadline-Date.now();remainingMs>0&&await abortableSleep(Math.min(intervalMs,remainingMs),options.signal,CANCEL_MESSAGE)}for(;Date.now()<deadline;){if(options.signal.aborted)throw new Error(CANCEL_MESSAGE);let result=await options.poll();if(result.status==="complete")return result.value;if(result.status==="failed")throw new Error(result.message);result.status==="slow_down"&&(slowDownResponses+=1,intervalMs=typeof result.intervalSeconds=="number"&&Number.isFinite(result.intervalSeconds)&&result.intervalSeconds>0?Math.max(1e3,Math.floor(result.intervalSeconds*1e3)):Math.max(1e3,intervalMs+5e3));let remainingMs=deadline-Date.now();if(remainingMs<=0)break;await abortableSleep(Math.min(intervalMs,remainingMs),options.signal,CANCEL_MESSAGE)}throw new Error(slowDownResponses>0?SLOW_DOWN_TIMEOUT_MESSAGE:TIMEOUT_MESSAGE)}function trustedHttpUrl(value){if(typeof value!="string"||!URL.canParse(value))return null;let url=new URL(value);return url.protocol==="https:"||url.protocol==="http:"?url.href:null}var CLIENT_ID="1031625952748946",AUTH_HOST="https://auth.meta.com",DEVICE_AUTHORIZATION_URL=`${AUTH_HOST}/oidc/device/authorization/`,DEVICE_TOKEN_URL=`${AUTH_HOST}/oidc/device/token/`,API_KEY_MINT_URL="https://api.meta.ai/muse-code/key",API_KEY_LIFETIME_MS=1440*60*1e3,REQUEST_TIMEOUT_MS=30*1e3;function requestSignal(signal){return AbortSignal.any([AbortSignal.timeout(REQUEST_TIMEOUT_MS),signal])}async function readJson(response){try{let json=await response.json();return json&&typeof json=="object"?json:null}catch{return null}}function errorDetail(json){for(let key of["error_description","detail","message","error"]){let value=json?.[key];if(typeof value=="string"&&value.trim())return`: ${value.trim()}`}return""}function positiveNumber(value){return typeof value=="number"&&Number.isFinite(value)&&value>0?value:void 0}async function startDeviceAuthorization(signal){let response=await fetch(DEVICE_AUTHORIZATION_URL,{method:"POST",headers:{"Content-Type":"application/x-www-form-urlencoded",Accept:"application/json"},body:new URLSearchParams({client_id:CLIENT_ID}).toString(),signal:requestSignal(signal)}),json=await readJson(response);if(!response.ok)throw new Error(`Meta device authorization failed with status ${response.status}${errorDetail(json)}`);let deviceCode=json?.device_code,userCode=json?.user_code,verificationUri=trustedHttpUrl(json?.verification_uri_complete)??trustedHttpUrl(json?.verification_uri);if(typeof deviceCode!="string"||!deviceCode||typeof userCode!="string"||!userCode||!verificationUri)throw new Error(`Invalid Meta device authorization response: ${JSON.stringify(json)}`);return{deviceCode,userCode,verificationUri,intervalSeconds:positiveNumber(json?.interval),expiresInSeconds:positiveNumber(json?.expires_in)}}async function pollForIdentityToken(device,signal){return pollOAuthDeviceCodeFlow({intervalSeconds:device.intervalSeconds,expiresInSeconds:device.expiresInSeconds,waitBeforeFirstPoll:!0,signal,poll:async()=>{let response=await fetch(DEVICE_TOKEN_URL,{method:"POST",headers:{"Content-Type":"application/x-www-form-urlencoded",Accept:"application/json"},body:new URLSearchParams({grant_type:"urn:ietf:params:oauth:grant-type:device_code",device_code:device.deviceCode,client_id:CLIENT_ID}).toString(),signal:requestSignal(signal)}),json=await readJson(response);if(response.ok&&typeof json?.access_token=="string"&&json.access_token)return{status:"complete",value:json.access_token};switch(json?.error){case"authorization_pending":return{status:"pending"};case"slow_down":return{status:"slow_down",intervalSeconds:positiveNumber(json?.interval)};case"access_denied":return{status:"failed",message:"Meta login was denied."};case"expired_token":return{status:"failed",message:"Meta device authorization expired. Please restart login."};default:return{status:"failed",message:`Meta device token request failed with status ${response.status}${errorDetail(json)}`}}}})}async function mintApiKey(identityToken,signal){let response=await fetch(API_KEY_MINT_URL,{method:"POST",headers:{Accept:"application/json",Authorization:`Bearer ${identityToken}`,"Content-Type":"application/json","x-api-version":"1.0.0"},body:"{}",signal:requestSignal(signal)}),json=await readJson(response);if(response.status===401||response.status===403)throw new Error(`Meta session expired (status ${response.status}). Run \`/login meta\` to sign in again.${errorDetail(json)}`);if(!response.ok)throw new Error(`Meta API key mint failed with status ${response.status}${errorDetail(json)}`);let apiKey=json?.api_key;if(typeof apiKey!="string"||!apiKey){let actionUrl=trustedHttpUrl(json?.action_url);throw new Error(`Meta did not issue an API key.${actionUrl?` Complete setup at ${actionUrl}`:""}`)}return{type:"oauth",refresh:identityToken,access:apiKey,expires:Date.now()+API_KEY_LIFETIME_MS}}async function loginMeta(interaction){try{let device=await startDeviceAuthorization(interaction.signal);interaction.notify({type:"device_code",userCode:device.userCode,verificationUri:device.verificationUri,intervalSeconds:device.intervalSeconds,expiresInSeconds:device.expiresInSeconds});let identityToken=await pollForIdentityToken(device,interaction.signal);return interaction.notify({type:"progress",message:"Enabling Meta Model API access..."}),await mintApiKey(identityToken,interaction.signal)}catch(e){throw interaction.signal.aborted?new Error("Login cancelled"):e}}var metaOAuth={name:"Meta (Muse subscription)",isSubscription:!0,loginLabel:"Sign in with Meta",login:loginMeta,refresh:(credential,signal)=>mintApiKey(credential.refresh,signal),async toAuth(credential){return{apiKey:credential.access}}};export{metaOAuth};