修复签名算法，增加每日积分自动领取

Author: Vinlic

src/api/consts/exceptions.ts

@@ -9,4 +9,5 @@ export default {
     API_CONTENT_FILTERED: [-2006, '内容由于合规问题已被阻止生成'],
     API_IMAGE_GENERATION_FAILED: [-2007, '图像生成失败'],
     API_VIDEO_GENERATION_FAILED: [-2008, '视频生成失败'],
+    API_IMAGE_GENERATION_INSUFFICIENT_POINTS: [-2009, '即梦积分不足'],
 }

src/api/controllers/core.ts

@@ -85,6 +85,51 @@ export function generateCookie(refreshToken: string) {
   ].join("; ");
 }
 
+/**
+ * 获取积分信息
+ *
+ * @param refreshToken 用于刷新access_token的refresh_token
+ */
+export async function getCredit(refreshToken: string) {
+  const {
+    credit: { gift_credit, purchase_credit, vip_credit }
+  } = await request("POST", "/commerce/v1/benefits/user_credit", refreshToken, {
+    data: {},
+    headers: {
+      // Cookie: 'x-web-secsdk-uid=ef44bd0d-0cf6-448c-b517-fd1b5a7267ba; s_v_web_id=verify_m4b1lhlu_DI8qKRlD_7mJJ_4eqx_9shQ_s8eS2QLAbc4n; passport_csrf_token=86f3619c0c4a9c13f24117f71dc18524; passport_csrf_token_default=86f3619c0c4a9c13f24117f71dc18524; n_mh=9-mIeuD4wZnlYrrOvfzG3MuT6aQmCUtmr8FxV8Kl8xY; sid_guard=a7eb745aec44bb3186dbc2083ea9e1a6%7C1733386629%7C5184000%7CMon%2C+03-Feb-2025+08%3A17%3A09+GMT; uid_tt=59a46c7d3f34bda9588b93590cca2e12; uid_tt_ss=59a46c7d3f34bda9588b93590cca2e12; sid_tt=a7eb745aec44bb3186dbc2083ea9e1a6; sessionid=a7eb745aec44bb3186dbc2083ea9e1a6; sessionid_ss=a7eb745aec44bb3186dbc2083ea9e1a6; is_staff_user=false; sid_ucp_v1=1.0.0-KGRiOGY2ODQyNWU1OTk3NzRhYTE2ZmZhYmFjNjdmYjY3NzRmZGRiZTgKHgjToPCw0cwbEIXDxboGGJ-tHyAMMITDxboGOAhAJhoCaGwiIGE3ZWI3NDVhZWM0NGJiMzE4NmRiYzIwODNlYTllMWE2; ssid_ucp_v1=1.0.0-KGRiOGY2ODQyNWU1OTk3NzRhYTE2ZmZhYmFjNjdmYjY3NzRmZGRiZTgKHgjToPCw0cwbEIXDxboGGJ-tHyAMMITDxboGOAhAJhoCaGwiIGE3ZWI3NDVhZWM0NGJiMzE4NmRiYzIwODNlYTllMWE2; store-region=cn-gd; store-region-src=uid; user_spaces_idc={"7444764277623653426":"lf"}; ttwid=1|cxHJViEev1mfkjntdMziir8SwbU8uPNVSaeh9QpEUs8|1733966961|d8d52f5f56607427691be4ac44253f7870a34d25dd05a01b4d89b8a7c5ea82ad; _tea_web_id=7444838473275573797; fpk1=fa6c6a4d9ba074b90003896f36b6960066521c1faec6a60bdcb69ec8ddf85e8360b4c0704412848ec582b2abca73d57a; odin_tt=efe9dc150207879b88509e651a1c4af4e7ffb4cfcb522425a75bd72fbf894eda570bbf7ffb551c8b1de0aa2bfa0bd1be6c4157411ecdcf4464fcaf8dd6657d66',
+      Referer: "https://jimeng.jianying.com/ai-tool/image/generate",
+      // "Device-Time": 1733966964,
+      // Sign: "f3dbb824b378abea7c03cbb152b3a365"
+    }
+  });
+  logger.info(`\n积分信息: \n赠送积分: ${gift_credit}, 购买积分: ${purchase_credit}, VIP积分: ${vip_credit}`);
+  return {
+    giftCredit: gift_credit,
+    purchaseCredit: purchase_credit,
+    vipCredit: vip_credit,
+    totalCredit: gift_credit + purchase_credit + vip_credit
+  }
+}
+
+/**
+ * 接收今日积分
+ *
+ * @param refreshToken 用于刷新access_token的refresh_token
+ */
+export async function receiveCredit(refreshToken: string) {
+  logger.info("正在收取今日积分...")
+  const { cur_total_credits, receive_quota  } = await request("POST", "/commerce/v1/benefits/credit_receive", refreshToken, {
+    data: {
+      time_zone: "Asia/Shanghai"
+    },
+    headers: {
+      Referer: "https://jimeng.jianying.com/ai-tool/image/generate"
+    }
+  });
+  logger.info(`\n今日${receive_quota}积分收取成功\n剩余积分: ${cur_total_credits}`);
+  return cur_total_credits;
+}
+
 /**
  * 请求jimeng
  *
@@ -102,7 +147,7 @@ export async function request(
   const token = await acquireToken(refreshToken);
   const deviceTime = util.unixTimestamp();
   const sign = util.md5(
-    `9e2c|_config|${PLATFORM_CODE}|${VERSION_CODE}|${deviceTime}|11ac`
+    `9e2c|${uri.slice(-7)}|${PLATFORM_CODE}|${VERSION_CODE}|${deviceTime}||11ac`
   );
   const response = await axios.request({
     method,
@@ -205,10 +250,11 @@ export async function uploadFile(
  * @param result 结果
  */
 export function checkResult(result: AxiosResponse) {
-  if (!result.data) return null;
   const { ret, errmsg, data } = result.data;
   if (!_.isFinite(Number(ret))) return result.data;
-  if (ret === "0") return data;
+  if (ret === '0') return data;
+  if (ret === '5000')
+    throw new APIException(EX.API_IMAGE_GENERATION_INSUFFICIENT_POINTS, `[无法生成图像]: 即梦积分可能不足，${errmsg}`);
   throw new APIException(EX.API_REQUEST_FAILED, `[请求jimeng失败]: ${errmsg}`);
 }
 

src/api/controllers/images.ts

@@ -3,7 +3,7 @@ import _ from "lodash";
 import APIException from "@/lib/exceptions/APIException.ts";
 import EX from "@/api/consts/exceptions.ts";
 import util from "@/lib/util.ts";
-import { request } from "./core.ts";
+import { getCredit, receiveCredit, request } from "./core.ts";
 import logger from "@/lib/logger.ts";
 
 const DEFAULT_ASSISTANT_ID = "513695";
@@ -39,6 +39,11 @@ export async function generateImages(
 ) {
   const model = getModel(_model);
   logger.info(`使用模型: ${_model} 映射模型: ${model} ${width}x${height} 精细度: ${sampleStrength}`);
+
+  const { totalCredit } = await getCredit(refreshToken);
+  if (totalCredit <= 0)
+    await receiveCredit(refreshToken);
+
   const componentId = util.uuid();
   const { aigc_data } = await request(
     "post",