MusicServer/sdk/auth/kg.ts

// 酷狗登录 —— token 绑定与验证（网关 android 签名，与 KuGouMusicApi 对齐）
// 完整的手机号/扫码登录需移植其 RSA/AES 全套协议，后续按需扩展；
// token 可从酷狗 App 抓包获取（login.token）

import crypto from 'node:crypto'
import { httpFetch } from '../common/request.ts'
import { getCredentials, setCredentials } from './store.ts'
import type { KgCredentials } from './store.ts'

const md5 = s => crypto.createHash('md5').update(s).digest('hex')
const SALT_KEY = '57ae12eb6890223e355ccfcb74edf70d'
const SALT_SIGN = 'OIlwieks28dk2k092lksi2UIkp'
const APPID = 1005
const CLIENTVER = 11430
const PUBLIC_KEY = `-----BEGIN PUBLIC KEY-----
MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDIAG7QOELSYoIJvTFJhMpe1s/gbjDJX51HBNnEl5HXqTW6lQ7LC8jr9fWZTwusknp+sVGzwd40MwP6U5yDE27M/X1+UR4tvOGOqp94TJtQ1EPnWGWXngpeIW5GxoQGao1rmYWAu6oi1z9XkChrsUdC6DJE5E221wf/4WLFxwAtRQIDAQAB
-----END PUBLIC KEY-----`

const calculateMid = (guid) => BigInt(`0x${md5(guid)}`).toString()
const randomString = (n) => {
  const chars = 'abcdefghijklmnopqrstuvwxyz0123456789'
  return Array.from({ length: n }, () => chars[Math.floor(Math.random() * chars.length)]).join('')
}

export const androidSign = (params, data = '') => {
  const s = Object.keys(params).sort()
    .map(k => `${k}=${typeof params[k] === 'object' ? JSON.stringify(params[k]) : params[k]}`)
    .join('')
  return md5(SALT_SIGN + s + data + SALT_SIGN)
}

/** 设备标识持久化（guid/mid/dfid 稳定复用，登录验证与取流共用同一指纹） */
export const getDevice = () => {
  const dev = getCredentials('_kg_device') ?? { guid: crypto.randomUUID(), mid: '', dfid: randomString(24) }
  if (!dev.mid) {
    dev.mid = calculateMid(dev.guid)
    setCredentials('_kg_device', dev)
  }
  return dev
}

/** 网关移动端签名请求（gateway.kugou.com + x-router） */
const gatewayRequest = async(url, { params: extraParams, data, router, headers: extraHeaders = {} }) => {
  const dev = getDevice()
  const params = {
    dfid: dev.dfid, mid: dev.mid, uuid: '-', appid: APPID, clientver: CLIENTVER,
    clienttime: Math.floor(Date.now() / 1000),
    ...extraParams,
  }
  const signature = androidSign(params, data ? JSON.stringify(data) : '')
  const qs = Object.keys(params).map(k => `${k}=${encodeURIComponent(params[k])}`).join('&')
  const r = await httpFetch(`https://gateway.kugou.com${url}?${qs}&signature=${signature}`, {
    method: 'post',
    headers: {
      dfid: dev.dfid, mid: dev.mid, clienttime: String(params.clienttime),
      'kg-rc': '1', 'kg-thash': '5d816a0', 'kg-rec': '1', 'kg-rf': 'B9EDA08A64250DEFFBCADDEE00F8F25F',
      'x-router': router,
      'User-Agent': 'Android15-1070-11083-46-0-DiscoveryDRADProtocol-wifi',
      ...extraHeaders,
    },
    body: data ?? undefined,
    timeout: 15000,
  }).promise
  return r.body ?? {}
}

/**
 * 验证 token 并返回账号资料（usercenter /v3/get_my_info）
 * p = RSA 原始加密（零填充 128 字节）{ token, clienttime }
 */
export const verifyToken = async({ token, userid }) => {
  const unixSec = Math.floor(Date.now() / 1000)
  const payload = Buffer.from(JSON.stringify({ token, clienttime: unixSec }))
  const padded = Buffer.concat([Buffer.alloc(128 - payload.length), payload])
  const p = crypto.publicEncrypt({ key: PUBLIC_KEY, padding: crypto.constants.RSA_NO_PADDING }, padded).toString('hex').toUpperCase()

  const body = await gatewayRequest('/v3/get_my_info', {
    router: 'usercenter.kugou.com',
    params: { plat: 1 },
    data: { visit_time: unixSec, usertype: 1, p, userid: Number(userid) || 0 },
  })
  if (body.status !== 1 || !body.data) {
    const err = new Error(body.error ?? 'token 无效或已过期')
    ;(err as any).status = 401
    throw err
  }
  return {
    userid: body.data.userid ?? Number(userid),
    nickname: body.data.nickname ?? body.data.username ?? null,
    vip: body.data.vip ? Boolean(body.data.vip) : null,
  }
}

