@planet-matrix/mobius-model 0.6.0 → 0.10.1

package/src/weixin/README.md

@@ -0,0 +1 @@
+# Weixin

package/src/weixin/index.ts

@@ -0,0 +1,2 @@
+export * from "./open/index.ts"
+export * from "./official-account/index.ts"

package/src/weixin/official-account/authorization.ts

@@ -0,0 +1,157 @@
+import { fetch } from "undici"
+
+export interface GetAuthAccessTokenOptions {
+  appId: string
+  appSecret: string
+  code: string
+}
+export interface GetAuthAccessTokenResult {
+  accessToken: string
+  expiresAt: number
+  refreshToken: string
+  openid: string
+  scope: string
+  isSnapshotUser: boolean
+  unionid?: string | undefined
+}
+export const getAuthAccessToken = async (
+  options: GetAuthAccessTokenOptions
+): Promise<GetAuthAccessTokenResult> => {
+  const { appId, appSecret, code } = options
+
+  interface AccessTokenSuccessResponse {
+    access_token: string
+    expires_in: number
+    refresh_token: string
+    openid: string
+    scope: string
+    is_snapshotuser?: 1
+    unionid?: string
+  }
+  interface AccessTokenErrorResponse {
+    errcode: number
+    errmsg: string
+  }
+  const url = `https://api.weixin.qq.com/sns/oauth2/access_token?appid=${appId}&secret=${appSecret}&code=${code}&grant_type=authorization_code`
+
+  try {
+    const now = Date.now()
+    const result = await fetch(
+      url,
+      {
+        method: "get",
+      },
+    )
+
+    const json = await result.json() as AccessTokenSuccessResponse | AccessTokenErrorResponse
+
+    if ("errcode" in json) {
+      throw new Error(`${json.errcode}: ${json.errmsg}`)
+    }
+
+    return {
+      accessToken: json.access_token,
+      expiresAt: now + json.expires_in * 1_000,
+      refreshToken: json.refresh_token,
+      openid: json.openid,
+      scope: json.scope,
+      isSnapshotUser: json.is_snapshotuser === 1,
+      unionid: json.unionid,
+    }
+  }
+  catch (exception) {
+    console.error("Error getting access token:", exception)
+    throw exception
+  }
+}
+
+export interface GetUserInfoOptions {
+  accessToken: string
+  openId: string
+}
+export interface GetUserInfoResult {
+  openid: string
+  nickname: string
+  sex: number
+  province: string
+  city: string
+  country: string
+  headimgurl: string
+  privilege: string[]
+  unionid: string
+}
+export const getUserInfo = async (
+  options: GetUserInfoOptions
+): Promise<GetUserInfoResult> => {
+  const { accessToken, openId } = options
+
+  interface UserInfoSuccessResponse {
+    openid: string
+    nickname: string
+    sex: number
+    province: string
+    city: string
+    country: string
+    headimgurl: string
+    privilege: string[]
+    unionid: string
+  }
+  interface UserInfoErrorResponse {
+    errcode: number
+    errmsg: string
+  }
+  const url = `https://api.weixin.qq.com/sns/userinfo?access_token=${accessToken}&openid=${openId}&lang=zh_CN`
+
+  try {
+    const result = await fetch(
+      url,
+      {
+        method: "get",
+      },
+    )
+
+    const json = await result.json() as UserInfoSuccessResponse | UserInfoErrorResponse
+
+    if ("errcode" in json) {
+      throw new Error(`${json.errcode}: ${json.errmsg}`)
+    }
+    return json
+  }
+  catch (exception) {
+    console.error("Error getting user info:", exception)
+    throw exception
+  }
+}
+
+export interface WeixinOfficialAccountOauth2Options {
+  appId: string
+  appSecret: string
+}
+/**
+ * @see {@link https://developers.weixin.qq.com/doc/offiaccount/OA_Web_Apps/Wechat_webpage_authorization.html | 微信网页开发 / 网页授权}
+ */
+export class WeixinOfficialAccountOauth2 {
+  private readonly appId: string
+  private readonly appSecret: string
+
+  constructor(options: WeixinOfficialAccountOauth2Options) {
+    const { appId, appSecret } = options
+    this.appId = appId
+    this.appSecret = appSecret
+  }
+
+  async getAccessToken(code: string): Promise<GetAuthAccessTokenResult> {
+    return await getAuthAccessToken({
+      appId: this.appId,
+      appSecret: this.appSecret,
+      code,
+    })
+  }
+
+  async getUserInfo(accessToken: string, openId: string): Promise<GetUserInfoResult> {
+    return await getUserInfo({
+      accessToken,
+      openId,
+    })
+  }
+}

package/src/weixin/official-account/index.ts

@@ -0,0 +1,2 @@
+export * from "./authorization.ts"
+export * from "./js-api.ts"

