@dan-uni/dan-any 0.0.2

package/src/index.ts

@@ -0,0 +1,282 @@
+import { XMLParser } from 'fast-xml-parser'
+import type { CommandDm as DM_JSON_BiliCommandGrpc } from './proto/gen/bili/dm_pb'
+// import type * as UniDMType from './utils/dm-gen'
+import type { platfrom } from './utils/id-gen'
+
+import { create, fromBinary, toBinary } from '@bufbuild/protobuf'
+import {
+  timestampDate,
+  timestampFromDate,
+  timestampNow,
+} from '@bufbuild/protobuf/wkt'
+
+import {
+  // DanmakuElem as DM_JSON_BiliGrpc,
+  DmSegMobileReplySchema,
+  DmWebViewReplySchema,
+} from './proto/gen/bili/dm_pb'
+import { DanmakuReplySchema } from './proto/gen/danuni_pb'
+// import type * as UniIDType from './utils/id-gen'
+
+import { UniDM } from './utils/dm-gen'
+import * as UniDMTools from './utils/dm-gen'
+import * as UniIDTools from './utils/id-gen'
+
+export interface DM_XML_Bili {
+  i: {
+    chatserver: string
+    chatid: number
+    mission: number
+    maxlimit: number
+    state: number
+    real_name: number
+    source: string
+    d: {
+      '#text': string
+      '@_p': string
+    }[]
+  }
+}
+export interface DM_JSON_Dplayer {
+  code: 0
+  /**
+   * progress,mode,color,midHash,content
+   */
+  data: [number, number, number, string, string][]
+}
+export interface DM_JSON_Artplayer {
+  text: string // 弹幕文本
+  time?: number // 弹幕时间, 默认为当前播放器时间
+  mode?: number // 弹幕模式: 0: 滚动(默认)，1: 顶部，2: 底部
+  color?: string // 弹幕颜色，默认为白色
+  border?: boolean // 弹幕是否有描边, 默认为 false
+  style?: {} // 弹幕自定义样式, 默认为空对象
+}
+export interface DM_JSON_DDPlay {
+  count: number | string
+  comments: {
+    cid: number
+    p: string
+    m: string
+  }[]
+}
+export class UniPool {
+  constructor(public dans: UniDM[]) {}
+  static fromPb(bin: Uint8Array | ArrayBuffer) {
+    const data = fromBinary(DanmakuReplySchema, new Uint8Array(bin))
+    return new UniPool(
+      data.danmakus.map(
+        (d) =>
+          new UniDM(
+            d.FCID,
+            d.progress,
+            d.mode as number,
+            d.fontsize,
+            d.color,
+            d.senderID,
+            d.content,
+            timestampDate(d.ctime || timestampNow()),
+            d.weight,
+            d.pool as number,
+            d.attr as UniDMTools.DMAttr[],
+            d.platform,
+            d.SPMO,
+            d.extra,
+            d.DMID,
+          ),
+      ),
+    )
+  }
+  /**
+   * 转为 protobuf 二进制
+   */
+  toPb() {
+    return toBinary(
+      DanmakuReplySchema,
+      create(DanmakuReplySchema, {
+        danmakus: this.dans.map((d) => {
+          return {
+            FCID: d.FCID,
+            DMID: d.DMID,
+            progress: d.progress,
+            mode: d.mode as number,
+            fontsize: d.fontsize,
+            color: d.color,
+            senderID: d.senderID,
+            content: d.content,
+            ctime: timestampFromDate(d.ctime),
+            weight: d.weight,
+            pool: d.pool as number,
+            attr: d.attr,
+            platform: d.platform,
+            SPMO: d.SPMO,
+            extra: d.extraStr,
+          }
+        }),
+      }),
+    )
+  }
+  static fromBiliXML(xml: string, SPMO?: string) {
+    const parser = new XMLParser({ ignoreAttributes: false }),
+      oriData: DM_XML_Bili = parser.parse(xml),
+      dans = oriData.i.d
+    return new UniPool(
+      dans.map((d) => {
+        const p_str = d['@_p'],
+          p_arr = p_str.split(',')
+        return UniDM.fromBili(
+          {
+            content: d['#text'],
+            progress: Number.parseFloat(p_arr[0]),
+            mode: Number.parseInt(p_arr[1]),
+            fontsize: Number.parseInt(p_arr[2]),
+            color: Number.parseInt(p_arr[3]),
+            ctime: BigInt(p_arr[4]),
+            pool: Number.parseInt(p_arr[5]),
+            midHash: p_arr[6],
+            id: BigInt(p_arr[7]),
+            weight: Number.parseInt(p_arr[8]),
+          },
+          SPMO,
+          BigInt(oriData.i.chatid),
+        )
+      }),
+    )
+  }
+  static fromBiliGrpc(bin: Uint8Array | ArrayBuffer, SPMO?: string) {
+    const data = fromBinary(DmSegMobileReplySchema, new Uint8Array(bin)),
+      json = data.elems
+    return new UniPool(
+      json.map((d) => {
+        return UniDM.fromBili(d, SPMO)
+      }),
+    )
+  }
+  /**
+   *
+   * @param bin 符合`DmWebViewReplySchema`(bili视频meta)的protobuf二进制
+   * @param SPMO
+   */
+  static fromBiliCommandGrpc(bin: Uint8Array | ArrayBuffer, SPMO?: string) {
+    const data = fromBinary(DmWebViewReplySchema, new Uint8Array(bin)),
+      json = data.commandDms
+    return new UniPool(
+      json.map((d) => {
+        return UniDM.fromBiliCommand(d, SPMO)
+      }),
+    )
+  }
+  static fromDplayer(
+    json: DM_JSON_Dplayer,
+    playerID: string,
+    domain = 'other',
+  ) {
+    return new UniPool(
+      json.data.map((d) => {
+        // let TYPE = 0
+        // if (d[1] === 1) TYPE = 5
+        // else if (d[1] === 2) TYPE = 4
+        return UniDM.fromDplayer(
+          {
+            content: d[4],
+            progress: d[0],
+            mode: d[1],
+            color: d[2],
+            midHash: d[3],
+          },
+          playerID,
+          domain,
+        )
+      }),
+    )
+  }
+  toDplayer(): DM_JSON_Dplayer {
+    return {
+      code: 0,
+      data: this.dans.map((dan) => {
+        const d = dan.toDplayer()
+        return [d.progress, d.mode, d.color, d.midHash, d.content]
+      }),
+    }
+  }
+  static fromArtplayer(
+    json: DM_JSON_Artplayer[],
+    playerID: string,
+    domain = 'other',
+  ) {
+    return new UniPool(
+      json.map((d) => {
+        // let TYPE = 0
+        // if (d.mode === 1) TYPE = 5
+        // else if (d.mode === 2) TYPE = 4
+        return UniDM.fromArtplayer(
+          {
+            content: d.text,
+            progress: d.time || 0,
+            mode: d.mode || 0,
+            color: Number((d.color || 'FFFFFF').replace('#', '0x')),
+            style: d.style,
+          },
+          playerID,
+          domain,
+        )
+      }),
+    )
+  }
+  toArtplayer(): DM_JSON_Artplayer[] {
+    return this.dans.map((dan) => {
+      const d = dan.toArtplayer()
+      return {
+        text: d.content,
+        time: d.progress,
+        mode: d.mode as 0 | 1 | 2,
+        color: `#${d.color.toString(16).toUpperCase() || 'FFFFFF'}`,
+        border: d.border,
+        style: d.style,
+      }
+    })
+  }
+  static fromDDPlay(json: DM_JSON_DDPlay, episodeId: string, domain?: string) {
+    return new UniPool(
+      json.comments.map((d) => {
+        const p_arr = d.p.split(',')
+        return UniDM.fromDDplay(
+          {
+            cid: d.cid,
+            color: Number.parseInt(p_arr[2]),
+            m: d.m,
+            mode: Number.parseInt(p_arr[1]),
+            progress: Number.parseFloat(p_arr[0]),
+            uid: p_arr[3],
+          },
+          episodeId,
+          domain,
+        )
+      }),
+    )
+  }
+  toDDplay(): DM_JSON_DDPlay {
+    return {
+      count: this.dans.length,
+      comments: this.dans.map((dan) => {
+        const d = dan.toDDplay()
+        return {
+          cid: d.cid,
+          p: `${d.progress},${d.mode},${d.color},${d.uid}`,
+          m: d.m,
+        }
+      }),
+    }
+  }
+}
+
+export {
+  // UniPool,
+  UniDM,
+  UniDMTools,
+  UniIDTools,
+  type DM_JSON_BiliCommandGrpc,
+  type platfrom,
+  // type UniDMType,
+  // type UniIDType,
+}