@uzooai/anime-ai-cli 0.0.0-stage → 0.5.0

package/src/protocol.js

@@ -0,0 +1,83 @@
+import crypto from 'node:crypto'
+function encodeRfc3986(value) {
+  return encodeURIComponent(value).replace(/[!'()*]/g, (char) => `%${char.charCodeAt(0).toString(16).toUpperCase()}`)
+}
+
+function decodeQueryPart(value) {
+  try {
+    return decodeURIComponent(value)
+  } catch {
+    return value
+  }
+}
+
+function canonicalQuery(search) {
+  const raw = String(search || '').replace(/^\?/, '')
+  if (!raw) return ''
+  const pairs = []
+  raw.split('&').forEach((segment, index) => {
+    if (!segment) return
+    const separator = segment.indexOf('=')
+    const rawKey = separator >= 0 ? segment.slice(0, separator) : segment
+    const rawValue = separator >= 0 ? segment.slice(separator + 1) : ''
+    pairs.push([encodeRfc3986(decodeQueryPart(rawKey)), encodeRfc3986(decodeQueryPart(rawValue)), index])
+  })
+  pairs.sort((a, b) => (a[0] === b[0] ? a[2] - b[2] : a[0] < b[0] ? -1 : 1))
+  return pairs.map(([key, value]) => `${key}=${value}`).join('&')
+}
+
+function b64url(value) {
+  return Buffer.from(value).toString('base64url')
+}
+
+function bodyBytes(body) {
+  if (body === undefined || body === null) return Buffer.alloc(0)
+  if (body instanceof URLSearchParams) return Buffer.from(body.toString())
+  return Buffer.from(typeof body === 'string' || Buffer.isBuffer(body) ? body : JSON.stringify(body))
+}
+
+function signRequest({ method, url, body, keyId, apiSecret }) {
+  const parsed = new URL(url)
+  const timestamp = String(Date.now())
+  const nonce = crypto.randomBytes(16).toString('base64url')
+  const key = `S1.${keyId}.${nonce}`
+  const bodyHash = crypto.createHash('sha256').update(bodyBytes(body)).digest('hex')
+  const canonical = [method.toUpperCase(), parsed.pathname || '/', canonicalQuery(parsed.search), bodyHash, timestamp, key].join('\n')
+  const signature = crypto.createHmac('sha256', String(apiSecret)).update(canonical).digest('base64url')
+  return { 'X-T': timestamp, 'X-K': key, 'X-Sign': signature }
+}
+
+function unwrap(payload) {
+  if (payload && typeof payload === 'object' && 'code' in payload) {
+    if (payload.code !== 0) {
+      const error = new Error(payload.message || `API error ${payload.code}`)
+      error.code = payload.code
+      throw error
+    }
+    return 'data' in payload ? payload.data : payload
+  }
+  return payload
+}
+
+function getSetCookie(headers) {
+  if (typeof headers.getSetCookie === 'function') return headers.getSetCookie()
+  const value = headers.get('set-cookie')
+  return value ? [value] : []
+}
+
+function mergeCookie(existing, setCookies) {
+  const jar = new Map()
+  for (const part of String(existing || '').split(';')) {
+    const [key, ...rest] = part.trim().split('=')
+    if (key && rest.length) jar.set(key, rest.join('='))
+  }
+  for (const cookie of setCookies) {
+    const first = String(cookie).split(';', 1)[0]
+    const [key, ...rest] = first.split('=')
+    if (key && rest.length) jar.set(key.trim(), rest.join('='))
+  }
+  return [...jar].map(([key, value]) => `${key}=${value}`).join('; ')
+}
+
+
+export { signRequest, bodyBytes, unwrap, getSetCookie, mergeCookie }