SOURCE

chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
  if (request.action === 'updateCount') {
    chrome.browserAction.setBadgeText({ text: request.count.toString() });
  }
});