browsh/webext/background.js

let tty_width, tty_height;

// Create WebSocket connection.
const socket = new WebSocket('ws://localhost:3334');

// Connection opened
socket.addEventListener('open', function (_event) {
  console.log('Connected to Browsh client');
});

// Listen for messages
socket.addEventListener('message', (event) => {
  console.log('Message from server ', event.data);
  const parts = event.data.split(',');
  const command = parts[0];
  if (command === '/tty_size') {
    tty_width = parts[1];
    tty_height = parts[2];
  } else {
    tabs[active_tab_id].postMessage(event.data);
  }
});

let active_tab_id;
function handleActivated(active_info) {
  console.log(active_info);
  active_tab_id = active_info.id
}
browser.tabs.onActivated.addListener(handleActivated);

function handleRegistration(_request, sender, sendResponse) {
  console.log('tab registered');
  sendResponse(sender.tab);
  if (sender.tab.active) active_tab_id = sender.tab.id;
}
browser.runtime.onMessage.addListener(handleRegistration);

let tabs = {};
function connected(channel) {
  console.log('tab connected');
  tabs[channel.name] = channel;
  resizeBrowserWindow();
  // Currently tabs will only ever send screen output over their channel
  channel.onMessage.addListener(function(screen) {
    socket.send(screen);
  });
}
browser.runtime.onConnect.addListener(connected);

setInterval(() => {
  if (tabs[active_tab_id] === undefined) return;
  tabs[active_tab_id].postMessage(`/send_frame,${tty_width},${tty_height}`);
}, 1000);

function resizeBrowserWindow() {
  // Does this include scrollbars???
  const width = parseInt(Math.round(tty_width * 9)) + 4;
  const height = parseInt(Math.round(tty_height * 20)) + 4; // this is actually line-height
  var getting = browser.windows.getCurrent();
  getting.then(
    (windowInfo) => {
      console.log('resizing window', windowInfo, width, height);
      const updating = browser.windows.update(
        windowInfo.id,
        {
          width: width,
          height: height
        }
      );

      updating(
        (info) => {
          console.log(info);
        },
        (error) => {
          console.error(error);
        }
      );
    },
    (error) => {
      console.error(error);
    }
  );
}