@kangwifi-pro/waliwa 1.0.0

package/src/messages/message-encoder.ts

@@ -0,0 +1,520 @@
+/**
+ * Waliwa - Message encoding/decoding
+ * Encode MessageContent ke WhatsApp protobuf Message format
+ * Decode incoming WA Message ke MessageContent
+ *
+ * Hybrid approach: manual binary encoding untuk messages (lebih hemat RAM),
+ * pakai protobufjs untuk complex nested message types
+ */
+
+import type {
+  MessageContent,
+  WAMessage,
+  WAMessageKey,
+  JIDLike,
+  ContextInfo
+} from '../types/index.js';
+import { concatBytes, getRandomBytes, sha256Hash } from '../core/crypto.js';
+import { jidToString, jidUser, isGroupJid } from '../utils/jid.js';
+
+// ============== Message ID Generator ==============
+
+let messageCounter = 0;
+
+/**
+ * Generate unique message ID
+ * Format: 3EB0 + 8 random hex chars + counter
+ * Contoh: "3EB0ABC1234567890A"
+ */
+export function generateMessageId(): string {
+  messageCounter++;
+  const rand = getRandomBytes(8);
+  const randHex = Array.from(rand).map(b => b.toString(16).padStart(2, '0')).join('');
+  const counterHex = messageCounter.toString(16).padStart(2, '0');
+  return `3EB0${randHex}${counterHex}`.toUpperCase();
+}
+
+/**
+ * Generate stanza ID untuk IQ requests
+ */
+export function generateStanzaId(): string {
+  const rand = getRandomBytes(4);
+  const randHex = Array.from(rand).map(b => b.toString(16).padStart(2, '0')).join('');
+  return `${Date.now().toString(36)}-${randHex}`;
+}
+
+// ============== Message Encoder ==============
+
+/**
+ * Encode MessageContent ke WA binary node
+ *
+ * WhatsApp message structure:
+ * <message to="jid" type="text" id="msgId">
+ *   <conversation>text content</conversation>
+ * </message>
+ *
+ * Atau untuk media:
+ * <message to="jid" type="media" id="msgId">
+ *   <imageMessage>...</imageMessage>
+ * </message>
+ */
+export function encodeMessageNode(
+  jid: JIDLike,
+  content: MessageContent,
+  options: EncodeMessageOptions = {}
+): any {
+  const id = options.messageId ?? generateMessageId();
+  const to = jidToString(jid);
+  const type = isGroupJid(jid) ? 'text' : 'text';
+
+  const messageNode: any = {
+    tag: 'message',
+    attrs: {
+      to,
+      type,
+      id
+    },
+    content: [] as any[]
+  };
+
+  // Add participant untuk group messages
+  if (options.participant) {
+    messageNode.attrs.participant = options.participant;
+  }
+
+  // Determine content type
+  if (content.text) {
+    messageNode.content.push({
+      tag: 'conversation',
+      attrs: {},
+      content: content.text
+    });
+  }
+
+  if (content.image) {
+    messageNode.content.push({
+      tag: 'imageMessage',
+      attrs: {
+        url: content.image.stream ? '' : '',
+        mimetype: content.image.mimetype,
+        fileSha256: '', // computed on upload
+        fileLength: (content.image.fileLength ?? 0).toString()
+      },
+      content: content.caption ? [{ tag: 'caption', attrs: {}, content: content.caption }] : undefined
+    });
+  }
+
+  if (content.video) {
+    messageNode.content.push({
+      tag: 'videoMessage',
+      attrs: {
+        mimetype: content.video.mimetype,
+        seconds: (content.video.seconds ?? 0).toString(),
+        gifPlayback: content.video.gifPlayback ? 'true' : 'false'
+      },
+      content: content.caption ? [{ tag: 'caption', attrs: {}, content: content.caption }] : undefined
+    });
+  }
+
+  if (content.audio) {
+    messageNode.content.push({
+      tag: 'audioMessage',
+      attrs: {
+        mimetype: content.audio.mimetype,
+        ptt: content.audio.ptt ? 'true' : 'false'
+      }
+    });
+  }
+
+  if (content.document) {
+    messageNode.content.push({
+      tag: 'documentMessage',
+      attrs: {
+        mimetype: content.document.mimetype,
+        fileName: content.document.fileName ?? 'document',
+        pageCount: content.document.pageCount?.toString() ?? '1'
+      }
+    });
+  }
+
+  if (content.sticker) {
+    messageNode.content.push({
+      tag: 'stickerMessage',
+      attrs: {
+        mimetype: content.sticker.mimetype,
+        isAnimated: content.sticker.isAnimated ? 'true' : 'false'
+      }
+    });
+  }
+
+  if (content.location) {
+    messageNode.content.push({
+      tag: 'locationMessage',
+      attrs: {
+        degreesLatitude: content.location.degreesLatitude.toString(),
+        degreesLongitude: content.location.degreesLongitude.toString(),
+        name: content.location.name ?? '',
+        address: content.location.address ?? ''
+      }
+    });
+  }
+
+  if (content.contact) {
+    messageNode.content.push({
+      tag: 'contactMessage',
+      attrs: {
+        displayName: content.contact.displayName
+      },
+      content: content.contact.vcard
+    });
+  }
+
+  if (content.poll) {
+    messageNode.content.push({
+      tag: 'pollCreationMessage',
+      attrs: {
+        name: content.poll.name
+      },
+      content: content.poll.values.map(v => ({
+        tag: 'option',
+        attrs: { name: v }
+      }))
+    });
+  }
+
+  // Quoted message (reply)
+  if (content.quoted) {
+    const contextInfo = buildContextInfo(content.quoted, content.mentions);
+    messageNode.content.push({
+      tag: 'contextInfo',
+      attrs: {},
+      content: contextInfo
+    });
+  }
+
+  // Mentions
+  if (content.mentions && content.mentions.length > 0 && !content.quoted) {
+    messageNode.content.push({
+      tag: 'mentionedJidList',
+      attrs: {},
+      content: content.mentions.map(m => jidToString(m))
+    });
+  }
+
+  // Ephemeral
+  if (content.ephemeralExpiration) {
+    messageNode.attrs.ephemeral = content.ephemeralExpiration.toString();
+  }
+
+  // View once
+  if (content.viewOnce) {
+    messageNode.attrs.viewOnce = 'true';
+  }
+
+  return messageNode;
+}
+
+interface EncodeMessageOptions {
+  messageId?: string;
+  participant?: string;
+}
+
+/**
+ * Build context info untuk quoted/reply messages
+ */
+function buildContextInfo(quoted: WAMessage, mentions?: JIDLike[]): any[] {
+  const parts: any[] = [];
+  parts.push({
+    tag: 'stanzaId',
+    attrs: {},
+    content: quoted.key.id
+  });
+  parts.push({
+    tag: 'participant',
+    attrs: {},
+    content: jidToString(quoted.key.participant ?? quoted.key.remoteJid)
+  });
+  if (mentions && mentions.length > 0) {
+    parts.push({
+      tag: 'mentionedJidList',
+      attrs: {},
+      content: mentions.map(m => jidToString(m))
+    });
+  }
+  // Quoted message content (partial)
+  if (quoted.message?.text) {
+    parts.push({
+      tag: 'quotedMessage',
+      attrs: {},
+      content: quoted.message.text
+    });
+  }
+  return parts;
+}
+
+// ============== Message Decoder ==============
+
+/**
+ * Decode incoming WA binary node ke WAMessage
+ */
+export function decodeMessageNode(node: any, fromMe: boolean = false): WAMessage | null {
+  if (!node || node.tag !== 'message') {
+    return null;
+  }
+
+  const attrs = node.attrs || {};
+  const jid = attrs.from ?? attrs.to ?? '';
+  const id = attrs.id;
+  const participant = attrs.participant;
+
+  const key: WAMessageKey = {
+    remoteJid: jid,
+    fromMe,
+    id,
+    participant
+  };
+
+  const message: MessageContent & { [k: string]: any } = {} as any;
+  let messageTimestamp = Date.now();
+  let pushName: string | undefined;
+
+  // Process children
+  if (Array.isArray(node.content)) {
+    for (const child of node.content) {
+      if (typeof child === 'string') continue;
+      if (!child.tag) continue;
+
+      switch (child.tag) {
+        case 'conversation':
+          message.text = typeof child.content === 'string' ? child.content : '';
+          break;
+
+        case 'imageMessage':
+          message.image = {
+            stream: undefined as any, // akan di-populate oleh media download
+            mimetype: child.attrs?.mimetype ?? 'image/jpeg',
+            fileLength: parseInt(child.attrs?.fileLength ?? '0', 10)
+          };
+          if (child.content && Array.isArray(child.content)) {
+            const captionNode = child.content.find((c: any) => c.tag === 'caption');
+            if (captionNode) message.caption = captionNode.content;
+          }
+          break;
+
+        case 'videoMessage':
+          message.video = {
+            stream: undefined as any,
+            mimetype: child.attrs?.mimetype ?? 'video/mp4',
+            seconds: parseInt(child.attrs?.seconds ?? '0', 10),
+            gifPlayback: child.attrs?.gifPlayback === 'true'
+          };
+          if (child.content && Array.isArray(child.content)) {
+            const captionNode = child.content.find((c: any) => c.tag === 'caption');
+            if (captionNode) message.caption = captionNode.content;
+          }
+          break;
+
+        case 'audioMessage':
+          message.audio = {
+            stream: undefined as any,
+            mimetype: child.attrs?.mimetype ?? 'audio/mpeg',
+            seconds: parseInt(child.attrs?.seconds ?? '0', 10),
+            ptt: child.attrs?.ptt === 'true'
+          };
+          break;
+
+        case 'documentMessage':
+          message.document = {
+            stream: undefined as any,
+            mimetype: child.attrs?.mimetype ?? 'application/octet-stream',
+            fileName: child.attrs?.fileName ?? 'document',
+            pageCount: parseInt(child.attrs?.pageCount ?? '1', 10)
+          };
+          break;
+
+        case 'stickerMessage':
+          message.sticker = {
+            stream: undefined as any,
+            mimetype: child.attrs?.mimetype ?? 'image/webp',
+            isAnimated: child.attrs?.isAnimated === 'true'
+          };
+          break;
+
+        case 'locationMessage':
+          message.location = {
+            degreesLatitude: parseFloat(child.attrs?.degreesLatitude ?? '0'),
+            degreesLongitude: parseFloat(child.attrs?.degreesLongitude ?? '0'),
+            name: child.attrs?.name,
+            address: child.attrs?.address
+          };
+          break;
+
+        case 'contactMessage':
+          message.contact = {
+            displayName: child.attrs?.displayName ?? '',
+            vcard: typeof child.content === 'string' ? child.content : ''
+          };
+          break;
+
+        case 'pollCreationMessage':
+          message.poll = {
+            name: child.attrs?.name ?? '',
+            values: Array.isArray(child.content)
+              ? child.content.filter((c: any) => c.tag === 'option').map((c: any) => c.attrs?.name)
+              : []
+          };
+          break;
+
+        case 'contextInfo':
+          // Process quoted message
+          if (Array.isArray(child.content)) {
+            const stanzaId = child.content.find((c: any) => c.tag === 'stanzaId');
+            const participantNode = child.content.find((c: any) => c.tag === 'participant');
+            const mentionedJids = child.content.find((c: any) => c.tag === 'mentionedJidList');
+            // Build quoted message stub
+            if (stanzaId) {
+              message.quoted = {
+                key: {
+                  remoteJid: jid,
+                  fromMe: false,
+                  id: stanzaId.content
+                },
+                message: {},
+                messageTimestamp: 0,
+                status: 'read' as const
+              } as WAMessage;
+              if (participantNode) {
+                message.quoted!.key.participant = participantNode.content;
+              }
+            }
+            if (mentionedJids && Array.isArray(mentionedJids.content)) {
+              message.mentions = mentionedJids.content;
+            }
+          }
+          break;
+
+        case 'pushName':
+          pushName = typeof child.content === 'string' ? child.content : undefined;
+          break;
+
+        case 'deviceSentTime':
+        case 'timestamp':
+          const ts = child.attrs?.value ?? child.content;
+          if (ts) messageTimestamp = parseInt(String(ts), 10);
+          break;
+      }
+    }
+  }
+
+  // Determine status from attrs
+  const status = attrs.type === 'media' ? 'sent' : 'pending';
+
+  return {
+    key,
+    message,
+    messageTimestamp,
+    pushName,
+    status: status as any
+  };
+}
+
+// ============== Receipt Encoder ==============
+
+export type ReceiptType = 'read' | 'receipt' | 'played' | 'inactive';
+
+/**
+ * Build receipt node untuk send read/delivered confirmation
+ */
+export function buildReceiptNode(
+  jid: JIDLike,
+  messageIds: string[],
+  type: ReceiptType = 'read',
+  participant?: JIDLike
+): any {
+  return {
+    tag: 'receipt',
+    attrs: {
+      id: messageIds[0],
+      type,
+      to: jidToString(jid),
+      t: Math.floor(Date.now() / 1000).toString()
+    },
+    content: messageIds.map(id => ({
+      tag: 'receipt',
+      attrs: { id }
+    }))
+  };
+}
+
+// ============== Status Encoder ==============
+
+export type MessageStatusUpdateType = 'read' | 'delivered' | 'played';
+
+/**
+ * Build status node untuk update message status (read receipts)
+ */
+export function buildStatusNode(
+  jid: JIDLike,
+  messageId: string,
+  type: MessageStatusUpdateType,
+  participant?: JIDLike
+): any {
+  const attrs: any = {
+    type,
+    to: jidToString(jid),
+    id: messageId
+  };
+  if (participant) attrs.participant = jidToString(participant);
+  return {
+    tag: 'status',
+    attrs,
+    content: undefined
+  };
+}
+
+// ============== Helpers ==============
+
+/**
+ * Extract text content dari message
+ */
+export function extractMessageText(message: any): string {
+  if (!message) return '';
+  if (typeof message.text === 'string') return message.text;
+  if (typeof message.conversation === 'string') return message.conversation;
+  if (message.caption) return message.caption;
+  return '';
+}
+
+/**
+ * Check apakah message punya media
+ */
+export function hasMedia(message: any): boolean {
+  return !!(message?.image || message?.video || message?.audio || message?.document || message?.sticker);
+}
+
+/**
+ * Get media info dari message
+ */
+export function getMediaInfo(message: any): {
+  type: 'image' | 'video' | 'audio' | 'document' | 'sticker';
+  mimetype: string;
+  fileName?: string;
+  fileLength?: number;
+} | null {
+  if (message?.image) {
+    return { type: 'image', mimetype: message.image.mimetype, fileLength: message.image.fileLength };
+  }
+  if (message?.video) {
+    return { type: 'video', mimetype: message.video.mimetype, fileLength: message.video.fileLength };
+  }
+  if (message?.audio) {
+    return { type: 'audio', mimetype: message.audio.mimetype, fileLength: message.audio.fileLength };
+  }
+  if (message?.document) {
+    return { type: 'document', mimetype: message.document.mimetype, fileName: message.document.fileName };
+  }
+  if (message?.sticker) {
+    return { type: 'sticker', mimetype: message.sticker.mimetype };
+  }
+  return null;
+}