@xbibzlibrary/whatsbibz 1.3.0

package/lib/Utils/reporting-utils.js

@@ -0,0 +1,257 @@
+import { createHmac } from 'crypto';
+import { proto } from '../../WAProto/index.js';
+import { hkdf } from './crypto.js';
+const reportingFields = [
+    { f: 1 },
+    {
+        f: 3,
+        s: [{ f: 2 }, { f: 3 }, { f: 8 }, { f: 11 }, { f: 17, s: [{ f: 21 }, { f: 22 }] }, { f: 25 }]
+    },
+    { f: 4, s: [{ f: 1 }, { f: 16 }, { f: 17, s: [{ f: 21 }, { f: 22 }] }] },
+    { f: 5, s: [{ f: 3 }, { f: 4 }, { f: 5 }, { f: 16 }, { f: 17, s: [{ f: 21 }, { f: 22 }] }] },
+    { f: 6, s: [{ f: 1 }, { f: 17, s: [{ f: 21 }, { f: 22 }] }, { f: 30 }] },
+    { f: 7, s: [{ f: 2 }, { f: 7 }, { f: 10 }, { f: 17, s: [{ f: 21 }, { f: 22 }] }, { f: 20 }] },
+    { f: 8, s: [{ f: 2 }, { f: 7 }, { f: 9 }, { f: 17, s: [{ f: 21 }, { f: 22 }] }, { f: 21 }] },
+    { f: 9, s: [{ f: 2 }, { f: 6 }, { f: 7 }, { f: 13 }, { f: 17, s: [{ f: 21 }, { f: 22 }] }, { f: 20 }] },
+    { f: 12, s: [{ f: 1 }, { f: 2 }, { f: 14, m: true }, { f: 15 }] },
+    { f: 18, s: [{ f: 6 }, { f: 16 }, { f: 17, s: [{ f: 21 }, { f: 22 }] }] },
+    { f: 26, s: [{ f: 4 }, { f: 5 }, { f: 8 }, { f: 13 }, { f: 17, s: [{ f: 21 }, { f: 22 }] }] },
+    { f: 28, s: [{ f: 1 }, { f: 2 }, { f: 4 }, { f: 5 }, { f: 6 }, { f: 7, s: [{ f: 21 }, { f: 22 }] }] },
+    { f: 37, s: [{ f: 1, m: true }] },
+    {
+        f: 49,
+        s: [
+            { f: 2 },
+            { f: 3, s: [{ f: 1 }, { f: 2 }] },
+            { f: 5, s: [{ f: 21 }, { f: 22 }] },
+            { f: 8, s: [{ f: 1 }, { f: 2 }] }
+        ]
+    },
+    { f: 53, s: [{ f: 1, m: true }] },
+    { f: 55, s: [{ f: 1, m: true }] },
+    { f: 58, s: [{ f: 1, m: true }] },
+    { f: 59, s: [{ f: 1, m: true }] },
+    {
+        f: 60,
+        s: [
+            { f: 2 },
+            { f: 3, s: [{ f: 1 }, { f: 2 }] },
+            { f: 5, s: [{ f: 21 }, { f: 22 }] },
+            { f: 8, s: [{ f: 1 }, { f: 2 }] }
+        ]
+    },
+    {
+        f: 64,
+        s: [
+            { f: 2 },
+            { f: 3, s: [{ f: 1 }, { f: 2 }] },
+            { f: 5, s: [{ f: 21 }, { f: 22 }] },
+            { f: 8, s: [{ f: 1 }, { f: 2 }] }
+        ]
+    },
+    { f: 66, s: [{ f: 2 }, { f: 6 }, { f: 7 }, { f: 13 }, { f: 17, s: [{ f: 21 }, { f: 22 }] }, { f: 20 }] },
+    { f: 74, s: [{ f: 1, m: true }] },
+    { f: 87, s: [{ f: 1, m: true }] },
+    { f: 88, s: [{ f: 1 }, { f: 2, s: [{ f: 1 }] }, { f: 3, s: [{ f: 21 }, { f: 22 }] }] },
+    { f: 92, s: [{ f: 1, m: true }] },
+    { f: 93, s: [{ f: 1, m: true }] },
+    { f: 94, s: [{ f: 1, m: true }] }
+];
+const compileReportingFields = (fields) => {
+    const map = new Map();
+    for (const f of fields) {
+        map.set(f.f, {
+            m: f.m,
+            children: f.s ? compileReportingFields(f.s) : undefined
+        });
+    }
+    return map;
+};
+const compiledReportingFields = compileReportingFields(reportingFields);
+const EMPTY_MAP = new Map();
+const ENC_SECRET_REPORT_TOKEN = 'Report Token';
+const WIRE = {
+    VARINT: 0,
+    FIXED64: 1,
+    BYTES: 2,
+    FIXED32: 5
+};
+export const shouldIncludeReportingToken = (message) => !message.reactionMessage &&
+    !message.encReactionMessage &&
+    !message.encEventResponseMessage &&
+    !message.pollUpdateMessage;
+const generateMsgSecretKey = (modificationType, origMsgId, origMsgSender, modificationSender, origMsgSecret) => {
+    const useCaseSecret = Buffer.concat([
+        Buffer.from(origMsgId, 'utf8'),
+        Buffer.from(origMsgSender, 'utf8'),
+        Buffer.from(modificationSender, 'utf8'),
+        Buffer.from(modificationType, 'utf8')
+    ]);
+    return hkdf(origMsgSecret, 32, { info: useCaseSecret.toString('latin1') });
+};
+const extractReportingTokenContent = (data, cfg) => {
+    const out = [];
+    let i = 0;
+    while (i < data.length) {
+        const tag = decodeVarint(data, i);
+        if (!tag.ok) {
+            return null;
+        }
+        const fieldNum = tag.value >> 3;
+        const wireType = tag.value & 0x7;
+        const fieldStart = i;
+        i += tag.bytes;
+        const fieldCfg = cfg.get(fieldNum);
+        const pushSlice = (end) => {
+            if (end > data.length) {
+                return false;
+            }
+            out.push({ num: fieldNum, bytes: data.subarray(fieldStart, end) });
+            i = end;
+            return true;
+        };
+        const skip = (end) => {
+            if (end > data.length) {
+                return false;
+            }
+            i = end;
+            return true;
+        };
+        if (wireType === WIRE.VARINT) {
+            const v = decodeVarint(data, i);
+            if (!v.ok) {
+                return null;
+            }
+            const end = i + v.bytes;
+            if (!fieldCfg) {
+                if (!skip(end)) {
+                    return null;
+                }
+                continue;
+            }
+            if (!pushSlice(end)) {
+                return null;
+            }
+            continue;
+        }
+        if (wireType === WIRE.FIXED64) {
+            const end = i + 8;
+            if (!fieldCfg) {
+                if (!skip(end)) {
+                    return null;
+                }
+                continue;
+            }
+            if (!pushSlice(end)) {
+                return null;
+            }
+            continue;
+        }
+        if (wireType === WIRE.FIXED32) {
+            const end = i + 4;
+            if (!fieldCfg) {
+                if (!skip(end)) {
+                    return null;
+                }
+                continue;
+            }
+            if (!pushSlice(end)) {
+                return null;
+            }
+            continue;
+        }
+        if (wireType === WIRE.BYTES) {
+            const len = decodeVarint(data, i);
+            if (!len.ok) {
+                return null;
+            }
+            const valStart = i + len.bytes;
+            const valEnd = valStart + len.value;
+            if (valEnd > data.length) {
+                return null;
+            }
+            if (!fieldCfg) {
+                i = valEnd;
+                continue;
+            }
+            if (fieldCfg.m || fieldCfg.children) {
+                const sub = extractReportingTokenContent(data.subarray(valStart, valEnd), fieldCfg.children ?? EMPTY_MAP);
+                if (sub === null) {
+                    return null;
+                }
+                if (sub.length > 0) {
+                    const newTag = encodeVarint(tag.value);
+                    const newLen = encodeVarint(sub.length);
+                    out.push({
+                        num: fieldNum,
+                        bytes: Buffer.concat([newTag, newLen, sub])
+                    });
+                }
+                i = valEnd;
+                continue;
+            }
+            out.push({ num: fieldNum, bytes: data.subarray(fieldStart, valEnd) });
+            i = valEnd;
+            continue;
+        }
+        return null;
+    }
+    if (out.length === 0) {
+        return Buffer.alloc(0);
+    }
+    out.sort((a, b) => a.num - b.num);
+    return Buffer.concat(out.map(f => f.bytes));
+};
+const decodeVarint = (buffer, offset) => {
+    let value = 0;
+    let bytes = 0;
+    let shift = 0;
+    while (offset + bytes < buffer.length) {
+        const current = buffer[offset + bytes];
+        value |= (current & 0x7f) << shift;
+        bytes++;
+        if ((current & 0x80) === 0) {
+            return { value, bytes, ok: true };
+        }
+        shift += 7;
+        if (shift > 35) {
+            return { value: 0, bytes: 0, ok: false };
+        }
+    }
+    return { value: 0, bytes: 0, ok: false };
+};
+const encodeVarint = (value) => {
+    const parts = [];
+    let remaining = value >>> 0;
+    while (remaining > 0x7f) {
+        parts.push((remaining & 0x7f) | 0x80);
+        remaining >>>= 7;
+    }
+    parts.push(remaining);
+    return Buffer.from(parts);
+};
+export const getMessageReportingToken = async (msgProtobuf, message, key) => {
+    const msgSecret = message.messageContextInfo?.messageSecret;
+    if (!msgSecret || !key.id) {
+        return null;
+    }
+    const from = key.fromMe ? key.remoteJid : key.participant || key.remoteJid;
+    const to = key.fromMe ? key.participant || key.remoteJid : key.remoteJid;
+    const reportingSecret = generateMsgSecretKey(ENC_SECRET_REPORT_TOKEN, key.id, from, to, msgSecret);
+    const content = extractReportingTokenContent(msgProtobuf, compiledReportingFields);
+    if (!content || content.length === 0) {
+        return null;
+    }
+    const reportingToken = createHmac('sha256', reportingSecret).update(content).digest().subarray(0, 16);
+    return {
+        tag: 'reporting',
+        attrs: {},
+        content: [
+            {
+                tag: 'reporting_token',
+                attrs: { v: '2' },
+                content: reportingToken
+            }
+        ]
+    };
+};