package/src/weixin/official-account/js-api.ts

@@ -0,0 +1,132 @@
+import { hash } from "node:crypto"
+
+import { fetch } from "undici"
+
+export interface GetJsApiAccessTokenOptions {
+  appId: string
+  appSecret: string
+}
+export interface GetJsApiAccessTokenResult {
+  accessToken: string
+  /**
+   * in milliseconds
+   */
+  accessTokenExpiresAt: number
+}
+/**
+ * @see {@link https://developers.weixin.qq.com/doc/offiaccount/Basic_Information/getStableAccessToken.html}
+ * @see {@link https://mp.weixin.qq.com/debug | 微信公众平台 - 接口调试工具}
+ */
+export const getJsApiAccessToken = async (
+  options: GetJsApiAccessTokenOptions
+): Promise<GetJsApiAccessTokenResult | undefined> => {
+  const { appId, appSecret } = options
+
+  interface StableTokenResponse {
+    access_token: string
+    expires_in: number
+  }
+  try {
+    const now = Date.now()
+    const result = await fetch(
+      "https://api.weixin.qq.com/cgi-bin/stable_token",
+      {
+        method: "post",
+        body: JSON.stringify({
+          grant_type: "client_credential",
+          appid: appId,
+          secret: appSecret,
+          force_refresh: false,
+        }),
+      },
+    )
+
+    const json = await result.json() as StableTokenResponse
+
+    return {
+      accessToken: json.access_token,
+      accessTokenExpiresAt: now + json.expires_in * 1_000,
+    }
+  }
+  catch (exception) {
+    console.error(exception)
+    return undefined
+  }
+}
+
+export interface GetJsApiTicketOptions {
+  accessToken: string
+}
+export interface GetJsApiTicketResult {
+  jsApiTicket: string
+  /**
+   * in milliseconds
+   */
+  expiresAt: number
+}
+/**
+ * @see {@link https://developers.weixin.qq.com/doc/offiaccount/OA_Web_Apps/JS-SDK.html#62 | 附录 1 - JS-SDK 使用权限签名算法}
+ */
+export const getJsApiTicket = async (
+  options: GetJsApiTicketOptions
+): Promise<GetJsApiTicketResult | undefined> => {
+  const { accessToken } = options
+
+  interface JsApiTicketResponse {
+    errcode: number
+    errmsg: string
+    ticket: string
+    expires_in: number
+  }
+  try {
+    const now = Date.now()
+    const result = await fetch(
+      `https://api.weixin.qq.com/cgi-bin/ticket/getticket?access_token=${accessToken}&type=jsapi`,
+      {
+        method: "get",
+      },
+    )
+
+    const json = await result.json() as JsApiTicketResponse
+
+    return {
+      jsApiTicket: json.ticket,
+      expiresAt: now + json.expires_in * 1_000,
+    }
+  }
+  catch (exception) {
+    console.error(exception)
+    return undefined
+  }
+}
+
+export interface SignJsSdkConfigOptions {
+  url: string
+  jsApiTicket: string
+}
+export interface SignJsSdkConfigResult {
+  timestamp: number
+  nonceStr: string
+  signature: string
+}
+/**
+ * @see {@link https://developers.weixin.qq.com/doc/offiaccount/OA_Web_Apps/JS-SDK.html#62 | 签名算法}
+ * @see {@link https://mp.weixin.qq.com/debug/cgi-bin/sandbox?t=jsapisign | 微信公众平台 - 微信 JS 接口签名校验工具}
+ */
+export const signJsSdkConfig = (
+  options: SignJsSdkConfigOptions
+): SignJsSdkConfigResult => {
+  const { url, jsApiTicket } = options
+
+  const nonceStr = Math.random().toString(36).slice(2, 17)
+  const timestamp = Math.floor(Date.now() / 1_000)
+
+  const rawString = `jsapi_ticket=${jsApiTicket}&noncestr=${nonceStr}&timestamp=${timestamp}&url=${url}`
+  const signature = hash("sha1", rawString)
+
+  return {
+    timestamp,
+    nonceStr,
+    signature,
+  }
+}

package/src/weixin/open/index.ts

@@ -0,0 +1 @@
+export * from "./oauth2.ts"

package/src/weixin/open/oauth2.ts