/** 绑定并验证 kg token */
export const loginByToken = async({ token, userid }) => {
  const profile = await verifyToken({ token, userid })
  setCredentials('kg', { token, userid: profile.userid, nickname: profile.nickname })
  return profile
}

// ---- token 自动续期 ----
// login_by_token 协议（与 lx-music-api-server 对齐）：
//   p3 = AES({clienttime, token})，密钥由固定串派生；params = AES({})（随机密钥）；
//   pk = RSA 裸加密 {clienttime_ms, key: 随机密钥}（右侧零填充 128 字节）；
//   成功响应 data.secu_params 用随机密钥 AES 解密得到新 token。

const AES_KEY_SEED = '90b8382a1bb4ccdcf063102053fd75b8'
const AES_IV = 'f063102053fd75b8'

const aesCbcEncryptHex = (data: string, keyHex: string, ivHex: string) => {
  const cipher = crypto.createCipheriv('aes-256-cbc', Buffer.from(keyHex, 'utf8'), Buffer.from(ivHex, 'utf8'))
  return Buffer.concat([cipher.update(data, 'utf8'), cipher.final()]).toString('hex')
}

const aesEncryptRandom = (data: string | object) => {
  const tempKey = randomString(16)
  const keyHex = md5(tempKey)
  return {
    str: aesCbcEncryptHex(typeof data === 'string' ? data : JSON.stringify(data), keyHex, keyHex.slice(16, 32)),
    key: tempKey,
  }
}

const aesDecryptHex = (hex: string, tempKey: string) => {
  const keyHex = md5(tempKey)
  const d = crypto.createDecipheriv('aes-256-cbc', Buffer.from(keyHex, 'utf8'), Buffer.from(keyHex.slice(16, 32), 'utf8'))
  const out = Buffer.concat([d.update(Buffer.from(hex, 'hex')), d.final()])
  const text = out.toString('utf8')
  try { return JSON.parse(text) } catch { return text }
}

/** RSA 裸加密（零填充 128 字节，与参考实现一致：负载在前、零在后） */
const rsaEncryptHex = (data: object) => {
  const payload = Buffer.from(JSON.stringify(data))
  if (payload.length > 128) throw new Error('RSA 负载超长')
  const padded = Buffer.concat([payload, Buffer.alloc(128 - payload.length)])
  return crypto.publicEncrypt({ key: PUBLIC_KEY, padding: crypto.constants.RSA_NO_PADDING }, padded).toString('hex')
}

/**
 * 刷新 kg token（login_by_token），成功后写回凭据存储。
 * 无登录态时返回 { skipped: true }；网关拒绝（token 已失效等）抛错。
 */
export const refreshLogin = async() => {
  const cred = getCredentials('kg') as KgCredentials | null
  if (!cred?.token || cred.userid === '' || cred.userid === 0 || cred.userid === '0') {
    return { skipped: true }
  }

  const clienttime = Math.floor(Date.now() / 1000)
  const p3 = aesCbcEncryptHex(JSON.stringify({ clienttime, token: cred.token }), md5(AES_KEY_SEED), AES_IV)
  const enc = aesEncryptRandom({})
  const pk = rsaEncryptHex({ clienttime_ms: clienttime * 1000, key: enc.key })

  const body = await gatewayRequest('/v5/login_by_token', {
    router: 'login.user.kugou.com',
    params: { clientver: 20349 },
    data: {
      p3,
      params: enc.str,
      userid: Number(cred.userid) || 0,
      need_toneinfo: 1,
      clienttime_ms: clienttime * 1000,
      dfid: '-',
      dev: 'SDY-AN00',
      plat: 1,
      pk,
      t1: '0',
      gitversion: 'a23c277',
      t2: '0',
      t3: 'MCwwLDEsMSwwLDYsMSw2LDA=',
    },
    headers: {
      'kg-thash': '6a6a1ba',
      'User-Agent': 'Android12-AndroidPhone-20349-201-0-ting#958959317/661004247-LOGIN-wifi',
      'support-calm': '1',
    },
  })

  if (body.error_code !== 0 || !body.data?.secu_params) {
    throw new Error(`刷新失败（code ${body.error_code}${body.error ? `: ${body.error}` : ''}）`)
  }
  const decrypted = aesDecryptHex(body.data.secu_params, enc.key)
  if (!decrypted?.token) throw new Error('刷新响应解密失败')
  setCredentials('kg', { token: decrypted.token })
  return { refreshed: true }
}

/** 当前 kg 登录资料（未登录返回 null） */
export const status = () => {
  const cred = getCredentials('kg')
  return cred?.token ? { userid: cred.userid, nickname: cred.nickname ?? null } : null
}