background.js

console.log('background listening...');

// Listen for messages from the popup
chrome.runtime.onMessage.addListener(function (message, sender, sendResponse) {
  if (message.type === "addSaveButton") {
    console.log('adding save buttons...');
    // Inject content script into the active tab
    chrome.tabs.query({ active: true, currentWindow: true }, function (tabs) {
      chrome.scripting.executeScript({
        target: { tabId: tabs[0].id },
        files: ["content.js"]
      }, function () {
        console.log('initiating addSaveButton message...');
        // Send message to the content script
        chrome.tabs.sendMessage(tabs[0].id, { type: "addSaveButton" });
      });
    });
  } else if (message.type === "codeFound") {
    // Initiate the code download
    console.log('download message filename: ', message.filename);

    const fileExtension = message.filename ? message.filename.split('.').pop() : 'txt';
    const mimeType = getFileMimeType(fileExtension);

    chrome.downloads.download({
      url: `data:${mimeType};charset=utf-8,` + encodeURIComponent(message.code),
      filename: message.filename || "code.txt",
      saveAs: true
    });
  }
});

// Function to get the MIME type based on the file extension
function getFileMimeType(extension) {
  switch (extension) {
    case 'txt':
      return 'text/plain';
    case 'js':
      return 'text/javascript';
    case 'html':
      return 'text/html';
    case 'css':
      return 'text/css';
    // Add more file extensions and corresponding MIME types as needed
    default:
      return 'application/octet-stream';
  }
}