Add plugin: Bilibili Video Searching

app/api/langchain-tools/bili_wbi_tools.ts

@@ -0,0 +1,85 @@
+import md5 from "md5";
+
+const mixinKeyEncTab: number[] = [
+  46, 47, 18, 2, 53, 8, 23, 32, 15, 50, 10, 31, 58, 3, 45, 35, 27, 43, 5, 49,
+  33, 9, 42, 19, 29, 28, 14, 39, 12, 38, 41, 13, 37, 48, 7, 16, 24, 55, 40, 61,
+  26, 17, 0, 1, 60, 51, 30, 4, 22, 25, 54, 21, 56, 59, 6, 63, 57, 62, 11, 36,
+  20, 34, 44, 52,
+];
+
+// 对 imgKey 和 subKey 进行字符顺序打乱编码
+const getMixinKey = (orig: string): string =>
+  mixinKeyEncTab
+    .map((n) => orig[n])
+    .join("")
+    .slice(0, 32);
+
+// 为请求参数进行 wbi 签名
+export function encWbi(
+  params: Record<string, string | number>,
+  img_key: string,
+  sub_key: string,
+): string {
+  const mixin_key: string = getMixinKey(img_key + sub_key);
+  const curr_time: number = Math.round(Date.now() / 1000);
+  const chr_filter: RegExp = /[!'()*]/g;
+
+  Object.assign(params, { wts: curr_time }); // 添加 wts 字段
+  // 按照 key 重排参数
+  const query: string = Object.keys(params)
+    .sort()
+    .map((key) => {
+      // 过滤 value 中的 "!'()*" 字符
+      const value: string = params[key].toString().replace(chr_filter, "");
+      return `${encodeURIComponent(key)}=${encodeURIComponent(value)}`;
+    })
+    .join("&");
+
+  const wbi_sign: string = md5(query + mixin_key); // 计算 w_rid
+
+  return query + "&w_rid=" + wbi_sign;
+}
+
+// 获取最新的 img_key 和 sub_key
+export async function getWbiKeys(): Promise<{
+  img_key: string;
+  sub_key: string;
+}> {
+  const res: Response = await fetch(
+    "https://api.bilibili.com/x/web-interface/nav",
+    {
+      headers: {
+        // SESSDATA 字段
+        Cookie: "SESSDATA=xxxxxx",
+        "User-Agent":
+          "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3",
+        Referer: "https://www.bilibili.com/", //对于直接浏览器调用可能不适用
+      },
+    },
+  );
+  const {
+    data: {
+      wbi_img: { img_url, sub_url },
+    },
+  } = await res.json();
+
+  return {
+    img_key: img_url.slice(
+      img_url.lastIndexOf("/") + 1,
+      img_url.lastIndexOf("."),
+    ),
+    sub_key: sub_url.slice(
+      sub_url.lastIndexOf("/") + 1,
+      sub_url.lastIndexOf("."),
+    ),
+  };
+}
+
+// async function main(): Promise<void> {
+//     const web_keys: { img_key: string, sub_key: string } = await getWbiKeys()
+//     const params: Record<string, string | number> = { foo: '114', bar: '514', baz: 1919810 }
+//     const img_key: string = web_keys.img_key
+//     const sub_key: string = web_keys.sub_key
+//     const query: string = encWbi(params, img_key, sub_key)
+//     console.log(query)
+// }