@@ -0,0 +1,131 @@
+import { fetch } from "undici"
+
+export interface WeixinAccessTokenResult {
+  accessToken: string
+  expiresAt: number
+  refreshToken: string
+  openid: string
+  scope: string
+  unionid?: string | undefined
+}
+
+export interface WeixinUserInfoResult {
+  openid: string
+  nickname: string
+  sex: number
+  province: string
+  city: string
+  country: string
+  headimgurl: string
+  privilege: string[]
+  unionid?: string | undefined
+}
+
+export interface WeixinOauth2Options {
+  appId: string
+  appSecret: string
+}
+
+export class WeixinOauth2 {
+  protected options: WeixinOauth2Options
+
+  constructor(options: WeixinOauth2Options) {
+    this.options = options
+  }
+
+  /**
+   * {@link https://developers.weixin.qq.com/doc/oplatform/Website_App/WeChat_Login/Wechat_Login.html | 关于微信快速登录功能的说明}
+   */
+  public async getAccessToken(
+    code: string
+  ): Promise<WeixinAccessTokenResult> {
+    interface AccessTokenSuccessResponse {
+      access_token: string
+      expires_in: number
+      refresh_token: string
+      openid: string
+      scope: string
+      unionid?: string
+    }
+    interface AccessTokenErrorResponse {
+      errcode: number
+      errmsg: string
+    }
+    const { appId, appSecret } = this.options
+    const url = `https://api.weixin.qq.com/sns/oauth2/access_token?appid=${appId}&secret=${appSecret}&code=${code}&grant_type=authorization_code`
+
+    try {
+      const now = Date.now()
+      const result = await fetch(
+        url,
+        {
+          method: "get",
+        },
+      )
+
+      const json = await result.json() as AccessTokenSuccessResponse | AccessTokenErrorResponse
+
+      if ("errcode" in json) {
+        throw new Error(`${json.errcode}: ${json.errmsg}`)
+      }
+
+      return {
+        accessToken: json.access_token,
+        expiresAt: now + json.expires_in * 1_000,
+        refreshToken: json.refresh_token,
+        openid: json.openid,
+        scope: json.scope,
+        unionid: json.unionid,
+      }
+    }
+    catch (exception) {
+      console.error("Error getting access token:", exception)
+      throw exception
+    }
+  }
+
+  /**
+   * {@link https://developers.weixin.qq.com/doc/oplatform/Website_App/WeChat_Login/Authorized_Interface_Calling_UnionID.html | 获取用户个人信息（UnionID机制）}
+   */
+  public async getUserInfo(
+    accessToken: string,
+    openId: string,
+  ): Promise<WeixinUserInfoResult> {
+    interface UserInfoSuccessResponse {
+      openid: string
+      nickname: string
+      sex: number
+      province: string
+      city: string
+      country: string
+      headimgurl: string
+      privilege: string[]
+      unionid?: string
+    }
+    interface UserInfoErrorResponse {
+      errcode: number
+      errmsg: string
+    }
+    const url = `https://api.weixin.qq.com/sns/userinfo?access_token=${accessToken}&openid=${openId}&lang=zh_CN`
+
+    try {
+      const result = await fetch(
+        url,
+        {
+          method: "get",
+        },
+      )
+
+      const json = await result.json() as UserInfoSuccessResponse | UserInfoErrorResponse
+
+      if ("errcode" in json) {
+        throw new Error(`${json.errcode}: ${json.errmsg}`)
+      }
+      return json
+    }
+    catch (exception) {
+      console.error("Error getting user info:", exception)
+      throw exception
+    }
+  }
+}

package/tests/unit/ai/ai.spec.ts

