x402 Create API key

import { createKeyPairSignerFromBytes } from '@solana/kit';
import { x402Client, x402HTTPClient } from '@x402/core/client';
import { registerExactSvmScheme } from '@x402/svm/exact/client';

const BASE_URL = process.env.API_URL ?? 'http://localhost:4058';
const PRIVATE_KEY_B58 = process.env.SOLANA_PRIVATE_KEY;
const endpoint = '/agent/x402/api-keys/create';

function base58ToBytes(b58: string): Uint8Array {
  const ALPHABET = '123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz';
  const map = new Uint8Array(256).fill(255);
  for (let i = 0; i < ALPHABET.length; i++) map[ALPHABET.charCodeAt(i)] = i;
  const bytes: number[] = [0];
  for (const char of b58) {
    const value = map[char.charCodeAt(0)];
    if (value === undefined || value === 255) throw new Error(`Invalid base58: ${char}`);
    let carry: number = value;
    for (let j = bytes.length - 1; j >= 0; j--) {
      carry += (bytes[j] as number) * 58;
      bytes[j] = carry & 0xff;
      carry >>= 8;
    }
    while (carry > 0) {
      bytes.unshift(carry & 0xff);
      carry >>= 8;
    }
  }
  for (const char of b58) {
    if (char !== '1') break;
    bytes.unshift(0);
  }
  return new Uint8Array(bytes);
}

async function main() {
  if (!PRIVATE_KEY_B58) {
    console.error('ERROR: Set SOLANA_PRIVATE_KEY (base58, same wallet used to subscribe)');
    process.exit(1);
  }

  const signer = await createKeyPairSignerFromBytes(base58ToBytes(PRIVATE_KEY_B58));

  console.log(`[1] GET ${BASE_URL}${endpoint} (no payment)`);
  const probeRes = await fetch(`${BASE_URL}${endpoint}`);
  if (probeRes.status !== 402) {
    console.error(probeRes.status, await probeRes.text());
    process.exit(1);
  }

  const paymentRequiredHeader = probeRes.headers.get('payment-required');
  if (!paymentRequiredHeader) {
    console.error('Missing payment-required header');
    process.exit(1);
  }

  const decoded = JSON.parse(Buffer.from(paymentRequiredHeader, 'base64').toString('utf-8'));
  const svmOption = decoded.accepts?.find((a: { network: string }) => a.network?.startsWith('solana:'));
  if (!svmOption) {
    console.error('No Solana option in 402');
    process.exit(1);
  }
  console.log('[2] Create API key price: $0.001 USDC');

  console.log('[3] Signing payment...');
  const client = new x402Client();
  registerExactSvmScheme(client, { signer });
  const httpClient = new x402HTTPClient(client);
  const paymentRequired = httpClient.getPaymentRequiredResponse((name) => probeRes.headers.get(name), decoded);

  let paymentHeaders: Record<string, string>;
  try {
    const payload = await httpClient.createPaymentPayload(paymentRequired);
    paymentHeaders = httpClient.encodePaymentSignatureHeader(payload);
  } catch (err) {
    console.error('Payment creation failed:', err);
    process.exit(1);
  }

  console.log(`[4] GET ${BASE_URL}${endpoint} (with payment)`);
  const paidRes = await fetch(`${BASE_URL}${endpoint}`, { headers: paymentHeaders });
  const body = await paidRes.text();

  if (paidRes.status === 404) {
    console.error('No agent found for this wallet. Subscribe first with agent-subscribe-solana.ts');
    process.exit(1);
  }
  if (paidRes.status !== 200) {
    console.error('Request failed:', paidRes.status, body);
    process.exit(1);
  }

  const data = JSON.parse(body) as { api_key: string; user_id: string };
  console.log('New API key created.');
  console.log('   api_key:', data.api_key);
  console.log('   user_id:', data.user_id);
  process.exit(0);
}

main().catch((err) => {
  console.error(err);
  process.exit(1);
});