/**
 * JWT generation for Enable Banking API authentication
 *
 * Enable Banking requires JWT tokens signed with RS256 using your private key.
 * The JWT is included in the Authorization header for all API calls.
 */

import * as crypto from 'crypto'

// Prefer _PRODUCTION variants when available (Vercel production deploys)
const APP_ID = process.env.ENABLE_BANKING_APP_ID_PRODUCTION || process.env.ENABLE_BANKING_APP_ID
const PRIVATE_KEY_RAW = process.env.ENABLE_BANKING_PRIVATE_KEY_PRODUCTION || process.env.ENABLE_BANKING_PRIVATE_KEY

interface JWTHeader {
  typ: string
  alg: string
  kid: string
}

interface JWTPayload {
  iss: string
  aud: string
  iat: number
  exp: number
}

function base64UrlEncode(data: Buffer | string): string {
  const str = typeof data === 'string' ? data : data.toString('base64')
  return str.replace(/\+/g, '-').replace(/\//g, '_').replace(/=+$/, '')
}

function getPrivateKey(): string {
  if (!PRIVATE_KEY_RAW) {
    throw new Error('ENABLE_BANKING_PRIVATE_KEY environment variable is not set')
  }

  // Try decoding as base64-encoded PEM (sandbox format: base64 wrapping a PEM string)
  const decoded = Buffer.from(PRIVATE_KEY_RAW, 'base64').toString('utf-8')
  if (decoded.startsWith('-----BEGIN')) {
    return decoded
  }

  // Otherwise treat as raw base64 DER key material — wrap in PEM headers
  const lines = PRIVATE_KEY_RAW.match(/.{1,64}/g) || []
  return `-----BEGIN PRIVATE KEY-----\n${lines.join('\n')}\n-----END PRIVATE KEY-----`
}

/**
 * Generate a JWT token for Enable Banking API authentication
 *
 * @param expiresInSeconds - Token validity in seconds (default: 3600 = 1 hour)
 * @returns Signed JWT token
 */
export function generateJWT(expiresInSeconds: number = 3600): string {
  if (!APP_ID) {
    throw new Error('ENABLE_BANKING_APP_ID environment variable is not set')
  }

  const now = Math.floor(Date.now() / 1000)

  const header: JWTHeader = {
    typ: 'JWT',
    alg: 'RS256',
    kid: APP_ID
  }

  const payload: JWTPayload = {
    iss: 'enablebanking.com',
    aud: 'api.enablebanking.com',
    iat: now,
    exp: now + expiresInSeconds
  }

  // Encode header and payload
  const headerBase64 = base64UrlEncode(Buffer.from(JSON.stringify(header)))
  const payloadBase64 = base64UrlEncode(Buffer.from(JSON.stringify(payload)))

  // Create signature
  const signatureInput = `${headerBase64}.${payloadBase64}`
  const privateKey = getPrivateKey()

  const sign = crypto.createSign('RSA-SHA256')
  sign.update(signatureInput)
  sign.end()

  const signature = sign.sign(privateKey)
  const signatureBase64 = base64UrlEncode(signature)

  return `${headerBase64}.${payloadBase64}.${signatureBase64}`
}

/**
 * Get the Authorization header value for Enable Banking API
 */
export function getAuthorizationHeader(): string {
  const token = generateJWT()
  return `Bearer ${token}`
}