package/lib/Utils/rich-messages.d.ts

@@ -0,0 +1,276 @@
+import { proto } from "../../WAProto/index.js";
+export declare const JS_KEYWORDS: Set<string>;
+export declare const PYTHON_KEYWORDS: Set<string>;
+export declare const GO_KEYWORDS: Set<string>;
+export declare const LUA_KEYWORDS: Set<string>;
+export declare const BASH_KEYWORDS: Set<string>;
+export declare const LANGUAGE_KEYWORDS: Record<string, Set<string>>;
+export declare enum CodeHighlightType {
+  DEFAULT = 0,
+  KEYWORD = 1,
+  METHOD = 2,
+  STRING = 3,
+  NUMBER = 4,
+  COMMENT = 5,
+}
+export declare enum RichSubMessageType {
+  UNKNOWN = 0,
+  GRID_IMAGE = 1,
+  TEXT = 2,
+  INLINE_IMAGE = 3,
+  TABLE = 4,
+  CODE = 5,
+  DYNAMIC = 6,
+  MAP = 7,
+  LATEX = 8,
+  CONTENT_ITEMS = 9,
+}
+export interface RichMessageOptions {
+  headerText?: string;
+  footer?: string;
+  title?: string;
+  language?: string;
+}
+export interface CodeBlockOptions extends RichMessageOptions {
+  language?: string;
+}
+export interface LatexExpression {
+  latexExpression: string;
+  url: string;
+  width: number;
+  height: number;
+  fontHeight?: number;
+  imageTopPadding?: number;
+  imageLeadingPadding?: number;
+  imageBottomPadding?: number;
+  imageTrailingPadding?: number;
+}
+export interface LatexOptions extends RichMessageOptions {
+  text?: string;
+  expressions: LatexExpression[];
+}
+export interface LatexImageOptions extends RichMessageOptions {
+  text?: string;
+  expressions: {
+    latexExpression: string;
+  }[];
+}
+export type LatexRenderFn = (latex: string) => Promise<{
+  buffer: Buffer;
+  width: number;
+  height: number;
+}>;
+export type MediaUploadFn = (
+  imageBuffer: Buffer,
+  mediaType: string,
+) => Promise<{
+  url: string;
+  directPath: string;
+}>;
+export interface UnifiedResponseOptions {
+  data: Buffer | Uint8Array;
+  headerText?: string;
+  footer?: string;
+}
+export interface CapturedUnifiedResponse {
+  unifiedResponse: {
+    data: Uint8Array;
+  };
+  submessages: any[];
+  contextInfo: any;
+}
+export interface QuotedMessage {
+  key: {
+    id?: string | null;
+    participant?: string | null;
+    remoteJid?: string | null;
+    fromMe?: boolean | null;
+  };
+  sender?: string;
+  message?: proto.IMessage | null;
+}
+interface CodeToken {
+  highlightType: number;
+  codeContent: string;
+}
+export declare const tokenizeCode: (
+  codeStr: string,
+  language?: string,
+) => CodeToken[];
+export declare const buildRichContextInfo: (
+  quoted?: QuotedMessage | null,
+) => Record<string, any>;
+export declare const buildBotForwardedMessage: (
+  submessages: any[],
+  contextInfo: Record<string, any>,
+  unifiedResponse?: {
+    data: Buffer | Uint8Array;
+  } | null,
+) => any;
+export declare const generateTableContent: (
+  title: string,
+  headers: string[],
+  rows: string[][],
+  quoted?: QuotedMessage | null,
+  options?: RichMessageOptions,
+) => {
+  message: any;
+  messageId: string;
+};
+export declare const generateListContent: (
+  title: string,
+  items: (string | string[])[],
+  quoted?: QuotedMessage | null,
+  options?: RichMessageOptions,
+) => {
+  message: any;
+  messageId: string;
+};
+export declare const generateCodeBlockContent: (
+  code: string,
+  quoted?: QuotedMessage | null,
+  options?: CodeBlockOptions,
+) => {
+  message: any;
+  messageId: string;
+};
+export declare const generateLatexContent: (
+  quoted: QuotedMessage | null | undefined,
+  options: LatexOptions,
+) => {
+  message: any;
+  messageId: string;
+};
+export declare const generateLatexImageContent: (
+  quoted: QuotedMessage | null | undefined,
+  options: LatexImageOptions,
+  uploadFn: MediaUploadFn,
+  renderLatexToPng: LatexRenderFn,
+) => Promise<{
+  message: any;
+  messageId: string;
+}>;
+export declare const generateLatexInlineImageContent: (
+  quoted: QuotedMessage | null | undefined,
+  options: LatexImageOptions,
+  uploadFn: MediaUploadFn,
+  renderLatexToPng: LatexRenderFn,
+) => Promise<{
+  message: any;
+  messageId: string;
+}>;
+export declare const captureUnifiedResponse: (
+  msg: proto.IMessage,
+) => CapturedUnifiedResponse | null;
+export declare const generateUnifiedResponseContent: (
+  quoted: QuotedMessage | null | undefined,
+  captured: CapturedUnifiedResponse,
+) => {
+  message: any;
+  messageId: string;
+};
+export declare const generateRichMessageContent: (
+  submessages: any[],
+  quoted?: QuotedMessage | null,
+) => {
+  message: any;
+  messageId: string;
+};
+export interface CodeBlockV2Options extends RichMessageOptions {
+  language?: string;
+  text?: string;
+}
+interface UnifiedCodeToken {
+  content: string;
+  type: "DEFAULT" | "KEYWORD" | "METHOD" | "STR" | "NUMBER" | "COMMENT";
+}
+interface TokenizeCodeV2Result {
+  codeBlock: CodeToken[];
+  unified_codeBlock: UnifiedCodeToken[];
+}
+export declare const tokenizeCodeV2: (
+  code: string,
+  language?: string,
+) => TokenizeCodeV2Result;
+interface UnifiedTableRow {
+  is_header: boolean;
+  cells: string[];
+}
+interface TableMetadataV2Result {
+  title: string;
+  rows: {
+    items: string[];
+    isHeading?: boolean;
+  }[];
+  unified_rows: UnifiedTableRow[];
+}
+export declare const toTableMetadataV2: (
+  arr: string[],
+) => TableMetadataV2Result;
+export interface TableV2Options extends RichMessageOptions {
+  text?: string;
+}
+export declare const generateTableContentV2: (
+  table: string[],
+  quoted?: QuotedMessage | null,
+  options?: TableV2Options,
+) => {
+  message: any;
+  messageId: string;
+};
+export declare const generateCodeBlockContentV2: (
+  code: string,
+  quoted?: QuotedMessage | null,
+  options?: CodeBlockV2Options,
+) => {
+  message: any;
+  messageId: string;
+};
+export interface Citation {
+  sourceQuery?: string;
+  faviconCdnUrl?: string;
+  citationNumber?: number;
+  sourceTitle?: string;
+}
+export interface Proof {
+  version?: number;
+  useCase?: number;
+  signature?: string;
+  certificateChain?: string[];
+}
+export interface LinkMessageOptions extends RichMessageOptions {
+  botJid?: string;
+  forwardingScore?: number;
+  citations?: Citation[];
+  proofs?: Proof[];
+}
+export interface LinkV2MessageOptions extends RichMessageOptions {
+  searchEngine?: string;
+}
+export declare const generateLinkContent: (
+  text: string,
+  links: (string | { url: string; displayName?: string })[],
+  quoted?: QuotedMessage | null,
+  options?: LinkMessageOptions,
+) => {
+  message: any;
+  messageId: string;
+};
+export declare const generateLinkContentV2: (
+  text: string,
+  links: (
+    | string
+    | {
+        url: string;
+        displayName?: string;
+        sourceDisplayName?: string;
+        sourceSubtitle?: string;
+      }
+  )[],
+  quoted?: QuotedMessage | null,
+  options?: LinkV2MessageOptions,
+) => {
+  message: any;
+  messageId: string;
+};
+export {};