native-messaging.js

// Native messaging port management

let port = null;
let pending = [];
let intentionalDisconnect = false;

/**
 * Establish connection to native host
 */
export function ensureNativePort(onMessage, onDisconnect) {
  if (port) return port;

  console.log("[BG] Connecting native host...");
  port = chrome.runtime.connectNative("com.redactosaurus.native");
  intentionalDisconnect = false;

  port.onMessage.addListener((msg) => {
    console.log("[BG] From native:", msg);
    if (onMessage) onMessage(msg);
  });

  port.onDisconnect.addListener(() => {
    const lastError = chrome.runtime.lastError;
    const wasIntentional = intentionalDisconnect;
    intentionalDisconnect = false;
    console.log("[BG] Native disconnected:", lastError, {
      intentional: wasIntentional,
    });
    port = null;
    if (onDisconnect) onDisconnect(lastError, { intentional: wasIntentional });
  });

  return port;
}

/**
 * Send message to native host
 */
export function sendToNative(message) {
  if (!port) {
    throw new Error("Native port not connected");
  }
  port.postMessage(message);
}

/**
 * Track pending promise for request/response
 */
export function addPending(promiseCallbacks) {
  pending.push(promiseCallbacks);
}

/**
 * Resolve all pending promises with a response
 */
export function flushPending(response) {
  const list = pending;
  pending = [];
  for (const p of list) p.resolve(response);
}

/**
 * Reject all pending promises with an error
 */
export function rejectAllPending(error) {
  const list = pending;
  pending = [];
  for (const p of list) p.reject(error);
}

/**
 * Disconnect from native host intentionally.
 */
export function disconnectNativePort() {
  if (!port) {
    intentionalDisconnect = false;
    return false;
  }

  intentionalDisconnect = true;
  try {
    port.disconnect();
  } catch {
    // Best effort.
  }
  return true;
}

/**
 * Check if a native port is currently connected.
 */
export function isNativePortConnected() {
  return port !== null;
}