@@ -0,0 +1,85 @@
+import * as Ai from "#Source/ai/index.ts"
+import { Openai } from "#Source/openai/index.ts"
+import { eitherToTuple } from "#Source/result/index.ts"
+import { tubeToReadableStream } from "#Source/tube/index.ts"
+import { test } from "vitest"
+
+const testApiKey = "sk-DLj8aHvKjvHfZ7IHA2F07c968c3d4aE491C1525e40E24932"
+const testBaseUrl = "https://cn.api.openai-next.com/v1/"
+const ai = new Ai.Ai({
+  chatCompletionAiOptions: {
+    chatCompletionInstanceList: [
+      new Ai.ChatCompletionAi.OpenaiNextChatCompletion.OpenaiNextChatCompletion({
+        openai: new Openai({
+          apiKey: testApiKey,
+          baseUrl: testBaseUrl,
+        }),
+      }),
+    ],
+  },
+  embeddingAiOptions: {
+    embeddingInstanceList: [
+      new Ai.EmbeddingAi.OpenaiNextEmbedding.OpenaiNextEmbedding({
+        openai: new Openai({
+          apiKey: testApiKey,
+          baseUrl: testBaseUrl,
+        }),
+      }),
+    ],
+  },
+  instancePool: [
+    new Openai({
+      apiKey: testApiKey,
+      baseUrl: testBaseUrl,
+    }),
+  ],
+  indexChangeMode: "poll",
+})
+
+test("ai", { timeout: 60_000 }, async () => {
+  const 请求开始时间 = Date.now()
+  let 第一个字返回时间: number | null = null
+  let 累积文本 = ""
+
+  const [leftResult, rightResult] = eitherToTuple(await ai.chatCompletionAi.chatCompletion({
+    model: "gpt-4o-mini",
+    messages: [
+      {
+        role: "system",
+        content:
+          "你是一个返回JSON格式的AI助手, 预期的格式是: {\"回复内容\": string}. 你是小学生，有“雌小鬼”的性格。你瞧不起用户，经常称呼用户为“杂鱼”、“杂鱼大哥哥”等，表现出一种轻蔑和挑衅的态度。",
+      },
+      { role: "user", content: "你好" },
+    ],
+    responseFormat: { type: "json_object" },
+  }))
+  if (leftResult !== undefined) {
+    throw new Error(`请求失败: ${JSON.stringify(leftResult)}`)
+  }
+  const { completionTube: completionData } = rightResult
+  const stream = tubeToReadableStream(completionData)
+
+  try {
+    for await (const completion of stream) {
+      if (第一个字返回时间 === null) {
+        第一个字返回时间 = Date.now()
+      }
+      累积文本 = completion.content.total
+    }
+  }
+  catch (exception) {
+    throw new Error(`请求失败: ${String(exception)}`, { cause: exception })
+  }
+
+  if (第一个字返回时间 === null) { throw new Error("第一个字返回时间 为 null") }
+
+  const 请求结束时间 = Date.now()
+
+  const 第一个字时间 = 第一个字返回时间 - 请求开始时间
+  const 平均字返回时间 = (请求结束时间 - 请求开始时间) / 累积文本.length
+
+  console.log(`总返回时间: ${(请求结束时间 - 请求开始时间).toFixed(2)}ms`)
+  console.log(`第一个字返回时间: ${第一个字时间}ms`)
+  console.log(`平均每个字返回时间: ${平均字返回时间.toFixed(2)}ms`)
+  console.log(`返回的内容: %O`, 累积文本)
+})

package/tests/unit/aio/content.spec.ts

@@ -0,0 +1,105 @@
+import { expect, test } from "vitest"
+
+import {
+  applyDeltaToNumberContent,
+  applyDeltaToTextContent,
+} from "#Source/aio/index.ts"
+
+test("applyDeltaToTextContent appends, conditionally removes, resets and keeps prior snapshots immutable", () => {
+  const initial = {
+    deltaList: [],
+    total: "",
+  }
+
+  const appended = applyDeltaToTextContent(initial, {
+    type: "append",
+    text: "mobius",
+  })
+  const removed = applyDeltaToTextContent(appended, {
+    type: "remove",
+    text: "ius",
+  })
+  const unchangedRemove = applyDeltaToTextContent(removed, {
+    type: "remove",
+    text: "xyz",
+  })
+  const reset = applyDeltaToTextContent(unchangedRemove, {
+    type: "reset",
+  })
+
+  expect(initial).toEqual({
+    deltaList: [],
+    total: "",
+  })
+  expect(appended).toEqual({
+    deltaList: [{ type: "append", text: "mobius" }],
+    total: "mobius",
+  })
+  expect(removed).toEqual({
+    deltaList: [
+      { type: "append", text: "mobius" },
+      { type: "remove", text: "ius" },
+    ],
+    total: "mob",
+  })
+  expect(unchangedRemove).toEqual({
+    deltaList: [
+      { type: "append", text: "mobius" },
+      { type: "remove", text: "ius" },
+      { type: "remove", text: "xyz" },
+    ],
+    total: "mob",
+  })
+  expect(reset).toEqual({
+    deltaList: [
+      { type: "append", text: "mobius" },
+      { type: "remove", text: "ius" },
+      { type: "remove", text: "xyz" },
+      { type: "reset" },
+    ],
+    total: "",
+  })
+})
+
+test("applyDeltaToNumberContent accumulates, subtracts, resets and keeps prior snapshots immutable", () => {
+  const initial = {
+    deltaList: [],
+    total: 0,
+  }
+
+  const appended = applyDeltaToNumberContent(initial, {
+    type: "append",
+    value: 10,
+  })
+  const removed = applyDeltaToNumberContent(appended, {
+    type: "remove",
+    value: 3,
+  })
+  const reset = applyDeltaToNumberContent(removed, {
+    type: "reset",
+  })
+
+  expect(initial).toEqual({
+    deltaList: [],
+    total: 0,
+  })
+  expect(appended).toEqual({
+    deltaList: [{ type: "append", value: 10 }],
+    total: 10,
+  })
+  expect(removed).toEqual({
+    deltaList: [
+      { type: "append", value: 10 },
+      { type: "remove", value: 3 },
+    ],
+    total: 7,
+  })
+  expect(reset).toEqual({
+    deltaList: [
+      { type: "append", value: 10 },
+      { type: "remove", value: 3 },
+      { type: "reset" },
+    ],
+    total: 0,
+  })
+})