chrome-extension/public/content-script.js

 async function connection(hostname) {
  const isConnected = await chrome.storage.local.get([hostname]);
  let connected = false
  if(isConnected && Object.keys(isConnected).length > 0 && isConnected[hostname]){
    connected = true
  }
  return connected
}

// In your content script
document.addEventListener('qortalExtensionRequests', async (event) => {
  const { type, payload, requestId, timeout } = event.detail; // Capture the requestId
  if (type === 'REQUEST_USER_INFO') {
    const hostname = window.location.hostname
    const res = await connection(hostname)

    if(!res){
      document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
        detail: { type: "USER_INFO", data: {
          error: "Not authorized"
        }, requestId }
      }));
      return
    }
    chrome?.runtime?.sendMessage({ action: "userInfo" }, (response) => {
      if (response.error) {
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "USER_INFO", data: {
            error: response.error
          }, requestId }
        }));
      } else {
        // Include the requestId in the detail when dispatching the response
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "USER_INFO", data: response, requestId }
        }));
      }
    });
  } else if (type === 'REQUEST_IS_INSTALLED') {
    chrome?.runtime?.sendMessage({ action: "version" }, (response) => {
      if (response.error) {
        console.error("Error:", response.error);
      } else {
        // Include the requestId in the detail when dispatching the response
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "IS_INSTALLED", data: response, requestId }
        }));
      }
    });
  } else if (type === 'REQUEST_CONNECTION') {
    const hostname = window.location.hostname
    chrome?.runtime?.sendMessage({ action: "connection", payload: {
      hostname
    }, timeout }, (response) => {
      if (response.error) {
        console.error("Error:", response.error);
      } else {
        // Include the requestId in the detail when dispatching the response
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "CONNECTION", data: response, requestId }
        }));
      }
    });
  } else if (type === 'REQUEST_OAUTH') {
    const hostname = window.location.hostname
    const res = await connection(hostname)
    if(!res){
      document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
        detail: { type: "OAUTH", data: {
          error: "Not authorized"
        }, requestId }
      }));
      return
    }

    chrome?.runtime?.sendMessage({ action: "oauth", payload: {
      nodeBaseUrl: payload.nodeBaseUrl,
      senderAddress: payload.senderAddress,
      senderPublicKey: payload.senderPublicKey, timestamp: payload.timestamp
    }}, (response) => {
      if (response.error) {
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "OAUTH", data: {
            error: response.error
          }, requestId }
        }));
      } else {
        // Include the requestId in the detail when dispatching the response
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "OAUTH", data: response, requestId }
        }));
      }
    });
  } else if (type === 'REQUEST_BUY_ORDER') {
    const hostname = window.location.hostname
    const res = await connection(hostname)
    if(!res){
      document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
        detail: { type: "BUY_ORDER", data: {
          error: "Not authorized"
        }, requestId }
      }));
      return
    }

    chrome?.runtime?.sendMessage({ action: "buyOrder", payload: {
      qortalAtAddresses: payload.qortalAtAddresses,
      hostname,
      useLocal: payload?.useLocal
      
    }, timeout}, (response) => {
      if (response.error) {
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "BUY_ORDER", data: {
            error: response.error
          }, requestId }
        }));
      } else {
        // Include the requestId in the detail when dispatching the response
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "BUY_ORDER", data: response, requestId }
        }));
      }
    });
  } else if(type === 'REQUEST_LTC_BALANCE'){
   
    
    const hostname = window.location.hostname
    const res = await connection(hostname)
    if(!res){
      document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
        detail: { type: "USER_INFO", data: {
          error: "Not authorized"
        }, requestId }
      }));
      return
    }
    chrome?.runtime?.sendMessage({ action: "ltcBalance", payload: {
      hostname
    },  timeout }, (response) => {
      
      if (response.error) {
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "LTC_BALANCE", data: {
            error: response.error
          }, requestId }
        }));
      } else {
        // Include the requestId in the detail when dispatching the response
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "LTC_BALANCE", data: response, requestId }
        }));
      }
    });
  } else if(type === 'CHECK_IF_LOCAL'){
   
    
    const hostname = window.location.hostname
    const res = await connection(hostname)
    if(!res){
      document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
        detail: { type: "USER_INFO", data: {
          error: "Not authorized"
        }, requestId }
      }));
      return
    }
    chrome?.runtime?.sendMessage({ action: "checkLocal", payload: {
      hostname
    },  timeout }, (response) => {
      
      if (response.error) {
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "CHECK_IF_LOCAL", data: {
            error: response.error
          }, requestId }
        }));
      } else {
        // Include the requestId in the detail when dispatching the response
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "CHECK_IF_LOCAL", data: response, requestId }
        }));
      }
    });
  }  else if (type === 'REQUEST_AUTHENTICATION') {
    const hostname = window.location.hostname
    const res = await connection(hostname)
    if(!res){
      document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
        detail: { type: "USER_INFO", data: {
          error: "Not authorized"
        }, requestId }
      }));
      return
    }
    chrome?.runtime?.sendMessage({ action: "authentication", payload: {
      hostname
    },  timeout }, (response) => {
      if (response.error) {
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "AUTHENTICATION", data: {
            error: response.error
          }, requestId }
        }));
      } else {
        // Include the requestId in the detail when dispatching the response
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "AUTHENTICATION", data: response, requestId }
        }));
      }
    });
  } else if (type === 'REQUEST_SEND_QORT') {
    const hostname = window.location.hostname
    const res = await connection(hostname)
    if(!res){
      document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
        detail: { type: "USER_INFO", data: {
          error: "Not authorized"
        }, requestId }
      }));
      return
    }
    chrome?.runtime?.sendMessage({ action: "sendQort", payload: {
      hostname,
      amount: payload.amount,
      description: payload.description,
      address: payload.address
    }, timeout }, (response) => {
      if (response.error) {
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "SEND_QORT", data: {
            error: response.error
          }, requestId }
        }));
      } else {
        // Include the requestId in the detail when dispatching the response
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "SEND_QORT", data: response, requestId }
        }));
      }
    });
  } else if (type === 'REQUEST_CLOSE_POPUP') {
    const hostname = window.location.hostname
    const res = await connection(hostname)
    if(!res){
      document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
        detail: { type: "USER_INFO", data: {
          error: "Not authorized"
        }, requestId }
      }));
      return
    }
    chrome?.runtime?.sendMessage({ action: "closePopup" }, (response) => {
      if (response.error) {
        
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "CLOSE_POPUP", data: {
            error: response.error
          }, requestId }
        }));
      } else {
        // Include the requestId in the detail when dispatching the response
        document.dispatchEvent(new CustomEvent('qortalExtensionResponses', {
          detail: { type: "CLOSE_POPUP", data: true, requestId }
        }));
      }
    });
  }
  // Handle other request types as needed...
});


chrome.runtime?.onMessage.addListener(function(message, sender, sendResponse) {
  if (message.type === "LOGOUT") {
      // Notify the web page
      window.postMessage({
          type: "LOGOUT",
          from: 'qortal'
      }, "*");
  } else if (message.type === "RESPONSE_FOR_TRADES") {
    // Notify the web page
    window.postMessage({
        type: "RESPONSE_FOR_TRADES",
        from: 'qortal',
        payload: message.message
    }, "*");
}
});