ZZZ-Plugin/lib/authkey.js

import fs from 'fs';
import MysZZZApi from './mysapi.js';
import YAML from 'yaml';
let User;
try {
  User = (await import('../../xiaoyao-cvs-plugin/model/user.js')).default;
} catch (e) {
  logger.warn('建议安装逍遥插件以获得更佳体验');
}

/**
 * 获取 Stoken
 * @param {*} e yunzai Event
 * @param {string} mysid 米游社ID
 * @returns
 */
export const getStoken = (e, mysid = '') => {
  // 获取QQ号
  const userId = e.user_id;
  // 实例化用户
  const user = new User(e);
  // 获取 sk 文件路径
  const filePath = `${user.stokenPath}${userId}.yaml`;
  try {
    // 读取文件
    const file = fs.readFileSync(filePath, 'utf-8');
    // 解析文件
    const cks = YAML.parse(file);
    for (const ck in cks) {
      if (cks?.[ck]?.['stuid'] && cks[ck]['stuid'] === mysid) {
        // 如果 ck 存在并且 stuid 与 mysid 相同则返回（这两者都是字符串，不需要类型转换，因此不需要使用弱比较，强比较速度更快，没有隐式转换一步骤，注意代码规范与代码风格）
        return cks[ck];
      }
    }
    return null;
  } catch (error) {
    logger.debug(`[zzz:error]getStoken`, error);
    return null;
  }
};

/**
 * 此方法依赖逍遥插件
 * @param {*} e yunzai Event
 * @param {*} _user
 * @param {string} zzzUid
 * @param {string} authAppid
 * @returns {Promise<string>}
 */
export const getAuthKey = async (e, _user, zzzUid, authAppid = 'csc') => {
  if (!User) {
    throw new Error('未安装逍遥插件，无法自动刷新抽卡链接');
  }
  // 获取 UID 数据
  const uidData = _user.getUidData(zzzUid, 'zzz', e);
  if (!uidData || uidData?.type != 'ck' || !uidData?.ltuid) {
    throw new Error(`当前UID${zzzUid}未绑定cookie，请切换UID后尝试`);
  }
  // 获取 sk
  let ck = getStoken(e, uidData.ltuid);
  if (!ck) {
    throw new Error(
      '获取cookie失败，请确认绑定了cookie或者查询的UID是否与cookie对应，请确认bot所使用的是绝区零UID'
    );
  }
  if (uidData.ltuid !== ck.stuid) {
    throw new Error(
      `当前UID${zzzUid}查询所使用的米游社ID${ck.stuid}与当前切换的米游社ID${uidData.ltuid}不匹配，请切换UID后尝试`
    );
  }
  // 拼接 sk
  ck = `stuid=${ck.stuid};stoken=${ck.stoken};mid=${ck.mid};`;
  // 实例化 API
  const api = new MysZZZApi(zzzUid, ck);
  let type = 'zzzAuthKey';
  switch (authAppid) {
    case 'csc': {
      type = 'zzzAuthKey';
      break;
    }
    default:
  }
  // 获取链接
  const { url, headers, body } = api.getUrl(type);
  // 发送请求
  let res = await fetch(url, {
    method: 'POST',
    headers,
    body,
  });
  // 获取数据
  res = await res.json();
  // 返回 authkey
  return res?.data?.authkey;
};