satyamark-react 0.0.12 → 0.0.14

package/dist/index.js

@@ -22,14 +22,290 @@ var src_exports = {};
 __export(src_exports, {
   init: () => init,
   onConnected: () => onConnected,
-  onReceive: () => onReceive,
-  process: () => process,
-  receiveData: () => receiveData,
-  registerStatus: () => registerStatus,
-  sendData: () => sendData
+  process: () => process
 });
 module.exports = __toCommonJS(src_exports);
 
+// src/core/socketClient.ts
+var SocketClient = class {
+  constructor(url, handlers) {
+    this.socket = null;
+    this.url = url;
+    this.handlers = handlers;
+  }
+  connect() {
+    this.socket = new WebSocket(this.url);
+    this.socket.onopen = () => this.handlers.onOpen();
+    this.socket.onmessage = (event) => this.handlers.onMessage(JSON.parse(event.data));
+    this.socket.onclose = () => this.handlers.onClose();
+    this.socket.onerror = () => this.handlers.onError();
+  }
+  send(payload) {
+    if (!this.socket || this.socket.readyState !== WebSocket.OPEN) {
+      throw new Error("Socket not ready");
+    }
+    this.socket.send(JSON.stringify(payload));
+  }
+  close() {
+    var _a;
+    (_a = this.socket) == null ? void 0 : _a.close();
+    this.socket = null;
+  }
+  isOpen() {
+    var _a;
+    return ((_a = this.socket) == null ? void 0 : _a.readyState) === WebSocket.OPEN;
+  }
+};
+
+// src/utils/generateIds.ts
+function generateTimestamp() {
+  const now = /* @__PURE__ */ new Date();
+  return now.getFullYear().toString() + String(now.getMonth() + 1).padStart(2, "0") + String(now.getDate()).padStart(2, "0") + String(now.getHours()).padStart(2, "0") + String(now.getMinutes()).padStart(2, "0") + String(now.getSeconds()).padStart(2, "0") + String(now.getMilliseconds()).padStart(3, "0") + Math.floor(Math.random() * 1e3).toString().padStart(3, "0");
+}
+function generateJobId(app_id, user_id, dataId) {
+  const timestamp = generateTimestamp();
+  const jobId = `${app_id}_${user_id}_${dataId}_${timestamp}`;
+  return jobId;
+}
+
+// src/core/eventBus.ts
+var messageListeners = /* @__PURE__ */ new Set();
+var connectionListeners = /* @__PURE__ */ new Set();
+function onMessage(cb) {
+  messageListeners.add(cb);
+  return () => messageListeners.delete(cb);
+}
+function emitMessage(data) {
+  messageListeners.forEach((cb) => {
+    try {
+      cb(data);
+    } catch (err) {
+      console.error("Message listener error:", err);
+    }
+  });
+}
+function onConnected(cb) {
+  connectionListeners.add(cb);
+  return () => connectionListeners.delete(cb);
+}
+function emitConnection(context2) {
+  connectionListeners.forEach((cb) => {
+    try {
+      cb(context2);
+    } catch (err) {
+      console.log("Connection listener error:", err);
+    }
+  });
+}
+
+// src/utils/encryption.ts
+var SECRET = "your-strong-secret";
+function strToBuf(str) {
+  return new TextEncoder().encode(str);
+}
+function bufToStr(buf) {
+  return new TextDecoder().decode(buf);
+}
+function bufToBase64(buf) {
+  return btoa(String.fromCharCode(...new Uint8Array(buf)));
+}
+function base64ToBuf(base64) {
+  const binary = atob(base64);
+  const bytes = new Uint8Array(binary.length);
+  for (let i = 0; i < binary.length; i++) {
+    bytes[i] = binary.charCodeAt(i);
+  }
+  return bytes.buffer;
+}
+async function getKey() {
+  const keyMaterial = await crypto.subtle.importKey(
+    "raw",
+    strToBuf(SECRET),
+    { name: "PBKDF2" },
+    false,
+    ["deriveKey"]
+  );
+  return crypto.subtle.deriveKey(
+    {
+      name: "PBKDF2",
+      salt: strToBuf("satya_salt"),
+      iterations: 1e5,
+      hash: "SHA-256"
+    },
+    keyMaterial,
+    { name: "AES-GCM", length: 256 },
+    false,
+    ["encrypt", "decrypt"]
+  );
+}
+async function encrypt(text) {
+  const iv = crypto.getRandomValues(new Uint8Array(12));
+  const key = await getKey();
+  const encrypted = await crypto.subtle.encrypt(
+    { name: "AES-GCM", iv },
+    key,
+    strToBuf(text)
+  );
+  const combined = new Uint8Array(iv.length + encrypted.byteLength);
+  combined.set(iv);
+  combined.set(new Uint8Array(encrypted), iv.length);
+  return bufToBase64(combined.buffer);
+}
+async function decrypt(cipherText) {
+  const combined = new Uint8Array(base64ToBuf(cipherText));
+  const iv = combined.slice(0, 12);
+  const data = combined.slice(12);
+  const key = await getKey();
+  const decrypted = await crypto.subtle.decrypt(
+    { name: "AES-GCM", iv },
+    key,
+    data
+  );
+  return bufToStr(decrypted);
+}
+
+// src/utils/storage.ts
+function getCookie(name) {
+  var _a;
+  const value = `; ${document.cookie}`;
+  const parts = value.split(`; ${name}=`);
+  if (parts.length === 2)
+    return (_a = parts.pop()) == null ? void 0 : _a.split(";").shift();
+}
+function setCookie(name, value, days = 7) {
+  const expires = new Date(Date.now() + days * 864e5).toUTCString();
+  document.cookie = `${name}=${value}; expires=${expires}; path=/; Secure; SameSite=Lax`;
+}
+
+// src/utils/manageSessions.ts
+async function getSessionId() {
+  const raw = getCookie("satya_session");
+  if (!raw)
+    return "";
+  try {
+    return await decrypt(raw);
+  } catch {
+    return "";
+  }
+}
+async function setSessionId(sessionId) {
+  const encrypted = await encrypt(sessionId);
+  setCookie("satya_session", encrypted);
+}
+function clearSession() {
+  setCookie("satya_session", "", -1);
+}
+
+// src/core/connectionManager.ts
+var isDev = false;
+var context = null;
+var socketClient = null;
+var isConnecting = false;
+var isConnected = false;
+var reconnectTimer = null;
+async function resolveWsUrl() {
+  if (isDev) {
+    const wsUrl2 = "ws://localhost:1000";
+    return wsUrl2;
+  }
+  const res = await fetch(
+    "https://dhirajkarangale.github.io/SatyaMark/ws.json",
+    { cache: "no-store" }
+  );
+  const data = await res.json();
+  const wsUrl = data.wsUrl;
+  if (!wsUrl)
+    throw new Error("WebSocket URL resolution failed");
+  return wsUrl;
+}
+async function init(newContext) {
+  if (!(newContext == null ? void 0 : newContext.app_id) || !(newContext == null ? void 0 : newContext.user_id)) {
+    throw new Error("init() requires valid app_id and user_id");
+  }
+  if (isConnecting || isConnected)
+    return;
+  context = newContext;
+  await connect();
+}
+async function connect() {
+  if (!context)
+    return;
+  if (isConnecting || isConnected)
+    return;
+  isConnecting = true;
+  const url = await resolveWsUrl();
+  socketClient = new SocketClient(url, {
+    onOpen: async () => {
+      const sessionId = await getSessionId();
+      socketClient == null ? void 0 : socketClient.send({
+        type: "handshake",
+        clientId: context == null ? void 0 : context.user_id,
+        app_id: context == null ? void 0 : context.app_id,
+        sessionId
+      });
+      isConnected = true;
+      isConnecting = false;
+      emitConnection(context);
+    },
+    onMessage: async (data) => {
+      if (data.type === "session_created" && data.sessionId) {
+        await setSessionId(data.sessionId);
+        return;
+      }
+      if (data.type === "RateLimiter") {
+        if (data.msg === "Invalid session") {
+          clearSession();
+          socketClient == null ? void 0 : socketClient.close();
+        }
+        return;
+      }
+      if (data.clientId === (context == null ? void 0 : context.user_id)) {
+        emitMessage(data);
+      }
+    },
+    onClose: () => {
+      if (!isConnected && !isConnecting)
+        return;
+      isConnected = false;
+      isConnecting = false;
+      emitConnection(null);
+      scheduleReconnect();
+    },
+    onError: () => {
+      socketClient == null ? void 0 : socketClient.close();
+    }
+  });
+  socketClient.connect();
+}
+function scheduleReconnect() {
+  if (reconnectTimer || !context)
+    return;
+  reconnectTimer = setTimeout(() => {
+    reconnectTimer = null;
+    connect();
+  }, 2e3);
+}
+async function sendJob(text, imageUrl, dataId) {
+  if (!context)
+    throw new Error("Satyamark is not ready Call init() first");
+  if (!text && !imageUrl) {
+    throw new Error("Provide text or imageUrl");
+  }
+  if (text && text.trim().length < 3) {
+    throw new Error("Text must be at least 3 characters");
+  }
+  const jobId = generateJobId(context.app_id, context.user_id, dataId);
+  const sessionId = await getSessionId();
+  socketClient == null ? void 0 : socketClient.send({
+    clientId: context.user_id,
+    sessionId,
+    jobId,
+    text,
+    image_url: imageUrl
+  });
+  return jobId;
+}
+
 // src/utils/iconRegistry.ts
 var ICON_URLS = {
   correct: "https://res.cloudinary.com/dfamljkyo/image/upload/v1768325442/correct_gjiadc.png",
@@ -61,218 +337,21 @@ function loadIcon(key) {
   iconLoadMap.set(key, promise);
   return promise;
 }
-function preloadIcons() {
-  const run = () => {
-    Object.keys(ICON_URLS).forEach(loadIcon);
-  };
-  if ("requestIdleCallback" in window) {
-    window.requestIdleCallback(run);
-  } else {
-    setTimeout(run, 0);
-  }
-}
 function ensureIconLoaded(key) {
   return loadIcon(key).catch(() => {
   });
 }
 
-// src/satyamark_connect.ts
-var wsUrl = null;
-var socket = null;
-var storedConnectionData = null;
-var listeners = [];
-var onConnectedCb = null;
-async function getWsUrl() {
-  if (wsUrl)
-    return wsUrl;
-  const res = await fetch(
-    "https://dhirajkarangale.github.io/SatyaMark/ws.json",
-    { cache: "no-store" }
-  );
-  const data = await res.json();
-  wsUrl = data.wsUrl;
-  return wsUrl;
-}
-function onReceive(cb) {
-  listeners.push(cb);
-  return () => {
-    const idx = listeners.indexOf(cb);
-    if (idx !== -1)
-      listeners.splice(idx, 1);
-  };
-}
-function onConnected(cb) {
-  onConnectedCb = cb;
-}
-async function init(connectionData, options) {
-  var _a;
-  preloadIcons();
-  if (socket && socket.readyState == WebSocket.OPEN && storedConnectionData == connectionData) {
-    console.log("Already Connected: ", connectionData);
-    return;
-  }
-  const url = await getWsUrl();
-  if (url) {
-    socket = new WebSocket(url);
-  } else {
-    console.error("WebSocket endpoint resolution failed. Unable to establish connection.");
-    return;
-  }
-  onConnectedCb = (_a = options == null ? void 0 : options.onConnected) != null ? _a : onConnectedCb;
-  socket.onopen = () => {
-    console.log("Connected to server: ", connectionData.user_id);
-    safeSend({
-      type: "handshake",
-      clientId: connectionData.user_id,
-      appId: connectionData.app_id
-    });
-    onConnectedCb == null ? void 0 : onConnectedCb(connectionData);
-  };
-  socket.onmessage = (event) => {
-    receiveData(JSON.parse(event.data));
-  };
-  socket.onclose = () => {
-    console.log("Server connection closed");
-  };
-  storedConnectionData = connectionData;
-}
-function isSocketOpen() {
-  return socket && socket.readyState === WebSocket.OPEN;
-}
-function assert(condition, message) {
-  if (!condition)
-    throw new Error(message);
-}
-function safeSend(msg) {
-  if (!storedConnectionData) {
-    console.warn("No connectionData found. Call connect() first.");
-    return;
-  }
-  if (!socket || socket.readyState !== WebSocket.OPEN) {
-    console.warn("Socket not ready");
-    return;
-  }
-  socket.send(JSON.stringify(msg));
-}
-function uniqueTimestamp() {
-  const now = /* @__PURE__ */ new Date();
-  const yyyy = now.getFullYear();
-  const MM = String(now.getMonth() + 1).padStart(2, "0");
-  const dd = String(now.getDate()).padStart(2, "0");
-  const hh = String(now.getHours()).padStart(2, "0");
-  const mm = String(now.getMinutes()).padStart(2, "0");
-  const ss = String(now.getSeconds()).padStart(2, "0");
-  const ms = String(now.getMilliseconds()).padStart(3, "0");
-  const micro = String(Math.floor(Math.random() * 1e3)).padStart(3, "0");
-  return `${yyyy}${MM}${dd}${hh}${mm}${ss}${ms}${micro}`;
-}
-function sendData(text, image_url, dataId) {
-  if (!storedConnectionData) {
-    console.log("No connectionData found. Call connect() first.");
-    return;
-  }
-  if (!socket || socket.readyState !== WebSocket.OPEN) {
-    console.log("Socket not ready");
-    return;
-  }
-  assert(storedConnectionData, "Call init() before sendData()");
-  assert(isSocketOpen(), "WebSocket is not ready");
-  assert(dataId, "dataId is required");
-  assert(text || image_url, "Provide text or image_url");
-  if (text)
-    assert(text.trim().length >= 3, "Text must be at least 3 characters");
-  const { app_id, user_id } = storedConnectionData;
-  const timestamp = uniqueTimestamp();
-  const jobId = `${app_id}_${user_id}_${dataId}_${timestamp}`;
-  const data = {
-    clientId: user_id,
-    jobId,
-    text,
-    image_url
-  };
-  socket.send(JSON.stringify(data));
-  return jobId;
-}
-function receiveData(data) {
-  if (!storedConnectionData || data.clientId != storedConnectionData.user_id)
-    return;
-  for (const cb of Array.from(listeners)) {
-    try {
-      cb(data);
-    } catch (err) {
-      console.log("listener error", err);
-    }
-  }
-}
-
-// src/satyamark_process.ts
-var mergeText = (texts) => texts.join(", ");
-var isValidImageUrl = (url) => {
-  return new Promise((resolve) => {
-    const img = new Image();
-    img.onload = () => resolve(true);
-    img.onerror = () => resolve(false);
-    img.src = url;
-  });
-};
-var getFirstValidImage = async (urls) => {
-  for (const url of urls) {
-    if (await isValidImageUrl(url))
-      return url;
-  }
-  return null;
-};
-var extractFromDiv = (root) => {
-  var _a;
-  const images = Array.from(root.querySelectorAll("img")).map((img) => img.src);
-  const text = [];
-  const walker = document.createTreeWalker(root, NodeFilter.SHOW_TEXT);
-  let node;
-  while (node = walker.nextNode()) {
-    const trimmed = (_a = node.textContent) == null ? void 0 : _a.trim();
-    if (trimmed)
-      text.push(trimmed);
-  }
-  return { text, images };
-};
-async function process(divRef, dataId) {
-  if (!divRef) {
-    throw new Error("Invalid root element");
-  }
-  if (!dataId) {
-    throw new Error("dataId is required");
-  }
-  const { text, images } = extractFromDiv(divRef);
-  const mergedText = mergeText(text);
-  const validImage = await getFirstValidImage(images);
-  if (!mergedText && !validImage) {
-    throw new Error("No valid text or image found in the element");
-  }
-  if (mergedText && mergedText.length < 3) {
-    throw new Error("Extracted text is too short");
-  }
-  const jobId = sendData(mergedText, validImage != null ? validImage : "", dataId);
-  return jobId;
-}
-
-// src/satyamark_status_controller.ts
-var jobMap = {};
+// src/core/status_controller.ts
 var DEFAULT_ICON_SIZE = 20;
 var satyamark_url = "https://satyamark.vercel.app/chat";
-function registerStatus(jobId, rootElement, options = {}) {
+function updateIcon(containerRef, data) {
   var _a;
-  jobMap[jobId] = {
-    root: rootElement,
-    iconSize: (_a = options.iconSize) != null ? _a : DEFAULT_ICON_SIZE
-  };
-  updateIcon(jobId, "pending", null);
-}
-function updateIcon(jobId, rawMark, data) {
-  const entry = jobMap[jobId];
-  if (!entry)
-    return;
-  const { root, iconSize } = entry;
-  const mark = rawMark in ICON_URLS ? rawMark : "pending";
+  const root = containerRef;
+  const iconSize = DEFAULT_ICON_SIZE;
+  let mark = "pending";
+  if (data && data.mark)
+    mark = (_a = data.mark) == null ? void 0 : _a.toLowerCase();
   ensureIconLoaded(mark);
   const container = root.querySelector("[data-satyamark-status-container]");
   if (!container)
@@ -328,19 +407,109 @@ function updateIcon(jobId, rawMark, data) {
     icon.onclick = null;
   }
 }
-onReceive((data) => {
-  var _a, _b;
-  if (!(data == null ? void 0 : data.jobId))
+
+// src/utils/process_data.ts
+var mergeText = (texts) => texts.join(" |#| ");
+var isValidImageUrl = (url) => {
+  return new Promise((resolve) => {
+    const img = new Image();
+    img.onload = () => resolve(true);
+    img.onerror = () => resolve(false);
+    img.src = url;
+  });
+};
+var getFirstValidImage = async (urls) => {
+  for (const url of urls) {
+    if (await isValidImageUrl(url))
+      return url;
+  }
+  return null;
+};
+var extractFromDiv = (root) => {
+  var _a;
+  const images = Array.from(root.querySelectorAll("img")).map((img) => img.src);
+  const text = [];
+  const walker = document.createTreeWalker(root, NodeFilter.SHOW_TEXT);
+  let node;
+  while (node = walker.nextNode()) {
+    const trimmed = (_a = node.textContent) == null ? void 0 : _a.trim();
+    if (trimmed)
+      text.push(trimmed);
+  }
+  return { text, images };
+};
+async function process_data(divRef, dataId) {
+  if (!divRef) {
+    throw new Error("Invalid root element");
+  }
+  if (!dataId) {
+    throw new Error("dataId is required");
+  }
+  const { text, images } = extractFromDiv(divRef);
+  const mergedText = mergeText(text);
+  const validImage = await getFirstValidImage(images);
+  if (!mergedText && !validImage) {
+    throw new Error("No valid text or image found in the element");
+  }
+  if (mergedText && mergedText.length < 3) {
+    throw new Error("Extracted text is too short");
+  }
+  const image_url = validImage ? validImage : "";
+  return { text: mergedText, image_url };
+}
+
+// src/core/process.ts
+var isConnected2 = false;
+var isSendingJobs = false;
+var jobMap = /* @__PURE__ */ new Map();
+var process_queue = [];
+function process(containerRef, dataId) {
+  process_queue.push({ containerRef, dataId });
+  void sendJobs();
+}
+async function sendJobs() {
+  if (isSendingJobs || !isConnected2)
+    return;
+  isSendingJobs = true;
+  while (process_queue.length > 0) {
+    const item = process_queue[0];
+    if (!item)
+      break;
+    const { containerRef, dataId } = item;
+    try {
+      const { text, image_url } = await process_data(containerRef, dataId);
+      const jobId = await sendJob(text, image_url, dataId);
+      jobMap.set(jobId, containerRef);
+      process_queue.shift();
+      updateIcon(containerRef, null);
+    } catch (error) {
+      isSendingJobs = false;
+      setTimeout(() => {
+        void sendJobs();
+      }, 1e3);
+      return;
+    }
+  }
+  isSendingJobs = false;
+}
+onMessage((data) => {
+  if (!data || !data.jobId)
     return;
-  updateIcon(data.jobId, (_b = (_a = data.mark) == null ? void 0 : _a.toLowerCase()) != null ? _b : "pending", data);
+  const containerRef = jobMap.get(data.jobId);
+  if (!containerRef)
+    return;
+  jobMap.delete(data.jobId);
+  updateIcon(containerRef, data);
+});
+onConnected((data) => {
+  isConnected2 = !!data;
+  if (isConnected2) {
+    void sendJobs();
+  }
 });
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   init,
   onConnected,
-  onReceive,
-  process,
-  receiveData,
-  registerStatus,
-  sendData
+  process
 });